Update app.py

app.py

@@ -10,35 +10,36 @@ 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.
+@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).
+    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:
-#         ddg = DuckDuckGoSearchTool()
-#         results = ddg.text(query, max_results=num_results)
+    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}'."
+        if not results:
+            return f"No results found for '{query}'."
 
-#         info_list = []
-#         for i, r in enumerate(results):
-#             title = r.get("title", "No title")
-#             body = r.get("body", "No summary available.")
-#             link = r.get("href", "")
-#             info_list.append(f"{i+1}. {title}\n{body}\n({link})")
+        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)
+        return f"Here are the top {num_results} results for '{query}':\n\n" + "\n\n".join(info_list)
 
-#     except Exception as e:
-#         return f"Error performing web search for '{query}': {str(e)}"
+    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}"
 
 
 
@@ -74,14 +75,12 @@ custom_role_conversions=None,
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-web_agent = DuckDuckGoSearchTool()
-
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, web_agent], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, web_search_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,