Update app.py

app.py

@@ -1,36 +1,17 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, load_tool, tool
 import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from ddgs import DDGS
+
 
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
-ddg_tool_instance = DDGS()
-
-@tool
-def web_search(query: str, max_results: int = 3) -> str:
-    """
-    Search the web for up-to-date information using DuckDuckGo via ddgs.
+ddg_tool = DuckDuckGoSearchTool()
 
-    Args:
-        query: The search query
-        max_results: Number of top results to return
 
-    Returns:
-        Top search results as a string with title + URL
-    """
-    try:
-        results = ddg_tool_instance.text(query, max_results=max_results)
-        if not results:
-            return "No results found! Try a less restrictive query."
-        top_results = results[:max_results]
-        return "\n".join([f"{r['title']}: {r['href']}" for r in top_results])
-    except Exception as e:
-        return f"Error during web search: {str(e)}"
         
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -73,7 +54,7 @@ agent = CodeAgent(
     tools=[
             final_answer, 
             get_current_time_in_timezone,
-            web_search 
+            ddg_tool 
     ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,