Update app.py

app.py

@@ -1,31 +1,38 @@
 import os
-from openai import OpenAI
 import gradio as gr
+from openai import OpenAI
 
-# --- Setup ---
-# Hugging Face Space will automatically inject your token from Secrets
+# --- Initialize client securely ---
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-# Create client for Hugging Face Inference Router
+if not HF_TOKEN:
+    raise ValueError("❌ HF_TOKEN not found. Please set it in your Hugging Face Space secrets.")
+
 client = OpenAI(
     base_url="https://router.huggingface.co/v1",
     api_key=HF_TOKEN,
 )
 
-# --- Chat Function ---
+# --- Chat handler ---
 def chat_with_model(message, history):
-    # Convert history into OpenAI-compatible messages
-    messages = [{"role": "system", "content": "You are a helpful assistant."}]
-    for human, ai in history:
-        messages.append({"role": "user", "content": human})
-        messages.append({"role": "assistant", "content": ai})
+    # Convert message history into OpenAI-style messages
+    messages = []
+
+    if history:
+        for msg in history:
+            if isinstance(msg, dict):  # New Gradio message format
+                messages.append(msg)
+            elif isinstance(msg, (list, tuple)) and len(msg) == 2:  # Old format fallback
+                messages.append({"role": "user", "content": msg[0]})
+                messages.append({"role": "assistant", "content": msg[1]})
+
     messages.append({"role": "user", "content": message})
 
+    # Query model
     try:
         response = client.chat.completions.create(
             model="openai/gpt-oss-120b:fireworks-ai",
             messages=messages,
-            max_tokens=500,
             temperature=0.7,
         )
         reply = response.choices[0].message.content
@@ -34,16 +41,14 @@ def chat_with_model(message, history):
 
     return reply
 
-# --- Gradio UI ---
-iface = gr.ChatInterface(
+# --- Gradio Interface ---
+chatbot_ui = gr.ChatInterface(
     fn=chat_with_model,
-    chatbot=gr.Chatbot(height=500, type="messages", show_label=False),
-    title="💬 GPT-OSS 120B Chat",
-    description="Chat with a massive open-source language model hosted via Hugging Face Router.",
-    theme="soft",
-    textbox=gr.Textbox(placeholder="Ask me anything...", scale=7),
+    title="🧠 GPT-OSS 120B (Fireworks)",
+    description="Chat with the OSS 120B model hosted via Hugging Face router.",
+    examples=["Hello!", "Tell me a joke.", "Explain quantum computing in simple terms."],
+    type="messages",  # ✅ required to use new format
 )
 
-# --- Launch ---
 if __name__ == "__main__":
-    iface.launch()
+    chatbot_ui.launch()