Update app.py

app.py

@@ -10,6 +10,47 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition --- 
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+import math
+
+class BasicAgent:
+    def __init__(self):
+        print("✅ BasicAgent initialized.")
+
+    def __call__(self, question: str) -> str:
+        print(f"🤔 Agent received question (first 50 chars): {question[:50]}...")
+
+        q = question.lower().strip()
+
+        # --- Simple math checks ---
+        if "2+2" in q or "two plus two" in q:
+            return "4"
+        if "sqrt" in q:
+            try:
+                # extract number after 'sqrt'
+                num = [int(s) for s in q.split() if s.isdigit()][0]
+                return str(math.isqrt(num))
+            except Exception:
+                return "The square root depends on the number provided."
+
+        # --- Geography checks ---
+        if "capital of france" in q:
+            return "Paris"
+        if "capital of germany" in q:
+            return "Berlin"
+        if "capital of italy" in q:
+            return "Rome"
+        if "capital of spain" in q:
+            return "Madrid"
+
+        # --- History / generic fallback ---
+        if "who wrote" in q:
+            return "It depends on the book or text mentioned."
+        if "who is" in q:
+            return "That depends on the person you are asking about."
+
+        # --- Default fallback ---
+        return f"I'm not fully sure, but here's my guess: {question}"
+
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")