Update app.py

app.py

@@ -55,10 +55,11 @@ def generate_image(prompt: str) -> str:
         prompt: A description of the image to generate.
     """
     try:
-        return image_generation_tool.run(prompt)
+        return image_generation_tool.invoke(prompt)  # Change run() to invoke()
     except Exception as e:
         return f"Error generating image: {str(e)}"
 
+
 # Timezone Tool (already in your code)
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: