Update app.py

app.py

@@ -16,6 +16,14 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
         arg1: the first argument
         arg2: the second argument
     """
+    if arg1 == "square":
+        return f"The square of {arg2} is {arg2 ** 2}"
+    elif arg1 == "double":
+        return f"The double of {arg2} is {arg2 * 2}"
+    elif arg1 == "triple":
+        return f"The triple of {arg2} is {arg2 * 3}"
+    else:
+        return "Unknown operation. Try 'square', 'double', or 'triple'."
     return "What magic will you build ?"
 
 @tool