Update app.py

app.py

@@ -29,19 +29,14 @@ class BasicAgent:
         }
     
     def _get_random_user_agent(self) -> str:
-        """Fallback user-agent generator if fake-useragent isn't installed"""
         browsers = [
-            # Chrome
             'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
-            # Firefox
             'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:89.0) Gecko/20100101 Firefox/89.0',
-            # Safari
             'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/14.1.1 Safari/605.1.15'
         ]
         return random.choice(browsers)
 
     def search(self, query: str, num_results: int = 3) -> List[Dict]:
-        """Perform Google search and return structured results"""
         encoded_query = urllib.parse.quote_plus(query)
         url = f"https://www.google.com/search?q={encoded_query}&num={num_results + 2}"
         
@@ -54,11 +49,9 @@ class BasicAgent:
             return []
 
     def _parse_results(self, html: str, max_results: int) -> List[Dict]:
-        """Parse HTML and extract search results"""
         soup = BeautifulSoup(html, 'html.parser')
         results = []
         
-        # Current Google result selectors (July 2024)
         for i, result in enumerate(soup.select('.tF2Cxc, .g')[:max_results]):
             title = result.select_one('h3, .LC20lb')
             link = result.find('a')['href']
@@ -71,11 +64,9 @@ class BasicAgent:
                     'link': link if link.startswith('http') else f"https://www.google.com{link}",
                     'snippet': snippet.get_text() if snippet else None
                 })
-        
         return results
 
     def pretty_print(self, results: List[Dict]) -> str:
-        """Format results for human-readable output"""
         output = []
         for res in results:
             output.append(
@@ -85,21 +76,20 @@ class BasicAgent:
             )
         return "\n".join(output)
 
+    def __call__(self, query: str) -> str:
+        """Added this to make the agent callable"""
+        return self.pretty_print(self.search(query))
+
 if __name__ == "__main__":
-    scraper = BasicAgent()
-    
-    # Example search
-    query = "Python programming language"
-    print(f"🔍 Searching Google for: '{query}'")
+    agent = BasicAgent()
     
-    results = scraper.search(query, num_results=3)
+    # Now you can use either approach:
+    # Option 1: Call as function
+    print(agent("Python programming language"))
     
-    if results:
-        print("\nTop Results:")
-        print(scraper.pretty_print(results))
-    else:
-        print("❌ No results found or search failed")
-
+    # Option 2: Use methods directly
+    results = agent.search("Python programming language")
+    print(agent.pretty_print(results))
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """