Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 import uuid
 import os
 
-API_KEY = "sk-ESkTZ28wTlIcedRXuq2KTYbgunw578RkXprkewhJpMsu69Ec"  # Replace this with your key
+API_KEY = "sk-k8c1MQmc6OLzrJRs9zJb9ox3VeLmO4Fho7bQo49uLxTX6OTs"  # Put your real Stability API key
 
 def generate_image(prompt):
     response = requests.post(
@@ -28,17 +28,17 @@ def generate_image(prompt):
     if response.status_code != 200:
         return f"Error: {response.status_code} - {response.text}", None
 
-    data = response.json()
-    b64_image = data["artifacts"][0]["base64"]
+    b64_image = response.json()["artifacts"][0]["base64"]
     image_data = base64.b64decode(b64_image)
     image = Image.open(BytesIO(image_data))
 
-    # Save image with a unique name to /tmp
+    # Save image to /tmp folder with a unique filename
     filename = f"/tmp/generated_{uuid.uuid4().hex}.png"
     image.save(filename)
 
-    return filename, filename  # One for display, one for download
+    return filename, filename  # One for image display, one for download
 
+# Gradio UI
 gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(label="Enter your prompt"),