Update app.py

app.py

@@ -139,7 +139,15 @@ def create_agent():
         logger.error(f"Failed to create agent: {str(e)}")
         raise
 
-def respond(message, history, temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round):
+def respond(chat_history, history, temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round):
+    if not chat_history:
+        return history + [{"role": "assistant", "content": "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:
+        return history + [{"role": "user", "content": message}, {"role": "assistant", "content": "Please provide a valid message with a string longer than 10 characters."}]
+
     updated_history = history + [{"role": "user", "content": message}]
     print("\n==== DEBUG ====")
     print("User Message:", message)
@@ -147,7 +155,6 @@ def respond(message, history, temperature, max_new_tokens, max_tokens, multi_age
     print("================\n")
 
     try:
-        # Ensure correct format for run_gradio_chat
         formatted_history = [(m["role"], m["content"]) for m in updated_history]
 
         response_generator = agent.run_gradio_chat(
@@ -160,7 +167,7 @@ def respond(message, history, temperature, max_new_tokens, max_tokens, multi_age
             max_round
         )
     except Exception as e:
-        return history + [{"role": "user", "content": message}, {"role": "assistant", "content": f"Error: {str(e)}"}]
+        return updated_history + [{"role": "assistant", "content": f"Error: {str(e)}"}]
 
     collected = ""
     for chunk in response_generator:
@@ -169,13 +176,13 @@ def respond(message, history, temperature, max_new_tokens, max_tokens, multi_age
         else:
             collected += str(chunk)
 
-    return history + [{"role": "user", "content": message}, {"role": "assistant", "content": collected}]
+    return updated_history + [{"role": "assistant", "content": collected}]
 
 def create_demo(agent):
     with gr.Blocks(css=chat_css) as demo:
         chatbot = gr.Chatbot(label="TxAgent", type="messages")
         with gr.Row():
-            msg = gr.Textbox(label="Your question")
+            msg = gr.Textbox(label="Your question")  # not used in inputs anymore
         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")
@@ -187,9 +194,10 @@ def create_demo(agent):
 
         submit.click(
             respond,
-            inputs=[msg, chatbot, temp, max_new_tokens, max_tokens, multi_agent, gr.State([]), max_rounds],
+            inputs=[chatbot, chatbot, temp, max_new_tokens, max_tokens, multi_agent, gr.State([]), max_rounds],
             outputs=[chatbot]
         )
+
     return demo
 
 def main():