Update app.py

app.py

@@ -19,6 +19,29 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 #Load environment variables
 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"
+
+    def __init__(self, lang="en"):
+        super().__init__()
+        self.wiki = wikipediaapi.Wikipedia(language=lang, user_agent="MinimalAgent/1.0")
+
+    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):
     name = "reverse_message"
     description = "When you received a strange text, try to reverse it and perform action described in reversed message."
@@ -110,6 +133,8 @@ def parse_excel_to_json(task_id: str) -> dict:
             "status": f"Error in parsing Excel file: {str(e)}"
         }
 
+
+
 class VideoTranscriptionTool(Tool):
     """Fetch transcripts from YouTube videos"""
     name = "transcript_video"