Update app.py

app.py

@@ -16,8 +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
     """
-    return "What magic will you build ?"
-
+    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'."
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.