Update app.py

app.py

@@ -11,6 +11,17 @@ from urllib.parse import quote_plus
 from scripts.text_web_browser import SimpleTextBrowser, SearchInformationTool
 from scripts.cookies import COOKIES
 
+class DuckDuckGoSearchTool(SearchInformationTool):
+    """Search tool that uses DuckDuckGo's HTML interface"""
+    
+    def forward(self, query: str, filter_year: Optional[int] = None) -> str:
+        """Performs search using DuckDuckGo"""
+        encoded_query = quote_plus(query)
+        url = f"https://duckduckgo.com/?t=h_&q={encoded_query}"
+        self.browser.visit_page(url)
+        header, content = self.browser._state()
+        return header.strip() + "\n=======================\n" + content
+
 @tool
 def analyze_content(text: str, analysis_type: str = "general") -> str:
     """Analyzes content for various aspects like key points, themes, or citations
@@ -48,7 +59,7 @@ def analyze_content(text: str, analysis_type: str = "general") -> str:
 
 class ResearchSystem:
     def __init__(self):
-        # Create browser with proper request_kwargs initialization
+        # Initialize browser with request settings
         self.browser = SimpleTextBrowser(
             viewport_size=4096,
             downloads_folder="./downloads",
@@ -69,10 +80,10 @@ class ResearchSystem:
             }
         )
 
-        # Initialize agent with web search tool
+        # Initialize agent with custom DuckDuckGo search tool
         self.researcher = CodeAgent(
             tools=[
-                SearchInformationTool(self.browser),
+                DuckDuckGoSearchTool(self.browser),
                 analyze_content
             ],
             model=self.model
@@ -121,8 +132,8 @@ class ResearchSystem:
         try:
             print(f"\nDEBUG: Processing query: {query}")
             
-            # Use the web search tool
-            search_results = self.researcher.run(f"web_search: {query}")
+            # Get search results using the browser
+            search_results = self.researcher.run(f"Search query: {query}")
             print(f"\nDEBUG: Search completed. Results:\n{search_results}")
             
             # Analyze the results