add even more tools

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import pandas as pd
 from smolagents import DuckDuckGoSearchTool, CodeAgent, VisitWebpageTool, FinalAnswerTool, OpenAIServerModel, \
-    WikipediaSearchTool
+    WikipediaSearchTool, PythonInterpreterTool
 from dotenv import load_dotenv
 # (Keep Constants as is)
 # --- Constants ---
@@ -15,15 +15,22 @@ load_dotenv()
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        search_tool = DuckDuckGoSearchTool()
-        model = OpenAIServerModel(model_id="gpt-4o", api_key=os.environ["OPENAI_API_KEY"])
-        wiki= WikipediaSearchTool()
+
         # model = HfApiModel()
-        visit_webpage = VisitWebpageTool()
-        final_answer = FinalAnswerTool()
+        model = OpenAIServerModel(model_id="gpt-4o", api_key=os.environ["OPENAI_API_KEY"])
+
+        # tools
+        search_tool = DuckDuckGoSearchTool()
+        visit_webpage_tool = VisitWebpageTool()
+        wiki_tool = WikipediaSearchTool()
+        code_tool = PythonInterpreterTool()
+        final_answer_tool = FinalAnswerTool()
+
         self.agent = CodeAgent(
             model=model,
-            tools=[search_tool, visit_webpage, wiki, final_answer],
+            tools=[search_tool, visit_webpage_tool, wiki_tool, code_tool, final_answer_tool],
+            add_base_tools=True, # probably redundant
+            additional_authorized_imports = ['requests', 'bs4'],
             managed_agents=[],
             max_steps=10,
             verbosity_level=2,