Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from PIL import Image
 
 from Gradio_UI import GradioUI
 
@@ -51,21 +52,31 @@ 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) -> str:
-    """Generates an image based on the given text prompt.
+def generate_image_from_text(prompt: str):
+    """Generates and displays an image based on the given text prompt.
     
     Args:
         prompt: A string describing the image to be generated.
     
     Returns:
-        A string indicating the success or failure of the image generation process.
+        Displays the generated image.
     """
     try:
+        # Load the image generation tool
         image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+        
+        # Generate the image and get the URL (if applicable)
         image_url = image_generation_tool(prompt)
-        return f"Image successfully generated: {image_url}"
+        
+        # Fetch the image from the URL
+        response = requests.get(image_url)
+        image = Image.open(io.BytesIO(response.content))
+        
+        # Display the image
+        image.show()
+        
     except Exception as e:
-        return f"Error generating image: {str(e)}"
+        print(f"Error generating image: {str(e)}")
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)