Update app.py

app.py

@@ -52,32 +52,22 @@ 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:
-    """Generates and displays an image based on the given text prompt.
-    
-    Args:
-        prompt: A string describing the image to be generated.
-    
-    Returns:
-        Displays the generated image.
-    """
+def generate_image_from_text(prompt: str) -> None:
+    """Generates and displays an image based on the given text prompt."""
     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)
-        
+
         # Fetch the image from the URL
         response = requests.get(image_url)
         image = Image.open(io.BytesIO(response.content))
-        
-        # Display the image
-        image.show()
-        
+
+        # Display the image inline in Jupyter Notebook
+        display.display(image)
+
     except Exception as e:
         print(f"Error generating image: {str(e)}")
-
+        
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)