Update app.py

app.py

@@ -3,23 +3,27 @@ import datetime
 import requests
 import pytz
 import yaml
-import wikipedia
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def wikipedia_search_tool(query: str)-> str: 
-    """A tool for searching Wikipedia to check topics where you
-    are unsure of the correct answer
-    Args:
-        query: the query to search on Wikipedia
+def wikipedia_page_tool(page: str)-> str: 
+    """
+        Retrieve a Wikipedia page. Useful for learning about a particular concept that isn't private information.
+
+        Args:
+            page (str): Title of the page to read.
     """
-    pages = wikipedia.search(query)
-    if len(pages) == 0:
-        return "No search results."
-    return self.load_data(pages[0], lang)
+    import wikipedia
+
+    try:
+        wikipedia_page = wikipedia.page(page, **load_kwargs, auto_suggest=False)
+    except wikipedia.PageError:
+        return "Unable to load page."
+        
+    return wikipedia_page.content
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +59,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, wikipedia_search_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, wikipedia_page_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,