Update app.py

app.py

@@ -19,18 +19,63 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 
 @tool
-def basic_search(q:str)-> str:
-    """A tool that takes in a query and returns the output from a DuckDuckGo search.
+def extended_calculator(operation: str) -> str:
+    """An extended calculator that safely evaluates math expressions.
+    
+    Supports:
+      - Basic operations: +, -, *, /, and parentheses.
+      - Exponentiation: using either '^' or '**'.
+      - Math functions: sqrt, sin, cos, tan, log, exp.
+      
     Args:
-        q: User query
+        operation: A string mathematical expression (e.g., "2 + 2", "sqrt(16)", "2^3").
+    
+    Returns:
+        str: The result of the calculation or an error message.
     """
-
     try:
-        search_tool = DuckDuckGoSearchTool()
-        results = search_tool(q)
+        # Replace caret '^' with Python's exponentiation operator '**'
+        operation = operation.replace("^", "**")
+        
+        # Define allowed names for math functions and constants.
+        safe_env = {
+            "sqrt": math.sqrt,
+            "sin": math.sin,
+            "cos": math.cos,
+            "tan": math.tan,
+            "log": math.log,   # natural logarithm by default
+            "exp": math.exp,
+            "pi": math.pi,
+            "e": math.e,
+            "abs": abs,
+            "pow": pow,
+        }
+        
+        # Use a regex to allow only safe characters (digits, letters, operators, parentheses, commas, and periods)
+        # This helps reject inputs with disallowed symbols.
+        if not re.match(r'^[0-9a-zA-Z\s+\-*/().,]+$', operation):
+            return "Error: Expression contains invalid characters."
+        
+        # Evaluate the expression using the restricted environment.
+        result = eval(operation, {"__builtins__": None}, safe_env)
+        return f"Result: {result}"
+    
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Error calculating {operation}: {str(e)}"
 
+# Example usage:
+if __name__ == "__main__":
+    expressions = [
+        "2 + 2",
+        "3 * (4 + 5)",
+        "2^3",           # using '^' for exponentiation
+        "sqrt(16)",
+        "sin(pi/2)",
+        "log(e)"
+    ]
+    
+    for expr in expressions:
+        print(f"{expr} = {extended_calculator(expr)}")
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.