Update app.py

app.py

@@ -139,18 +139,18 @@ 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 + [["assistant", "Please provide a valid message with a string longer than 10 characters."]]
+        return chat_history + [{"role": "assistant", "content": "Please provide a valid message with a string longer than 10 characters."}]
 
-    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")
 
-    # Format for TxAgent
-    formatted_history = [(role, content) for role, content in chat_history]
-
     try:
+        # Convert to tuples for the agent
+        formatted_history = [(m["role"], m["content"]) for m in chat_history]
         response_generator = agent.run_gradio_chat(
             formatted_history,
             temperature,
@@ -160,23 +160,21 @@ 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})
     except Exception as e:
-        chat_history.append(["assistant", f"Error: {str(e)}"])
-        return chat_history
-
-    collected = ""
-    for chunk in response_generator:
-        if isinstance(chunk, dict):
-            collected += chunk.get("content", "")
-        else:
-            collected += str(chunk)
+        chat_history.append({"role": "assistant", "content": f"Error: {str(e)}"})
 
-    chat_history.append(["assistant", collected])
     return chat_history
 
 def create_demo(agent):
     with gr.Blocks(css=chat_css) as demo:
-        chatbot = gr.Chatbot(label="TxAgent", type="messages")
+        chatbot = gr.Chatbot(label="TxAgent", render_markdown=True)
         msg = gr.Textbox(label="Your question", placeholder="Type your biomedical query...", scale=6)
         with gr.Row():
             temp = gr.Slider(0, 1, value=0.3, label="Temperature")