Update app.py

app.py

@@ -48,26 +48,43 @@ class TxAgentApplication:
     def __init__(self):
         self.agent = None
         self.is_initialized = False
+        self.initialization_error = None
 
     def initialize_agent(self):
         if self.is_initialized:
             return "Model already initialized"
         
         try:
+            # 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']
             )
-            self.agent.init_model()
+            
+            # Initialize model with error handling
+            try:
+                self.agent.init_model()
+            except Exception as e:
+                # Handle specific tool embedding error
+                if "No such file or directory" in str(e) and "tool_embedding" in str(e):
+                    return ("Error: Missing tool embedding file. "
+                           "Please ensure the RAG model files are properly downloaded.")
+                raise
+            
             self.is_initialized = True
+            self.initialization_error = None
             return "TxAgent initialized successfully"
+            
         except Exception as e:
+            self.initialization_error = str(e)
             return f"Initialization failed: {str(e)}"
 
     def chat(self, message, chat_history):
         if not self.is_initialized:
+            if self.initialization_error:
+                return chat_history + [(message, f"System Error: {self.initialization_error}")]
             return chat_history + [(message, "Error: Please initialize the model first")]
         
         try:
@@ -93,7 +110,7 @@ class TxAgentApplication:
             
             return chat_history + [(message, response)]
         except Exception as e:
-            return chat_history + [(message, f"Error: {str(e)}")]
+            return chat_history + [(message, f"Error during processing: {str(e)}")]
 
 # ========== Gradio Interface ==========
 def create_interface():
@@ -107,15 +124,11 @@ 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
         chatbot = gr.Chatbot(
             height=600,
             label="Conversation",
             show_label=True,
-            avatar_images=(
-                None,  # User avatar (None for default)
-                None   # Bot avatar (None for default)
-            ),
             show_copy_button=True
         )
         
@@ -147,15 +160,13 @@ def create_interface():
         msg.submit(
             app.chat,
             [msg, chatbot],
-            chatbot,
-            queue=True  # Enable queue for this event
+            chatbot
         )
         
         submit_btn.click(
             app.chat,
             [msg, chatbot],
-            chatbot,
-            queue=True  # Enable queue for this event
+            chatbot
         ).then(
             lambda: "", None, msg
         )
@@ -164,12 +175,19 @@ def create_interface():
 
 # ========== Main Execution ==========
 if __name__ == "__main__":
+    # Create and configure the interface
     interface = create_interface()
     
-    # Correct launch configuration
-    interface.queue()  # Enable queue globally
-    interface.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True
-    )
+    # Launch configuration
+    launch_params = {
+        'server_name': '0.0.0.0',
+        'server_port': 7860,
+        'share': True
+    }
+    
+    # Enable queue if needed (for production)
+    try:
+        interface.queue().launch(**launch_params)
+    except Exception as e:
+        print(f"Error launching interface: {e}")
+        interface.launch(**launch_params)  # Fallback without queue