Update app.py

app.py

@@ -2,24 +2,25 @@ import gradio as gr
 from PIL import Image, ImageDraw
 import io
 
-# Define a function to generate an image from a prompt
+# Function to generate an image from a prompt
 def generate_image(prompt):
     try:
-        # Create a simple image for testing
+        # Create a simple image with a blue background and text
         img = Image.new('RGB', (256, 256), color='blue')
         draw = ImageDraw.Draw(img)
         draw.text((10, 10), prompt, fill='white')
         
-        # Convert image to bytes and then return
+        # Convert the PIL image to a format Gradio can use
         buffered = io.BytesIO()
         img.save(buffered, format="PNG")
         img_bytes = buffered.getvalue()
         
-        return Image.open(io.BytesIO(img_bytes))  # Ensure image is in PIL format
+        # Return the image as a PIL Image object for Gradio
+        return Image.open(io.BytesIO(img_bytes))
     except Exception as e:
         return f"Error: {e}"
 
-# Create Gradio interface
+# Create and launch the Gradio interface
 iface = gr.Interface(fn=generate_image, inputs="text", outputs="image")
 
 # Launch the Gradio interface