Update app.py

Debugging to get to compile

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool,tool
+from smolagents import DuckDuckGoSearchTool, VisitWebpageTool, HfApiModel
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -17,13 +17,22 @@ model = HfApiModel(
     "Qwen/Qwen2.5-Coder-32B-Instruct", max_tokens=8096
 )
 
-## The web_agent we will work with - they can search the web and visit a website
+statement = f"""
+      You are a general AI assistant. 
+      I will ask you a question. 
+      Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. 
+      YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. 
+      If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 
+      If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. 
+      If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+            
+      {question}
+      """
 web_agent = CodeAgent(
     model=model,
     tools=[
         DuckDuckGoSearchTool(),
         VisitWebpageTool(),
-        final_answer,
     ],
     name="web_agent",
     description="Browses the web to find information by using search or by visiting a specific website",
@@ -32,28 +41,34 @@ web_agent = CodeAgent(
     prompt_templates=None
 )
 
+manager_agent = CodeAgent(
+    model=model,
+    tools=[],
+    managed_agents=[web_agent],
+    additional_authorized_imports=[
+        "geopandas",
+        "plotly",
+        "shapely",
+        "json",
+        "pandas",
+        "numpy",
+    ],
+    planning_interval=5,
+    verbosity_level=2,
+    #final_answer_checks=[check_reasoning_and_plot],
+    max_steps=15,
+)
+
 class BasicAgent:
 
     def __init__(self):
         print("BasicAgent initialized.")
-        self.agent = web_agent()      
+        self.agent = manager_agent()      
     
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         fixed_answer = "This is a default answer."
-
-        statement = f"""
-            You are a general AI assistant. 
-            I will ask you a question. 
-            Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. 
-            YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. 
-            If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 
-            If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. 
-            If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
-            
-            {question}
-            """
-        
+        print(statement)       
         agent_answer = self.agent.run(statement)
         print(f"Agent returning answer: {agent_answer}")
         return agent_answer