Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,get_tool_schema,tools
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
@@ -6,21 +6,32 @@ import yaml
 import gradio as gr
 from tools.final_answer import FinalAnswerTool
  
-from Gradio_UI import GradioUI
+from smolagents import tool
 
 @tool
-def calculator(a: int, b: int) -> int:
-    """Multiply two intergers.
-        Args:
+def calculator(a: int, b: int, operation: str = "multiply") -> float:
+    """
+    Perform basic arithmetic operations on two numbers.
+
+    Args:
         a (int): First number
         b (int): Second number
+        operation (str): Operation to perform ("add", "subtract", "multiply", "divide")
 
     Returns:
-        int: The product of a and b
+        float: Result of the operation
     """
-    return a * b
-    
-print(get_tool_schema(calculator))
+    if operation == "add":
+        return a + b
+    elif operation == "subtract":
+        return a - b
+    elif operation == "multiply":
+        return a * b
+    elif operation == "divide":
+        return a / b if b != 0 else float("inf")
+    else:
+        raise ValueError(f"Unsupported operation: {operation}")
+        
        
 @tool 
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -65,7 +76,7 @@ from smolagents import load_tool
 tool_registry = {
     "calculator": {
         "tool": calculator,
-        "schema": calculator_tool_schema(),
+        "schema": calculator.tool_schema(),
         "enabled": True
     },
     "get_current_time_in_timezone": {