Update app.py

app.py

@@ -139,18 +139,20 @@ def create_agent():
 
 def respond(msg, chat_history, temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round):
     if not msg or len(msg.strip()) <= 10:
-        return chat_history + [{"role": "assistant", "content": "Please provide a valid message with a string longer than 10 characters."}]
+        return chat_history + [["assistant", "Please provide a valid message with a string longer than 10 characters."]]
+
+    # Append user message in tuple format
+    chat_history = chat_history + [["user", msg]]
 
-    # Append new user message
-    chat_history = chat_history + [{"role": "user", "content": msg}]
     print("\n==== DEBUG ====")
     print("User Message:", msg)
     print("Chat History:", chat_history)
     print("================\n")
 
     try:
-        # Convert to tuples for the agent
-        formatted_history = [(m["role"], m["content"]) for m in chat_history]
+        # Convert to TxAgent format
+        formatted_history = [(role, content) for role, content in chat_history]
+
         response_generator = agent.run_gradio_chat(
             formatted_history,
             temperature,
@@ -160,15 +162,17 @@ def respond(msg, chat_history, temperature, max_new_tokens, max_tokens, multi_ag
             conversation,
             max_round
         )
+
         collected = ""
         for chunk in response_generator:
             if isinstance(chunk, dict):
                 collected += chunk.get("content", "")
             else:
                 collected += str(chunk)
-        chat_history.append({"role": "assistant", "content": collected})
+
+        chat_history.append(["assistant", collected])
     except Exception as e:
-        chat_history.append({"role": "assistant", "content": f"Error: {str(e)}"})
+        chat_history.append(["assistant", f"Error: {str(e)}"])
 
     return chat_history