Update app.py

app.py

@@ -1,26 +1,28 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool,HfApiModel,load_tool,tool
+from smolagents import CodeAgent,HfApiModel,load_tool,tool
+# from smolagents import CodeAgent,DuckDuckGoSearchTool,HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 from tools.visit_webpage import VisitWebpageTool
+from tools.web_search import DuckDuckGoSearchTool
 
 from Gradio_UI import GradioUI
 
-ddg = DuckDuckGoSearchTool(max_results=5)
+# ddg = DuckDuckGoSearchTool(max_results=5)
 
-@tool
-def search_web(query:str)-> str:
-    """A tool that runs a DuckDuckGo search and returns a formatted result.
-    Args:
-        query: A string representing the query to search for.
-    """
-    try:
-        results = ddg(query)
-        return results
-    except Exception as e:
-        return f"Error while searching: {str(e)}"
+# @tool
+# def search_web(query:str)-> str:
+#    """A tool that runs a DuckDuckGo search and returns a formatted result.
+#    Args:
+#        query: A string representing the query to search for.
+#    """
+#    try:
+#        results = ddg(query)
+#        return results
+#    except Exception as e:
+#        return f"Error while searching: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -39,6 +41,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 
 final_answer = FinalAnswerTool()
+web_search = DuckDuckGoSearchTool()
 visit_webpage = VisitWebpageTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -60,7 +63,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, search_web, visit_webpage, image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, web_search, visit_webpage, image_generation_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,