Update app.py

app.py

@@ -7,7 +7,7 @@ BASE_URL = "https://router.huggingface.co/v1"
 def get_headers():
     key = os.environ.get("HF_API_KEY")
     if not key:
-        return None, "HF_API_KEY missing (Settings → Variables)"
+        return None, "HF_API_KEY missing (Space Settings → Variables)"
     return {
         "Authorization": f"Bearer {key}",
         "Content-Type": "application/json",
@@ -20,7 +20,7 @@ def list_models(headers):
     data = r.json()
     models = [m["id"] for m in data.get("data", []) if "id" in m]
     if not models:
-        return None, "No models available for this account"
+        return None, "No models available for this account (enable a provider in HF settings)"
     return models, None
 
 def chat_call(headers, model, history, user_msg):
@@ -33,7 +33,12 @@ def chat_call(headers, model, history, user_msg):
     r = requests.post(
         f"{BASE_URL}/chat/completions",
         headers=headers,
-        json={"model": model, "messages": messages, "max_tokens": 200},
+        json={
+            "model": model,
+            "messages": messages,
+            "temperature": 0.7,
+            "max_tokens": 200,
+        },
         timeout=60,
     )
     if r.status_code != 200:
@@ -42,31 +47,56 @@ def chat_call(headers, model, history, user_msg):
 
 with gr.Blocks(title="My AI Chatbot") as demo:
     status = gr.Markdown("Starting…")
-    model_dd = gr.Dropdown(label="Model", choices=[], interactive=True)
-    chat = gr.Chatbot(height=500)
-    msg = gr.Textbox(placeholder="Type a message…")
+    model_dd = gr.Dropdown(label="Model", choices=[], value=None, interactive=True)
+    chat = gr.Chatbot(height=520)
+    msg = gr.Textbox(placeholder="Type a message…", show_label=False)
     send = gr.Button("Send")
+    clear = gr.Button("Clear")
 
     def init():
         headers, err = get_headers()
         if err:
-            return f"❌ {err}", []
+            status_md = f"❌ {err}"
+            return status_md, gr.update(choices=[], value=None)
+
         models, err = list_models(headers)
         if err:
-            return f"❌ {err}", []
-        return f"✅ Ready (model: `{models[0]}`)", models
+            status_md = f"❌ {err}"
+            return status_md, gr.update(choices=[], value=None)
+
+        default_model = models[0]
+        status_md = f"✅ Ready. Using model: `{default_model}`"
+        return status_md, gr.update(choices=models, value=default_model)
 
     def respond(user_msg, history, model):
-        headers, err = get_headers()
-        if err:
-            history.append([user_msg, err])
+        try:
+            history = history or []
+
+            headers, err = get_headers()
+            if err:
+                history.append([user_msg, f"Setup error: {err}"])
+                return "", history
+
+            # If model is not selected for any reason, pick the first available
+            if not model:
+                models, m_err = list_models(headers)
+                if m_err:
+                    history.append([user_msg, f"Model error: {m_err}"])
+                    return "", history
+                model = models[0]
+
+            bot = chat_call(headers, model, history, user_msg)
+            history.append([user_msg, bot])
+            return "", history
+
+        except Exception as e:
+            history = history or []
+            history.append([user_msg, f"Exception: {type(e).__name__}: {e}"])
             return "", history
-        bot = chat_call(headers, model, history, user_msg)
-        history.append([user_msg, bot])
-        return "", history
 
     demo.load(init, outputs=[status, model_dd])
     send.click(respond, inputs=[msg, chat, model_dd], outputs=[msg, chat])
     msg.submit(respond, inputs=[msg, chat, model_dd], outputs=[msg, chat])
+    clear.click(lambda: [], outputs=chat)
 
 demo.launch()