Update app.py

app.py

@@ -19,25 +19,25 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 
 @tool
-def time_converter(time_str: str) -> str:
-    """A tool that converts a time string from 12-hour format (with AM/PM) to 24-hour format.
+@tool
+def calculator(operation: str) -> str:
+    """A simple calculator tool that safely evaluates basic math expressions.
     
     Args:
-        time_str: A time string in 12-hour format (e.g., "02:30 PM").
+        operation: The mathematical expression to evaluate (e.g., "2 + 2", "5 * 3").
+                   Supports basic operations (+, -, *, /) and parentheses.
     
     Returns:
-        str: The time in 24-hour format (e.g., "14:30") or an error message.
+        str: The result of the calculation or an error message.
     """
-    from datetime import datetime
     try:
-        dt = datetime.strptime(time_str, "%I:%M %p")
-        result = dt.strftime("%H:%M")
-        print(f"Converted time: {result}")  # For debugging/logging
-        return result
+        allowed_chars = set("0123456789+-*/ .()")
+        if not all(c in allowed_chars for c in operation):
+            return "Error: Only basic math operations allowed"
+        result = eval(operation, {"__builtins__": {}})
+        return f"Result: {result}"
     except Exception as e:
-        error_msg = f"Error converting time: {str(e)}"
-        print(error_msg)  # For debugging/logging
-        return error_msg
+        return f"Error calculating {operation}: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: