Update app.py

app.py

@@ -12,6 +12,9 @@ from tools.visit_webpage import VisitWebpageTool
 
 from Gradio_UI import GradioUI
 
+from smolagents import WebSearchTool
+search_tool = WebSearchTool()
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -131,7 +134,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,  my_custom_tool, get_current_time_in_timezone, web_search, visit_webpage, 
+    tools=[final_answer,  my_custom_tool, get_current_time_in_timezone, web_search, visit_webpage, search_tool,
            python_tool, image_generation_tool, image_generation_tool2,
           suggest_menu, catering_service_tool, SuperheroPartyThemeTool()], ## add your tools here (don't remove final answer)
     max_steps=10,