Update app.py

app.py

@@ -137,24 +137,20 @@ def create_agent():
         logger.error(f"Failed to create agent: {str(e)}")
         raise
 
-def respond(chat_history, history, temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round):
-    if not chat_history:
-        return [["assistant", "Please provide a message."]]
-
-    message = chat_history[-1][1] if isinstance(chat_history[-1], (list, tuple)) else chat_history[-1]
-
-    if not isinstance(message, str) or len(message.strip()) <= 10:
+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."]]
 
-    updated_history = history + [{"role": "user", "content": message}]
+    chat_history = chat_history + [["user", msg]]
     print("\n==== DEBUG ====")
-    print("User Message:", message)
-    print("Full History:", updated_history)
+    print("User Message:", msg)
+    print("Chat History:", chat_history)
     print("================\n")
 
-    try:
-        formatted_history = [(m["role"], m["content"]) for m in updated_history]
+    # Format for TxAgent
+    formatted_history = [(role, content) for role, content in chat_history]
 
+    try:
         response_generator = agent.run_gradio_chat(
             formatted_history,
             temperature,
@@ -165,22 +161,23 @@ def respond(chat_history, history, temperature, max_new_tokens, max_tokens, mult
             max_round
         )
     except Exception as e:
-        updated_history.append({"role": "assistant", "content": f"Error: {str(e)}"})
-    else:
-        collected = ""
-        for chunk in response_generator:
-            if isinstance(chunk, dict):
-                collected += chunk.get("content", "")
-            else:
-                collected += str(chunk)
-        updated_history.append({"role": "assistant", "content": collected})
+        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)
 
-    # Return formatted history to Gradio
-    return [(m["role"], m["content"]) for m in updated_history]
+    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")
+        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")
             max_new_tokens = gr.Slider(128, 4096, value=1024, label="Max New Tokens")
@@ -192,7 +189,7 @@ def create_demo(agent):
 
         submit.click(
             respond,
-            inputs=[chatbot, chatbot, temp, max_new_tokens, max_tokens, multi_agent, gr.State([]), max_rounds],
+            inputs=[msg, chatbot, temp, max_new_tokens, max_tokens, multi_agent, gr.State([]), max_rounds],
             outputs=[chatbot]
         )