Update app.py

app.py

@@ -5,11 +5,10 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 import markdownify
 import re
-from smolagents.tools.web import web_search as web_search_tool  # Import web search from SmolAgents
 
 @tool
 def web_search(query: str) -> str:
-    """Performs a web search and returns the top search results.
+    """Performs a web search using DuckDuckGo and returns the top search results.
     
     Args:
         query: The search query string.
@@ -18,10 +17,25 @@ def web_search(query: str) -> str:
         A formatted string with the top search results, or an error message if the request fails.
     """
     try:
-        results = web_search_tool(query)
-        return results if results else "No relevant results found."
-    except Exception as e:
+        url = f"https://duckduckgo.com/html/?q={query.replace(' ', '+')}"
+        headers = {"User-Agent": "Mozilla/5.0"}
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()
+
+        # Extract search results
+        search_results = re.findall(r'<a rel="nofollow" class="result__url" href="(.*?)">', response.text)
+
+        if not search_results:
+            return "No relevant results found."
+
+        # Format the top results
+        results_str = "\n".join([f"{i+1}. {result}" for i, result in enumerate(search_results[:5])])
+        return f"Top search results:\n{results_str}"
+
+    except RequestException as e:
         return f"Error performing web search: {str(e)}"
+    except Exception as e:
+        return f"An unexpected error occurred: {str(e)}"
 
 @tool
 def visit_webpage(url: str) -> str:
@@ -75,7 +89,7 @@ final_answer = FinalAnswerTool()
 
 # Initialize the model
 model = HfApiModel(
-    max_tokens=1000,  # Adjusted to avoid token limit errors
+    max_tokens=1100,  # Adjusted to avoid token limit errors
     temperature=0.5,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     custom_role_conversions=None,