Updated BasicAgent logic

app.py

@@ -13,11 +13,41 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
+
+    def search_wikipedia(self, query):
+        try:
+            url = "https://en.wikipedia.org/api/rest_v1/page/summary/" + query.replace(" ", "_")
+            response = requests.get(url, timeout=10)
+
+            if response.status_code == 200:
+                data = response.json()
+                return data.get("extract", "")
+
+            return ""
+
+        except Exception as e:
+            print(f"Search error: {e}")
+            return ""
+
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+
+        print(f"Question: {question}")
+
+        try:
+            context = self.search_wikipedia(question)
+
+            if context:
+                answer = context[:300]
+            else:
+                answer = "Could not determine the answer."
+
+            print(f"Answer: {answer}")
+
+            return answer
+
+        except Exception as e:
+            print(f"Agent error: {e}")
+            return f"Error: {str(e)}"
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """

requirements.txt

@@ -1,2 +1,3 @@
 gradio
-requests
+requests
+pandas