Update app.py

app.py

@@ -2,51 +2,43 @@ import gradio as gr
 import os
 from huggingface_hub import InferenceClient
 
-# 🔑 Your Hugging Face Space secret name = "images"
+# 🔑 Load token from Space secret
 HF_TOKEN = os.getenv("images")
 
-# 🚨 Safety check (very important)
+# ❌ DO NOT crash app
 if not HF_TOKEN:
-    raise ValueError("❌ Token not found. Make sure Space Secret name is 'images'")
+    HF_TOKEN = ""
 
-# 🤖 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 or not prompt.strip():
-        return None, "⚠️ Please enter a prompt"
+    if not prompt.strip():
+        return None, "⚠️ Enter a prompt"
+
+    if not HF_TOKEN:
+        return None, "❌ Token not loaded. Restart Space or check secret name."
 
     try:
         image = client.text_to_image(prompt)
-        return image, "✅ Image generated successfully"
+        return image, "✅ Image generated"
 
     except Exception as e:
         return None, f"❌ Error: {str(e)}"
 
 
-# 🖥️ Gradio UI
 with gr.Blocks() as app:
-    gr.Markdown("# 🎨 AI Text → Image Generator")
-    gr.Markdown("Generate images using Stable Diffusion (Hugging Face API)")
+    gr.Markdown("# 🎨 Text → Image Generator")
 
-    prompt = gr.Textbox(
-        label="Enter Prompt",
-        placeholder="A futuristic city at night, ultra detailed..."
-    )
+    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(
-        fn=generate_image,
-        inputs=prompt,
-        outputs=[output_image, status]
-    )
+    btn.click(generate_image, prompt, [img, status])
 
 app.launch()