Update app.py

app.py

@@ -2,43 +2,49 @@ import os
 from groq import Groq
 import gradio as gr
 
-# Initialize Groq client (API key from HF Secrets)
+# Initialize client
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# System prompt (bot personality)
+# System prompt
 SYSTEM_PROMPT = {
     "role": "system",
-    "content": "You are a helpful, friendly, and intelligent AI assistant. Answer clearly and concisely."
+    "content": "You are a helpful, friendly AI assistant."
 }
 
+# Clean message (IMPORTANT FIX)
+def clean_message(msg):
+    return {
+        "role": msg.get("role"),
+        "content": msg.get("content")
+    }
+
 # Chatbot function
 def chatbot(message, history):
     try:
         messages = [SYSTEM_PROMPT]
 
-        # Limit history (prevents token overflow)
+        # Limit history
         history = history[-6:] if history else []
 
-        # Handle BOTH formats safely
         for item in history:
-            # Case 1: (user, bot)
+            # Case 1: tuple (user, bot)
             if isinstance(item, (list, tuple)) and len(item) == 2:
                 user_msg, bot_msg = item
 
                 if user_msg:
-                    messages.append({"role": "user", "content": user_msg})
+                    messages.append({"role": "user", "content": str(user_msg)})
                 if bot_msg:
-                    messages.append({"role": "assistant", "content": bot_msg})
+                    messages.append({"role": "assistant", "content": str(bot_msg)})
 
-            # Case 2: dict format
+            # Case 2: dict (REMOVE metadata)
             elif isinstance(item, dict):
                 if "role" in item and "content" in item:
-                    messages.append(item)
+                    messages.append(clean_message(item))
 
         # Add current message
-        messages.append({"role": "user", "content": message})
+        messages.append({"role": "user", "content": str(message)})
 
-        # Call Groq API
+        # Call Groq
         response = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
             messages=messages,
@@ -46,20 +52,19 @@ def chatbot(message, history):
             max_tokens=1024,
         )
 
-        reply = response.choices[0].message.content.strip()
-        return reply
+        return response.choices[0].message.content.strip()
 
     except Exception as e:
         return f"⚠️ Error: {str(e)}"
 
 
-# Create UI (NO theme here to avoid error)
+# UI
 demo = gr.ChatInterface(
     fn=chatbot,
     title="🚀 Groq AI Chatbot",
     description="Fast chatbot powered by Groq",
 )
 
-# Launch app
+# Launch
 if __name__ == "__main__":
     demo.launch()