Update app.py

app.py

@@ -4,9 +4,9 @@ import logging
 import torch
 import gradio as gr
 from tooluniverse import ToolUniverse
+from txagent import TxAgent  # Updated import statement
 import warnings
 from typing import List, Dict, Any
-from txagent import TxAgent
 
 # Suppress specific warnings
 warnings.filterwarnings("ignore", category=UserWarning)
@@ -66,7 +66,7 @@ def patch_embedding_loading():
                 self.tool_desc_embedding = safe_load_embeddings(CONFIG["embedding_filename"])
                 
                 # Handle tool count mismatch
-                tools = tooluniverse.get_all_tools()  # Use get_all_tools() instead of direct access
+                tools = tooluniverse.get_all_tools()
                 current_count = len(tools)
                 embedding_count = len(self.tool_desc_embedding)
                 
@@ -163,7 +163,7 @@ class TxAgentApp:
                 history=tx_history,
                 temperature=0.3,
                 max_new_tokens=1024,
-                max_token=8192,  # Note: Using max_token instead of max_length
+                max_token=8192,
                 call_agent=False,
                 conversation=None,
                 max_round=30
@@ -198,8 +198,7 @@ def create_interface() -> gr.Blocks:
         
         chatbot = gr.Chatbot(
             height=500, 
-            label="Conversation",
-            bubble_full_width=False
+            label="Conversation"
         )
         msg = gr.Textbox(label="Your clinical question")
         clear_btn = gr.Button("Clear Chat")