Update app.py

app.py

@@ -10,14 +10,79 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+from smolagent import Agent, Tool
+from duckduckgo_search import ddg
+import sympy
+import re
+
+class WikipediaSearchTool(Tool):
+    name = "wikipedia_search"
+    description = "Searches for answers using Wikipedia"
+
+    def use(self, input: str) -> str:
+        results = ddg(input, max_results=3)
+        if results:
+            return results[0]["body"]
+        return "No information found."
+
+class MathSolverTool(Tool):
+    name = "math_solver"
+    description = "Solves math expressions and equations"
+
+    def use(self, input: str) -> str:
+        try:
+            expression = sympy.sympify(input)
+            result = expression.evalf()
+            return str(result)
+        except Exception as e:
+            return f"Math Error: {e}"
+
+class CodeExecutionTool(Tool):
+    name = "python_executor"
+    description = "Executes Python code snippets and returns output"
+
+    def use(self, input: str) -> str:
+        try:
+            local_vars = {}
+            exec(f"result = {input}", {}, local_vars)
+            return str(local_vars["result"])
+        except Exception as e:
+            return f"Code Error: {e}"
+
+def classify_question(q: str) -> str:
+    q_lower = q.lower()
+    if any(keyword in q_lower for keyword in ["solve", "evaluate", "calculate", "math"]):
+        return "math"
+    elif any(keyword in q_lower for keyword in ["code", "python", "return", "function"]):
+        return "code"
+    elif any(keyword in q_lower for keyword in ["who", "what", "where", "when", "why", "how", "capital", "invented"]):
+        return "search"
+    else:
+        return "default"
+
 class BasicAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
+        self.agent = Agent(
+            tools=[
+                WikipediaSearchTool(),
+                MathSolverTool(),
+                CodeExecutionTool(),
+            ]
+        )
+
     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
+        category = classify_question(question)
+        print(f"Routing question: {question[:50]}... -> [{category}]")
+
+        if category == "math":
+            return MathSolverTool().use(question)
+        elif category == "code":
+            return CodeExecutionTool().use(question)
+        elif category == "search":
+            return WikipediaSearchTool().use(question)
+        else:
+            return self.agent.run(question)
+
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """