Update tools logic in app.py

app.py

@@ -13,8 +13,9 @@ def calculator(args:dict)-> dict: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that does nothing yet 
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        args: A dictionary containing:
+            - arg1: the first argument
+            - arg2: the second argument
     """
     arg1=args["arg1"]
     arg2=args["arg2"]
@@ -38,6 +39,11 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 @tool
 def generate_image(args):
+    """A tool that generate the image based on text prompt.
+    Args:
+        args: A dictionary containing:
+            - prompt(str): The text description used to generate the image.
+    """
     try:
         prompt = args["prompt"]
         image = image_generation_tool(prompt)