Update app.py

app.py

@@ -430,7 +430,9 @@ class BasicAgent:
             token = os.environ.get("HF_API_TOKEN")
             self.model = HfApiModel(temperature=0.1, token=token)
         except Exception as e:
-            raise RuntimeError(f"Error initializing model: {e}")
+            print("Error initializing model:")
+            traceback.print_exc()
+            raise
 
         try:
         # Initialize all tool instances
@@ -442,7 +444,9 @@ class BasicAgent:
             self.visit_webpage_tool = VisitWebpageTool() # Ensure this class is defined/imported
             self.final_answer_tool = FinalAnswerTool()
         except Exception as e:
-            raise RuntimeError(f"Error initializing tools: {e}")
+            print("Error initializing model:")
+            traceback.print_exc()
+            raise
 
         
         # Custom tools - ensure these classes are defined and imported
@@ -477,7 +481,9 @@ If the answer is a comma-separated list, apply the above rules for each element
                 description="Runs web searches and analyzes images",
             )
         except Exception as e:
-            raise RuntimeError(f"Error initializing web agent: {e}")
+            print("Error initializing model:")
+            traceback.print_exc()
+            raise
 
         # Create main agent with all capabilities
         try:
@@ -500,13 +506,22 @@ If the answer is a comma-separated list, apply the above rules for each element
                 add_base_tools=True # Consider what this adds, ensure it doesn't duplicate.
             )
         except Exception as e:
-            raise RuntimeError(f"Error initializing main agent: {e}")
+            print("Error initializing model:")
+            traceback.print_exc()
+            raise
 
         # Update system prompt
         # It's generally better to pass the system prompt directly if possible
         # or manage it through prompt templates defined by smolagents.
         # If smolagents adds its own system prompt, this appends to it.
-        self.agent.prompt_templates["system_prompt"] = self.agent.prompt_templates["system_prompt"] + system_prompt_template
+        # self.agent.prompt_templates["system_prompt"] = self.agent.prompt_templates["system_prompt"] + system_prompt_template
+        try:
+            if hasattr(self.agent, "prompt_templates") and "system_prompt" in self.agent.prompt_templates:
+                self.agent.prompt_templates["system_prompt"] += system_prompt_template
+        except Exception as e:
+            print("Error updating prompt:")
+            traceback.print_exc()
+            raise
 
 
     def __call__(self, question: str) -> str: