Update app.py

app.py

@@ -6,7 +6,7 @@ import pandas as pd
 
 # Import our custom tools from their modules
 from huggingface_hub import login
-from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel
+from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel, GoogleSearchTool
 from tools import web_search, visit_webpage, final_answer
 # from tools import web_search, visit_webpage, final_answer, go_back, close_popups, search_item_ctrl_f
 from retriever import load_guest_dataset
@@ -29,7 +29,7 @@ class BasicAgent:
         login(token=api_key)
         
         # Create the search tool
-        self.web_search = web_search
+        # self.web_search = web_search
 
         self.visit_webpage = visit_webpage
 
@@ -51,7 +51,7 @@ class BasicAgent:
         
         # Create the agent with tools
         self.agent = CodeAgent(
-            tools=[self.web_search, self.visit_webpage, self.final_answer],
+            tools=[GoogleSearchTool(), self.visit_webpage, self.final_answer],
             # tools=[self.web_search, self.visit_webpage, self.final_answer, go_back, close_popups, search_item_ctrl_f],
             model=self.model,
             additional_authorized_imports=["pandas", "openpyxl", "yt_dlp", "requests", "io", "json", "whisper", "bs4"],