Update app.py

app.py

@@ -4,9 +4,10 @@ from PIL import Image
 import io
 import os
 
-# 🔑 Your HF Space secret name is "images"
+# 🔑 your secret token name
 HF_TOKEN = os.getenv("images")
 
+# ✅ CORRECT inference API endpoint
 API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
 
 headers = {
@@ -14,8 +15,8 @@ headers = {
 }
 
 def generate_image(prompt):
-    if not prompt or not prompt.strip():
-        return None, "⚠️ Please enter a prompt"
+    if not prompt.strip():
+        return None, "⚠️ Enter a prompt"
 
     try:
         response = requests.post(
@@ -24,34 +25,33 @@ def generate_image(prompt):
             json={"inputs": prompt}
         )
 
-        # Handle API errors / loading
+        # 🔴 IMPORTANT: handle loading state
+        if response.status_code == 503:
+            return None, "⏳ Model is loading... try again in 20 seconds"
+
         if response.status_code != 200:
             return None, f"Error: {response.text}"
 
-        # Convert response bytes → image
+        # convert bytes → image
         image = Image.open(io.BytesIO(response.content))
 
-        return image, "✅ Image generated successfully"
+        return image, "✅ Image generated"
 
     except Exception as e:
-        return None, f"❌ Error: {str(e)}"
+        return None, str(e)
 
 
-# 🎨 Gradio UI
+# 🎨 UI
 with gr.Blocks() as app:
-    gr.Markdown("# 🎨 Text → Image Generator (HF API)")
-    gr.Markdown("Enter a prompt and generate AI images using Hugging Face API")
+    gr.Markdown("# 🎨 Text → Image Generator (Fixed HF API)")
 
-    prompt = gr.Textbox(
-        label="Enter Prompt",
-        placeholder="A futuristic cyberpunk city at night..."
-    )
+    prompt = gr.Textbox(label="Prompt")
 
-    btn = gr.Button("Generate Image")
+    btn = gr.Button("Generate")
 
-    output_image = gr.Image(type="pil")
+    img = gr.Image(type="pil")
     status = gr.Textbox()
 
-    btn.click(generate_image, prompt, [output_image, status])
+    btn.click(generate_image, prompt, [img, status])
 
 app.launch()