Update app.py

app.py

@@ -26,24 +26,41 @@ def generate_image(prompt, width, height):
         return None, f"❌ Error: {str(e)}"
 
 
-# 🎨 UI (LIGHT + MODERN)
+# 🎨 CLEAN MODERN UI
 with gr.Blocks(
-    theme=gr.themes.Soft(primary_hue="blue"),
-    css="body {background-color:#f4f7ff !important;}"
+    theme=gr.themes.Soft(
+        primary_hue="blue",
+        secondary_hue="sky",
+        neutral_hue="gray"
+    )
 ) as app:
 
-    gr.Markdown("# 🎨 AI Text → Image Generator")
+    # ✅ MAIN TITLE (fixed visibility)
+    gr.Markdown(
+        """
+        # 🎨 AI Text → Image Generator
+        ### ✨ Generate high-quality AI images using Stable Diffusion
+        """,
+        elem_id="title"
+    )
 
-    prompt = gr.Textbox(label="Prompt")
+    with gr.Row():
+        prompt = gr.Textbox(
+            label="✍️ Enter Prompt",
+            placeholder="A futuristic city at night, cinematic lighting...",
+            lines=3
+        )
 
-    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():
+        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", variant="primary")
+    with gr.Row():
+        btn = gr.Button("🚀 Generate Image", variant="primary")
 
-    img = gr.Image(type="pil")
-    status = gr.Textbox()
+    output_img = gr.Image(label="🖼️ Result", type="pil")
+    status = gr.Textbox(label="📢 Status")
 
-    btn.click(generate_image, [prompt, width, height], [img, status])
+    btn.click(generate_image, [prompt, width, height], [output_img, status])
 
 app.launch()