Update app.py

app.py

@@ -3,6 +3,8 @@ import torch
 from diffusers import StableDiffusionPipeline, EulerAncestralDiscreteScheduler
 import gradio as gr
 from PIL import Image
+import io
+import base64
 import time
 
 # Force CPU usage
@@ -60,11 +62,11 @@ def generate_image(prompt):
     # Generate the image with better settings
     image = pipe(
         prompt=enhanced_prompt,
-        negative_prompt=negative_prompt,  # Added negative prompt
+        negative_prompt=negative_prompt,
         width=512,
         height=512,
-        guidance_scale=9.0,  # Slightly higher for better prompt adherence
-        num_inference_steps=25,  # More steps for better quality
+        guidance_scale=9.0,
+        num_inference_steps=25,
         generator=torch.Generator(device=device)
     ).images[0]
     
@@ -73,9 +75,15 @@ def generate_image(prompt):
         image = image.convert('RGB')
     
     print("Image generated successfully!")
-    return image
+    
+    # SOLUTION: Convert image to bytes and return as file
+    img_bytes = io.BytesIO()
+    image.save(img_bytes, format='PNG')
+    img_bytes.seek(0)
+    
+    return img_bytes
 
-# Create the Gradio Interface
+# Create the Gradio Interface - CHANGE OUTPUT TO "file"
 demo = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(
@@ -83,7 +91,7 @@ demo = gr.Interface(
         lines=2,
         placeholder="A dragon reading a book under a magical tree"
     ),
-    outputs=gr.Image(label="Generated Illustration", type="pil"),
+    outputs=gr.File(label="Download Generated Illustration"),  # Changed to File output
     title="Premium Children's Book Illustrator 🤖🎨",
     description="Generating high-quality, sharp, detailed images for your stories. Enter a scene description."
 )