Update app.py

app.py

@@ -68,9 +68,8 @@ class TxAgentApplication:
 
     def chat(self, message, chat_history):
         if not self.is_initialized:
-            yield "Error: Please initialize the model first"
-            return
-
+            return chat_history + [(message, "Error: Please initialize the model first")]
+        
         try:
             # Convert to messages format
             messages = []
@@ -79,8 +78,8 @@ class TxAgentApplication:
                 messages.append({"role": "assistant", "content": assistant})
             messages.append({"role": "user", "content": message})
 
-            # Stream response
-            full_response = ""
+            # Get response
+            response = ""
             for chunk in self.agent.run_gradio_chat(
                 messages,
                 temperature=0.3,
@@ -90,11 +89,11 @@ class TxAgentApplication:
                 conversation=[],
                 max_round=30
             ):
-                full_response += chunk
-                yield [(message, full_response)]
-                
+                response += chunk
+            
+            return chat_history + [(message, response)]
         except Exception as e:
-            yield [(message, f"Error: {str(e)}")]
+            return chat_history + [(message, f"Error: {str(e)}")]
 
 # ========== Gradio Interface ==========
 def create_interface():
@@ -108,21 +107,24 @@ def create_interface():
             init_btn = gr.Button("Initialize TxAgent", variant="primary")
             init_status = gr.Textbox(label="Status", interactive=False)
         
-        # Chat Interface (using modern messages format)
+        # Chat Interface using modern messages format
         chatbot = gr.Chatbot(
             height=600,
             label="Conversation",
+            show_label=True,
             avatar_images=(
-                "https://example.com/user.png",  # User avatar
-                "https://example.com/bot.png"    # Bot avatar
-            )
+                None,  # User avatar (None for default)
+                None   # Bot avatar (None for default)
+            ),
+            show_copy_button=True
         )
         
         with gr.Row():
             msg = gr.Textbox(
                 label="Your Question",
                 placeholder="Ask about drug interactions or treatments...",
-                scale=4
+                scale=4,
+                container=False
             )
             submit_btn = gr.Button("Submit", variant="primary", scale=1)
         
@@ -130,7 +132,8 @@ def create_interface():
         gr.Examples(
             examples=EXAMPLE_QUESTIONS,
             inputs=msg,
-            label="Try these examples:"
+            label="Try these examples:",
+            examples_per_page=3
         )
         
         gr.Markdown(UI_CONFIG['disclaimer'])
@@ -144,13 +147,15 @@ def create_interface():
         msg.submit(
             app.chat,
             [msg, chatbot],
-            [chatbot]
+            chatbot,
+            queue=True  # Enable queue for this event
         )
         
         submit_btn.click(
             app.chat,
             [msg, chatbot],
-            [chatbot]
+            chatbot,
+            queue=True  # Enable queue for this event
         ).then(
             lambda: "", None, msg
         )
@@ -162,9 +167,9 @@ if __name__ == "__main__":
     interface = create_interface()
     
     # Correct launch configuration
+    interface.queue()  # Enable queue globally
     interface.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=True,
-        enable_queue=True  # Enable queue without deprecated parameters
+        share=True
     )