Update app.py

app.py

@@ -3,16 +3,25 @@ import os
 import requests
 import tempfile
 
+# Validate HF_TOKEN environment variable
 HF_API_TOKEN = os.getenv("HF_TOKEN")
+if not HF_API_TOKEN:
+    raise ValueError("HF_TOKEN environment variable is not set. Please set it to your Hugging Face API token.")
+
 MODEL_ID = "rohitnagareddy/Qwen3-0.6B-Coding-Finetuned-v1"
 API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
 HEADERS = {"Authorization": f"Bearer {HF_API_TOKEN}"}
 
 def query_hf_api(prompt):
-    response = requests.post(API_URL, headers=HEADERS, json={"inputs": prompt})
-    if response.status_code == 200:
-        return response.json()[0]["generated_text"]
-    return f"[Error] {response.status_code}: {response.text}"
+    try:
+        response = requests.post(API_URL, headers=HEADERS, json={"inputs": prompt})
+        response.raise_for_status()  # Raise an exception for HTTP errors
+        data = response.json()
+        if isinstance(data, list) and "generated_text" in data[0]:
+            return data[0]["generated_text"]
+        return "[Error] Unexpected response format."
+    except requests.exceptions.RequestException as e:
+        return f"[Error] API request failed: {e}"
 
 def chat_fn(prompt, chat_history):
     response = query_hf_api(prompt)
@@ -21,18 +30,21 @@ def chat_fn(prompt, chat_history):
     return chat_history, chat_history
 
 def save_chat(chat_history):
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8") as f:
-        for entry in chat_history:
-            f.write(f"{entry['role'].capitalize()}: {entry['content']}\n\n")
-        return f.name  # return path to file
+    try:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8") as f:
+            for entry in chat_history:
+                f.write(f"{entry['role'].capitalize()}: {entry['content']}\n\n")
+            return gr.File.update(value=f.name, visible=True)
+    except Exception as e:
+        return f"[Error] Failed to save chat: {e}"
 
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
     gr.Markdown("# 🤖 Qwen3 Coding Chatbot (Gradio + HF API)")
-    
+
     with gr.Row():
         clear = gr.Button("🧹 Clear Chat")
         download_btn = gr.Button("⬇️ Download Chat")
-    
+
     chat = gr.Chatbot(label="Qwen Chat", type="messages")
     msg = gr.Textbox(label="Your message", placeholder="Ask me something...")
     submit = gr.Button("🚀 Send")
@@ -45,5 +57,3 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
     download_btn.click(save_chat, [history], download_file)
 
 demo.launch()
-
-