Update app.py

app.py

@@ -20,26 +20,33 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 load_dotenv()
 
 
-from smolagents import Tool
-import wikipediaapi
 
-class WikipediaSearchTool(Tool):
-    name = "wikipedia_search"
-    description = "Fetches a summary of a Wikipedia page based on a given search query (only one word or group of words)."
-    inputs = {
-        "query": {"type": "string", "description": "The search term for the Wikipedia page (only one word or group of words)."}
-    }
-    output_type = "string"
+@tool
+def search_wikipedia(query: str) -> str:
+    """
+    Fetches a summary of a Wikipedia page for a given query.
+    Args:
+        query: The search term to look up on Wikipedia.
+    Returns:
+        str: A summary of the Wikipedia page if successful, or an error message if the request fails.
+    Raises:
+        requests.exceptions.RequestException: If there is an issue with the HTTP request.
+    """
+    url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{query}"
+
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
 
-    def __init__(self, lang="en"):
-        super().__init__()
-        self.wiki = wikipediaapi.Wikipedia(language=lang, user_agent="MinimalAgent/1.0")
+        data = response.json()
+        title = data["title"]
+        extract = data["extract"]
+
+        return f"Summary for {title}: {extract}"
+
+    except requests.exceptions.RequestException as e:
+        return f"Error fetching Wikipedia data: {str(e)}"
 
-    def forward(self, query: str):
-        page = self.wiki.page(query)
-        if not page.exists():
-            return "No Wikipedia page found."
-        return page.summary[:1000]
 
 
 class StringReverseTool(Tool):
@@ -180,7 +187,7 @@ class BasicAgent:
         )
 
         search_tool = DuckDuckGoSearchTool()
-        wiki_search_tool = WikipediaSearchTool()
+        wiki_search_tool = search_wikipedia()
         str_reverse_tool = StringReverseTool()
         keywords_extract_tool = KeywordsExtractorTool()
         speech_to_text_tool = SpeechToTextTool()