Update app.py

app.py

@@ -10,17 +10,15 @@ import base64
 import json
 import html as html_lib
 from io import BytesIO
-from diffusers import Flux2KleinPipeline, AutoencoderKLFlux2
 from PIL import Image
-from typing import Iterable
+from pathlib import Path
+from diffusers import Flux2KleinPipeline, AutoencoderKLFlux2
 
 MAX_SEED       = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 dtype          = torch.bfloat16
 device         = "cuda" if torch.cuda.is_available() else "cpu"
 
-LANCZOS = getattr(Image, "Resampling", Image).LANCZOS
-
 print("Loading 4B Distilled model (Standard VAE)...")
 pipe_standard = Flux2KleinPipeline.from_pretrained(
     "black-forest-labs/FLUX.2-klein-4B",
@@ -49,6 +47,7 @@ EXAMPLES_CONFIG = [
     {"images": ["examples/4.jpg"], "prompt": "Make the texture high-resolution."},
 ]
 
+LANCZOS = getattr(Image, "Resampling", Image).LANCZOS
 
 def make_thumb_b64(path, max_dim=220):
     if not os.path.exists(path):
@@ -63,7 +62,6 @@ def make_thumb_b64(path, max_dim=220):
         print(f"Thumbnail error for {path}: {e}")
         return ""
 
-
 def encode_full_image(path):
     if not os.path.exists(path):
         return ""
@@ -77,7 +75,6 @@ def encode_full_image(path):
         print(f"Encode error for {path}: {e}")
         return ""
 
-
 def build_example_cards_html():
     cards = ""
     for i, ex in enumerate(EXAMPLES_CONFIG):
@@ -100,7 +97,6 @@ def build_example_cards_html():
         </div>'''
     return cards
 
-
 def load_example_data(idx_str):
     try:
         idx = int(float(idx_str)) if idx_str and idx_str.strip() else -1
@@ -117,12 +113,10 @@ def load_example_data(idx_str):
             names.append(os.path.basename(path))
     return json.dumps({"images": b64_list, "prompt": ex["prompt"], "names": names, "status": "ok"})
 
-
 print("Building example thumbnails...")
 EXAMPLE_CARDS_HTML = build_example_cards_html()
 print(f"Built {len(EXAMPLES_CONFIG)} example cards.")
 
-
 def b64_to_pil_list(b64_json_str):
     if not b64_json_str or b64_json_str.strip() in ("", "[]"):
         return []
@@ -145,15 +139,6 @@ def b64_to_pil_list(b64_json_str):
             print(f"Error decoding image: {e}")
     return pil_images
 
-
-def format_time(seconds: float) -> str:
-    if seconds < 60:
-        return f"{seconds:.2f}s"
-    minutes = int(seconds // 60)
-    secs    = seconds % 60
-    return f"{minutes}m {secs:.2f}s"
-
-
 def update_dimensions_on_upload(pil_images):
     if not pil_images:
         return 1024, 1024
@@ -165,8 +150,14 @@ def update_dimensions_on_upload(pil_images):
     else:
         nh = 1024
         nw = int(1024 * iw / ih)
-    return max(256, min(1024, (nw // 8) * 8)), max(256, min(1024, (nh // 8) * 8))
+    return (max(256, (nw // 8) * 8), max(256, (nh // 8) * 8))
 
+def format_time(seconds: float) -> str:
+    if seconds < 60:
+        return f"{seconds:.2f}s"
+    minutes = int(seconds // 60)
+    secs    = seconds % 60
+    return f"{minutes}m {secs:.2f}s"
 
 @spaces.GPU(duration=240)
 def infer(
@@ -176,7 +167,7 @@ def infer(
     randomize_seed,
     width,
     height,
-    steps,
+    num_inference_steps,
     guidance_scale,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -197,48 +188,49 @@ def infer(
         prompt=prompt,
         height=height,
         width=width,
-        num_inference_steps=steps,
+        num_inference_steps=num_inference_steps,
         guidance_scale=guidance_scale,
     )
     if pil_images:
         shared_kwargs["image"] = pil_images
 
-    progress(0.05, desc="Running Standard VAE generation...")
+    progress(0.05, desc="🟦 Running Standard VAE generation...")
     t0_std       = time.perf_counter()
     gen_std      = torch.Generator(device="cpu").manual_seed(seed)
     out_standard = pipe_standard(**shared_kwargs, generator=gen_std).images[0]
-    time_std_str = format_time(time.perf_counter() - t0_std)
+    t1_std       = time.perf_counter()
+    time_std_str = format_time(t1_std - t0_std)
     print(f"Standard VAE done in {time_std_str}")
 
+    progress(0.55, desc=f"🟦 Standard VAE done in {time_std_str} — now running 🟩 Small Decoder VAE...")
     gc.collect()
     torch.cuda.empty_cache()
 
-    progress(0.55, desc=f"Standard VAE done in {time_std_str} — now running Small Decoder VAE...")
     t0_small       = time.perf_counter()
     gen_small      = torch.Generator(device="cpu").manual_seed(seed)
     out_small      = pipe_small_decoder(**shared_kwargs, generator=gen_small).images[0]
-    time_small_str = format_time(time.perf_counter() - t0_small)
+    t1_small       = time.perf_counter()
+    time_small_str = format_time(t1_small - t0_small)
     print(f"Small Decoder VAE done in {time_small_str}")
 
+    progress(1.0, desc=f"✅ Both done! Standard: {time_std_str} | Small: {time_small_str}")
     gc.collect()
     torch.cuda.empty_cache()
 
-    label_std   = f"Standard VAE — {time_std_str}"
-    label_small = f"Small Decoder VAE — {time_small_str}"
+    label_std   = f"🟦 Standard VAE — ⏱ {time_std_str}"
+    label_small = f"🟩 Small Decoder VAE — ⏱ {time_small_str}"
 
     return out_standard, out_small, seed, label_std, label_small
 
-
-# ── SVG / Icon constants ───────────────────────────────────────────────────────
-
-FLUX_LOGO_SVG = '''<svg viewBox="0 0 24 24" fill="white" xmlns="http://www.w3.org/2000/svg">
-<path d="M13 2L3 14h9l-1 8 10-12h-9l1-8z"/>
-</svg>'''
+# ── SVG Icons ─────────────────────────────────────────────────────────────────
 
 UPLOAD_SVG = '<svg class="tb-svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M21 15v4a2 2 0 01-2 2H5a2 2 0 01-2-2v-4"/><polyline points="17 8 12 3 7 8"/><line x1="12" y1="3" x2="12" y2="15"/></svg>'
-REMOVE_SVG = '<svg class="tb-svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><circle cx="12" cy="12" r="10"/><line x1="15" y1="9" x2="9" y2="15"/><line x1="9" y1="9" x2="15" y2="15"/></svg>'
 CLEAR_SVG  = '<svg class="tb-svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><polyline points="3 6 5 6 21 6"/><path d="M19 6v14a2 2 0 01-2 2H7a2 2 0 01-2-2V6m3 0V4a2 2 0 012-2h4a2 2 0 012 2v2"/><line x1="10" y1="11" x2="10" y2="17"/><line x1="14" y1="11" x2="14" y2="17"/></svg>'
-DOWNLOAD_SVG = '<svg viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg"><path d="M12 16l-5-5h3V4h4v7h3l-5 5z"/><path d="M20 18H4v2h16v-2z"/></svg>'
+REMOVE_SVG = '<svg class="tb-svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><circle cx="12" cy="12" r="10"/><line x1="15" y1="9" x2="9" y2="15"/><line x1="9" y1="9" x2="15" y2="15"/></svg>'
+DL_SVG     = '<svg viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg"><path d="M12 16l-5-5h3V4h4v7h3l-5 5z"/><path d="M20 18H4v2h16v-2z"/></svg>'
+FLUX_LOGO  = '<svg viewBox="0 0 24 24" fill="white" xmlns="http://www.w3.org/2000/svg"><path d="M13 2L3 14h9l-1 8 10-12h-9l1-8z"/></svg>'
+
+# ── CSS ───────────────────────────────────────────────────────────────────────
 
 css = r"""
 @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700;800&family=JetBrains+Mono:wght@400;500;600&display=swap');
@@ -258,18 +250,21 @@ footer{display:none!important}
     position:absolute;left:-9999px;top:-9999px;width:1px;height:1px;opacity:0.01;pointer-events:none;
 }
 
+/* ── Shell ── */
 .app-shell{
     background:#18181b;border:1px solid #27272a;border-radius:16px;
     margin:12px auto;max-width:1500px;overflow:hidden;
     box-shadow:0 25px 50px -12px rgba(0,0,0,.6),0 0 0 1px rgba(255,255,255,.03);
 }
+
+/* ── Header ── */
 .app-header{
     background:linear-gradient(135deg,#18181b,#1e1e24);border-bottom:1px solid #27272a;
     padding:14px 24px;display:flex;align-items:center;justify-content:space-between;flex-wrap:wrap;gap:12px;
 }
 .app-header-left{display:flex;align-items:center;gap:12px}
 .app-logo{
-    width:36px;height:36px;background:linear-gradient(135deg,#FF6B00,#FF4500,#FF8C00);
+    width:36px;height:36px;background:linear-gradient(135deg,#FF4500,#FF8533,#FFB366);
     border-radius:10px;display:flex;align-items:center;justify-content:center;
     box-shadow:0 4px 12px rgba(255,69,0,.35);
 }
@@ -280,11 +275,12 @@ footer{display:none!important}
 }
 .app-badge{
     font-size:11px;font-weight:600;padding:3px 10px;border-radius:20px;
-    background:rgba(255,69,0,.15);color:#FF8C00;border:1px solid rgba(255,69,0,.25);letter-spacing:.3px;
+    background:rgba(255,69,0,.15);color:#FF8533;border:1px solid rgba(255,69,0,.25);letter-spacing:.3px;
 }
-.app-badge.std{background:rgba(30,100,255,.12);color:#60A5FA;border:1px solid rgba(30,100,255,.25)}
-.app-badge.small{background:rgba(34,197,94,.12);color:#4ade80;border:1px solid rgba(34,197,94,.25)}
+.app-badge.klein{background:rgba(255,69,0,.12);color:#FF6A33;border:1px solid rgba(255,69,0,.25)}
+.app-badge.fast{background:rgba(34,197,94,.12);color:#4ade80;border:1px solid rgba(34,197,94,.25)}
 
+/* ── Toolbar ── */
 .app-toolbar{
     background:#18181b;border-bottom:1px solid #27272a;padding:8px 16px;
     display:flex;gap:4px;align-items:center;flex-wrap:wrap;
@@ -304,26 +300,28 @@ footer{display:none!important}
 .modern-tb-btn .tb-svg,.modern-tb-btn .tb-svg *{stroke:#ffffff!important;fill:none!important}
 .tb-info{font-family:'JetBrains Mono',monospace;font-size:12px;color:#71717a;padding:0 8px;display:flex;align-items:center}
 
+/* ── Main Layout ── */
 .app-main-row{display:flex;gap:0;flex:1;overflow:hidden}
-.app-main-left{flex:0 0 360px;display:flex;flex-direction:column;min-width:0;border-right:1px solid #27272a;background:#18181b}
-.app-main-right{flex:1;display:flex;flex-direction:column;min-width:0;background:#0f0f13}
+.app-main-left{flex:1;display:flex;flex-direction:column;min-width:0;border-right:1px solid #27272a}
+.app-main-right{width:460px;display:flex;flex-direction:column;flex-shrink:0;background:#18181b}
 
-/* ── Input panel ── */
-#gallery-drop-zone{position:relative;background:#09090b;min-height:260px;overflow:auto}
+/* ── Gallery / Drop Zone ── */
+#gallery-drop-zone{position:relative;background:#09090b;min-height:320px;overflow:auto}
 #gallery-drop-zone.drag-over{outline:2px solid #FF4500;outline-offset:-2px;background:rgba(255,69,0,.04)}
-.upload-prompt-modern{position:absolute;top:50%;left:50%;transform:translate(-50%,-50%);z-index:20;width:90%}
+.upload-prompt-modern{position:absolute;top:50%;left:50%;transform:translate(-50%,-50%);z-index:20}
 .upload-click-area{
     display:flex;flex-direction:column;align-items:center;justify-content:center;
-    cursor:pointer;padding:28px 24px;border:2px dashed #3f3f46;border-radius:16px;
+    cursor:pointer;padding:36px 52px;border:2px dashed #3f3f46;border-radius:16px;
     background:rgba(255,69,0,.03);transition:all .2s ease;gap:8px;
 }
-.upload-click-area:hover{background:rgba(255,69,0,.08);border-color:#FF4500;transform:scale(1.02)}
-.upload-click-area svg{width:64px;height:64px}
-.upload-main-text{color:#71717a;font-size:13px;font-weight:500;margin-top:4px;text-align:center}
-.upload-sub-text{color:#52525b;font-size:11px;text-align:center}
+.upload-click-area:hover{background:rgba(255,69,0,.08);border-color:#FF4500;transform:scale(1.03)}
+.upload-click-area:active{background:rgba(255,69,0,.12);transform:scale(.98)}
+.upload-click-area svg{width:80px;height:80px}
+.upload-main-text{color:#71717a;font-size:14px;font-weight:500;margin-top:4px}
+.upload-sub-text{color:#52525b;font-size:12px}
 .image-gallery-grid{
-    display:grid;grid-template-columns:repeat(auto-fill,minmax(120px,1fr));
-    gap:10px;padding:12px;align-content:start;
+    display:grid;grid-template-columns:repeat(auto-fill,minmax(140px,1fr));
+    gap:12px;padding:16px;align-content:start;
 }
 .gallery-thumb{
     position:relative;aspect-ratio:1;border-radius:10px;overflow:hidden;
@@ -352,26 +350,72 @@ footer{display:none!important}
 .gallery-add-card .add-icon{font-size:28px;color:#71717a;font-weight:300}
 .gallery-add-card .add-text{font-size:12px;color:#71717a;font-weight:500}
 
+/* ── Hint bar ── */
 .hint-bar{
     background:rgba(255,69,0,.06);border-top:1px solid #27272a;border-bottom:1px solid #27272a;
-    padding:8px 16px;font-size:12px;color:#a1a1aa;line-height:1.6;
+    padding:10px 20px;font-size:13px;color:#a1a1aa;line-height:1.7;
 }
-.hint-bar b{color:#FF8C00;font-weight:600}
+.hint-bar b{color:#FF8533;font-weight:600}
 .hint-bar kbd{
     display:inline-block;padding:1px 6px;background:#27272a;border:1px solid #3f3f46;
     border-radius:4px;font-family:'JetBrains Mono',monospace;font-size:11px;color:#a1a1aa;
 }
 
+/* ── Quick Prompts ── */
+.suggestions-section{border-top:1px solid #27272a;padding:12px 16px}
+.suggestions-title,.examples-title{
+    font-size:12px;font-weight:600;color:#71717a;text-transform:uppercase;
+    letter-spacing:.8px;margin-bottom:10px;
+}
+.suggestions-wrap{display:flex;flex-wrap:wrap;gap:6px}
+.suggestion-chip{
+    display:inline-flex;align-items:center;gap:4px;padding:5px 12px;
+    background:rgba(255,69,0,.08);border:1px solid rgba(255,69,0,.2);border-radius:20px;
+    color:#FF8533;font-size:12px;font-weight:500;font-family:'Inter',sans-serif;
+    cursor:pointer;transition:all .15s;white-space:nowrap;
+}
+.suggestion-chip:hover{background:rgba(255,69,0,.15);border-color:rgba(255,69,0,.35);color:#FFB366;transform:translateY(-1px)}
+
+/* ── Example Cards ── */
+.examples-section{border-top:1px solid #27272a;padding:12px 16px}
+.examples-scroll{display:flex;gap:10px;overflow-x:auto;padding-bottom:8px}
+.examples-scroll::-webkit-scrollbar{height:6px}
+.examples-scroll::-webkit-scrollbar-track{background:#09090b;border-radius:3px}
+.examples-scroll::-webkit-scrollbar-thumb{background:#27272a;border-radius:3px}
+.examples-scroll::-webkit-scrollbar-thumb:hover{background:#3f3f46}
+.example-card{
+    flex-shrink:0;width:210px;background:#09090b;border:1px solid #27272a;
+    border-radius:10px;overflow:hidden;cursor:pointer;transition:all .2s ease;
+}
+.example-card:hover{border-color:#FF4500;transform:translateY(-2px);box-shadow:0 4px 12px rgba(255,69,0,.15)}
+.example-card.loading{opacity:.5;pointer-events:none}
+.example-thumbs{display:flex;height:110px;overflow:hidden;background:#18181b}
+.example-thumbs img{flex:1;object-fit:cover;min-width:0;border-bottom:1px solid #27272a}
+.example-thumb-placeholder{
+    flex:1;display:flex;align-items:center;justify-content:center;
+    background:#18181b;color:#3f3f46;font-size:11px;min-width:0;
+}
+.example-meta{padding:6px 10px;display:flex;align-items:center;gap:6px}
+.example-badge{
+    display:inline-flex;padding:2px 7px;background:rgba(255,69,0,.1);border-radius:4px;
+    font-size:10px;font-weight:600;color:#FF8533;font-family:'JetBrains Mono',monospace;white-space:nowrap;
+}
+.example-prompt-text{
+    padding:0 10px 8px;font-size:11px;color:#a1a1aa;line-height:1.4;
+    display:-webkit-box;-webkit-line-clamp:2;-webkit-box-orient:vertical;overflow:hidden;
+}
+
+/* ── Right Panel ── */
 .panel-card{border-bottom:1px solid #27272a}
 .panel-card-title{
-    padding:10px 16px;font-size:12px;font-weight:600;color:#71717a;
+    padding:12px 20px;font-size:12px;font-weight:600;color:#71717a;
     text-transform:uppercase;letter-spacing:.8px;border-bottom:1px solid rgba(39,39,42,.6);
 }
-.panel-card-body{padding:12px 16px;display:flex;flex-direction:column;gap:8px}
+.panel-card-body{padding:16px 20px;display:flex;flex-direction:column;gap:8px}
 .modern-label{font-size:13px;font-weight:500;color:#a1a1aa;margin-bottom:4px;display:block}
 .modern-textarea{
     width:100%;background:#09090b;border:1px solid #27272a;border-radius:8px;
-    padding:10px 14px;font-family:'Inter',sans-serif;font-size:13px;color:#e4e4e7;
+    padding:10px 14px;font-family:'Inter',sans-serif;font-size:14px;color:#e4e4e7;
     resize:vertical;outline:none;min-height:80px;transition:border-color .2s;
 }
 .modern-textarea:focus{border-color:#FF4500;box-shadow:0 0 0 3px rgba(255,69,0,.15)}
@@ -381,170 +425,120 @@ footer{display:none!important}
 }
 @keyframes shake{0%,100%{transform:translateX(0)}20%,60%{transform:translateX(-4px)}40%,80%{transform:translateX(4px)}}
 
+/* ── Run Button ── */
 .btn-run{
     display:flex;align-items:center;justify-content:center;gap:8px;width:100%;
-    background:linear-gradient(135deg,#FF4500,#CC3700);border:none;border-radius:10px;
+    background:linear-gradient(135deg,#FF4500,#E63E00);border:none;border-radius:10px;
     padding:12px 24px;cursor:pointer;font-size:15px;font-weight:600;font-family:'Inter',sans-serif;
-    color:#ffffff!important;-webkit-text-fill-color:#ffffff!important;transition:all .2s ease;
+    color:#ffffff!important;-webkit-text-fill-color:#ffffff!important;transition:all .2s ease;letter-spacing:-.2px;
     box-shadow:0 4px 16px rgba(255,69,0,.3),inset 0 1px 0 rgba(255,255,255,.1);
 }
 .btn-run:hover{
-    background:linear-gradient(135deg,#FF6B00,#FF4500);transform:translateY(-1px);
+    background:linear-gradient(135deg,#FF6A33,#FF4500);transform:translateY(-1px);
     box-shadow:0 6px 24px rgba(255,69,0,.45),inset 0 1px 0 rgba(255,255,255,.15);
 }
 .btn-run:active{transform:translateY(0);box-shadow:0 2px 8px rgba(255,69,0,.3)}
 .btn-run svg{width:18px;height:18px;fill:#ffffff!important}
 
-/* ── Output panels ── */
-.output-area{flex:1;display:flex;flex-direction:column;gap:0;overflow:hidden}
-.output-cols{display:flex;flex:1;gap:0;overflow:hidden}
-.output-col{
-    flex:1;display:flex;flex-direction:column;border-left:1px solid #27272a;
-    background:#0f0f13;min-width:0;
-}
-.output-col:first-child{border-left:none}
-.out-col-header{
-    padding:10px 16px;border-bottom:1px solid #27272a;
-    display:flex;align-items:center;justify-content:space-between;background:#18181b;
+/* ── Toast ── */
+.toast-notification{
+    position:fixed;top:24px;left:50%;transform:translateX(-50%) translateY(-120%);
+    z-index:9999;padding:10px 24px;border-radius:10px;font-family:'Inter',sans-serif;
+    font-size:14px;font-weight:600;display:flex;align-items:center;gap:8px;
+    box-shadow:0 8px 24px rgba(0,0,0,.5);
+    transition:transform .35s cubic-bezier(.34,1.56,.64,1),opacity .35s ease;opacity:0;pointer-events:none;
 }
-.out-col-title{font-size:12px;font-weight:700;letter-spacing:.5px;text-transform:uppercase}
-.out-col-title.std{color:#60A5FA}
-.out-col-title.small{color:#4ade80}
-.out-col-badge{
-    font-size:10px;font-weight:600;padding:2px 8px;border-radius:10px;
-    font-family:'JetBrains Mono',monospace;white-space:nowrap;
+.toast-notification.visible{transform:translateX(-50%) translateY(0);opacity:1;pointer-events:auto}
+.toast-notification.error{background:linear-gradient(135deg,#dc2626,#b91c1c);color:#fff;border:1px solid rgba(255,255,255,.15)}
+.toast-notification.warning{background:linear-gradient(135deg,#d97706,#b45309);color:#fff;border:1px solid rgba(255,255,255,.15)}
+.toast-notification.info{background:linear-gradient(135deg,#FF4500,#E63E00);color:#fff;border:1px solid rgba(255,255,255,.15)}
+
+/* ── Output Panel ── */
+.output-panel{padding:16px;display:flex;flex-direction:column;gap:12px;flex:1}
+.output-compare-row{display:grid;grid-template-columns:1fr 1fr;gap:12px}
+.output-card{
+    background:#09090b;border:1px solid #27272a;border-radius:10px;overflow:hidden;
+    display:flex;flex-direction:column;transition:border-color .2s;
 }
-.out-col-badge.std{background:rgba(30,100,255,.12);color:#60A5FA;border:1px solid rgba(30,100,255,.25)}
-.out-col-badge.small{background:rgba(34,197,94,.12);color:#4ade80;border:1px solid rgba(34,197,94,.25)}
-.out-col-body{
-    flex:1;background:#09090b;display:flex;align-items:center;justify-content:center;
-    overflow:hidden;min-height:300px;position:relative;
+.output-card:hover{border-color:#3f3f46}
+.output-card-header{
+    padding:8px 12px;border-bottom:1px solid #27272a;display:flex;align-items:center;
+    justify-content:space-between;background:rgba(24,24,27,.8);
 }
-.out-col-body img.modern-out-img{max-width:100%;max-height:500px;image-rendering:auto;display:block}
-.out-col-placeholder{color:#3f3f46;font-size:12px;text-align:center;padding:16px}
-.out-col-footer{
-    padding:8px 16px;border-top:1px solid #27272a;background:#18181b;
-    display:flex;align-items:center;justify-content:space-between;min-height:38px;
+.output-card-label{font-size:12px;font-weight:700;color:#ffffff!important;-webkit-text-fill-color:#ffffff!important}
+.output-card-label.std{color:#60A5FA!important;-webkit-text-fill-color:#60A5FA!important}
+.output-card-label.small{color:#4ADE80!important;-webkit-text-fill-color:#4ADE80!important}
+.output-card-body{
+    flex:1;display:flex;align-items:center;justify-content:center;
+    background:#09090b;overflow:hidden;min-height:200px;position:relative;
 }
-.timing-display{
-    font-family:'JetBrains Mono',monospace;font-size:11px;color:#71717a;
+.output-card-body img{max-width:100%;max-height:300px;object-fit:contain}
+.out-placeholder-text{color:#3f3f46;font-size:12px;text-align:center}
+.timing-badge{
+    font-family:'JetBrains Mono',monospace;font-size:10px;font-weight:600;
+    padding:2px 8px;border-radius:4px;background:rgba(255,255,255,.05);color:#71717a;
+    border:1px solid #27272a;white-space:nowrap;
 }
 .out-download-btn{
     display:none;align-items:center;justify-content:center;background:rgba(255,69,0,.1);
     border:1px solid rgba(255,69,0,.2);border-radius:6px;cursor:pointer;padding:3px 10px;
-    font-size:11px;font-weight:500;color:#FF8C00!important;gap:4px;height:24px;transition:all .15s;
+    font-size:11px;font-weight:500;color:#FF8533!important;gap:4px;height:24px;transition:all .15s;
 }
 .out-download-btn:hover{background:rgba(255,69,0,.2);border-color:rgba(255,69,0,.35);color:#ffffff!important}
 .out-download-btn.visible{display:inline-flex}
-.out-download-btn svg{width:12px;height:12px;fill:#FF8C00}
+.out-download-btn svg{width:12px;height:12px;fill:#FF8533}
 
 /* ── Loader ── */
 .modern-loader{
     display:none;position:absolute;top:0;left:0;right:0;bottom:0;background:rgba(9,9,11,.92);
-    z-index:15;flex-direction:column;align-items:center;justify-content:center;gap:16px;backdrop-filter:blur(4px);
+    z-index:15;flex-direction:column;align-items:center;justify-content:center;gap:12px;backdrop-filter:blur(4px);
 }
 .modern-loader.active{display:flex}
 .modern-loader .loader-spinner{
-    width:36px;height:36px;border:3px solid #27272a;border-top-color:#FF4500;
+    width:32px;height:32px;border:3px solid #27272a;border-top-color:#FF4500;
     border-radius:50%;animation:spin .8s linear infinite;
 }
 @keyframes spin{to{transform:rotate(360deg)}}
-.modern-loader .loader-text{font-size:12px;color:#a1a1aa;font-weight:500;text-align:center}
-.loader-bar-track{width:160px;height:4px;background:#27272a;border-radius:2px;overflow:hidden}
+.modern-loader .loader-text{font-size:12px;color:#a1a1aa;font-weight:500}
+.loader-bar-track{width:180px;height:4px;background:#27272a;border-radius:2px;overflow:hidden}
 .loader-bar-fill{
-    height:100%;background:linear-gradient(90deg,#FF4500,#FF8C00,#FF4500);
+    height:100%;background:linear-gradient(90deg,#FF4500,#FF8533,#FF4500);
     background-size:200% 100%;animation:shimmer 1.5s ease-in-out infinite;border-radius:2px;
 }
 @keyframes shimmer{0%{background-position:200% 0}100%{background-position:-200% 0}}
 
-/* ── Toast ── */
-.toast-notification{
-    position:fixed;top:24px;left:50%;transform:translateX(-50%) translateY(-120%);
-    z-index:9999;padding:10px 24px;border-radius:10px;font-family:'Inter',sans-serif;
-    font-size:14px;font-weight:600;display:flex;align-items:center;gap:8px;
-    box-shadow:0 8px 24px rgba(0,0,0,.5);
-    transition:transform .35s cubic-bezier(.34,1.56,.64,1),opacity .35s ease;opacity:0;pointer-events:none;
-}
-.toast-notification.visible{transform:translateX(-50%) translateY(0);opacity:1;pointer-events:auto}
-.toast-notification.error{background:linear-gradient(135deg,#dc2626,#b91c1c);color:#fff;border:1px solid rgba(255,255,255,.15)}
-.toast-notification.warning{background:linear-gradient(135deg,#d97706,#b45309);color:#fff;border:1px solid rgba(255,255,255,.15)}
-.toast-notification.info{background:linear-gradient(135deg,#FF4500,#CC3700);color:#fff;border:1px solid rgba(255,255,255,.15)}
-.toast-notification .toast-icon{font-size:16px;line-height:1}
-.toast-notification .toast-text{line-height:1.3}
-
-/* ── Settings ── */
-.settings-group{border:1px solid #27272a;border-radius:10px;margin:12px 16px;padding:0;overflow:hidden}
+/* ── Settings Group ── */
+.settings-group{border:1px solid #27272a;border-radius:10px;margin:12px 16px;overflow:hidden}
 .settings-group-title{
     font-size:12px;font-weight:600;color:#71717a;text-transform:uppercase;letter-spacing:.8px;
-    padding:8px 14px;border-bottom:1px solid #27272a;background:rgba(24,24,27,.5);
+    padding:10px 16px;border-bottom:1px solid #27272a;background:rgba(24,24,27,.5);
 }
-.settings-group-body{padding:12px 14px;display:flex;flex-direction:column;gap:10px}
-.slider-row{display:flex;align-items:center;gap:10px;min-height:26px}
-.slider-row label{font-size:12px;font-weight:500;color:#a1a1aa;min-width:80px;flex-shrink:0}
+.settings-group-body{padding:14px 16px;display:flex;flex-direction:column;gap:12px}
+.slider-row{display:flex;align-items:center;gap:10px;min-height:28px}
+.slider-row label{font-size:13px;font-weight:500;color:#a1a1aa;min-width:72px;flex-shrink:0}
 .slider-row input[type="range"]{
     flex:1;-webkit-appearance:none;appearance:none;height:6px;background:#27272a;
     border-radius:3px;outline:none;min-width:0;
 }
 .slider-row input[type="range"]::-webkit-slider-thumb{
-    -webkit-appearance:none;width:16px;height:16px;background:linear-gradient(135deg,#FF4500,#CC3700);
+    -webkit-appearance:none;width:16px;height:16px;background:linear-gradient(135deg,#FF4500,#E63E00);
     border-radius:50%;cursor:pointer;box-shadow:0 2px 6px rgba(255,69,0,.4);transition:transform .15s;
 }
 .slider-row input[type="range"]::-webkit-slider-thumb:hover{transform:scale(1.2)}
 .slider-row input[type="range"]::-moz-range-thumb{
-    width:16px;height:16px;background:linear-gradient(135deg,#FF4500,#CC3700);
+    width:16px;height:16px;background:linear-gradient(135deg,#FF4500,#E63E00);
     border-radius:50%;cursor:pointer;border:none;
 }
 .slider-row .slider-val{
-    min-width:52px;text-align:right;font-family:'JetBrains Mono',monospace;font-size:11px;
+    min-width:52px;text-align:right;font-family:'JetBrains Mono',monospace;font-size:12px;
     font-weight:500;padding:3px 8px;background:#09090b;border:1px solid #27272a;
     border-radius:6px;color:#a1a1aa;flex-shrink:0;
 }
 .checkbox-row{display:flex;align-items:center;gap:8px;font-size:13px;color:#a1a1aa}
 .checkbox-row input[type="checkbox"]{accent-color:#FF4500;width:16px;height:16px;cursor:pointer}
+.slider-row-2col{display:grid;grid-template-columns:1fr 1fr;gap:12px}
 
-/* ── Suggestions / Examples ── */
-.suggestions-section{border-top:1px solid #27272a;padding:10px 14px}
-.suggestions-title,.examples-title{
-    font-size:11px;font-weight:600;color:#71717a;text-transform:uppercase;
-    letter-spacing:.8px;margin-bottom:8px;
-}
-.suggestions-wrap{display:flex;flex-wrap:wrap;gap:5px}
-.suggestion-chip{
-    display:inline-flex;align-items:center;gap:4px;padding:4px 10px;
-    background:rgba(255,69,0,.08);border:1px solid rgba(255,69,0,.2);border-radius:20px;
-    color:#FF8C00;font-size:11px;font-weight:500;font-family:'Inter',sans-serif;
-    cursor:pointer;transition:all .15s;white-space:nowrap;
-}
-.suggestion-chip:hover{background:rgba(255,69,0,.15);border-color:rgba(255,69,0,.35);color:#FFB366;transform:translateY(-1px)}
-
-.examples-section{border-top:1px solid #27272a;padding:10px 14px}
-.examples-scroll{display:flex;gap:8px;overflow-x:auto;padding-bottom:6px}
-.examples-scroll::-webkit-scrollbar{height:4px}
-.examples-scroll::-webkit-scrollbar-track{background:#09090b;border-radius:3px}
-.examples-scroll::-webkit-scrollbar-thumb{background:#27272a;border-radius:3px}
-.example-card{
-    flex-shrink:0;width:180px;background:#09090b;border:1px solid #27272a;
-    border-radius:10px;overflow:hidden;cursor:pointer;transition:all .2s ease;
-}
-.example-card:hover{border-color:#FF4500;transform:translateY(-2px);box-shadow:0 4px 12px rgba(255,69,0,.15)}
-.example-card.loading{opacity:.5;pointer-events:none}
-.example-thumbs{display:flex;height:90px;overflow:hidden;background:#18181b}
-.example-thumbs img{flex:1;object-fit:cover;min-width:0;border-bottom:1px solid #27272a}
-.example-thumb-placeholder{
-    flex:1;display:flex;align-items:center;justify-content:center;
-    background:#18181b;color:#3f3f46;font-size:11px;min-width:0;
-}
-.example-meta{padding:5px 8px;display:flex;align-items:center;gap:6px}
-.example-badge{
-    display:inline-flex;padding:2px 6px;background:rgba(255,69,0,.1);border-radius:4px;
-    font-size:10px;font-weight:600;color:#FF8C00;font-family:'JetBrains Mono',monospace;white-space:nowrap;
-}
-.example-prompt-text{
-    padding:0 8px 7px;font-size:10px;color:#a1a1aa;line-height:1.4;
-    display:-webkit-box;-webkit-line-clamp:2;-webkit-box-orient:vertical;overflow:hidden;
-}
-
-/* ── Statusbar ── */
+/* ── Status Bar ── */
 .app-statusbar{
     background:#18181b;border-top:1px solid #27272a;padding:6px 20px;
     display:flex;gap:12px;height:34px;align-items:center;font-size:12px;
@@ -555,10 +549,10 @@ footer{display:none!important}
 }
 .app-statusbar .sb-section.sb-fixed{
     flex:0 0 auto;min-width:90px;text-align:center;justify-content:center;
-    padding:3px 12px;background:rgba(255,69,0,.08);border-radius:6px;color:#FF8C00;font-weight:500;
+    padding:3px 12px;background:rgba(255,69,0,.08);border-radius:6px;color:#FF8533;font-weight:500;
 }
 .exp-note{padding:10px 20px;font-size:12px;color:#52525b;border-top:1px solid #27272a;text-align:center}
-.exp-note a{color:#FF8C00;text-decoration:none}
+.exp-note a{color:#FF8533;text-decoration:none}
 .exp-note a:hover{text-decoration:underline}
 
 ::-webkit-scrollbar{width:8px;height:8px}
@@ -568,12 +562,14 @@ footer{display:none!important}
 
 @media(max-width:900px){
     .app-main-row{flex-direction:column}
-    .app-main-left{flex:0 0 auto;border-right:none;border-bottom:1px solid #27272a}
-    .output-cols{flex-direction:column}
-    .output-col{border-left:none;border-top:1px solid #27272a}
+    .app-main-right{width:100%}
+    .app-main-left{border-right:none;border-bottom:1px solid #27272a}
+    .output-compare-row{grid-template-columns:1fr}
 }
 """
 
+# ── Gallery JS ─────────────────────────────────────────────────────────────────
+
 gallery_js = r"""
 () => {
 function init() {
@@ -613,7 +609,9 @@ function init() {
         const icon = toast.querySelector('.toast-icon');
         const text = toast.querySelector('.toast-text');
         toast.className = 'toast-notification ' + (type || 'error');
-        icon.textContent = type === 'warning' ? '\u26A0' : type === 'info' ? '\u2139' : '\u2717';
+        if (type === 'warning') icon.textContent = '\u26A0';
+        else if (type === 'info') icon.textContent = '\u2139';
+        else icon.textContent = '\u2717';
         text.textContent = message;
         if (toastTimer) clearTimeout(toastTimer);
         void toast.offsetWidth;
@@ -660,11 +658,11 @@ function init() {
         const n   = images.length;
         const txt = n > 0 ? n + ' image' + (n > 1 ? 's' : '') : 'No images';
         if (imgCountTb) imgCountTb.textContent = txt;
-        if (imgCountSb) imgCountSb.textContent = n > 0 ? txt + ' uploaded' : 'No images uploaded';
+        if (imgCountSb) imgCountSb.textContent = n > 0 ? txt + ' loaded' : 'No images loaded';
     }
 
     function addImage(b64, name) {
-        images.push({id: Date.now() + Math.random(), b64, name});
+        images.push({id: Date.now() + Math.random(), b64: b64, name: name});
         renderGallery();
         syncImagesToGradio();
     }
@@ -687,6 +685,11 @@ function init() {
     }
     window.__clearAll = clearAll;
 
+    function selectImage(idx) {
+        selectedIdx = (selectedIdx === idx) ? -1 : idx;
+        renderGallery();
+    }
+
     function renderGallery() {
         if (images.length === 0) {
             galleryGrid.innerHTML = '';
@@ -696,30 +699,30 @@ function init() {
         }
         if (uploadPrompt) uploadPrompt.style.display = 'none';
         galleryGrid.style.display = 'grid';
-
         let html = '';
         images.forEach((img, i) => {
             const sel = i === selectedIdx ? ' selected' : '';
-            html += `<div class="gallery-thumb${sel}" data-idx="${i}">
-                <img src="${img.b64}" alt="${img.name||'image'}">
-                <span class="thumb-badge">#${i+1}</span>
-                <button class="thumb-remove" data-remove="${i}">\u2715</button>
-            </div>`;
+            html += '<div class="gallery-thumb' + sel + '" data-idx="' + i + '">'
+                  + '<img src="' + img.b64 + '" alt="' + (img.name||'image') + '">'
+                  + '<span class="thumb-badge">#' + (i+1) + '</span>'
+                  + '<button class="thumb-remove" data-remove="' + i + '">\u2715</button>'
+                  + '</div>';
         });
-        html += `<div class="gallery-add-card" id="gallery-add-card">
-            <span class="add-icon">+</span><span class="add-text">Add</span>
-        </div>`;
+        html += '<div class="gallery-add-card" id="gallery-add-card">'
+              + '<span class="add-icon">+</span><span class="add-text">Add</span></div>';
         galleryGrid.innerHTML = html;
 
         galleryGrid.querySelectorAll('.gallery-thumb').forEach(thumb => {
-            thumb.addEventListener('click', e => {
+            thumb.addEventListener('click', (e) => {
                 if (e.target.closest('.thumb-remove')) return;
-                selectedIdx = selectedIdx === parseInt(thumb.dataset.idx) ? -1 : parseInt(thumb.dataset.idx);
-                renderGallery();
+                selectImage(parseInt(thumb.dataset.idx));
             });
         });
         galleryGrid.querySelectorAll('.thumb-remove').forEach(btn => {
-            btn.addEventListener('click', e => { e.stopPropagation(); removeImage(parseInt(btn.dataset.remove)); });
+            btn.addEventListener('click', (e) => {
+                e.stopPropagation();
+                removeImage(parseInt(btn.dataset.remove));
+            });
         });
         const addCard = document.getElementById('gallery-add-card');
         if (addCard) addCard.addEventListener('click', () => fileInput.click());
@@ -729,20 +732,22 @@ function init() {
         Array.from(files).forEach(file => {
             if (!file.type.startsWith('image/')) return;
             const reader = new FileReader();
-            reader.onload = e => addImage(e.target.result, file.name);
+            reader.onload = (e) => addImage(e.target.result, file.name);
             reader.readAsDataURL(file);
         });
     }
 
-    fileInput.addEventListener('change', e => { processFiles(e.target.files); e.target.value = ''; });
+    fileInput.addEventListener('change', (e) => { processFiles(e.target.files); e.target.value = ''; });
     if (uploadClick) uploadClick.addEventListener('click', () => fileInput.click());
     if (btnUpload)   btnUpload.addEventListener('click', () => fileInput.click());
-    if (btnRemove)   btnRemove.addEventListener('click', () => { if (selectedIdx >= 0) removeImage(selectedIdx); });
+    if (btnRemove)   btnRemove.addEventListener('click', () => {
+        if (selectedIdx >= 0 && selectedIdx < images.length) removeImage(selectedIdx);
+    });
     if (btnClear)    btnClear.addEventListener('click', clearAll);
 
-    dropZone.addEventListener('dragover',  e => { e.preventDefault(); dropZone.classList.add('drag-over'); });
-    dropZone.addEventListener('dragleave', e => { e.preventDefault(); dropZone.classList.remove('drag-over'); });
-    dropZone.addEventListener('drop', e => {
+    dropZone.addEventListener('dragover', (e) => { e.preventDefault(); dropZone.classList.add('drag-over'); });
+    dropZone.addEventListener('dragleave', (e) => { e.preventDefault(); dropZone.classList.remove('drag-over'); });
+    dropZone.addEventListener('drop', (e) => {
         e.preventDefault(); dropZone.classList.remove('drag-over');
         if (e.dataTransfer.files.length) processFiles(e.dataTransfer.files);
     });
@@ -753,7 +758,7 @@ function init() {
         if (promptInput) { promptInput.value = text; syncPromptToGradio(); }
     };
 
-    // Example cards
+    /* ── Example cards ── */
     document.querySelectorAll('.example-card[data-idx]').forEach(card => {
         card.addEventListener('click', () => {
             const idx = card.getAttribute('data-idx');
@@ -770,13 +775,13 @@ function init() {
         });
     });
 
-    // Sliders
+    /* ── Slider sync ── */
     function syncSlider(customId, gradioId) {
         const slider = document.getElementById(customId);
         const valSpan = document.getElementById(customId + '-val');
         if (!slider) return;
         slider.addEventListener('input', () => {
-            if (valSpan) valSpan.textContent = slider.value;
+            if (valSpan) valSpan.textContent = parseFloat(slider.value).toFixed(slider.step && slider.step < 1 ? 1 : 0);
             const container = document.getElementById(gradioId);
             if (!container) return;
             container.querySelectorAll('input[type="range"],input[type="number"]').forEach(el => {
@@ -789,34 +794,43 @@ function init() {
             });
         });
     }
-    syncSlider('custom-seed',     'gradio-seed');
+    syncSlider('custom-seed', 'gradio-seed');
     syncSlider('custom-guidance', 'gradio-guidance');
-    syncSlider('custom-steps',    'gradio-steps');
-    syncSlider('custom-width',    'gradio-width');
-    syncSlider('custom-height',   'gradio-height');
+    syncSlider('custom-steps', 'gradio-steps');
+    syncSlider('custom-width', 'gradio-width');
+    syncSlider('custom-height', 'gradio-height');
 
     const randCheck = document.getElementById('custom-randomize');
-    if (randCheck) randCheck.addEventListener('change', () => {
-        const container = document.getElementById('gradio-randomize');
-        if (!container) return;
-        const cb = container.querySelector('input[type="checkbox"]');
-        if (cb && cb.checked !== randCheck.checked) cb.click();
-    });
+    if (randCheck) {
+        randCheck.addEventListener('change', () => {
+            const container = document.getElementById('gradio-randomize');
+            if (!container) return;
+            const cb = container.querySelector('input[type="checkbox"]');
+            if (cb && cb.checked !== randCheck.checked) cb.click();
+        });
+    }
 
-    // Loader
+    /* ── Loader ── */
     function showLoader() {
-        document.querySelectorAll('.modern-loader').forEach(l => l.classList.add('active'));
+        ['output-loader-std','output-loader-small'].forEach(id => {
+            const l = document.getElementById(id);
+            if (l) l.classList.add('active');
+        });
         const sb = document.querySelector('.sb-fixed');
         if (sb) sb.textContent = 'Processing...';
     }
     function hideLoader() {
-        document.querySelectorAll('.modern-loader').forEach(l => l.classList.remove('active'));
+        ['output-loader-std','output-loader-small'].forEach(id => {
+            const l = document.getElementById(id);
+            if (l) l.classList.remove('active');
+        });
         const sb = document.querySelector('.sb-fixed');
         if (sb) sb.textContent = 'Done';
     }
     window.__showLoader = showLoader;
     window.__hideLoader = hideLoader;
 
+    /* ── Validate & run ── */
     function validateBeforeRun() {
         const promptVal = promptInput ? promptInput.value.trim() : '';
         if (!promptVal) { showToast('Please enter a prompt', 'warning'); flashPromptError(); return false; }
@@ -846,79 +860,77 @@ init();
 wire_outputs_js = r"""
 () => {
 function watchOutputs() {
-    // Watch both result containers
-    const containers = [
-        {id: 'gradio-result-std',   bodyId: 'output-std-body',   dlId: 'dl-btn-std',   timingId: 'timing-std',   suffix: 'std_output.png'},
-        {id: 'gradio-result-small', bodyId: 'output-small-body', dlId: 'dl-btn-small', timingId: 'timing-small', suffix: 'small_output.png'},
-    ];
-
-    let ready = containers.every(c => document.getElementById(c.id) && document.getElementById(c.bodyId));
-    if (!ready) { setTimeout(watchOutputs, 500); return; }
-
-    containers.forEach(c => {
-        const dlBtn = document.getElementById(c.dlId);
-        if (dlBtn) {
-            dlBtn.addEventListener('click', e => {
-                e.stopPropagation();
-                const img = document.getElementById(c.bodyId).querySelector('img.modern-out-img');
-                if (img && img.src) {
-                    const a = document.createElement('a');
-                    a.href = img.src; a.download = c.suffix;
-                    document.body.appendChild(a); a.click(); document.body.removeChild(a);
-                }
-            });
-        }
-    });
+    const stdContainer   = document.getElementById('gradio-result-std');
+    const smallContainer = document.getElementById('gradio-result-small');
+    const stdBody        = document.getElementById('output-body-std');
+    const smallBody      = document.getElementById('output-body-small');
+    const dlStd          = document.getElementById('dl-btn-std');
+    const dlSmall        = document.getElementById('dl-btn-small');
+
+    if (!stdContainer || !smallContainer) { setTimeout(watchOutputs, 500); return; }
+
+    function makeDownload(btn, bodyEl, filename) {
+        if (!btn) return;
+        btn.addEventListener('click', (e) => {
+            e.stopPropagation();
+            const img = bodyEl.querySelector('img.modern-out-img');
+            if (img && img.src) {
+                const a = document.createElement('a');
+                a.href = img.src; a.download = filename;
+                document.body.appendChild(a); a.click(); document.body.removeChild(a);
+            }
+        });
+    }
+    makeDownload(dlStd,   stdBody,   'flux_standard_vae.png');
+    makeDownload(dlSmall, smallBody, 'flux_small_decoder.png');
 
-    function syncImage(c) {
-        const resultContainer = document.getElementById(c.id);
-        const outBody = document.getElementById(c.bodyId);
-        if (!resultContainer || !outBody) return;
-        const resultImg = resultContainer.querySelector('img');
+    function syncOne(container, body, loaderId, dlBtn) {
+        const resultImg = container.querySelector('img');
         if (resultImg && resultImg.src) {
-            const ph = outBody.querySelector('.out-col-placeholder');
+            const ph = body.querySelector('.out-placeholder-text');
             if (ph) ph.style.display = 'none';
-            let existing = outBody.querySelector('img.modern-out-img');
-            if (!existing) { existing = document.createElement('img'); existing.className = 'modern-out-img'; outBody.appendChild(existing); }
+            let existing = body.querySelector('img.modern-out-img');
+            if (!existing) { existing = document.createElement('img'); existing.className = 'modern-out-img'; body.appendChild(existing); }
             if (existing.src !== resultImg.src) {
                 existing.src = resultImg.src;
-                const dlBtn = document.getElementById(c.dlId);
                 if (dlBtn) dlBtn.classList.add('visible');
-                if (window.__hideLoader) window.__hideLoader();
+                const l = document.getElementById(loaderId);
+                if (l) l.classList.remove('active');
             }
         }
     }
 
-    containers.forEach(c => {
-        const resultContainer = document.getElementById(c.id);
-        if (!resultContainer) return;
-        const observer = new MutationObserver(() => syncImage(c));
-        observer.observe(resultContainer, {childList:true, subtree:true, attributes:true, attributeFilter:['src']});
-        setInterval(() => syncImage(c), 800);
-    });
+    function syncAll() {
+        syncOne(stdContainer,   stdBody,   'output-loader-std',   dlStd);
+        syncOne(smallContainer, smallBody, 'output-loader-small', dlSmall);
+    }
+
+    const obs = new MutationObserver(syncAll);
+    obs.observe(stdContainer,   {childList:true,subtree:true,attributes:true,attributeFilter:['src']});
+    obs.observe(smallContainer, {childList:true,subtree:true,attributes:true,attributeFilter:['src']});
+    setInterval(syncAll, 800);
 }
 watchOutputs();
 
 function watchTimings() {
-    const timingMap = [
-        {gradioId: 'gradio-timing-std',   displayId: 'timing-std-display'},
-        {gradioId: 'gradio-timing-small', displayId: 'timing-small-display'},
-    ];
-    timingMap.forEach(t => {
-        function sync() {
-            const container = document.getElementById(t.gradioId);
-            const display   = document.getElementById(t.displayId);
-            if (!container || !display) return;
-            const el = container.querySelector('textarea, input');
-            if (el && el.value && el.value.trim()) display.textContent = el.value;
-        }
-        const container = document.getElementById(t.gradioId);
-        if (container) {
-            const obs = new MutationObserver(sync);
-            obs.observe(container, {childList:true, subtree:true, characterData:true, attributes:true});
-        }
-        setInterval(sync, 600);
-    });
+    function syncTiming(gradioId, badgeId) {
+        const container = document.getElementById(gradioId);
+        const badge     = document.getElementById(badgeId);
+        if (!container || !badge) return;
+        const obs = new MutationObserver(() => {
+            const el = container.querySelector('p, span, div');
+            if (el && el.textContent && el.textContent.trim()) {
+                badge.textContent = el.textContent.trim();
+            }
+        });
+        obs.observe(container, {childList:true, subtree:true, characterData:true});
+        setInterval(() => {
+            const el = container.querySelector('p, span, div');
+            if (el && el.textContent && el.textContent.trim()) badge.textContent = el.textContent.trim();
+        }, 800);
+    }
+    syncTiming('gradio-timing-std',   'timing-badge-std');
+    syncTiming('gradio-timing-small', 'timing-badge-small');
 }
 watchTimings();
 
@@ -931,8 +943,7 @@ function watchSeed() {
         const el = seedContainer.querySelector('input[type="range"],input[type="number"]');
         if (el && el.value) { seedSlider.value = el.value; if (seedVal) seedVal.textContent = el.value; }
     }
-    const obs = new MutationObserver(sync);
-    obs.observe(seedContainer, {childList:true, subtree:true, attributes:true});
+    new MutationObserver(sync).observe(seedContainer, {childList:true,subtree:true,attributes:true});
     setInterval(sync, 1000);
 }
 watchSeed();
@@ -964,59 +975,61 @@ function watchExampleResults() {
                 document.querySelectorAll('.example-card.loading').forEach(c => c.classList.remove('loading'));
                 if (window.__showToast) window.__showToast('Could not load example images', 'error');
             }
-        } catch(e) {}
+        } catch(e) { console.error('Example parse error:', e); }
     }
-    const obs = new MutationObserver(checkResult);
-    obs.observe(container, {childList:true, subtree:true, characterData:true, attributes:true});
+    new MutationObserver(checkResult).observe(container, {childList:true,subtree:true,characterData:true,attributes:true});
     setInterval(checkResult, 500);
 }
 watchExampleResults();
 }
 """
 
+# ── Gradio App ────────────────────────────────────────────────────────────────
+
 with gr.Blocks(css=css) as demo:
 
-    # ── Hidden Gradio state ───────────────────────────────────────────────────
-    hidden_images_b64  = gr.Textbox(value="[]",  elem_id="hidden-images-b64",      elem_classes="hidden-input", container=False)
-    prompt_hidden      = gr.Textbox(value="",    elem_id="prompt-gradio-input",     elem_classes="hidden-input", container=False)
-    seed_slider        = gr.Slider(minimum=0, maximum=MAX_SEED, step=1, value=0,    elem_id="gradio-seed",       elem_classes="hidden-input", container=False)
-    randomize_seed     = gr.Checkbox(value=True,                                    elem_id="gradio-randomize",  elem_classes="hidden-input", container=False)
-    guidance_scale     = gr.Slider(minimum=0.0, maximum=10.0, step=0.1, value=1.0, elem_id="gradio-guidance",   elem_classes="hidden-input", container=False)
-    steps_slider       = gr.Slider(minimum=1, maximum=20, step=1, value=4,         elem_id="gradio-steps",      elem_classes="hidden-input", container=False)
-    width_slider       = gr.Slider(minimum=256, maximum=1024, step=8, value=1024,  elem_id="gradio-width",      elem_classes="hidden-input", container=False)
-    height_slider      = gr.Slider(minimum=256, maximum=1024, step=8, value=1024,  elem_id="gradio-height",     elem_classes="hidden-input", container=False)
-
-    result_std         = gr.Image(elem_id="gradio-result-std",   elem_classes="hidden-input", container=False, format="png")
-    result_small_dec   = gr.Image(elem_id="gradio-result-small", elem_classes="hidden-input", container=False, format="png")
-    seed_used          = gr.Number(value=0,  elem_id="gradio-seed-used",     elem_classes="hidden-input", container=False)
-    timing_std_hidden  = gr.Textbox(value="", elem_id="gradio-timing-std",   elem_classes="hidden-input", container=False)
-    timing_small_hidden= gr.Textbox(value="", elem_id="gradio-timing-small", elem_classes="hidden-input", container=False)
-
-    example_idx        = gr.Textbox(value="", elem_id="example-idx-input",   elem_classes="hidden-input", container=False)
-    example_result     = gr.Textbox(value="", elem_id="example-result-data", elem_classes="hidden-input", container=False)
-    example_load_btn   = gr.Button("Load Example", elem_id="example-load-btn")
+    # Hidden Gradio state components
+    hidden_images_b64 = gr.Textbox(value="[]",  elem_id="hidden-images-b64",    elem_classes="hidden-input", container=False)
+    prompt            = gr.Textbox(value="",    elem_id="prompt-gradio-input",   elem_classes="hidden-input", container=False)
+    seed              = gr.Slider(minimum=0, maximum=MAX_SEED, step=1, value=0,  elem_id="gradio-seed",       elem_classes="hidden-input", container=False)
+    randomize_seed    = gr.Checkbox(value=True,                                  elem_id="gradio-randomize",  elem_classes="hidden-input", container=False)
+    guidance_scale    = gr.Slider(minimum=0.0, maximum=10.0, step=0.1, value=1.0,elem_id="gradio-guidance",   elem_classes="hidden-input", container=False)
+    steps             = gr.Slider(minimum=1, maximum=20, step=1, value=4,        elem_id="gradio-steps",      elem_classes="hidden-input", container=False)
+    width_gr          = gr.Slider(minimum=256, maximum=1024, step=8, value=1024, elem_id="gradio-width",      elem_classes="hidden-input", container=False)
+    height_gr         = gr.Slider(minimum=256, maximum=1024, step=8, value=1024, elem_id="gradio-height",     elem_classes="hidden-input", container=False)
+
+    result_std        = gr.Image(elem_id="gradio-result-std",   elem_classes="hidden-input", container=False, format="png")
+    result_small_gr   = gr.Image(elem_id="gradio-result-small", elem_classes="hidden-input", container=False, format="png")
+    seed_output       = gr.Number(elem_id="gradio-seed-out",    elem_classes="hidden-input", container=False)
+    timing_std_gr     = gr.Markdown(elem_id="gradio-timing-std",   elem_classes="hidden-input")
+    timing_small_gr   = gr.Markdown(elem_id="gradio-timing-small", elem_classes="hidden-input")
+
+    example_idx       = gr.Textbox(value="", elem_id="example-idx-input",  elem_classes="hidden-input", container=False)
+    example_result    = gr.Textbox(value="", elem_id="example-result-data", elem_classes="hidden-input", container=False)
+    example_load_btn  = gr.Button("Load Example", elem_id="example-load-btn")
 
     gr.HTML(f"""
     <div class="app-shell">
 
+        <!-- Header -->
         <div class="app-header">
             <div class="app-header-left">
-                <div class="app-logo">{FLUX_LOGO_SVG}</div>
-                <span class="app-title">FLUX.2-klein-4B Comparator</span>
-                <span class="app-badge">v1.0</span>
-                <span class="app-badge std">Standard VAE</span>
-                <span class="app-badge small">Small Decoder</span>
+                <div class="app-logo">{FLUX_LOGO}</div>
+                <span class="app-title">FLUX.2 Klein VAE Comparator</span>
+                <span class="app-badge klein">4B Distilled</span>
+                <span class="app-badge fast">4-Step Fast</span>
             </div>
         </div>
 
+        <!-- Toolbar -->
         <div class="app-toolbar">
-            <button id="tb-upload" class="modern-tb-btn" title="Upload images">
+            <button id="tb-upload" class="modern-tb-btn" title="Upload reference images">
                 {UPLOAD_SVG}<span class="tb-label">Upload</span>
             </button>
-            <button id="tb-remove" class="modern-tb-btn" title="Remove selected">
+            <button id="tb-remove" class="modern-tb-btn" title="Remove selected image">
                 {REMOVE_SVG}<span class="tb-label">Remove</span>
             </button>
-            <button id="tb-clear" class="modern-tb-btn" title="Clear all">
+            <button id="tb-clear" class="modern-tb-btn" title="Clear all images">
                 {CLEAR_SVG}<span class="tb-label">Clear All</span>
             </button>
             <div class="tb-sep"></div>
@@ -1025,7 +1038,7 @@ with gr.Blocks(css=css) as demo:
 
         <div class="app-main-row">
 
-            <!-- LEFT: Input Panel -->
+            <!-- LEFT: gallery + prompts + examples -->
             <div class="app-main-left">
 
                 <div id="gallery-drop-zone">
@@ -1036,8 +1049,8 @@ with gr.Blocks(css=css) as demo:
                                 <polygon points="12,62 30,40 42,50 54,34 68,62" fill="rgba(255,69,0,0.15)" stroke="#FF4500" stroke-width="1.5"/>
                                 <circle cx="28" cy="30" r="6" fill="rgba(255,69,0,0.2)" stroke="#FF4500" stroke-width="1.5"/>
                             </svg>
-                            <span class="upload-main-text">Click or drag image here</span>
-                            <span class="upload-sub-text">Optional reference image for editing tasks</span>
+                            <span class="upload-main-text">Click or drag reference images here (optional)</span>
+                            <span class="upload-sub-text">Supports multiple images for image-guided generation</span>
                         </div>
                     </div>
                     <input id="custom-file-input" type="file" accept="image/*" multiple style="display:none;" />
@@ -1045,49 +1058,105 @@ with gr.Blocks(css=css) as demo:
                 </div>
 
                 <div class="hint-bar">
-                    <b>Optional:</b> Upload a reference image for editing &nbsp;&middot;&nbsp;
-                    Text-to-image also works without any input &nbsp;&middot;&nbsp;
-                    <kbd>Clear All</kbd> to reset
+                    <b>Text-to-Image:</b> Leave gallery empty for pure text generation &nbsp;&middot;&nbsp;
+                    <b>Image-guided:</b> Upload reference images to condition generation &nbsp;&middot;&nbsp;
+                    <kbd>Clear All</kbd> removes all references
                 </div>
 
+                <div class="suggestions-section">
+                    <div class="suggestions-title">Quick Prompts</div>
+                    <div class="suggestions-wrap">
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Change the weather to stormy with dark clouds and lightning.')">Stormy Weather</button>
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Transform the scene into a snowy winter day.')">Snowy Winter</button>
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Relight with soft golden sunset lighting.')">Sunset Lighting</button>
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Make the texture high-resolution and ultra-detailed.')">HD Texture</button>
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Convert to cinematic film look with anamorphic lens flares.')">Cinematic</button>
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Transform into anime style illustration with vibrant colors.')">Anime Style</button>
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Apply dramatic neon cyberpunk lighting and atmosphere.')">Cyberpunk</button>
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Render as hyperrealistic oil painting with visible brush strokes.')">Oil Painting</button>
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Add dramatic fog and moody atmospheric perspective.')">Foggy Mood</button>
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Transform into a vibrant watercolor painting with soft edges.')">Watercolor</button>
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Apply vintage retro film look with faded colors and grain.')">Vintage Retro</button>
+                        <button class="suggestion-chip" onclick="window.__setPrompt('Render with dramatic volumetric god rays and dust particles.')">God Rays</button>
+                    </div>
+                </div>
+
+                <div class="examples-section">
+                    <div class="examples-title">Quick Examples</div>
+                    <div class="examples-scroll">
+                        {EXAMPLE_CARDS_HTML}
+                    </div>
+                </div>
+
+            </div><!-- end app-main-left -->
+
+            <!-- RIGHT: prompt + run + outputs + settings -->
+            <div class="app-main-right">
+
                 <div class="panel-card">
-                    <div class="panel-card-title">Prompt</div>
+                    <div class="panel-card-title">Generation Prompt</div>
                     <div class="panel-card-body">
+                        <label class="modern-label" for="custom-prompt-input">Prompt</label>
                         <textarea id="custom-prompt-input" class="modern-textarea" rows="4"
                             placeholder="e.g., Transform the scene into a snowy winter day..."></textarea>
                     </div>
                 </div>
 
-                <div style="padding:12px 16px;">
+                <div style="padding:12px 20px;">
                     <button id="custom-run-btn" class="btn-run">
                         <svg viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg">
-                            <path d="M13 2L3 14h9l-1 8 10-12h-9l1-8z" fill="white"/>
+                            <path d="M13 2L3 14h9l-1 8 10-12h-9l1-8z" fill="#ffffff"/>
                         </svg>
-                        <span>Run Comparison</span>
+                        <span id="run-btn-label">⚡ Run Comparison</span>
                     </button>
                 </div>
 
-                <div class="suggestions-section">
-                    <div class="suggestions-title">Quick Prompts</div>
-                    <div class="suggestions-wrap">
-                        <button class="suggestion-chip" onclick="window.__setPrompt('Change the weather to stormy.')">Stormy Weather</button>
-                        <button class="suggestion-chip" onclick="window.__setPrompt('Transform the scene into a snowy winter day.')">Snowy Winter</button>
-                        <button class="suggestion-chip" onclick="window.__setPrompt('Relight with soft golden sunset lighting.')">Golden Sunset</button>
-                        <button class="suggestion-chip" onclick="window.__setPrompt('Make the texture high-resolution and detailed.')">HD Texture</button>
-                        <button class="suggestion-chip" onclick="window.__setPrompt('Convert to oil painting style.')">Oil Painting</button>
-                        <button class="suggestion-chip" onclick="window.__setPrompt('Add cinematic film grain and color grading.')">Cinematic</button>
-                        <button class="suggestion-chip" onclick="window.__setPrompt('Transform into anime illustration style.')">Anime Style</button>
-                        <button class="suggestion-chip" onclick="window.__setPrompt('Apply dramatic noir black and white style.')">Noir B&W</button>
-                    </div>
-                </div>
+                <!-- Dual output cards -->
+                <div class="output-panel">
+                    <div class="output-compare-row">
 
-                <div class="examples-section">
-                    <div class="examples-title">Quick Examples</div>
-                    <div class="examples-scroll">
-                        {EXAMPLE_CARDS_HTML}
-                    </div>
-                </div>
+                        <!-- Standard VAE -->
+                        <div class="output-card">
+                            <div class="output-card-header">
+                                <span class="output-card-label std">🟦 Standard VAE</span>
+                                <span id="timing-badge-std" class="timing-badge">⏱ —</span>
+                                <span id="dl-btn-std" class="out-download-btn" title="Download">
+                                    {DL_SVG} Save
+                                </span>
+                            </div>
+                            <div class="output-card-body" id="output-body-std">
+                                <div class="modern-loader" id="output-loader-std">
+                                    <div class="loader-spinner"></div>
+                                    <div class="loader-text">Generating...</div>
+                                    <div class="loader-bar-track"><div class="loader-bar-fill"></div></div>
+                                </div>
+                                <div class="out-placeholder-text">Standard VAE result</div>
+                            </div>
+                        </div>
+
+                        <!-- Small Decoder VAE -->
+                        <div class="output-card">
+                            <div class="output-card-header">
+                                <span class="output-card-label small">🟩 Small Decoder</span>
+                                <span id="timing-badge-small" class="timing-badge">⏱ —</span>
+                                <span id="dl-btn-small" class="out-download-btn" title="Download">
+                                    {DL_SVG} Save
+                                </span>
+                            </div>
+                            <div class="output-card-body" id="output-body-small">
+                                <div class="modern-loader" id="output-loader-small">
+                                    <div class="loader-spinner"></div>
+                                    <div class="loader-text">Generating...</div>
+                                    <div class="loader-bar-track"><div class="loader-bar-fill"></div></div>
+                                </div>
+                                <div class="out-placeholder-text">Small Decoder result</div>
+                            </div>
+                        </div>
+
+                    </div><!-- end output-compare-row -->
+                </div><!-- end output-panel -->
 
+                <!-- Advanced Settings -->
                 <div class="settings-group">
                     <div class="settings-group-title">Advanced Settings</div>
                     <div class="settings-group-body">
@@ -1098,7 +1167,7 @@ with gr.Blocks(css=css) as demo:
                         </div>
                         <div class="checkbox-row">
                             <input type="checkbox" id="custom-randomize" checked>
-                            <label for="custom-randomize">Randomize seed</label>
+                            <label for="custom-randomize">Randomize seed each run</label>
                         </div>
                         <div class="slider-row">
                             <label>Guidance</label>
@@ -1110,88 +1179,38 @@ with gr.Blocks(css=css) as demo:
                             <input type="range" id="custom-steps" min="1" max="20" step="1" value="4">
                             <span class="slider-val" id="custom-steps-val">4</span>
                         </div>
-                        <div class="slider-row">
-                            <label>Width</label>
-                            <input type="range" id="custom-width" min="256" max="1024" step="8" value="1024">
-                            <span class="slider-val" id="custom-width-val">1024</span>
-                        </div>
-                        <div class="slider-row">
-                            <label>Height</label>
-                            <input type="range" id="custom-height" min="256" max="1024" step="8" value="1024">
-                            <span class="slider-val" id="custom-height-val">1024</span>
-                        </div>
-                    </div>
-                </div>
-
-            </div>
-
-            <!-- RIGHT: Output Area -->
-            <div class="app-main-right">
-                <div class="output-area">
-
-                    <div class="output-cols">
-
-                        <!-- Standard VAE -->
-                        <div class="output-col">
-                            <div class="out-col-header">
-                                <span class="out-col-title std">&#128310; Standard VAE</span>
-                                <span class="out-col-badge std">Generated First</span>
-                            </div>
-                            <div class="out-col-body" id="output-std-body">
-                                <div class="modern-loader" id="output-loader-std">
-                                    <div class="loader-spinner"></div>
-                                    <div class="loader-text">Running Standard VAE...</div>
-                                    <div class="loader-bar-track"><div class="loader-bar-fill"></div></div>
-                                </div>
-                                <div class="out-col-placeholder">Standard VAE output will appear here</div>
-                            </div>
-                            <div class="out-col-footer">
-                                <span class="timing-display" id="timing-std-display">⏱ Waiting...</span>
-                                <span id="dl-btn-std" class="out-download-btn" title="Download">
-                                    {DOWNLOAD_SVG} Save
-                                </span>
-                            </div>
-                        </div>
-
-                        <!-- Small Decoder VAE -->
-                        <div class="output-col">
-                            <div class="out-col-header">
-                                <span class="out-col-title small">&#128312; Small Decoder VAE</span>
-                                <span class="out-col-badge small">Generated Second</span>
-                            </div>
-                            <div class="out-col-body" id="output-small-body">
-                                <div class="modern-loader" id="output-loader-small">
-                                    <div class="loader-spinner"></div>
-                                    <div class="loader-text">Running Small Decoder VAE...</div>
-                                    <div class="loader-bar-track"><div class="loader-bar-fill"></div></div>
-                                </div>
-                                <div class="out-col-placeholder">Small Decoder VAE output will appear here</div>
+                        <div class="slider-row-2col">
+                            <div class="slider-row" style="flex:1">
+                                <label>Width</label>
+                                <input type="range" id="custom-width" min="256" max="1024" step="8" value="1024">
+                                <span class="slider-val" id="custom-width-val">1024</span>
                             </div>
-                            <div class="out-col-footer">
-                                <span class="timing-display" id="timing-small-display">⏱ Waiting...</span>
-                                <span id="dl-btn-small" class="out-download-btn" title="Download">
-                                    {DOWNLOAD_SVG} Save
-                                </span>
+                            <div class="slider-row" style="flex:1">
+                                <label>Height</label>
+                                <input type="range" id="custom-height" min="256" max="1024" step="8" value="1024">
+                                <span class="slider-val" id="custom-height-val">1024</span>
                             </div>
                         </div>
-
                     </div>
                 </div>
-            </div>
 
-        </div>
+            </div><!-- end app-main-right -->
+        </div><!-- end app-main-row -->
 
         <div class="exp-note">
-            Experimental Space — <a href="https://huggingface.co/black-forest-labs/FLUX.2-klein-4B" target="_blank">FLUX.2-klein-4B</a>
-            &middot; <a href="https://huggingface.co/black-forest-labs/FLUX.2-small-decoder" target="_blank">FLUX.2-small-decoder</a>
+            Experimental Space —
+            <a href="https://huggingface.co/black-forest-labs/FLUX.2-klein-4B" target="_blank">FLUX.2-klein-4B</a>
+            &middot;
+            <a href="https://huggingface.co/black-forest-labs/FLUX.2-small-decoder" target="_blank">FLUX.2-small-decoder</a>
+            &middot; Compare Standard vs Small Decoder VAE side-by-side
         </div>
 
         <div class="app-statusbar">
-            <div class="sb-section" id="sb-image-count">No images uploaded</div>
+            <div class="sb-section" id="sb-image-count">No images loaded</div>
             <div class="sb-section sb-fixed">Ready</div>
         </div>
 
-    </div>
+    </div><!-- end app-shell -->
     """)
 
     run_btn = gr.Button("Run", elem_id="gradio-run-btn")
@@ -1201,24 +1220,27 @@ with gr.Blocks(css=css) as demo:
 
     run_btn.click(
         fn=infer,
-        inputs=[
-            hidden_images_b64,
-            prompt_hidden,
-            seed_slider,
-            randomize_seed,
-            width_slider,
-            height_slider,
-            steps_slider,
-            guidance_scale,
-        ],
-        outputs=[result_std, result_small_dec, seed_used, timing_std_hidden, timing_small_hidden],
+        inputs=[hidden_images_b64, prompt, seed, randomize_seed, width_gr, height_gr, steps, guidance_scale],
+        outputs=[result_std, result_small_gr, seed_output, timing_std_gr, timing_small_gr],
         js=r"""(imgs, p, s, rs, w, h, st, gs) => {
             const images   = window.__uploadedImages || [];
             const b64Array = images.map(img => img.b64);
             const imgsJson = JSON.stringify(b64Array);
             const promptEl = document.getElementById('custom-prompt-input');
             const promptVal = promptEl ? promptEl.value : p;
-            return [imgsJson, promptVal, s, rs, w, h, st, gs];
+            const seedEl   = document.getElementById('custom-seed');
+            const sv       = seedEl ? parseFloat(seedEl.value) : s;
+            const randEl   = document.getElementById('custom-randomize');
+            const rv       = randEl ? randEl.checked : rs;
+            const wEl      = document.getElementById('custom-width');
+            const hEl      = document.getElementById('custom-height');
+            const wv       = wEl ? parseFloat(wEl.value) : w;
+            const hv       = hEl ? parseFloat(hEl.value) : h;
+            const stEl     = document.getElementById('custom-steps');
+            const stv      = stEl ? parseFloat(stEl.value) : st;
+            const gsEl     = document.getElementById('custom-guidance');
+            const gsv      = gsEl ? parseFloat(gsEl.value) : gs;
+            return [imgsJson, promptVal, sv, rv, wv, hv, stv, gsv];
         }""",
     )
 
@@ -1231,7 +1253,6 @@ with gr.Blocks(css=css) as demo:
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch(
-        css=css,
         ssr_mode=False,
         show_error=True,
         allowed_paths=["examples"],