Update app.py

app.py

@@ -8,6 +8,40 @@ import pandas as pd
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
+import requests
+import re
+from math import sqrt
+
+class MathTool:
+    def __call__(self, question: str) -> str:
+        try:
+            allowed = {"sqrt": sqrt}
+            result = eval(question, {"__builtins__": None}, allowed)
+            return str(result)
+        except:
+            return "I can't compute this."
+
+class WikipediaTool:
+    def __call__(self, query: str) -> str:
+        try:
+            # Step 1: Use Wikipedia search
+            search_url = f"https://en.wikipedia.org/w/rest.php/v1/search/page?q={requests.utils.quote(query)}&limit=1"
+            res = requests.get(search_url, timeout=5).json()
+            if not res.get("pages"):
+                return "N/A"
+
+            title = res["pages"][0]["title"]
+            summary_url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{requests.utils.quote(title)}"
+            summary = requests.get(summary_url, timeout=5).json()
+            extract = summary.get("extract", "")
+
+            # Step 2: Extract short fact — e.g. first date, number, or name
+            match = re.search(r'\b([A-Z][a-z]+(?:\s[A-Z][a-z]+)?)\b', extract)
+            return match.group(1) if match else extract.split(".")[0]
+
+        except Exception:
+            return "N/A"
+
 class BasicAgent:
     def __init__(self):
         self.math_tool = MathTool()