Update app.py

app.py

@@ -1,7 +1,11 @@
 import gradio as gr
 import requests
+from PIL import Image
+import io
+import os
 
-HF_TOKEN = "images"
+# 🔑 Your HF Space secret name is "images"
+HF_TOKEN = os.getenv("images")
 
 API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
 
@@ -10,29 +14,44 @@ headers = {
 }
 
 def generate_image(prompt):
-    if not prompt.strip():
-        return None, "Enter prompt"
+    if not prompt or not prompt.strip():
+        return None, "⚠️ Please enter a prompt"
 
-    response = requests.post(
-        API_URL,
-        headers=headers,
-        json={"inputs": prompt}
-    )
+    try:
+        response = requests.post(
+            API_URL,
+            headers=headers,
+            json={"inputs": prompt}
+        )
+
+        # Handle API errors / loading
+        if response.status_code != 200:
+            return None, f"Error: {response.text}"
+
+        # Convert response bytes → image
+        image = Image.open(io.BytesIO(response.content))
 
-    if response.status_code != 200:
-        return None, response.text
+        return image, "✅ Image generated successfully"
 
-    return response.content, "Done"
+    except Exception as e:
+        return None, f"❌ Error: {str(e)}"
 
+
+# 🎨 Gradio UI
 with gr.Blocks() as app:
-    gr.Markdown("# Text to Image AI")
+    gr.Markdown("# 🎨 Text → Image Generator (HF API)")
+    gr.Markdown("Enter a prompt and generate AI images using Hugging Face API")
+
+    prompt = gr.Textbox(
+        label="Enter Prompt",
+        placeholder="A futuristic cyberpunk city at night..."
+    )
 
-    prompt = gr.Textbox(label="Prompt")
-    btn = gr.Button("Generate")
+    btn = gr.Button("Generate Image")
 
-    img = gr.Image(type="pil")
+    output_image = gr.Image(type="pil")
     status = gr.Textbox()
 
-    btn.click(generate_image, prompt, [img, status])
+    btn.click(generate_image, prompt, [output_image, status])
 
 app.launch()