Update app.py

app.py

@@ -1,19 +1,24 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
 import os
+from huggingface_hub import InferenceClient
 
-# 🔑 your HF secret name = images
+# 🔑 Your Hugging Face Space secret name = "images"
 HF_TOKEN = os.getenv("images")
 
-# ✅ Use official client (THIS FIXES YOUR ERROR)
+# 🚨 Safety check (very important)
+if not HF_TOKEN:
+    raise ValueError("❌ Token not found. Make sure Space Secret name is 'images'")
+
+# 🤖 Stable Diffusion model (fast + reliable)
 client = InferenceClient(
     model="stabilityai/stable-diffusion-xl-base-1.0",
     token=HF_TOKEN
 )
 
+# 🎨 Image generator function
 def generate_image(prompt):
-    if not prompt.strip():
-        return None, "⚠️ Enter a prompt"
+    if not prompt or not prompt.strip():
+        return None, "⚠️ Please enter a prompt"
 
     try:
         image = client.text_to_image(prompt)
@@ -23,17 +28,25 @@ def generate_image(prompt):
         return None, f"❌ Error: {str(e)}"
 
 
-# 🎨 UI
+# 🖥️ Gradio UI
 with gr.Blocks() as app:
-    gr.Markdown("# 🎨 Text → Image Generator (FIXED VERSION)")
+    gr.Markdown("# 🎨 AI Text → Image Generator")
+    gr.Markdown("Generate images using Stable Diffusion (Hugging Face API)")
 
-    prompt = gr.Textbox(label="Enter Prompt")
+    prompt = gr.Textbox(
+        label="Enter Prompt",
+        placeholder="A futuristic city at night, ultra detailed..."
+    )
 
     btn = gr.Button("Generate Image")
 
-    output = gr.Image(type="pil")
+    output_image = gr.Image(type="pil")
     status = gr.Textbox()
 
-    btn.click(generate_image, prompt, [output, status])
+    btn.click(
+        fn=generate_image,
+        inputs=prompt,
+        outputs=[output_image, status]
+    )
 
 app.launch()