Redesign UI with modern studio-style interface

- Add hero section with gradient title and feature pills
- Implement numbered sections (Prompt, Result, Settings)
- Add custom CSS with purple/violet theme for anime aesthetic
- Add negative prompt as visible field with default value
- Add styled generate button with hover effects
- Add styled accordions for aspect ratio and metadata
- Improve overall layout with columns
- Add ZeroGPU badge and model info pills
- Use custom fonts (Outfit, Fira Code)
- Dark theme with gradient background effects

app.py

@@ -16,7 +16,7 @@ from diffusers import StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipelin
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-DESCRIPTION = "PonyDiffusion V6 XL"
+DESCRIPTION = "Pony Studio"
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU. </p>"
 IS_COLAB = utils.is_google_colab() or os.getenv("IS_COLAB") == "1"
@@ -28,6 +28,11 @@ USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
 OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./outputs")
 
+DEFAULT_NEGATIVE = (
+    "(worst quality, low quality, normal quality, lowres, low details), "
+    "(watermark, signature, text, logo, words, letters), blurry, jpeg artifacts"
+)
+
 MODEL = os.getenv(
     "MODEL",
     "https://huggingface.co/AstraliteHeart/pony-diffusion-v6/blob/main/v6.safetensors",
@@ -176,52 +181,265 @@ if torch.cuda.is_available():
 else:
     pipe = None
 
-with gr.Blocks(css="style.css") as demo:
-    title = gr.HTML(
-        f"""<h1><span>{DESCRIPTION}</span></h1>""",
-        elem_id="title",
-    )
-    gr.Markdown(
-        f"""Gradio demo for [Pony Diffusion V6](https://civitai.com/models/257749/pony-diffusion-v6-xl/)""",
-        elem_id="subtitle",
-    )
+CSS = """
+@import url('https://fonts.googleapis.com/css2?family=Outfit:wght@300;400;500;600;700;800&family=Fira+Code:wght@400;500&display=swap');
+
+:root {
+    --bg:      #080a0e;
+    --surf:    #0d1017;
+    --card:    #111520;
+    --border:  #1c2133;
+    --border2: #252d45;
+    --purple:  #a78bfa;
+    --violet:  #7c3aed;
+    --lavender: #c4b5fd;
+    --text:    #e2e8f8;
+    --muted:   #4a5578;
+    --r:       14px;
+    --r-sm:    8px;
+}
+
+*, *::before, *::after { box-sizing: border-box; }
+
+body, .gradio-container {
+    background: var(--bg) !important;
+    font-family: 'Outfit', sans-serif !important;
+    color: var(--text) !important;
+}
+
+.gradio-container::before {
+    content: '';
+    position: fixed; inset: 0; pointer-events: none; z-index: 0;
+    background:
+        radial-gradient(ellipse 70% 50% at 50% -10%, rgba(167,139,250,0.07) 0%, transparent 65%),
+        radial-gradient(ellipse 40% 30% at 90% 90%, rgba(124,58,237,0.04) 0%, transparent 60%);
+}
+
+.app-hero { padding: 52px 0 28px; text-align: center; }
+
+.app-hero h1 {
+    font-size: 3rem; font-weight: 800; letter-spacing: -0.05em;
+    line-height: 1; margin: 0 0 12px;
+    background: linear-gradient(135deg, var(--lavender) 0%, var(--purple) 40%, var(--violet) 100%);
+    -webkit-background-clip: text; -webkit-text-fill-color: transparent; background-clip: text;
+}
+
+.app-hero .tagline {
+    color: var(--muted); font-size: 0.88rem; font-weight: 300;
+    letter-spacing: 0.06em; text-transform: uppercase; margin: 0 0 20px;
+}
+
+.app-hero .pills { display: flex; justify-content: center; gap: 8px; flex-wrap: wrap; }
+
+.app-hero .pill {
+    background: var(--card); border: 1px solid var(--border2); border-radius: 100px;
+    padding: 4px 14px; font-size: 0.74rem; font-weight: 500; color: var(--muted);
+    font-family: 'Fira Code', monospace;
+}
+
+.app-hero .pill.violet { color: var(--purple); border-color: rgba(167,139,250,0.3); }
+
+.sec-label {
+    font-size: 0.62rem !important; font-weight: 700 !important;
+    letter-spacing: 0.15em !important; text-transform: uppercase !important;
+    color: var(--purple) !important; margin: 0 0 8px !important; display: block;
+}
+
+label > span {
+    font-family: 'Outfit', sans-serif !important; font-size: 0.72rem !important;
+    font-weight: 500 !important; color: var(--muted) !important;
+    text-transform: uppercase; letter-spacing: 0.08em;
+}
+
+textarea, input[type="text"] {
+    background: var(--surf) !important; border: 1px solid var(--border) !important;
+    border-radius: var(--r-sm) !important; color: var(--text) !important;
+    font-family: 'Outfit', sans-serif !important; font-size: 0.95rem !important;
+    transition: border-color 0.2s, box-shadow 0.2s;
+}
+
+textarea:focus, input[type="text"]:focus {
+    border-color: var(--purple) !important;
+    box-shadow: 0 0 0 3px rgba(167,139,250,0.12) !important;
+    outline: none !important;
+}
+
+.gen-btn {
+    background: linear-gradient(135deg, var(--purple), var(--violet)) !important;
+    border: none !important; border-radius: var(--r) !important;
+    color: #fff !important; font-family: 'Outfit', sans-serif !important;
+    font-weight: 700 !important; font-size: 1rem !important;
+    height: 54px !important; width: 100% !important;
+    letter-spacing: 0.02em !important; cursor: pointer !important;
+    transition: opacity 0.18s, transform 0.15s, box-shadow 0.2s !important;
+    box-shadow: 0 4px 20px rgba(167,139,250,0.28) !important;
+}
+
+.gen-btn:hover {
+    opacity: 0.88 !important; transform: translateY(-1px) !important;
+    box-shadow: 0 8px 30px rgba(167,139,250,0.48) !important;
+}
+
+.gen-btn:active { transform: translateY(0) !important; }
+
+.result-gallery .grid-wrap {
+    background: var(--surf) !important;
+    border: 1px solid var(--border) !important;
+    border-radius: var(--r) !important;
+}
+
+.result-gallery img { border-radius: 10px !important; }
+
+.gr-accordion {
+    background: var(--card) !important; border: 1px solid var(--border) !important;
+    border-radius: var(--r) !important; margin-top: 10px !important;
+}
+
+.aspect-radio .wrap {
+    display: flex !important; flex-wrap: wrap !important; gap: 6px !important;
+    background: transparent !important; padding: 0 !important; border: none !important;
+}
+
+.aspect-radio label {
+    background: var(--surf) !important; border: 1px solid var(--border) !important;
+    border-radius: 100px !important; padding: 5px 12px !important;
+    font-size: 0.78rem !important; font-family: 'Fira Code', monospace !important;
+    color: var(--muted) !important; cursor: pointer !important;
+    transition: all 0.15s !important; white-space: nowrap !important;
+}
+
+.aspect-radio label:hover { border-color: var(--purple) !important; color: var(--text) !important; }
+
+.aspect-radio label:has(input:checked) {
+    background: var(--purple) !important; border-color: var(--purple) !important;
+    color: #fff !important; font-weight: 600 !important;
+}
+
+.ghost-btn button {
+    background: transparent !important; border: 1px solid var(--border2) !important;
+    border-radius: var(--r-sm) !important; color: var(--muted) !important;
+    font-family: 'Outfit', sans-serif !important; font-size: 0.82rem !important;
+    height: 36px !important; transition: border-color 0.18s, color 0.18s !important;
+}
+
+.ghost-btn button:hover { border-color: var(--purple) !important; color: var(--text) !important; }
+
+.upscaler-checkbox input:checked {
+    background-color: var(--purple) !important;
+}
+
+::-webkit-scrollbar { width: 5px; }
+::-webkit-scrollbar-track { background: var(--surf); }
+::-webkit-scrollbar-thumb { background: var(--border2); border-radius: 3px; }
+::-webkit-scrollbar-thumb:hover { background: var(--purple); }
+
+#duplicate-button {
+    margin: auto;
+    color: #fff;
+    background: var(--violet);
+    border-radius: 100vh;
+    font-family: 'Outfit', sans-serif;
+}
+"""
+
+with gr.Blocks(theme=gr.themes.Base(), css=CSS) as demo:
+
+    gr.HTML("""
+    <div class="app-hero">
+        <h1>Pony Studio</h1>
+        <p class="tagline">Anime-style · High-fidelity · Text-to-image</p>
+        <div class="pills">
+            <span class="pill violet">ZeroGPU ⚡</span>
+            <span class="pill">Pony V6 XL</span>
+            <span class="pill">SDXL · fp16</span>
+            <span class="pill">safetensors</span>
+        </div>
+    </div>
+    """)
+
     gr.DuplicateButton(
         value="Duplicate Space for private use",
         elem_id="duplicate-button",
         visible=os.getenv("SHOW_DUPLICATE_BUTTON") == "1",
     )
-    with gr.Group():
-        with gr.Row():
-            prompt = gr.Text(
-                label="Prompt",
-                show_label=False,
-                max_lines=5,
-                placeholder="Enter your prompt",
+
+    with gr.Row():
+
+        with gr.Column(scale=2):
+            gr.HTML('<span class="sec-label">① Prompt</span>')
+            prompt = gr.Textbox(
+                label="",
+                lines=4,
+                placeholder="score_9, score_8_up, score_7_up, ... your anime-style prompt here",
                 container=False,
             )
-            run_button = gr.Button(
-                "Generate", 
-                variant="primary", 
-                scale=0
+            gr.HTML('<div style="height:8px"></div>')
+            negative_prompt = gr.Textbox(
+                label="Negative prompt", 
+                lines=2, 
+                value=DEFAULT_NEGATIVE
             )
-        result = gr.Gallery(
-            label="Result", 
-            columns=1, 
-            preview=True, 
-            show_label=False
-        )
-    with gr.Accordion(label="Advanced Settings", open=False):
-        negative_prompt = gr.Text(
-            label="Negative Prompt",
-            max_lines=5,
-            placeholder="Enter a negative prompt",
-            value=""
-        )
+
+            gr.HTML('<div style="height:10px"></div>')
+            run_btn = gr.Button("▶  Generate", variant="primary", elem_classes=["gen-btn"])
+
+            gr.HTML('<div style="height:14px"></div>')
+            gr.HTML('<span class="sec-label">② Result</span>')
+            result = gr.Gallery(
+                label="", columns=1, preview=True,
+                container=False, elem_classes=["result-gallery"], height=600,
+            )
+
+        with gr.Column(scale=1, min_width=280):
+            gr.HTML('<span class="sec-label">③ Settings</span>')
+            
+            use_upscaler = gr.Checkbox(label="Use Upscaler", value=False, elem_classes=["upscaler-checkbox"])
+            with gr.Row(visible=False) as upscaler_row:
+                upscaler_strength = gr.Slider(
+                    label="Strength",
+                    minimum=0,
+                    maximum=1,
+                    step=0.05,
+                    value=0.55,
+                )
+                upscale_by = gr.Slider(
+                    label="Upscale by",
+                    minimum=1,
+                    maximum=1.5,
+                    step=0.1,
+                    value=1.5,
+                )
+
+            gr.HTML('<div style="height:8px"></div>')
+            sampler = gr.Dropdown(
+                label="Sampler", 
+                choices=config.sampler_list, 
+                value="DPM++ 2M SDE Karras",
+            )
+            
+            with gr.Row():
+                guidance_scale = gr.Slider(
+                    label="Guidance scale",
+                    minimum=1,
+                    maximum=12,
+                    step=0.1,
+                    value=7.0,
+                )
+                num_inference_steps = gr.Slider(
+                    label="Steps",
+                    minimum=1,
+                    maximum=50,
+                    step=1,
+                    value=28,
+                )
+
+            seed = gr.Slider(label="Seed", minimum=0, maximum=utils.MAX_SEED, step=1, value=0)
+            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+
+    with gr.Accordion("⚙  Aspect ratio", open=False):
         aspect_ratio_selector = gr.Radio(
-            label="Aspect Ratio",
-            choices=config.aspect_ratios,
-            value="1024 x 1024",
-            container=True,
+            label="", choices=config.aspect_ratios, value="1024 x 1024",
+            container=False, elem_classes=["aspect-radio"],
         )
         with gr.Group(visible=False) as custom_resolution:
             with gr.Row():
@@ -239,68 +457,28 @@ with gr.Blocks(css="style.css") as demo:
                     step=8,
                     value=1024,
                 )
-        use_upscaler = gr.Checkbox(label="Use Upscaler", value=False)
-        with gr.Row() as upscaler_row:
-            upscaler_strength = gr.Slider(
-                label="Strength",
-                minimum=0,
-                maximum=1,
-                step=0.05,
-                value=0.55,
-                visible=False,
-            )
-            upscale_by = gr.Slider(
-                label="Upscale by",
-                minimum=1,
-                maximum=1.5,
-                step=0.1,
-                value=1.5,
-                visible=False,
-            )
 
-        sampler = gr.Dropdown(
-            label="Sampler",
-            choices=config.sampler_list,
-            interactive=True,
-            value="DPM++ 2M SDE Karras",
-        )
-        with gr.Row():
-            seed = gr.Slider(
-                label="Seed", minimum=0, maximum=utils.MAX_SEED, step=1, value=0
-            )
-            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-        with gr.Group():
-            with gr.Row():
-                guidance_scale = gr.Slider(
-                    label="Guidance scale",
-                    minimum=1,
-                    maximum=12,
-                    step=0.1,
-                    value=7.0,
-                )
-                num_inference_steps = gr.Slider(
-                    label="Number of inference steps",
-                    minimum=1,
-                    maximum=50,
-                    step=1,
-                    value=28,
-                )
-    with gr.Accordion(label="Generation Parameters", open=False):
-        gr_metadata = gr.JSON(label="Metadata", show_label=False)
+    with gr.Accordion("📋  Generation metadata", open=False):
+        gr_metadata = gr.JSON(label="", show_label=False)
+
+    gr.HTML('<div style="height:12px"></div>')
+    gr.HTML('<span class="sec-label">Examples</span>')
     gr.Examples(
         examples=config.examples,
         inputs=prompt,
         outputs=[result, gr_metadata],
-        fn=lambda *args, **kwargs: generate(*args, use_upscaler=True, **kwargs),
+        fn=lambda *args, **kwargs: generate(*args, use_upscaler=False, **kwargs),
         cache_examples=CACHE_EXAMPLES,
     )
+
     use_upscaler.change(
-        fn=lambda x: [gr.update(visible=x), gr.update(visible=x)],
+        fn=lambda x: gr.update(visible=x),
         inputs=use_upscaler,
-        outputs=[upscaler_strength, upscale_by],
+        outputs=upscaler_row,
         queue=False,
         api_name=False,
     )
+
     aspect_ratio_selector.change(
         fn=lambda x: gr.update(visible=x == "Custom"),
         inputs=aspect_ratio_selector,
@@ -333,9 +511,10 @@ with gr.Blocks(css="style.css") as demo:
     ).then(
         fn=generate,
         inputs=inputs,
-        outputs=result,
+        outputs=[result, gr_metadata],
         api_name="run",
     )
+
     negative_prompt.submit(
         fn=utils.randomize_seed_fn,
         inputs=[seed, randomize_seed],
@@ -345,10 +524,11 @@ with gr.Blocks(css="style.css") as demo:
     ).then(
         fn=generate,
         inputs=inputs,
-        outputs=result,
+        outputs=[result, gr_metadata],
         api_name=False,
     )
-    run_button.click(
+
+    run_btn.click(
         fn=utils.randomize_seed_fn,
         inputs=[seed, randomize_seed],
         outputs=seed,
@@ -360,4 +540,6 @@ with gr.Blocks(css="style.css") as demo:
         outputs=[result, gr_metadata],
         api_name=False,
     )
-demo.queue(max_size=20).launch(debug=IS_COLAB, share=IS_COLAB)
+
+demo.queue(max_size=20)
+demo.launch(debug=IS_COLAB, share=IS_COLAB)