Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 from langchain_google_genai import ChatGoogleGenerativeAI
-from tools import InternetSearchTool, WeatherInfoTool
+from tools import InternetSearchTool, WebscraperTool, ExcelTool, WikipediaTool, SummarizeTool
 from smolagents import CodeAgent, InferenceClientModel
 
 # (Keep Constants as is)
@@ -19,18 +19,21 @@ class BasicAgent:
         # Initialize the Hugging Face model
         model = ChatGoogleGenerativeAI(model="gemini-1.5-flash-latest")
         
-        # Initialize the web search tool
+        # Initialize tools
         search_tool = InternetSearchTool()
-        
-        # Initialize the weather tool
-        weather_info_tool = WeatherInfoTool()
+        webscraper_tool = WebscraperTool()
+        excel_tool = ExcelTool()
+        python_tool = PythonTool()
+        wikipedia_tool = WikipediaTool()
+        summarize_tool = SummarizeTool()
 
         # Create agent
         self.agent= CodeAgent(
-            tools=[weather_info_tool, search_tool],
+            tools=[search_tool, webscraper_tool, excel_tool, python_tool, wikipedia_tool, summarize_tool],
             model=InferenceClientModel(token=os.getenv("HF_TOKEN"), provider="together"),
             add_base_tools=True,  # Add any additional base tools
-            planning_interval=3   # Enable planning every 3 steps
+            planning_interval=3,   # Enable planning every 3 steps
+            max_steps=10
         )
 
     def __call__(self, question: str) -> str: