Update app.py

app.py

@@ -1,43 +1,50 @@
-import gradio as gr
-import os
-from huggingface_hub import InferenceClient
-
-# 🔑 load secret
-HF_TOKEN = os.getenv("image")
-
-def debug():
-    return f"DEBUG - TOKEN LOADED: {HF_TOKEN is not None}"
-
-client = None
-if HF_TOKEN:
-    client = InferenceClient(
-        model="stabilityai/stable-diffusion-xl-base-1.0",
-        token=HF_TOKEN
+with gr.Blocks(
+    theme=gr.themes.Soft(
+        primary_hue="blue",
+        secondary_hue="indigo",
+        neutral_hue="gray"
+    ),
+    css="""
+    body {
+        background-color: #f4f7ff !important;
+    }
+
+    .gradio-container {
+        background-color: #f4f7ff !important;
+    }
+
+    button {
+        border-radius: 10px !important;
+    }
+"""
+) as app:
+
+    gr.Markdown(
+        """
+        # 🎨 AI Text → Image Generator
+        ### ✨ Create beautiful AI images instantly
+        """
     )
 
-def generate_image(prompt):
-    if not prompt.strip():
-        return None, "⚠️ Enter prompt"
-
-    if client is None:
-        return None, "❌ TOKEN STILL NOT LOADED (check secrets + restart)"
-
-    image = client.text_to_image(prompt)
-    return image, "✅ Success"
-
-
-with gr.Blocks() as app:
-    gr.Markdown("# 🎨 DEBUG IMAGE GENERATOR")
-
-    gr.Textbox(value=debug(), label="System Status")
+    with gr.Row():
+        prompt = gr.Textbox(
+            label="✍️ Enter Prompt",
+            placeholder="A futuristic city at sunset, ultra detailed, cinematic lighting",
+            lines=3
+        )
 
-    prompt = gr.Textbox(label="Prompt")
+    with gr.Row():
+        width = gr.Slider(256, 1024, value=768, step=64, label="📏 Width")
+        height = gr.Slider(256, 1024, value=768, step=64, label="📐 Height")
 
-    btn = gr.Button("Generate")
+    with gr.Row():
+        generate_btn = gr.Button("🚀 Generate Image", variant="primary")
+        clear_btn = gr.Button("🧹 Clear")
 
-    img = gr.Image(type="pil")
-    status = gr.Textbox()
+    output_image = gr.Image(label="🖼️ Generated Image", type="pil")
+    status = gr.Textbox(label="📢 Status")
 
-    btn.click(generate_image, prompt, [img, status])
+    generate_btn.click(generate_image, [prompt, width, height], [output_image, status])
 
-app.launch()
+    clear_btn.click(lambda: ("", 768, 768, None, ""), None,
+                    [prompt, width, height, output_image, status])