Update app.py

Trying to get generate_image working

app.py

@@ -52,14 +52,14 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def generate_image(description: str) -> Image:
+def generate_image(description: str) -> str:
     """A tool that generates an image from a text description.
     Args:
         description: A string representing the text description to generate an image of.
     """
     image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
     image = image_generation_tool(description)
-    return image
+    return f"Image generated for prompt: {description}."