Update app.py

app.py

@@ -19,6 +19,53 @@ class BasicAgent:
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
+
+
+def calculator_tool(expression: str) -> str:
+    """Evaluates complex mathematical expressions. Input should be a single string expression, e.g., '10 * (2 + 3) / 5'."""
+    print(f"--- TOOL CALL: Calculator with expression: {expression} ---")
+    try:
+        # Use sympify for safe and robust parsing/evaluation of math
+        result = sympify(expression).evalf()
+        return str(result)
+    except Exception as e:
+        return f"ERROR: Could not evaluate expression '{expression}'. Error: {e}"
+
+def search_tool(query: str) -> str:
+    """Performs a web search using Serper.dev (Google results)."""
+    print(f"--- TOOL CALL: Search with query: {query} ---")
+    
+    api_key = os.environ.get("SERPER_API_KEY")
+    if not api_key:
+        return "ERROR: SERPER_API_KEY environment variable not set."
+
+    try:
+        url = "https://google.serper.dev/search"
+        headers = {
+            "X-API-KEY": api_key,
+            "Content-Type": "application/json"
+        }
+        payload = {"q": query}
+
+        resp = requests.post(url, headers=headers, json=payload, timeout=10)
+        resp.raise_for_status()
+        data = resp.json()
+
+        # Extract snippets from top results
+        if "organic" in data:
+            snippets = [item.get("snippet", "") for item in data["organic"][:3]]
+            return " | ".join(snippets) if snippets else "No results found."
+        else:
+            return "No search results found."
+    except Exception as e:
+        return f"ERROR: Search request failed. {e}"
+
+
+# A dictionary to map tool names to the actual functions
+AVAILABLE_TOOLS = {
+    "calculator": calculator_tool,
+    "search": search_tool,
+}
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,