Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import logging
 from txagent import TxAgent
 from tooluniverse import ToolUniverse
+from importlib.resources import files
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -13,22 +14,24 @@ class TxAgentApp:
         self.agent = self._initialize_agent()
 
     def _initialize_agent(self):
-        """Initialize the TxAgent with proper parameters"""
+        """Initialize the TxAgent with proper tool file paths"""
         try:
             logger.info("Initializing TxAgent...")
             
-            # Initialize default tool files
+            # Get absolute paths to tool files from package installation
             tool_files = {
-                "opentarget": "opentarget_tools.json",
-                "fda_drug_label": "fda_drug_labeling_tools.json",
-                "special_tools": "special_tools.json",
-                "monarch": "monarch_tools.json"
+                "opentarget": str(files('tooluniverse.data').joinpath('opentarget_tools.json')),
+                "fda_drug_label": str(files('tooluniverse.data').joinpath('fda_drug_labeling_tools.json')),
+                "special_tools": str(files('tooluniverse.data').joinpath('special_tools.json')),
+                "monarch": str(files('tooluniverse.data').joinpath('monarch_tools.json'))
             }
             
+            logger.info(f"Using tool files at: {tool_files}")
+            
             agent = TxAgent(
                 model_name="mims-harvard/TxAgent-T1-Llama-3.1-8B",
                 rag_model_name="mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B",
-                tool_files_dict=tool_files,  # This is critical!
+                tool_files_dict=tool_files,
                 enable_finish=True,
                 enable_rag=True,
                 enable_summary=False,
@@ -98,8 +101,7 @@ def create_demo():
         
         chatbot = gr.Chatbot(
             label="Conversation",
-            height=600,
-            bubble_full_width=False
+            height=600
         )
         
         msg = gr.Textbox(