Update app.py

app.py

@@ -10,38 +10,6 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
-@tool
-def web_search_tool(query: str, num_results: int = 3) -> str:
-    """A tool that performs a web search and fetches up-to-date information using DuckDuckGo.
-    
-    Args:
-        query: The search query or topic to look up on the web.
-        num_results: The number of top search results to return (default is 3).
-    
-    Returns:
-        A string containing summarized up-to-date information from the web.
-    """
-    try:
-        with DDGS() as ddg:
-            results = list(ddg.text(query, max_results=num_results))
-
-        if not results:
-            return f"No results found for '{query}'."
-
-        info_list = []
-        for i, r in enumerate(results):
-            title = r.get("title") or "No title"
-            body = r.get("body") or "No summary available."
-            link = r.get("href") or r.get("url", "")
-            info_list.append(f"{i+1}. {title}\n{body}\n({link})")
-
-        return f"Here are the top {num_results} results for '{query}':\n\n" + "\n\n".join(info_list)
-
-    except Exception as e:
-        # return the exception text so debugging in the UI is easier
-        return f"Error performing web search for '{query}': {type(e).__name__}: {e}"
-
-
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -78,9 +46,10 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
+duckduckgo_tool = DuckDuckGoSearchTool()
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, web_search_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, duckduckgo_tool], 
     max_steps=6,
     verbosity_level=1,
     grammar=None,