Update app.py

app.py

@@ -1,5 +1,5 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
-from duckduckgo_search import ddg
+from duckduckgo_search import DDGS
 from typing import List
 import time
 import requests
@@ -13,6 +13,7 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
+
 @tool
 def search_internet(query: str, max_results: int = 3) -> str:
     """A tool that searches the internet for information about a given query.
@@ -22,24 +23,28 @@ def search_internet(query: str, max_results: int = 3) -> str:
         max_results: Maximum number of results to return (default: 3)
     """
     try:
-        # Perform the search
-        results = ddg(query, max_results=max_results)
-        
-        if not results:
-            return f"No results found for '{query}'"
-        
-        # Format the results
-        formatted_results = f"Search results for '{query}':\n\n"
-        
-        for i, result in enumerate(results, 1):
-            formatted_results += f"{i}. {result['title']}\n"
-            formatted_results += f"   {result['link']}\n"
-            formatted_results += f"   {result['body']}\n\n"
+        # Initialize the DuckDuckGo search
+        with DDGS() as ddgs:
+            # Perform the search
+            results = list(ddgs.text(query, max_results=max_results))
+            
+            if not results:
+                return f"No results found for '{query}'"
+            
+            # Format the results
+            formatted_results = f"Search results for '{query}':\n\n"
             
-        return formatted_results
+            for i, result in enumerate(results, 1):
+                formatted_results += f"{i}. {result['title']}\n"
+                formatted_results += f"   {result['link']}\n"
+                formatted_results += f"   {result['body']}\n\n"
+                
+            return formatted_results
     
     except Exception as e:
         return f"Error performing search: {str(e)}"
+
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.