Update app.py

app.py

@@ -15,20 +15,25 @@ client = OpenAI(
 
 # --- Chat handler ---
 def chat_with_model(message, history):
-    # Convert message history into OpenAI-style messages
+    # Build messages list safely
     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
+            # Handle both dict and tuple formats
+            if isinstance(msg, dict):
+                # Keep only allowed keys
+                messages.append({
+                    "role": msg.get("role", "user"),
+                    "content": msg.get("content", "")
+                })
+            elif isinstance(msg, (list, tuple)) and len(msg) == 2:
                 messages.append({"role": "user", "content": msg[0]})
                 messages.append({"role": "assistant", "content": msg[1]})
 
+    # Add latest user message
     messages.append({"role": "user", "content": message})
 
-    # Query model
     try:
         response = client.chat.completions.create(
             model="openai/gpt-oss-120b:fireworks-ai",
@@ -46,8 +51,8 @@ chatbot_ui = gr.ChatInterface(
     fn=chat_with_model,
     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
+    examples=["Hello!", "Tell me a joke.", "Explain AI in simple terms."],
+    type="messages",  # required
 )
 
 if __name__ == "__main__":