Update app.py

app.py

@@ -3,14 +3,13 @@ from dotenv import load_dotenv
 import gradio as gr
 import requests
 
-from typing import List, Dict, Union
-import pandas as pd
-import wikipediaapi
+import os
+import inspect
+import gradio as gr
 import requests
-from bs4 import BeautifulSoup
-import urllib.parse
-import random
-from typing import List, Dict
+import pandas as pd
+from langchain_core.messages import HumanMessage
+from agent import build_graph
 
 load_dotenv()
 
@@ -23,66 +22,15 @@ 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)
-
-    def search(self, query: str, num_results: int = 3) -> List[Dict]:
-        encoded_query = urllib.parse.quote_plus(query)
-        url = f"https://www.google.com/search?q={encoded_query}&num={num_results + 2}"
-        
-        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')
-            
-            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))
-
-
+        print("BasicAgent initialized.")
+        self.graph = build_graph()
+
+    def __call__(self, question: str) -> str:
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        messages = [HumanMessage(content=question)]
+        result = self.graph.invoke({"messages": messages})
+        answer = result['messages'][-1].content
+        return answer
     
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):