Update app.py

app.py

@@ -2,10 +2,12 @@ import gradio as gr
 import os
 from huggingface_hub import InferenceClient
 
-# 🔑 Load Hugging Face token from Space secret
-HF_TOKEN = os.environ.get("image")
+# 🔑 load secret
+HF_TOKEN = os.getenv("image")
+
+def debug():
+    return f"DEBUG - TOKEN LOADED: {HF_TOKEN is not None}"
 
-# 🤖 Create client only if token exists
 client = None
 if HF_TOKEN:
     client = InferenceClient(
@@ -14,39 +16,28 @@ if HF_TOKEN:
     )
 
 def generate_image(prompt):
-    if not prompt or not prompt.strip():
-        return None, "⚠️ Please enter a prompt"
+    if not prompt.strip():
+        return None, "⚠️ Enter prompt"
 
     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 successfully"
+        return None, "❌ TOKEN STILL NOT LOADED (check secrets + restart)"
 
-    except Exception as e:
-        return None, f"❌ Error: {str(e)}"
+    image = client.text_to_image(prompt)
+    return image, "✅ Success"
 
 
-# 🎨 UI
 with gr.Blocks() as app:
-    gr.Markdown("# 🎨 AI Text → Image Generator")
-    gr.Markdown("Generate images using Stable Diffusion (Hugging Face API)")
+    gr.Markdown("# 🎨 DEBUG IMAGE GENERATOR")
 
-    prompt = gr.Textbox(
-        label="Enter Prompt",
-        placeholder="A futuristic city at night, ultra detailed..."
-    )
+    gr.Textbox(value=debug(), label="System Status")
 
-    btn = gr.Button("Generate Image")
+    prompt = gr.Textbox(label="Prompt")
 
-    output_image = gr.Image(type="pil")
+    btn = gr.Button("Generate")
+
+    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()