Update app.py

app.py

@@ -2,42 +2,51 @@ import gradio as gr
 import os
 from huggingface_hub import InferenceClient
 
-# 🔑 Read secret safely
-HF_TOKEN = os.environ.get("images")
+# 🔑 Load Hugging Face token from Space secret
+HF_TOKEN = os.environ.get("image")
 
-def check_token():
-    return HF_TOKEN is not None
-
-client = InferenceClient(
-    model="stabilityai/stable-diffusion-xl-base-1.0",
-    token=HF_TOKEN
-)
+# 🤖 Create client only if token exists
+client = None
+if HF_TOKEN:
+    client = InferenceClient(
+        model="stabilityai/stable-diffusion-xl-base-1.0",
+        token=HF_TOKEN
+    )
 
 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"
 
-    if not HF_TOKEN:
-        return None, "❌ Secret not loaded. Restart Space again."
+    if client is None:
+        return None, "❌ Token not loaded. Check secret name = 'image' and restart Space."
 
     try:
         image = client.text_to_image(prompt)
-        return image, "✅ Image generated"
+        return image, "✅ Image generated successfully"
 
     except Exception as e:
         return None, f"❌ Error: {str(e)}"
 
 
+# 🎨 UI
 with gr.Blocks() as app:
-    gr.Markdown("# 🎨 Text → Image Generator")
+    gr.Markdown("# 🎨 AI Text → Image Generator")
+    gr.Markdown("Generate images using Stable Diffusion (Hugging Face API)")
 
-    prompt = gr.Textbox(label="Prompt")
+    prompt = gr.Textbox(
+        label="Enter Prompt",
+        placeholder="A futuristic city at night, ultra detailed..."
+    )
 
-    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(
+        fn=generate_image,
+        inputs=prompt,
+        outputs=[output_image, status]
+    )
 
 app.launch()