Update app.py

app.py

@@ -9,33 +9,18 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def calculator(arg1: int, arg2: str, arg3: int) -> int:
+def calculator(arg1: int, arg2: int) -> int:
     """
-    Performs basic arithmetic operations (addition, subtraction, multiplication, and division).
+    Performs addition of two numbers.
 
     Args:
         arg1 (int): The first operand.
-        arg2 (str): The operator ('+', '-', '*', '/').
-        arg3 (int): The second operand.
+        arg2 (int): The second operand.
 
     Returns:
-        int: The result of the calculation.
-
-    Raises:
-        ValueError: If an invalid operator is provided.
+        int: The sum of the two numbers.
     """
-    if arg2 == "+":
-        return arg1 + arg3
-    elif arg2 == "-":
-        return arg1 - arg3
-    elif arg2 == "*":
-        return arg1 * arg3
-    elif arg2 == "/":
-        if arg3 == 0:
-            raise ValueError("Division by zero is not allowed.")
-        return arg1 / arg3   
-    else:
-        raise ValueError("Invalid operator! Only '+', '-', '*', and '/' are allowed.")
+    return arg1 + arg2