Update app.py

app.py

@@ -38,14 +38,12 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def generate_image(args):
+def generate_image(prompt:str)-> dict:
     """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.
+        prompt: The text description used to generate the image.
     """
     try:
-        prompt = args["prompt"]
         image = image_generation_tool(prompt)
         return {"image":image}
     except Exception as e: