Update app.py

app.py

@@ -9,6 +9,7 @@ from duckduckgo_search import DDGS
 from transformers import pipeline
 from sklearn.metrics.pairwise import cosine_similarity
 import numpy as np
+import wikipedia
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -46,6 +47,16 @@ class BasicAgent:
         return best_answer
 
     def search(self, question: str) -> str:
+         # Try Wikipedia first for reliable context
+        try:
+            wiki_titles = wikipedia.search(question)
+            if wiki_titles:
+                page = wikipedia.page(wiki_titles[0])
+                wiki_content = page.content[:4000]  # Truncate to 4000 chars for the QA model
+                result = self.search_pipeline(question=question, context=wiki_content)
+                return result["answer"]
+        except Exception as e:
+            print(f"Wikipedia lookup failed: {e}")
         try:
             with DDGS() as ddgs:
                 results = list(ddgs.text(question, max_results=3))  # Fetch top 3 results