Update app.py

app.py

@@ -10,20 +10,20 @@ 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:
-    '''
-    A simple calculator that performs basic arithmetic operations.
+    """
+    Performs basic arithmetic operations (addition, subtraction, multiplication, and division).
 
     Args:
-        arg1 (int): The first operand, an integer number.
-        arg2 (str): The operator, which can be '+', '-', '*' or '/'.
-        arg3 (int): The second operand, an integer number.
+        arg1 (int): The first operand.
+        arg2 (str): The operator ('+', '-', '*', '/').
+        arg3 (int): The second operand.
 
     Returns:
         int: The result of the calculation.
 
     Raises:
         ValueError: If an invalid operator is provided.
-    '''
+    """
     if arg2 == "+":
         return arg1 + arg3
     elif arg2 == "-":
@@ -31,9 +31,12 @@ def calculator(arg1: int, arg2: str, arg3: int) -> int:
     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.")
+        raise ValueError("Invalid operator! Only '+', '-', '*', and '/' are allowed.")
+
 
 
 @tool