Update app.py

app.py

@@ -1,50 +1,49 @@
+import gradio as gr
+import os
+from huggingface_hub import InferenceClient
+
+# 🔑 token
+HF_TOKEN = os.getenv("image")
+
+client = InferenceClient(
+    model="stabilityai/stable-diffusion-xl-base-1.0",
+    token=HF_TOKEN
+)
+
+def generate_image(prompt, width, height):
+    if not prompt.strip():
+        return None, "⚠️ Enter prompt"
+
+    try:
+        image = client.text_to_image(
+            prompt,
+            width=int(width),
+            height=int(height)
+        )
+        return image, "✅ Image generated"
+
+    except Exception as e:
+        return None, f"❌ Error: {str(e)}"
+
+
+# 🎨 UI (LIGHT + MODERN)
 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;
-    }
-"""
+    theme=gr.themes.Soft(primary_hue="blue"),
+    css="body {background-color:#f4f7ff !important;}"
 ) as app:
 
-    gr.Markdown(
-        """
-        # 🎨 AI Text → Image Generator
-        ### ✨ Create beautiful AI images instantly
-        """
-    )
-
-    with gr.Row():
-        prompt = gr.Textbox(
-            label="✍️ Enter Prompt",
-            placeholder="A futuristic city at sunset, ultra detailed, cinematic lighting",
-            lines=3
-        )
+    gr.Markdown("# 🎨 AI Text → Image Generator")
+
+    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")
+    width = gr.Slider(256, 1024, value=768, step=64, label="Width")
+    height = gr.Slider(256, 1024, value=768, step=64, label="Height")
 
-    with gr.Row():
-        generate_btn = gr.Button("🚀 Generate Image", variant="primary")
-        clear_btn = gr.Button("🧹 Clear")
+    btn = gr.Button("Generate", variant="primary")
 
-    output_image = gr.Image(label="🖼️ Generated Image", type="pil")
-    status = gr.Textbox(label="📢 Status")
+    img = gr.Image(type="pil")
+    status = gr.Textbox()
 
-    generate_btn.click(generate_image, [prompt, width, height], [output_image, status])
+    btn.click(generate_image, [prompt, width, height], [img, status])
 
-    clear_btn.click(lambda: ("", 768, 768, None, ""), None,
-                    [prompt, width, height, output_image, status])
+app.launch()