Update app.py

app.py

@@ -1,9 +1,10 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool, wikipedia
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+import wikipedia
 
 from Gradio_UI import GradioUI
 
@@ -41,7 +42,9 @@ def get_wiki_summary(query1: str) -> str:
     """
     try:
         wikipedia.set_lang("en")
-        summary = get_wiki_summary(query1)
+        search_results = wikipedia.search(query1)
+        page = wikipedia.page(search_results[0])
+        summary = page.summary()
         return summary
     except Exception as e:
         return f"No page found for that topic"