Update app.py

app.py

@@ -3,12 +3,11 @@ import base64
 from PIL import Image
 from io import BytesIO
 import gradio as gr
-import uuid
-import os
 
-API_KEY = "sk-k8c1MQmc6OLzrJRs9zJb9ox3VeLmO4Fho7bQo49uLxTX6OTs"  # Replace with your actual key
+API_KEY = "sk-ESkTZ28wTlIcedRXuq2KTYbgunw578RkXprkewhJpMsu69Ec"  # REPLACE THIS!
 
 def generate_image(prompt):
+    # Step 1: Make the API request
     response = requests.post(
         "https://api.stability.ai/v1/generation/stable-diffusion-v1-5/text-to-image",
         headers={
@@ -25,20 +24,23 @@ def generate_image(prompt):
         }
     )
 
+    # Step 2: Error check
     if response.status_code != 200:
         return f"Error: {response.status_code} - {response.text}"
 
-    # Decode base64 image and return directly as PIL Image (NO need to save!)
-    b64_image = response.json()["artifacts"][0]["base64"]
-    image_data = base64.b64decode(b64_image)
-    image = Image.open(BytesIO(image_data))
+    # Step 3: Convert base64 to PIL.Image
+    image_data = response.json()["artifacts"][0]["base64"]
+    image = Image.open(BytesIO(base64.b64decode(image_data))).convert("RGB")  # Ensure RGB
 
     return image
 
-# Just show the image, no download
-gr.Interface(
+# Step 4: Gradio Interface — PIL Image Return Only
+demo = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(label="Enter your prompt"),
-    outputs=gr.Image(type="pil", label="Generated Image"),  # <- THIS is key!
-    title="Speech2Image Generator (API Powered)"
-).launch()
+    inputs=gr.Textbox(label="Enter a prompt"),
+    outputs=gr.Image(type="pil", label="Generated Image"),
+    title="✨ Speech2Image Generator ✨"
+)
+
+demo.launch()
+