Update app.py

app.py

@@ -6,7 +6,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
-import wikipedia
+import wikipediaapi
 
 @tool
 def wikipedia_summary(topic: str) -> str:
@@ -20,13 +20,14 @@ def wikipedia_summary(topic: str) -> str:
     Returns:
         str: A summary of the Wikipedia article.
     """
-    try:
-        summary = wikipedia.summary(topic, sentences=3)
-        return summary
-    except wikipedia.exceptions.DisambiguationError as e:
-        return f"Topic is ambiguous. Did you mean: {', '.join(e.options[:5])}?"
-    except wikipedia.exceptions.PageError:
-        return "No matching Wikipedia page found."
+    wiki = wikipediaapi.Wikipedia(language="en", user_agent="MyWikipediaBot/1.0")
+    page = wiki.page(topic)
+    
+    if page.exists():
+        page_content = page.summary
+        return page_content
+    else:
+        return "Page not found"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: