Update app.py

app.py

@@ -53,16 +53,27 @@ custom_role_conversions=None,
 # 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."""
+    """
+    Generates and displays an image based on the given text prompt.
+
+    Args:
+        prompt (str): A string describing the image to be generated.
+
+    Returns:
+        None: The function 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
         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 inline in Jupyter Notebook
+        # Display the image inline (for Jupyter Notebook)
         display.display(image)
 
     except Exception as e: