Update app.py

app.py

@@ -2,19 +2,41 @@ import os
 import gradio as gr
 from groq import Groq
 
-# Load Groq API Key (HF Spaces uses environment variables)
+# Initialize Groq client
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
+def normalize_chat_history(chat_history):
+    """
+    Converts chat_history into list of tuples (user, assistant)
+    Works with both tuple and dict formats returned by Gradio.
+    """
+    normalized = []
+    for item in chat_history:
+        if isinstance(item, dict):
+            # If it's a dict, convert accordingly
+            if item.get("role") == "user":
+                normalized.append((item.get("content"), ""))
+            elif item.get("role") == "assistant":
+                # merge assistant into last user entry
+                if normalized:
+                    last_user, _ = normalized[-1]
+                    normalized[-1] = (last_user, item.get("content"))
+        elif isinstance(item, (list, tuple)) and len(item) == 2:
+            normalized.append((item[0], item[1]))
+    return normalized
+
 def chat_with_groq(user_input, chat_history):
     if not user_input.strip():
         return chat_history, ""
 
+    chat_history = normalize_chat_history(chat_history)
+
     messages = []
     for u, a in chat_history:
-        messages += [
-            {"role": "user", "content": u},
-            {"role": "assistant", "content": a},
-        ]
+        if u:
+            messages.append({"role": "user", "content": u})
+        if a:
+            messages.append({"role": "assistant", "content": a})
 
     messages.append({"role": "user", "content": user_input})
 
@@ -23,6 +45,7 @@ def chat_with_groq(user_input, chat_history):
             model="llama-3.3-70b-versatile",
             messages=messages,
         )
+
         assistant_response = chat_completion.choices[0].message.content
         chat_history.append((user_input, assistant_response))
 
@@ -31,7 +54,7 @@ def chat_with_groq(user_input, chat_history):
 
     return chat_history, ""
 
-# UI Styling (Accessible + Responsive)
+# UI Styling
 custom_css = """
 .gradio-container {
     max-width: 1000px;