Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import random
 import gradio as gr
 from txagent import TxAgent
 
@@ -26,22 +25,22 @@ MODEL_CONFIG = {
 UI_CONFIG = {
     'description': '''
     <div>
-    <h1 style="text-align: center;">TxAgent: An AI Agent for Therapeutic Reasoning</h1>
-    <p style="text-align: center;">Precision therapeutics AI with multi-step reasoning and real-time biomedical knowledge</p>
+    <h1 style="text-align: center;">TxAgent: Therapeutic Reasoning AI</h1>
+    <p style="text-align: center;">Precision therapeutics with multi-step reasoning</p>
     </div>
     ''',
     'disclaimer': '''
     <div style="color: #666; font-size: 0.9em; margin-top: 20px;">
-    <strong>Disclaimer:</strong> This tool provides informational support only and is not a substitute for professional medical advice.
+    <strong>Disclaimer:</strong> For informational purposes only, not medical advice.
     </div>
     '''
 }
 
 # Example questions
 EXAMPLE_QUESTIONS = [
-    "How should dosage be adjusted for a 50-year-old with hepatic impairment taking Journavx?",
-    "Is Xolremdi suitable for a patient with WHIM syndrome already taking Prozac?",
-    "What are the contraindications for combining Warfarin with Amiodarone?"
+    "How should dosage be adjusted for hepatic impairment with Journavx?",
+    "Is Xolremdi suitable with Prozac for WHIM syndrome?",
+    "What are Warfarin-Amiodarone contraindications?"
 ]
 
 # ========== Application Class ==========
@@ -49,114 +48,91 @@ class TxAgentApplication:
     def __init__(self):
         self.agent = None
         self.is_initialized = False
-        self.loading = False
 
-    def initialize_agent(self, progress=gr.Progress()):
+    def initialize_agent(self):
         if self.is_initialized:
-            return True, "Model already initialized"
+            return "Model already initialized"
         
-        self.loading = True
         try:
-            progress(0.1, desc="Initializing TxAgent...")
-            
-            # Initialize the agent
             self.agent = TxAgent(
                 MODEL_CONFIG['model_name'],
                 MODEL_CONFIG['rag_model_name'],
                 tool_files_dict=MODEL_CONFIG['tool_files'],
                 **MODEL_CONFIG['default_params']
             )
-            
-            progress(0.3, desc="Loading language model...")
             self.agent.init_model()
-            
-            progress(0.8, desc="Finalizing setup...")
             self.is_initialized = True
-            self.loading = False
-            
-            return True, "TxAgent initialized successfully"
+            return "TxAgent initialized successfully"
         except Exception as e:
-            self.loading = False
-            return False, f"Initialization failed: {str(e)}"
+            return f"Initialization failed: {str(e)}"
 
-    def chat(self, message, chat_history, temperature, max_new_tokens):
+    def chat(self, message, chat_history):
         if not self.is_initialized:
-            yield "Error: Model not initialized. Please initialize first."
+            yield "Error: Please initialize the model first"
             return
 
         try:
-            # Convert Gradio chat history to agent format
+            # Convert to messages format
             messages = []
-            for turn in chat_history:
-                messages.append({"role": "user", "content": turn[0]})
-                messages.append({"role": "assistant", "content": turn[1]})
+            for user, assistant in chat_history:
+                messages.append({"role": "user", "content": user})
+                messages.append({"role": "assistant", "content": assistant})
             messages.append({"role": "user", "content": message})
 
-            # Stream the response
+            # Stream response
             full_response = ""
             for chunk in self.agent.run_gradio_chat(
                 messages,
-                temperature=temperature,
-                max_new_tokens=max_new_tokens,
+                temperature=0.3,
+                max_new_tokens=1024,
                 max_tokens=8192,
                 multi_agent=False,
                 conversation=[],
                 max_round=30
             ):
                 full_response += chunk
-                yield full_response
-
+                yield [(message, full_response)]
+                
         except Exception as e:
-            yield f"Error during chat: {str(e)}"
+            yield [(message, f"Error: {str(e)}")]
 
 # ========== Gradio Interface ==========
 def create_interface():
     app = TxAgentApplication()
     
     with gr.Blocks(title="TxAgent", theme=gr.themes.Soft()) as demo:
-        # Header Section
         gr.Markdown(UI_CONFIG['description'])
         
-        # Initialization Section
+        # Initialization
         with gr.Row():
             init_btn = gr.Button("Initialize TxAgent", variant="primary")
-            init_status = gr.Textbox(label="Initialization Status", interactive=False)
+            init_status = gr.Textbox(label="Status", interactive=False)
         
-        # Chat Interface
+        # Chat Interface (using modern messages format)
         chatbot = gr.Chatbot(
             height=600,
-            bubble_full_width=False,
-            placeholder="Type your medical question below..."
+            label="Conversation",
+            avatar_images=(
+                "https://example.com/user.png",  # User avatar
+                "https://example.com/bot.png"    # Bot avatar
+            )
         )
         
         with gr.Row():
             msg = gr.Textbox(
                 label="Your Question",
-                placeholder="Ask about drug interactions, dosage, or treatment options...",
+                placeholder="Ask about drug interactions or treatments...",
                 scale=4
             )
             submit_btn = gr.Button("Submit", variant="primary", scale=1)
         
-        # Settings
-        with gr.Accordion("Advanced Settings", open=False):
-            with gr.Row():
-                temperature = gr.Slider(
-                    minimum=0.1, maximum=1.0, value=0.3, step=0.1,
-                    label="Temperature (higher = more creative)"
-                )
-                max_new_tokens = gr.Slider(
-                    minimum=128, maximum=4096, value=1024, step=128,
-                    label="Max Response Length"
-                )
-        
         # Examples
         gr.Examples(
             examples=EXAMPLE_QUESTIONS,
             inputs=msg,
-            label="Example Questions (click to try)"
+            label="Try these examples:"
         )
         
-        # Footer
         gr.Markdown(UI_CONFIG['disclaimer'])
 
         # Event Handlers
@@ -167,13 +143,13 @@ def create_interface():
         
         msg.submit(
             app.chat,
-            [msg, chatbot, temperature, max_new_tokens],
+            [msg, chatbot],
             [chatbot]
         )
         
         submit_btn.click(
             app.chat,
-            [msg, chatbot, temperature, max_new_tokens],
+            [msg, chatbot],
             [chatbot]
         ).then(
             lambda: "", None, msg
@@ -183,20 +159,12 @@ def create_interface():
 
 # ========== Main Execution ==========
 if __name__ == "__main__":
-    # Create and launch the interface
     interface = create_interface()
     
-    # Launch settings
-    launch_config = {
-        'server_name': '0.0.0.0',
-        'server_port': 7860,
-        'share': True,
-        'favicon_path': None,
-        'auth': None,
-        'auth_message': None,
-        'enable_queue': True,
-        'max_threads': 40,
-        'show_error': True
-    }
-    
-    interface.queue(concurrency_count=5).launch(**launch_config)
+    # Correct launch configuration
+    interface.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=True,
+        enable_queue=True  # Enable queue without deprecated parameters
+    )