Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import pandas as pd
 import re
 import sympy as sp
-from wikipedia_api import Wikipedia
+import wikipedia
 from bs4 import BeautifulSoup
 
 # --- Constants ---
@@ -20,9 +20,7 @@ class BasicAgent:
         if not self.api_token:
             raise ValueError("HF_TOKEN environment variable not set.")
         self.headers = {"Authorization": f"Bearer {self.api_token}"}
-        # Initialize Wikipedia API
-        self.wiki = Wikipedia(language='en', user_agent='Mozilla/5.0')
-        print("BasicAgent initialized with Flan-T5-Base, SymPy, Wikipedia API, and DuckDuckGo search.")
+        print("BasicAgent initialized with Flan-T5-Base, SymPy, Wikipedia, and DuckDuckGo search.")
 
     def __call__(self, question: str) -> tuple[str, str]:
         print(f"Question: {question[:50]}...")
@@ -51,30 +49,29 @@ class BasicAgent:
 
         # Step 2: Try Wikipedia for factual questions
         try:
+            wikipedia.set_lang("en")
             key_terms = " ".join(question.split()[:5]).strip("?")
-            wiki_page = self.wiki.page(key_terms)
-            if wiki_page.exists():
-                wiki_summary = wiki_page.summary[:500]  # Limit to 500 chars
-                prompt = (
-                    f"Question: {question}\n"
-                    f"Context: {wiki_summary}\n"
-                    "Provide a concise answer (e.g., a number or short phrase) based on the context."
-                )
-                wiki_answer = self._query_llm(prompt)
-                answer_match = re.search(r"Answer: (.*?)(?:\n|$)", wiki_answer, re.DOTALL)
-                if answer_match:
-                    concise_answer = answer_match.group(1).strip()
-                    wiki_reasoning = wiki_answer if wiki_answer else "No reasoning provided."
-                else:
-                    concise_answer = self._extract_concise_answer(wiki_answer)
-                    wiki_reasoning = wiki_answer
-                reasoning.append(f"Wikipedia API: Searched '{key_terms}'. Summary: {wiki_summary[:100]}... Answer: {concise_answer}")
-                print(f"Returning concise answer (Wikipedia): {concise_answer}")
-                return concise_answer, "\n".join(reasoning)
+            wiki_summary = wikipedia.summary(key_terms, sentences=2, auto_suggest=True, parser_features="html.parser")
+            prompt = (
+                f"Question: {question}\n"
+                f"Context: {wiki_summary}\n"
+                "Provide a concise answer (e.g., a number or short phrase) based on the context."
+            )
+            wiki_answer = self._query_llm(prompt)
+            answer_match = re.search(r"Answer: (.*?)(?:\n|$)", wiki_answer, re.DOTALL)
+            if answer_match:
+                concise_answer = answer_match.group(1).strip()
+                wiki_reasoning = wiki_answer if wiki_answer else "No reasoning provided."
             else:
-                reasoning.append(f"Wikipedia API: No page found for '{key_terms}'.")
+                concise_answer = self._extract_concise_answer(wiki_answer)
+                wiki_reasoning = wiki_answer
+            reasoning.append(f"Wikipedia: Searched '{key_terms}'. Summary: {wiki_summary[:100]}... Answer: {concise_answer}")
+            print(f"Returning concise answer (Wikipedia): {concise_answer}")
+            return concise_answer, "\n".join(reasoning)
+        except wikipedia.exceptions.DisambiguationError:
+            reasoning.append("Wikipedia: Disambiguation error, multiple results found.")
         except Exception as e:
-            reasoning.append(f"Wikipedia API failed: {e}")
+            reasoning.append(f"Wikipedia failed: {e}")
 
         # Step 3: Try web search with DuckDuckGo
         try: