fix response format

app.py

@@ -257,10 +257,16 @@ def generate_response(persona_key: str, message: str, history: list, max_tokens:
             messages.append({"role": "system", "content": system_prompt})
         
         # Add conversation history (last 5 exchanges to avoid too long context)
-        # History is in tuple format: list of [user_msg, assistant_msg] pairs
-        for user_msg, assistant_msg in history[-5:]:
-            messages.append({"role": "user", "content": user_msg})
-            messages.append({"role": "assistant", "content": assistant_msg})
+        # Handle both tuple format [user_msg, assistant_msg] and messages format
+        for item in history[-5:]:
+            if isinstance(item, dict) and "role" in item:
+                # Messages format
+                messages.append(item)
+            else:
+                # Tuple format [user_msg, assistant_msg]
+                user_msg, assistant_msg = item
+                messages.append({"role": "user", "content": user_msg})
+                messages.append({"role": "assistant", "content": assistant_msg})
         
         # Add current message
         messages.append({"role": "user", "content": message})
@@ -315,8 +321,9 @@ def generate_response(persona_key: str, message: str, history: list, max_tokens:
         response = " ".join(response.split())
         response = response.strip()
         
-        # Update history with tuple format [user_msg, assistant_msg]
-        history.append([message, response])
+        # Update history with messages format (Gradio expects this format)
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": response})
         
         return history, ""
         
@@ -328,7 +335,7 @@ def generate_response(persona_key: str, message: str, history: list, max_tokens:
 
 def clear_chat():
     """Clear the chat history."""
-    return [], ""  # Empty list for tuple format
+    return [], ""  # Empty list for messages format
 
 
 # Create Gradio interface