Update app.py

app.py

@@ -1,25 +1,41 @@
+import os
 import gradio as gr
 from openai import OpenAI
-import os
 
+# Initialize HuggingFace router client
 client = OpenAI(
     base_url="https://router.huggingface.co/v1",
-    api_key=os.getenv("HF_TOKEN")
+    api_key=os.getenv("HF_TOKEN")  # Set in HF Spaces Secrets
 )
 
-def chat_with_model(user_input):
-    completion = client.chat.completions.create(
-        model="openai/gpt-oss-20b:nebius",
-        messages=[{"role": "user", "content": user_input}],
-    )
-    return completion.choices[0].message.content
-
-iface = gr.Interface(
-    fn=chat_with_model,
-    inputs=gr.Textbox(lines=5, placeholder="Type your message here..."),
-    outputs=gr.Textbox(label="Response"),
-    title="Hugging Face GPT-OSS Chat",
-    description="Chat with GPT-OSS 20B model deployed via Hugging Face API"
-)
+def chat_with_model(message: str):
+    try:
+        response = client.chat.completions.create(
+            model="openai/gpt-oss-20b:nebius",
+            messages=[{"role": "user", "content": message}],
+        )
+
+        # Correct extraction of the text
+        return response.choices[0].message.content
+
+    except Exception as e:
+        return f"❌ Error: {e}"
+
 
-iface.launch(ssr_mode=False)
+# Build simple HF-friendly UI
+with gr.Blocks(title="GPT-OSS 20B Chat") as demo:
+    gr.Markdown("## Chat with GPT-OSS 20B (HF Router)")
+    input_box = gr.Textbox(lines=4, label="Your Message")
+    output_box = gr.Textbox(label="Model Reply")
+
+    submit = gr.Button("Send")
+
+    submit.click(chat_with_model, inputs=input_box, outputs=output_box)
+
+
+# Launch for HF Spaces (no SSR, no hot reload)
+demo.launch(
+    server_name="0.0.0.0",
+    server_port=7860,
+    show_error=True
+)