Update app.py

app.py

@@ -23,62 +23,28 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class BasicAgent:
     def __init__(self):
-        self.headers = {
-            'User-Agent': self._get_random_user_agent(),
-            'Accept-Language': 'en-US,en;q=0.5',
-        }
-    
-    def _get_random_user_agent(self) -> str:
-        browsers = [
-            'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
-            'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:89.0) Gecko/20100101 Firefox/89.0',
-            '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)
+        self.headers = {'User-Agent': random.choice([
+            'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
+            'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15'
+        ])}
 
-    def search(self, query: str, num_results: int = 5) -> List[Dict]:
-        encoded_query = urllib.parse.quote_plus(query)
-        url = f"https://www.google.com/search?q={encoded_query}&num={num_results + 2}"
-        
+    def __call__(self, query: str) -> str:
+        """Returns 1-3 word answer when possible"""
         try:
-            response = requests.get(url, headers=self.headers, timeout=10)
-            response.raise_for_status()
-            return self._parse_results(response.text, num_results)
-        except Exception as e:
-            print(f"Search failed: {str(e)}")
-            return []
-
-    def _parse_results(self, html: str, max_results: int) -> List[Dict]:
-        soup = BeautifulSoup(html, 'html.parser')
-        results = []
-        
-        for i, result in enumerate(soup.select('.tF2Cxc, .g')[:max_results]):
-            title = result.select_one('h3, .LC20lb')
-            link = result.find('a')['href']
-            snippet = result.select_one('.IsZvec, .VwiC3b')
+            url = f"https://www.google.com/search?q={requests.utils.quote(query)}"
+            html = requests.get(url, headers=self.headers, timeout=3).text
+            soup = BeautifulSoup(html, 'html.parser')
+            
+            # Try to extract short answer
+            short = (soup.select_one('.LGOjhe, .kno-rdesc span') or 
+                    soup.select_one('.hgKElc') or 
+                    soup.select_one('.Z0LcW'))
+            
+            return short.get_text()[:50].split('.')[0] if short else "No short answer found"
+            
+        except Exception:
+            return "Search failed"
             
-            if title and link:
-                results.append({
-                    'position': i + 1,
-                    'title': title.get_text(),
-                    '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:
-        output = []
-        for res in results:
-            output.append(
-                f"{res['position']}. {res['title']}\n"
-                f"   🔗 {res['link']}\n"
-                f"   📝 {res['snippet'] or 'No description available'}\n"
-            )
-        return "\n".join(output)
-
-    def __call__(self, query: str) -> str:
-        """Added this to make the agent callable"""
-        return self.pretty_print(self.search(query))
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):