Update app.py

app.py

@@ -10,15 +10,101 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+import requests
+import re
+import ast
+import operator
+
+# --- Safe math evaluation ---
+def safe_eval(expr: str):
+    """
+    Safely evaluate a math expression containing +, -, *, /, (), decimals.
+    """
+    allowed_operators = {
+        ast.Add: operator.add,
+        ast.Sub: operator.sub,
+        ast.Mult: operator.mul,
+        ast.Div: operator.truediv,
+        ast.USub: operator.neg,
+    }
+
+    def _eval(node):
+        if isinstance(node, ast.Num):  # <number>
+            return node.n
+        elif isinstance(node, ast.BinOp):
+            return allowed_operators[type(node.op)](_eval(node.left), _eval(node.right))
+        elif isinstance(node, ast.UnaryOp):
+            return allowed_operators[type(node.op)](_eval(node.operand))
+        else:
+            raise ValueError(f"Unsupported expression: {expr}")
+
+    node = ast.parse(expr, mode='eval').body
+    return _eval(node)
+
+
+# --- Enhanced Basic Agent ---
 class BasicAgent:
+    COUNTRY_CAPITALS = {
+        "france": "Paris",
+        "morocco": "Rabat",
+        "usa": "Washington, D.C.",
+        "germany": "Berlin",
+        "italy": "Rome",
+        "spain": "Madrid",
+        "japan": "Tokyo",
+        "china": "Beijing",
+        "egypt": "Cairo",
+        # add more if needed
+    }
+
     def __init__(self):
-        print("BasicAgent initialized.")
+        print("Enhanced BasicAgent initialized.")
+
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        q = question.lower().strip()
+
+        # 1️⃣ Math
+        if any(op in q for op in ["+", "-", "*", "/"]):
+            numbers_expr = re.findall(r'[\d\.\+\-\*\/\(\) ]+', q)
+            if numbers_expr:
+                try:
+                    result = safe_eval(numbers_expr[0])
+                    return f"Final Answer: {result}"
+                except:
+                    pass
+
+        # 2️⃣ Capital questions
+        if "capital of" in q:
+            country = q.split("capital of")[-1].strip()
+            capital = self.COUNTRY_CAPITALS.get(country.lower())
+            if capital:
+                return f"Final Answer: The capital of {country.capitalize()} is {capital}."
+            else:
+                return f"Final Answer: The capital of {country.capitalize()} is unknown."
+
+        # 3️⃣ Simple count / length
+        if "how many" in q:
+            numbers = re.findall(r'\d+', q)
+            if numbers:
+                return f"Final Answer: {numbers[0]}"
+
+        # 4️⃣ Wikipedia lookup
+        try:
+            query = q.replace("who is", "").replace("what is", "").replace("?", "").strip().replace(" ", "_")
+            url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{query}"
+            res = requests.get(url, timeout=5).json()
+            if "extract" in res:
+                text = res["extract"]
+                # Limit length to 300 chars to avoid super long answers
+                if len(text) > 300:
+                    text = text[:300] + "..."
+                return f"Final Answer: {text}"
+        except:
+            pass
 
+        # 5️⃣ Fallback
+        return "Final Answer: I don't know"
+        
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,