Update app.py

app.py

@@ -426,12 +426,13 @@ class VideoTranscriptionTool(Tool):
 
 class BasicAgent:
     def __init__(self):
-        token = os.environ.get("HF_API_TOKEN")
-        self.model = HfApiModel( # Store model as self.model if you need to access it later
-            temperature=0.1,
-            token=token
-        )
+        try:
+            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}")
 
+        try:
         # Initialize all tool instances
         self.search_tool = DuckDuckGoSearchTool()
         self.wiki_search_tool = WikiSearchTool() # Ensure this class is defined/imported
@@ -440,13 +441,17 @@ class BasicAgent:
         self.speech_to_text_tool = SpeechToTextTool() # Ensure this class is defined/imported
         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}")
 
+        
         # Custom tools - ensure these classes are defined and imported
         self.video_transcription_tool = VideoTranscriptionTool()
         self.image_analysis_tool = ImageAnalysisTool() # Renamed for clarity
         self.analyse_attachment_tool = AnalyseAttachmentTool() # Renamed for clarity
         self.code_llama_tool = CodeLlamaTool() # Ensure this class is defined/imported
         self.parse_excel_to_json_tool = ParseExcelToJsonTool()
+        
 
         
         system_prompt_template = """
@@ -460,6 +465,7 @@ If the answer is a comma-separated list, apply the above rules for each element
 """
 
         # Create web agent with image analysis capability
+        try:
         self.web_agent = ToolCallingAgent(
             tools=[
                 self.search_tool, # Use the initialized DuckDuckGoSearchTool instance
@@ -471,8 +477,11 @@ If the answer is a comma-separated list, apply the above rules for each element
             name="web_search_agent",
             description="Runs web searches and analyzes images",
         )
+        except Exception as e:
+            raise RuntimeError(f"Error initializing web agent: {e}")
 
         # Create main agent with all capabilities
+        try:
         self.agent = CodeAgent(
             model=self.model, # Use self.model
             tools=[
@@ -491,6 +500,8 @@ 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 web agent: {e}")
 
         # Update system prompt
         # It's generally better to pass the system prompt directly if possible