Update app.py

app.py

@@ -52,7 +52,7 @@ custom_role_conversions=None,
 # @tool
 # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 @tool
-def generate_image_from_text(prompt: str) -> None:
+def generate_image_from_text(prompt: str) -> Image:
     """
     A tool that generates an image based on the given text prompt and displays it.
     Args:
@@ -73,6 +73,8 @@ def generate_image_from_text(prompt: str) -> None:
         display.display(image)
         image.save("generated_image.png")
         print("Image saved as generated_image.png. Open the file to view it.")
+        # Return the image object
+        return image
 
     except Exception as e:
         print(f"Error generating image: {str(e)}")