Update app.py

app.py

@@ -4,7 +4,8 @@ from diffusers import StableDiffusionPipeline, EulerAncestralDiscreteScheduler
 import gradio as gr
 from PIL import Image
 import io
-import base64
+import tempfile
+import os
 import time
 
 # Force CPU usage
@@ -76,14 +77,14 @@ def generate_image(prompt):
     
     print("Image generated successfully!")
     
-    # SOLUTION: Convert image to bytes and return as file
-    img_bytes = io.BytesIO()
-    image.save(img_bytes, format='PNG')
-    img_bytes.seek(0)
+    # Create a temporary file and save the image
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as tmp_file:
+        image.save(tmp_file, format='PNG')
+        tmp_file_path = tmp_file.name
     
-    return img_bytes
+    return tmp_file_path
 
-# Create the Gradio Interface - CHANGE OUTPUT TO "file"
+# Create the Gradio Interface
 demo = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(
@@ -91,7 +92,7 @@ demo = gr.Interface(
         lines=2,
         placeholder="A dragon reading a book under a magical tree"
     ),
-    outputs=gr.File(label="Download Generated Illustration"),  # Changed to File output
+    outputs=gr.File(label="Download Generated Illustration"),
     title="Premium Children's Book Illustrator 🤖🎨",
     description="Generating high-quality, sharp, detailed images for your stories. Enter a scene description."
 )