update app

app.py

@@ -5,19 +5,19 @@ import numpy as np
 import random
 import spaces
 import torch
-import time
-import base64
-import json
-import html as html_lib
-from io import BytesIO
+from diffusers import Flux2KleinPipeline, AutoencoderKLFlux2
 from PIL import Image
 from pathlib import Path
-from diffusers import Flux2KleinPipeline, AutoencoderKLFlux2
+import concurrent.futures
+import threading
+from typing import Iterable
+
+dtype   = torch.bfloat16
+device  = "cuda" if torch.cuda.is_available() else "cpu"
 
 MAX_SEED       = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
-dtype          = torch.bfloat16
-device         = "cuda" if torch.cuda.is_available() else "cpu"
+EXAMPLES_DIR   = Path("examples")
 
 print("Loading 4B Distilled model (Standard VAE)...")
 pipe_standard = Flux2KleinPipeline.from_pretrained(
@@ -40,135 +40,93 @@ pipe_small_decoder = Flux2KleinPipeline.from_pretrained(
 )
 pipe_small_decoder.enable_model_cpu_offload()
 
-EXAMPLES_CONFIG = [
-    {"images": ["examples/1.jpg"], "prompt": "Change the weather to stormy."},
-    {"images": ["examples/2.jpg"], "prompt": "Transform the scene into a snowy winter day while preserving the original subject identity, framing, and composition."},
-    {"images": ["examples/3.jpg"], "prompt": "Relight the image with soft golden sunset lighting while keeping all structures and subject details consistent."},
-    {"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):
-        return ""
-    try:
-        img = Image.open(path).convert("RGB")
-        img.thumbnail((max_dim, max_dim), LANCZOS)
-        buf = BytesIO()
-        img.save(buf, format="JPEG", quality=65)
-        return f"data:image/jpeg;base64,{base64.b64encode(buf.getvalue()).decode()}"
-    except Exception as e:
-        print(f"Thumbnail error for {path}: {e}")
-        return ""
-
-def encode_full_image(path):
-    if not os.path.exists(path):
-        return ""
-    try:
-        with open(path, "rb") as f:
-            data = f.read()
-        ext  = path.rsplit(".", 1)[-1].lower()
-        mime = {"jpg": "image/jpeg", "jpeg": "image/jpeg", "png": "image/png", "webp": "image/webp"}.get(ext, "image/jpeg")
-        return f"data:{mime};base64,{base64.b64encode(data).decode()}"
-    except Exception as e:
-        print(f"Encode error for {path}: {e}")
-        return ""
+pipe_lock_standard = threading.Lock()
+pipe_lock_small    = threading.Lock()
 
-def build_example_cards_html():
-    cards = ""
-    for i, ex in enumerate(EXAMPLES_CONFIG):
-        thumbs_html = ""
-        for path in ex["images"]:
-            thumb = make_thumb_b64(path)
-            if thumb:
-                thumbs_html += f'<img src="{thumb}" alt="">'
-            else:
-                thumbs_html += '<div class="example-thumb-placeholder">Preview</div>'
-        n     = len(ex["images"])
-        badge = f'{n} image{"s" if n > 1 else ""}'
-        prompt_short = html_lib.escape(ex["prompt"][:90])
-        if len(ex["prompt"]) > 90:
-            prompt_short += "..."
-        cards += f'''<div class="example-card" data-idx="{i}">
-            <div class="example-thumbs">{thumbs_html}</div>
-            <div class="example-meta"><span class="example-badge">{badge}</span></div>
-            <div class="example-prompt-text">{prompt_short}</div>
-        </div>'''
-    return cards
+def update_dimensions_from_image(image_list):
+    if image_list is None or len(image_list) == 0:
+        return 1024, 1024
 
-def load_example_data(idx_str):
-    try:
-        idx = int(float(idx_str)) if idx_str and idx_str.strip() else -1
-    except (ValueError, TypeError):
-        idx = -1
-    if idx < 0 or idx >= len(EXAMPLES_CONFIG):
-        return json.dumps({"images": [], "prompt": "", "names": [], "status": "error"})
-    ex       = EXAMPLES_CONFIG[idx]
-    b64_list, names = [], []
-    for path in ex["images"]:
-        b64 = encode_full_image(path)
-        if b64:
-            b64_list.append(b64)
-            names.append(os.path.basename(path))
-    return json.dumps({"images": b64_list, "prompt": ex["prompt"], "names": names, "status": "ok"})
+    item = image_list[0]
+    img  = item[0] if isinstance(item, tuple) else item
 
-print("Building example thumbnails...")
-EXAMPLE_CARDS_HTML = build_example_cards_html()
-print(f"Built {len(EXAMPLES_CONFIG)} example cards.")
+    if isinstance(img, str):
+        img = Image.open(img).convert("RGB")
 
-def b64_to_pil_list(b64_json_str):
-    if not b64_json_str or b64_json_str.strip() in ("", "[]"):
-        return []
-    try:
-        b64_list = json.loads(b64_json_str)
-    except Exception:
-        return []
-    pil_images = []
-    for b64_str in b64_list:
-        if not b64_str or not isinstance(b64_str, str):
-            continue
-        try:
-            if b64_str.startswith("data:image"):
-                _, data = b64_str.split(",", 1)
-            else:
-                data = b64_str
-            image_data = base64.b64decode(data)
-            pil_images.append(Image.open(BytesIO(image_data)).convert("RGB"))
-        except Exception as e:
-            print(f"Error decoding image: {e}")
-    return pil_images
+    iw, ih       = img.size
+    aspect_ratio = iw / ih
 
-def update_dimensions_on_upload(pil_images):
-    if not pil_images:
-        return 1024, 1024
-    img = pil_images[0]
-    iw, ih = img.size
-    if iw >= ih:
-        nw = 1024
-        nh = int(1024 * ih / iw)
+    if aspect_ratio >= 1:
+        new_width  = 1024
+        new_height = int(1024 / aspect_ratio)
     else:
-        nh = 1024
-        nw = int(1024 * iw / ih)
-    return (max(256, (nw // 8) * 8), max(256, (nh // 8) * 8))
+        new_height = 1024
+        new_width  = int(1024 * aspect_ratio)
+
+    new_width  = max(256, min(1024, round(new_width  / 8) * 8))
+    new_height = max(256, min(1024, round(new_height / 8) * 8))
+    return new_width, new_height
 
-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 get_example_items():
+    example_prompts = {
+        "1.jpg": "Change the weather to stormy.",
+        "2.jpg": "Transform the scene into a snowy winter day while preserving the original subject identity, framing, and composition.",
+        "3.jpg": "Relight the image with soft golden sunset lighting while keeping all structures and subject details consistent.",
+        "4.jpg": "Make the texture high-resolution.",
+    }
+    items = []
+    if EXAMPLES_DIR.exists():
+        for name in sorted(os.listdir(EXAMPLES_DIR)):
+            if name.lower().endswith((".png", ".jpg", ".jpeg", ".webp")):
+                items.append({
+                    "file":   name,
+                    "path":   str(EXAMPLES_DIR / name),
+                    "prompt": example_prompts.get(
+                        name, "Edit this image while preserving composition."
+                    ),
+                })
+    return items
+
+
+def parse_input_images(input_images):
+    """Safely parse gallery / filepath / PIL inputs → list[PIL.Image] or None."""
+    if input_images is None:
+        return None
+    if isinstance(input_images, str):
+        return [Image.open(input_images).convert("RGB")] if os.path.exists(input_images) else None
+    if isinstance(input_images, list) and len(input_images) > 0:
+        parsed = []
+        for item in input_images:
+            try:
+                src = item[0] if isinstance(item, tuple) else item
+                if isinstance(src, str):
+                    parsed.append(Image.open(src).convert("RGB"))
+                elif isinstance(src, Image.Image):
+                    parsed.append(src.convert("RGB"))
+                elif hasattr(src, "name"):
+                    parsed.append(Image.open(src.name).convert("RGB"))
+            except Exception as e:
+                print(f"Skipping invalid image: {e}")
+        return parsed or None
+    return None
+
+def run_pipeline(pipe, lock, kwargs, seed):
+    with lock:
+        gen    = torch.Generator(device="cpu").manual_seed(seed)
+        result = pipe(**kwargs, generator=gen).images[0]
+    return result
+
+@spaces.GPU(duration=120)
 def infer(
-    images_b64_json,
     prompt,
-    seed,
-    randomize_seed,
-    width,
-    height,
-    num_inference_steps,
-    guidance_scale,
+    input_images=None,
+    seed=42,
+    randomize_seed=False,
+    width=1024,
+    height=1024,
+    num_inference_steps=4,
+    guidance_scale=1.0,
     progress=gr.Progress(track_tqdm=True),
 ):
     gc.collect()
@@ -180,9 +138,7 @@ def infer(
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
-    pil_images = b64_to_pil_list(images_b64_json)
-    if pil_images:
-        width, height = update_dimensions_on_upload(pil_images)
+    image_list = parse_input_images(input_images)
 
     shared_kwargs = dict(
         prompt=prompt,
@@ -191,1070 +147,222 @@ def infer(
         num_inference_steps=num_inference_steps,
         guidance_scale=guidance_scale,
     )
-    if pil_images:
-        shared_kwargs["image"] = pil_images
+    if image_list is not None:
+        shared_kwargs["image"] = image_list
 
-    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]
-    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.05, desc="⚡ Launching both pipelines simultaneously...")
 
-    progress(0.55, desc=f"🟦 Standard VAE done in {time_std_str} — now running 🟩 Small Decoder VAE...")
-    gc.collect()
-    torch.cuda.empty_cache()
+    with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
+        future_std   = executor.submit(run_pipeline, pipe_standard,     pipe_lock_standard, shared_kwargs, seed)
+        future_small = executor.submit(run_pipeline, pipe_small_decoder, pipe_lock_small,    shared_kwargs, seed)
+        concurrent.futures.wait(
+            [future_std, future_small],
+            return_when=concurrent.futures.ALL_COMPLETED,
+        )
 
-    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]
-    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(0.95, desc="✅ Both pipelines done!")
+
+    out_standard = future_std.result()
+    out_small    = future_small.result()
 
-    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}"
-
-    return out_standard, out_small, seed, label_std, label_small
-
-# ── 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>'
-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>'
-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');
-*{box-sizing:border-box;margin:0;padding:0}
-body,.gradio-container{
-    background:#0f0f13!important;font-family:'Inter',system-ui,-apple-system,sans-serif!important;
-    font-size:14px!important;color:#e4e4e7!important;min-height:100vh;
-}
-footer{display:none!important}
-.hidden-input{display:none!important;height:0!important;overflow:hidden!important;margin:0!important;padding:0!important}
-
-#example-load-btn{
-    position:absolute!important;left:-9999px!important;top:-9999px!important;
-    width:1px!important;height:1px!important;opacity:0.01!important;pointer-events:none!important;
-}
-#gradio-run-btn{
-    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,#FF4500,#FF8533,#FFB366);
-    border-radius:10px;display:flex;align-items:center;justify-content:center;
-    box-shadow:0 4px 12px rgba(255,69,0,.35);
-}
-.app-logo svg{width:20px;height:20px;fill:#fff;flex-shrink:0}
-.app-title{
-    font-size:18px;font-weight:700;background:linear-gradient(135deg,#e4e4e7,#a1a1aa);
-    -webkit-background-clip:text;-webkit-text-fill-color:transparent;letter-spacing:-.3px;
-}
-.app-badge{
-    font-size:11px;font-weight:600;padding:3px 10px;border-radius:20px;
-    background:rgba(255,69,0,.15);color:#FF8533;border:1px solid rgba(255,69,0,.25);letter-spacing:.3px;
-}
-.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;
-}
-.tb-sep{width:1px;height:28px;background:#27272a;margin:0 8px}
-.modern-tb-btn{
-    display:inline-flex;align-items:center;justify-content:center;gap:6px;
-    min-width:32px;height:34px;background:transparent;border:1px solid transparent;
-    border-radius:8px;cursor:pointer;font-size:13px;font-weight:600;padding:0 12px;
-    font-family:'Inter',sans-serif;color:#ffffff!important;-webkit-text-fill-color:#ffffff!important;
-    transition:all .15s ease;
-}
-.modern-tb-btn:hover{background:rgba(255,69,0,.15);border-color:rgba(255,69,0,.3)}
-.modern-tb-btn:active{background:rgba(255,69,0,.25);border-color:rgba(255,69,0,.45)}
-.modern-tb-btn .tb-label{font-size:13px;color:#ffffff!important;-webkit-text-fill-color:#ffffff!important;font-weight:600}
-.modern-tb-btn .tb-svg{width:15px;height:15px;flex-shrink:0}
-.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: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}
-
-/* ── 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}
-.upload-click-area{
-    display:flex;flex-direction:column;align-items:center;justify-content:center;
-    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.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(140px,1fr));
-    gap:12px;padding:16px;align-content:start;
-}
-.gallery-thumb{
-    position:relative;aspect-ratio:1;border-radius:10px;overflow:hidden;
-    cursor:pointer;border:2px solid #27272a;transition:all .2s ease;background:#18181b;
-}
-.gallery-thumb:hover{border-color:#3f3f46;transform:translateY(-2px);box-shadow:0 4px 12px rgba(0,0,0,.4)}
-.gallery-thumb.selected{border-color:#FF4500!important;box-shadow:0 0 0 3px rgba(255,69,0,.2)}
-.gallery-thumb img{width:100%;height:100%;object-fit:cover}
-.thumb-badge{
-    position:absolute;top:6px;left:6px;background:#FF4500;color:#fff;
-    padding:2px 8px;border-radius:4px;font-family:'JetBrains Mono',monospace;font-size:11px;font-weight:600;
-}
-.thumb-remove{
-    position:absolute;top:6px;right:6px;width:24px;height:24px;background:rgba(0,0,0,.75);
-    color:#fff;border:1px solid rgba(255,255,255,.15);border-radius:50%;cursor:pointer;
-    display:none;align-items:center;justify-content:center;font-size:12px;transition:all .15s;line-height:1;
-}
-.gallery-thumb:hover .thumb-remove{display:flex}
-.thumb-remove:hover{background:#FF4500;border-color:#FF4500}
-.gallery-add-card{
-    aspect-ratio:1;border-radius:10px;border:2px dashed #3f3f46;
-    display:flex;flex-direction:column;align-items:center;justify-content:center;
-    cursor:pointer;transition:all .2s ease;background:rgba(255,69,0,.03);gap:4px;
-}
-.gallery-add-card:hover{border-color:#FF4500;background:rgba(255,69,0,.08)}
-.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:10px 20px;font-size:13px;color:#a1a1aa;line-height:1.7;
-}
-.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: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: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: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)}
-.modern-textarea::placeholder{color:#3f3f46}
-.modern-textarea.error-flash{
-    border-color:#ef4444!important;box-shadow:0 0 0 3px rgba(239,68,68,.2)!important;animation:shake .4s ease;
-}
-@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,#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;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,#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}
-
-/* ── 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,#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;
-}
-.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);
-}
-.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;
-}
-.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:#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:#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:12px;backdrop-filter:blur(4px);
-}
-.modern-loader.active{display:flex}
-.modern-loader .loader-spinner{
-    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}
-.loader-bar-track{width:180px;height:4px;background:#27272a;border-radius:2px;overflow:hidden}
-.loader-bar-fill{
-    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}}
-
-/* ── 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:10px 16px;border-bottom:1px solid #27272a;background:rgba(24,24,27,.5);
-}
-.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,#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,#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: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}
-
-/* ── 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;
-}
-.app-statusbar .sb-section{
-    padding:0 12px;flex:1;display:flex;align-items:center;font-family:'JetBrains Mono',monospace;
-    font-size:12px;color:#52525b;overflow:hidden;white-space:nowrap;
-}
-.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:#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:#FF8533;text-decoration:none}
-.exp-note a:hover{text-decoration:underline}
-
-::-webkit-scrollbar{width:8px;height:8px}
-::-webkit-scrollbar-track{background:#09090b}
-::-webkit-scrollbar-thumb{background:#27272a;border-radius:4px}
-::-webkit-scrollbar-thumb:hover{background:#3f3f46}
-
-@media(max-width:900px){
-    .app-main-row{flex-direction:column}
-    .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() {
-    if (window.__fluxInitDone) return;
-
-    const galleryGrid  = document.getElementById('image-gallery-grid');
-    const dropZone     = document.getElementById('gallery-drop-zone');
-    const uploadPrompt = document.getElementById('upload-prompt');
-    const uploadClick  = document.getElementById('upload-click-area');
-    const fileInput    = document.getElementById('custom-file-input');
-    const btnUpload    = document.getElementById('tb-upload');
-    const btnRemove    = document.getElementById('tb-remove');
-    const btnClear     = document.getElementById('tb-clear');
-    const promptInput  = document.getElementById('custom-prompt-input');
-    const runBtnEl     = document.getElementById('custom-run-btn');
-    const imgCountTb   = document.getElementById('tb-image-count');
-    const imgCountSb   = document.getElementById('sb-image-count');
-
-    if (!galleryGrid || !fileInput || !dropZone) { setTimeout(init, 250); return; }
-
-    window.__fluxInitDone = true;
-
-    let images = [];
-    window.__uploadedImages = images;
-    let selectedIdx = -1;
-    let toastTimer  = null;
+    return out_standard, out_small, seed
 
-    function showToast(message, type) {
-        let toast = document.getElementById('app-toast');
-        if (!toast) {
-            toast = document.createElement('div');
-            toast.id = 'app-toast';
-            toast.className = 'toast-notification';
-            toast.innerHTML = '<span class="toast-icon"></span><span class="toast-text"></span>';
-            document.body.appendChild(toast);
-        }
-        const icon = toast.querySelector('.toast-icon');
-        const text = toast.querySelector('.toast-text');
-        toast.className = 'toast-notification ' + (type || 'error');
-        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;
-        toast.classList.add('visible');
-        toastTimer = setTimeout(() => toast.classList.remove('visible'), 3500);
-    }
-    window.__showToast = showToast;
-
-    function flashPromptError() {
-        if (!promptInput) return;
-        promptInput.classList.add('error-flash');
-        promptInput.focus();
-        setTimeout(() => promptInput.classList.remove('error-flash'), 800);
-    }
-
-    function setGradioValue(containerId, value) {
-        const container = document.getElementById(containerId);
-        if (!container) return;
-        container.querySelectorAll('input, textarea').forEach(el => {
-            if (el.type === 'file' || el.type === 'range' || el.type === 'checkbox') return;
-            const proto = el.tagName === 'TEXTAREA' ? HTMLTextAreaElement.prototype : HTMLInputElement.prototype;
-            const ns = Object.getOwnPropertyDescriptor(proto, 'value');
-            if (ns && ns.set) {
-                ns.set.call(el, value);
-                el.dispatchEvent(new Event('input',  {bubbles:true, composed:true}));
-                el.dispatchEvent(new Event('change', {bubbles:true, composed:true}));
-            }
-        });
-    }
-    window.__setGradioValue = setGradioValue;
-
-    function syncImagesToGradio() {
-        window.__uploadedImages = images;
-        const b64Array = images.map(img => img.b64);
-        setGradioValue('hidden-images-b64', JSON.stringify(b64Array));
-        updateCounts();
-    }
-
-    function syncPromptToGradio() {
-        if (promptInput) setGradioValue('prompt-gradio-input', promptInput.value);
-    }
-
-    function updateCounts() {
-        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 + ' loaded' : 'No images loaded';
-    }
-
-    function addImage(b64, name) {
-        images.push({id: Date.now() + Math.random(), b64: b64, name: name});
-        renderGallery();
-        syncImagesToGradio();
-    }
-    window.__addImage = addImage;
-
-    function removeImage(idx) {
-        images.splice(idx, 1);
-        if (selectedIdx === idx) selectedIdx = -1;
-        else if (selectedIdx > idx) selectedIdx--;
-        renderGallery();
-        syncImagesToGradio();
-    }
-
-    function clearAll() {
-        images = [];
-        window.__uploadedImages = images;
-        selectedIdx = -1;
-        renderGallery();
-        syncImagesToGradio();
-    }
-    window.__clearAll = clearAll;
-
-    function selectImage(idx) {
-        selectedIdx = (selectedIdx === idx) ? -1 : idx;
-        renderGallery();
-    }
-
-    function renderGallery() {
-        if (images.length === 0) {
-            galleryGrid.innerHTML = '';
-            galleryGrid.style.display = 'none';
-            if (uploadPrompt) uploadPrompt.style.display = '';
-            return;
-        }
-        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-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) => {
-                if (e.target.closest('.thumb-remove')) return;
-                selectImage(parseInt(thumb.dataset.idx));
-            });
-        });
-        galleryGrid.querySelectorAll('.thumb-remove').forEach(btn => {
-            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());
-    }
-
-    function processFiles(files) {
-        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.readAsDataURL(file);
-        });
-    }
-
-    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 && 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) => {
-        e.preventDefault(); dropZone.classList.remove('drag-over');
-        if (e.dataTransfer.files.length) processFiles(e.dataTransfer.files);
-    });
-
-    if (promptInput) promptInput.addEventListener('input', syncPromptToGradio);
-
-    window.__setPrompt = function(text) {
-        if (promptInput) { promptInput.value = text; syncPromptToGradio(); }
-    };
-
-    /* ── Example cards ── */
-    document.querySelectorAll('.example-card[data-idx]').forEach(card => {
-        card.addEventListener('click', () => {
-            const idx = card.getAttribute('data-idx');
-            document.querySelectorAll('.example-card.loading').forEach(c => c.classList.remove('loading'));
-            card.classList.add('loading');
-            showToast('Loading example...', 'info');
-            setGradioValue('example-result-data', '');
-            setGradioValue('example-idx-input', idx);
-            setTimeout(() => {
-                const btn = document.getElementById('example-load-btn');
-                if (btn) { const b = btn.querySelector('button'); if (b) b.click(); else btn.click(); }
-            }, 150);
-            setTimeout(() => card.classList.remove('loading'), 12000);
-        });
-    });
-
-    /* ── 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 = 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 => {
-                const ns = Object.getOwnPropertyDescriptor(HTMLInputElement.prototype, 'value');
-                if (ns && ns.set) {
-                    ns.set.call(el, slider.value);
-                    el.dispatchEvent(new Event('input',  {bubbles:true, composed:true}));
-                    el.dispatchEvent(new Event('change', {bubbles:true, composed:true}));
-                }
-            });
-        });
-    }
-    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');
-
-    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();
-        });
-    }
-
-    /* ── Loader ── */
-    function showLoader() {
-        ['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() {
-        ['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; }
-        return true;
-    }
-
-    window.__clickGradioRunBtn = function() {
-        if (!validateBeforeRun()) return;
-        syncPromptToGradio(); syncImagesToGradio(); showLoader();
-        setTimeout(() => {
-            const gradioBtn = document.getElementById('gradio-run-btn');
-            if (!gradioBtn) return;
-            const btn = gradioBtn.querySelector('button');
-            if (btn) btn.click(); else gradioBtn.click();
-        }, 200);
-    };
-
-    if (runBtnEl) runBtnEl.addEventListener('click', () => window.__clickGradioRunBtn());
-
-    renderGallery();
-    updateCounts();
-}
-init();
-}
-"""
-
-wire_outputs_js = r"""
-() => {
-function watchOutputs() {
-    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 syncOne(container, body, loaderId, dlBtn) {
-        const resultImg = container.querySelector('img');
-        if (resultImg && resultImg.src) {
-            const ph = body.querySelector('.out-placeholder-text');
-            if (ph) ph.style.display = 'none';
-            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;
-                if (dlBtn) dlBtn.classList.add('visible');
-                const l = document.getElementById(loaderId);
-                if (l) l.classList.remove('active');
-            }
-        }
-    }
+@spaces.GPU(duration=120)
+def infer_example(prompt):
+    out_std, out_small, seed_used = infer(
+        prompt=prompt,
+        input_images=None,
+        seed=0,
+        randomize_seed=True,
+        width=1024,
+        height=1024,
+        num_inference_steps=4,
+        guidance_scale=1.0,
+    )
+    return out_std, out_small, seed_used
 
-    function syncAll() {
-        syncOne(stdContainer,   stdBody,   'output-loader-std',   dlStd);
-        syncOne(smallContainer, smallBody, 'output-loader-small', dlSmall);
-    }
+EXAMPLE_ITEMS = get_example_items()
 
-    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);
+css = """
+#col-container {
+    margin: 0 auto;
+    max-width: 1100px;
 }
-watchOutputs();
-
-function watchTimings() {
-    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');
+#main-title h1 {
+    font-size: 2.4em !important;
 }
-watchTimings();
-
-function watchSeed() {
-    const seedContainer = document.getElementById('gradio-seed');
-    const seedSlider    = document.getElementById('custom-seed');
-    const seedVal       = document.getElementById('custom-seed-val');
-    if (!seedContainer || !seedSlider) { setTimeout(watchSeed, 500); return; }
-    function sync() {
-        const el = seedContainer.querySelector('input[type="range"],input[type="number"]');
-        if (el && el.value) { seedSlider.value = el.value; if (seedVal) seedVal.textContent = el.value; }
-    }
-    new MutationObserver(sync).observe(seedContainer, {childList:true,subtree:true,attributes:true});
-    setInterval(sync, 1000);
-}
-watchSeed();
-
-function watchExampleResults() {
-    const container = document.getElementById('example-result-data');
-    if (!container) { setTimeout(watchExampleResults, 500); return; }
-    let lastProcessed = '';
-    function checkResult() {
-        const el = container.querySelector('textarea') || container.querySelector('input');
-        if (!el) return;
-        const val = el.value;
-        if (!val || val === lastProcessed || val.length < 20) return;
-        try {
-            const data = JSON.parse(val);
-            if (data.status === 'ok' && data.images && data.images.length > 0) {
-                lastProcessed = val;
-                if (window.__clearAll) window.__clearAll();
-                if (window.__setPrompt && data.prompt) window.__setPrompt(data.prompt);
-                data.images.forEach((b64, i) => {
-                    if (b64 && window.__addImage) {
-                        const name = (data.names && data.names[i]) ? data.names[i] : ('example_' + (i+1) + '.jpg');
-                        window.__addImage(b64, name);
-                    }
-                });
-                document.querySelectorAll('.example-card.loading').forEach(c => c.classList.remove('loading'));
-                if (window.__showToast) window.__showToast('Example loaded — ' + data.images.length + ' image(s)', 'info');
-            } else if (data.status === 'error') {
-                document.querySelectorAll('.example-card.loading').forEach(c => c.classList.remove('loading'));
-                if (window.__showToast) window.__showToast('Could not load example images', 'error');
-            }
-        } catch(e) { console.error('Example parse error:', e); }
-    }
-    new MutationObserver(checkResult).observe(container, {childList:true,subtree:true,characterData:true,attributes:true});
-    setInterval(checkResult, 500);
-}
-watchExampleResults();
+.vae-badge {
+    font-weight: 700;
+    font-size: 0.95em;
+    text-align: center;
+    padding: 4px 16px;
+    border-radius: 20px;
+    display: block;
+    margin-bottom: 6px;
 }
 """
 
-# ── Gradio App ────────────────────────────────────────────────────────────────
-
 with gr.Blocks() as demo:
 
-    # 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}</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 reference images">
-                {UPLOAD_SVG}<span class="tb-label">Upload</span>
-            </button>
-            <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 images">
-                {CLEAR_SVG}<span class="tb-label">Clear All</span>
-            </button>
-            <div class="tb-sep"></div>
-            <span id="tb-image-count" class="tb-info">No images</span>
-        </div>
-
-        <div class="app-main-row">
-
-            <!-- LEFT: gallery + prompts + examples -->
-            <div class="app-main-left">
-
-                <div id="gallery-drop-zone">
-                    <div id="upload-prompt" class="upload-prompt-modern">
-                        <div id="upload-click-area" class="upload-click-area">
-                            <svg viewBox="0 0 80 80" fill="none" xmlns="http://www.w3.org/2000/svg">
-                                <rect x="8" y="14" width="64" height="52" rx="6" fill="none" stroke="#FF4500" stroke-width="2" stroke-dasharray="4 3"/>
-                                <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 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;" />
-                    <div id="image-gallery-grid" class="image-gallery-grid" style="display:none;"></div>
-                </div>
-
-                <div class="hint-bar">
-                    <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">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 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="#ffffff"/>
-                        </svg>
-                        <span id="run-btn-label">⚡ Run Comparison</span>
-                    </button>
-                </div>
-
-                <!-- Dual output cards -->
-                <div class="output-panel">
-                    <div class="output-compare-row">
-
-                        <!-- 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">
-                        <div class="slider-row">
-                            <label>Seed</label>
-                            <input type="range" id="custom-seed" min="0" max="2147483647" step="1" value="0">
-                            <span class="slider-val" id="custom-seed-val">0</span>
-                        </div>
-                        <div class="checkbox-row">
-                            <input type="checkbox" id="custom-randomize" checked>
-                            <label for="custom-randomize">Randomize seed each run</label>
-                        </div>
-                        <div class="slider-row">
-                            <label>Guidance</label>
-                            <input type="range" id="custom-guidance" min="0" max="10" step="0.1" value="1.0">
-                            <span class="slider-val" id="custom-guidance-val">1.0</span>
-                        </div>
-                        <div class="slider-row">
-                            <label>Steps</label>
-                            <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-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="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><!-- 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>
-            &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 loaded</div>
-            <div class="sb-section sb-fixed">Ready</div>
-        </div>
-
-    </div><!-- end app-shell -->
-    """)
-
-    run_btn = gr.Button("Run", elem_id="gradio-run-btn")
-
-    demo.load(fn=None, js=gallery_js)
-    demo.load(fn=None, js=wire_outputs_js)
-
-    run_btn.click(
-        fn=infer,
-        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;
-            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];
-        }""",
+    with gr.Column(elem_id="col-container"):
+
+        gr.Markdown(
+            "# ⚡ **Flux.2-4B-Encoder-Comparator**",
+            elem_id="main-title",
+        )
+        gr.Markdown(
+            "Compare **FLUX.2-klein-4B** side-by-side with two VAE decoders — "
+            "generated **simultaneously** from the **same seed**. "
+            "🟦 **Standard Decoder** vs 🟩 **Small Decoder** "
+            "([FLUX.2-small-decoder](https://huggingface.co/black-forest-labs/FLUX.2-small-decoder)) · "
+            "[[model](https://huggingface.co/black-forest-labs/FLUX.2-klein-4B)] · "
+            "[[blog](https://bfl.ai/blog/flux-2)]"
+        )
+
+        with gr.Row(equal_height=True):
+
+            with gr.Column():
+                input_images = gr.Gallery(
+                    label="Input Image(s) for Editing (optional)",
+                    type="pil",
+                    columns=2,
+                    rows=1,
+                    height=280,
+                    allow_preview=True,
+                )
+
+                prompt = gr.Text(
+                    label="Prompt",
+                    show_label=True,
+                    placeholder="e.g., A black cat holding a sign that says hello world...",
+                )
+
+                run_button = gr.Button("Run Comparison", variant="primary")
+
+            with gr.Column():
+                with gr.Row():
+                    with gr.Column():
+                        gr.HTML(
+                            '<span class="vae-badge" '
+                            'style="background:#dbeafe;color:#1d4ed8;">'
+                            '🟦 Standard VAE</span>'
+                        )
+                        result_standard = gr.Image(
+                            label="Standard VAE",
+                            show_label=False,
+                            interactive=False,
+                            format="png",
+                            height=280,
+                        )
+
+                    with gr.Column():
+                        gr.HTML(
+                            '<span class="vae-badge" '
+                            'style="background:#d1fae5;color:#065f46;">'
+                            '🟩 Small Decoder VAE</span>'
+                        )
+                        result_small = gr.Image(
+                            label="Small Decoder VAE",
+                            show_label=False,
+                            interactive=False,
+                            format="png",
+                            height=280,
+                        )
+
+                seed_output = gr.Number(label="Seed Used", precision=0)
+
+                with gr.Accordion("⚙️ Advanced Settings", open=False):
+                    seed = gr.Slider(
+                        label="Seed",
+                        minimum=0,
+                        maximum=MAX_SEED,
+                        step=1,
+                        value=0,
+                    )
+                    randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
+
+                    with gr.Row():
+                        width = gr.Slider(
+                            label="Width",
+                            minimum=256,
+                            maximum=MAX_IMAGE_SIZE,
+                            step=8,
+                            value=1024,
+                        )
+                        height_slider = gr.Slider(
+                            label="Height",
+                            minimum=256,
+                            maximum=MAX_IMAGE_SIZE,
+                            step=8,
+                            value=1024,
+                        )
+
+                    with gr.Row():
+                        num_inference_steps = gr.Slider(
+                            label="Inference Steps",
+                            minimum=1,
+                            maximum=20,
+                            step=1,
+                            value=4,
+                        )
+                        guidance_scale = gr.Slider(
+                            label="Guidance Scale",
+                            minimum=0.0,
+                            maximum=10.0,
+                            step=0.1,
+                            value=1.0,
+                        )
+
+        gr.Examples(
+            examples=[
+                [["examples/1.jpg"], "Change the weather to stormy."],
+                [["examples/2.jpg"], "Transform the scene into a snowy winter day while preserving the original subject identity, framing, and composition."],
+                [["examples/3.jpg"], "Relight the image with soft golden sunset lighting while keeping all structures and subject details consistent."],
+                [["examples/4.jpg"], "Make the texture high-resolution."],
+                [None, "A beautiful cyberpunk cityscape at night, neon lights, highly detailed."],
+            ],
+            inputs=[input_images, prompt],
+            outputs=[result_standard, result_small, seed_output],
+            fn=infer_example,
+            cache_examples=False,
+            label="Examples",
+        )
+
+        gr.Markdown(
+            "[*](https://huggingface.co/black-forest-labs/FLUX.2-klein-4B) "
+            "Experimental Space — FLUX.2 [klein] 4B VAE Decoder Comparison."
+        )
+        
+    input_images.upload(
+        fn=update_dimensions_from_image,
+        inputs=[input_images],
+        outputs=[width, height_slider],
     )
 
-    example_load_btn.click(
-        fn=load_example_data,
-        inputs=[example_idx],
-        outputs=[example_result],
-        queue=False,
+    gr.on(
+        triggers=[run_button.click, prompt.submit],
+        fn=infer,
+        inputs=[
+            prompt,
+            input_images,
+            seed,
+            randomize_seed,
+            width,
+            height_slider,
+            num_inference_steps,
+            guidance_scale,
+        ],
+        outputs=[result_standard, result_small, seed_output],
     )
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch(
+        theme=gr.themes.Citrus(),
         css=css,
         ssr_mode=False,
         show_error=True,
-        allowed_paths=["examples"],
     )