Update app.py

app.py

@@ -1,9 +1,8 @@
 import os
-import gc
 import io
+import gc
 import uuid
 import json
-import base64
 import random
 import threading
 import concurrent.futures
@@ -18,9 +17,9 @@ from PIL import Image
 from gradio import Server
 from fastapi import Request, UploadFile, File, Form
 from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
-
 from diffusers import Flux2KleinPipeline, AutoencoderKLFlux2
 
+# --- App Configuration & Directories ---
 app = Server()
 
 BASE_DIR = Path(__file__).resolve().parent
@@ -34,78 +33,69 @@ OUTPUT_DIR.mkdir(exist_ok=True)
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
-dtype = torch.bfloat16
-DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+dtype  = torch.bfloat16
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 if torch.cuda.is_available():
+    print("current device:", torch.cuda.current_device())
     DEVICE_LABEL = torch.cuda.get_device_name(torch.cuda.current_device()).lower()
 else:
-    DEVICE_LABEL = DEVICE
+    DEVICE_LABEL = str(device).lower()
 
+# --- Model Loading ---
 print("Loading 4B Distilled model (Standard VAE)...")
 pipe_standard = Flux2KleinPipeline.from_pretrained(
     "black-forest-labs/FLUX.2-klein-4B",
     torch_dtype=dtype,
-)
+).to(device)
 pipe_standard.enable_model_cpu_offload()
 
 print("Loading Small Decoder VAE...")
 vae_small = AutoencoderKLFlux2.from_pretrained(
     "black-forest-labs/FLUX.2-small-decoder",
     torch_dtype=dtype,
-)
+).to(device)
 
 print("Loading 4B Distilled model (Small Decoder VAE)...")
 pipe_small_decoder = Flux2KleinPipeline.from_pretrained(
     "black-forest-labs/FLUX.2-klein-4B",
     vae=vae_small,
     torch_dtype=dtype,
-)
+).to(device)
 pipe_small_decoder.enable_model_cpu_offload()
 
 pipe_lock_standard = threading.Lock()
-pipe_lock_small = threading.Lock()
-
-
-def image_to_base64(img: Image.Image) -> str:
-    buf = io.BytesIO()
-    img.save(buf, format="PNG")
-    return base64.b64encode(buf.getvalue()).decode("utf-8")
-
-
-def save_image(img: Image.Image, prefix: str = "output") -> str:
-    filename = f"{prefix}_{uuid.uuid4().hex}.png"
-    path = OUTPUT_DIR / filename
-    img.save(path, format="PNG")
-    return filename
-
+pipe_lock_small    = threading.Lock()
 
+# --- Utility Functions ---
 def calc_dimensions(pil_img: Image.Image):
     iw, ih = pil_img.size
     aspect = iw / ih
+
     if aspect >= 1:
-        new_width = 1024
+        new_width  = 1024
         new_height = int(round(1024 / aspect))
     else:
         new_height = 1024
-        new_width = int(round(1024 * aspect))
-    new_width = max(256, min(1024, round(new_width / 8) * 8))
+        new_width  = int(round(1024 * aspect))
+
+    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 parse_and_resize_images(image_paths: list, width: int, height: int):
-    raw_list = []
+def parse_and_resize_images(image_paths: List[str], width: int, height: int):
+    if not image_paths:
+        return None
+    
+    resized = []
     for path in image_paths:
         try:
             img = Image.open(path).convert("RGB")
-            raw_list.append(img)
+            resized.append(img.resize((width, height), Image.LANCZOS))
         except Exception as e:
             print(f"Skipping invalid image: {e}")
-    if not raw_list:
-        return None
-    return [img.resize((width, height), Image.LANCZOS) for img in raw_list]
-
+            
+    return resized if resized else None
 
 def run_pipeline(pipe, lock, kwargs, seed):
     with lock:
@@ -113,80 +103,67 @@ def run_pipeline(pipe, lock, kwargs, seed):
         result = pipe(**kwargs, generator=gen).images[0]
     return result
 
-
-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,
-                    "url": f"/example-file/{name}",
-                    "prompt": example_prompts.get(name, "Edit this image while preserving composition."),
-                })
-    return items
-
-
-@app.get("/example-file/{filename}")
-async def example_file(filename: str):
-    path = EXAMPLES_DIR / filename
-    if not path.exists():
-        return JSONResponse({"error": "Example not found"}, status_code=404)
-    return FileResponse(path)
-
-
-@app.get("/download/{filename}")
-async def download_file(filename: str):
+def save_image(img: Image.Image, prefix: str = "output") -> str:
+    filename = f"{prefix}_{uuid.uuid4().hex}.png"
     path = OUTPUT_DIR / filename
-    if not path.exists():
-        return JSONResponse({"error": "File not found"}, status_code=404)
-    return FileResponse(path, filename=filename, media_type="image/png")
-
+    img.save(path, format="PNG")
+    return filename
 
+# --- Inference Function ---
 @spaces.GPU(duration=120)
-def run_inference(image_paths, prompt, seed, randomize_seed, width, height, steps, guidance_scale):
+def infer(
+    prompt: str,
+    image_paths: List[str] = None,
+    seed: int = 42,
+    randomize_seed: bool = False,
+    width: int = 1024,
+    height: int = 1024,
+    num_inference_steps: int = 4,
+    guidance_scale: float = 1.0,
+):
     gc.collect()
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
 
+    if not prompt or not prompt.strip():
+        raise ValueError("Please enter a prompt.")
+
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
     image_list = None
-    if image_paths:
+    if image_paths and len(image_paths) > 0:
         try:
             first_pil = Image.open(image_paths[0]).convert("RGB")
             width, height = calc_dimensions(first_pil)
-        except Exception:
-            pass
-        image_list = parse_and_resize_images(image_paths, width, height)
+            image_list = parse_and_resize_images(image_paths, width, height)
+        except Exception as e:
+            print(f"Error processing upload: {e}")
 
-    width = max(256, min(MAX_IMAGE_SIZE, round(int(width) / 8) * 8))
+    width  = max(256, min(MAX_IMAGE_SIZE, round(int(width)  / 8) * 8))
     height = max(256, min(MAX_IMAGE_SIZE, round(int(height) / 8) * 8))
 
     shared_kwargs = dict(
         prompt=prompt,
         height=height,
         width=width,
-        num_inference_steps=steps,
+        num_inference_steps=num_inference_steps,
         guidance_scale=guidance_scale,
     )
     if image_list is not None:
         shared_kwargs["image"] = image_list
 
     with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
-        future_std = executor.submit(run_pipeline, pipe_standard, pipe_lock_standard, shared_kwargs, seed)
+        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)
+        
+        concurrent.futures.wait(
+            [future_std, future_small],
+            return_when=concurrent.futures.ALL_COMPLETED,
+        )
 
     out_standard = future_std.result()
-    out_small = future_small.result()
+    out_small    = future_small.result()
 
     gc.collect()
     if torch.cuda.is_available():
@@ -195,6 +172,51 @@ def run_inference(image_paths, prompt, seed, randomize_seed, width, height, step
     return out_standard, out_small, seed
 
 
+# --- FastAPI Endpoints ---
+def get_example_items():
+    # Explicitly structured dictionary to handle multi-image examples properly
+    return [
+        {
+            "files": ["I1.jpg", "I2.jpg"],
+            "urls": ["/example-file/I1.jpg", "/example-file/I2.jpg"],
+            "prompt": "Make her wear these glasses in Image 2."
+        },
+        {
+            "files": ["1.jpg"],
+            "urls": ["/example-file/1.jpg"],
+            "prompt": "Change the weather to stormy."
+        },
+        {
+            "files": ["2.jpg"],
+            "urls": ["/example-file/2.jpg"],
+            "prompt": "Transform the scene into a snowy winter day while preserving the original subject identity, framing, and composition."
+        },
+        {
+            "files": ["3.jpg"],
+            "urls": ["/example-file/3.jpg"],
+            "prompt": "Relight the image with soft golden sunset lighting while keeping all structures and subject details consistent."
+        },
+        {
+            "files": ["4.jpg"],
+            "urls": ["/example-file/4.jpg"],
+            "prompt": "Make the texture high-resolution."
+        }
+    ]
+
+@app.get("/example-file/{filename}")
+async def example_file(filename: str):
+    path = EXAMPLES_DIR / filename
+    if not path.exists():
+        return JSONResponse({"error": "Example not found"}, status_code=404)
+    return FileResponse(path)
+
+@app.get("/download/{filename}")
+async def download_file(filename: str):
+    path = OUTPUT_DIR / filename
+    if not path.exists():
+        return JSONResponse({"error": "File not found"}, status_code=404)
+    return FileResponse(path, filename=filename, media_type="image/png")
+
 @app.post("/api/compare")
 async def compare_images(
     prompt: str = Form(...),
@@ -203,7 +225,7 @@ async def compare_images(
     width: str = Form("1024"),
     height: str = Form("1024"),
     steps: str = Form("4"),
-    guidance_scale: str = Form("1.0"),
+    guidance: str = Form("1.0"),
     images: Optional[List[UploadFile]] = File(None),
 ):
     temp_paths = []
@@ -211,40 +233,34 @@ async def compare_images(
         image_paths = []
         if images:
             for upload in images:
+                if not upload.filename: continue
                 suffix = Path(upload.filename).suffix or ".png"
-                temp_name = f"upload_{uuid.uuid4().hex}{suffix}"
-                temp_path = OUTPUT_DIR / temp_name
+                temp_path = OUTPUT_DIR / f"upload_{uuid.uuid4().hex}{suffix}"
                 content = await upload.read()
                 with open(temp_path, "wb") as f:
                     f.write(content)
                 temp_paths.append(str(temp_path))
                 image_paths.append(str(temp_path))
 
-        out_standard, out_small, used_seed = run_inference(
-            image_paths=image_paths,
+        result_std, result_small, used_seed = infer(
             prompt=prompt,
+            image_paths=image_paths,
             seed=int(seed),
-            randomize_seed=randomize_seed.lower() == "true",
+            randomize_seed=(randomize_seed.lower() == "true"),
             width=int(width),
             height=int(height),
-            steps=int(steps),
-            guidance_scale=float(guidance_scale),
+            num_inference_steps=int(steps),
+            guidance_scale=float(guidance),
         )
 
-        std_filename = save_image(out_standard, prefix="standard")
-        small_filename = save_image(out_small, prefix="small")
+        std_filename = save_image(result_std, prefix="std")
+        small_filename = save_image(result_small, prefix="small")
 
         return JSONResponse({
             "success": True,
             "seed": used_seed,
-            "standard": {
-                "image_url": f"/download/{std_filename}",
-                "download_url": f"/download/{std_filename}",
-            },
-            "small": {
-                "image_url": f"/download/{small_filename}",
-                "download_url": f"/download/{small_filename}",
-            },
+            "std_url": f"/download/{std_filename}",
+            "small_url": f"/download/{small_filename}",
             "device": DEVICE_LABEL,
         })
 
@@ -252,1832 +268,555 @@ async def compare_images(
         return JSONResponse({"success": False, "error": str(e)}, status_code=500)
     finally:
         for p in temp_paths:
-            try:
-                if os.path.exists(p):
-                    os.remove(p)
-            except Exception:
-                pass
-
+            if os.path.exists(p):
+                try: os.remove(p)
+                except: pass
 
+# --- Frontend ---
 @app.get("/", response_class=HTMLResponse)
 async def homepage(request: Request):
     examples = get_example_items()
     examples_json = json.dumps(examples)
 
-    return f"""<!DOCTYPE html>
+    return f"""
+<!DOCTYPE html>
 <html lang="en">
 <head>
-  <meta charset="UTF-8"/>
-  <meta name="viewport" content="width=device-width, initial-scale=1.0"/>
-  <title>FLUX.2 Decoder Comparator — Terminal</title>
+  <meta charset="UTF-8" />
+  <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+  <title>Flux.2-4B-Decoder-Comparator</title>
+  <link href="https://fonts.googleapis.com/css2?family=Ubuntu:wght@300;400;500;700&display=swap" rel="stylesheet">
   <style>
-    @import url('https://fonts.googleapis.com/css2?family=Ubuntu+Mono:ital,wght@0,400;0,700;1,400&family=Ubuntu:wght@300;400;500;700&display=swap');
-
     :root {{
-      --term-bg:       #300a24;
-      --term-bg2:      #3a0e2e;
-      --term-bg3:      #2a0820;
-      --term-chrome:   #1a0514;
-      --term-border:   #6b2050;
-      --term-border2:  #8b2a68;
-      --text:          #d8c8d4;
-      --text-dim:      #a0889c;
-      --text-muted:    #6b4d64;
-      --green:         #4caf50;
-      --green2:        #8bc34a;
-      --orange:        #e95420;
-      --orange2:       #f4a460;
-      --orange-soft:   rgba(233,84,32,0.18);
-      --orange-border: rgba(233,84,32,0.45);
-      --yellow:        #f0c060;
-      --cyan:          #17a589;
-      --cyan2:         #76d7c4;
-      --red:           #c0392b;
-      --white:         #f0e8ec;
-      --prompt-color:  #8bc34a;
-      --cursor-color:  #e95420;
-      --sel-bg:        rgba(233,84,32,0.25);
-      --scrollbar:     #6b2050;
-      --radius:        6px;
-      --mono:          'Ubuntu Mono', 'Courier New', monospace;
-      --sans:          'Ubuntu', sans-serif;
-    }}
-
-    *, *::before, *::after {{ box-sizing: border-box; margin: 0; padding: 0; }}
-
-    html, body {{
-      background: #1a001a;
-      color: var(--text);
-      font-family: var(--mono);
-      min-height: 100vh;
-      overflow-x: hidden;
-    }}
-
-    ::-webkit-scrollbar {{ width: 5px; height: 5px; }}
-    ::-webkit-scrollbar-track {{ background: var(--term-chrome); }}
-    ::-webkit-scrollbar-thumb {{ background: var(--scrollbar); border-radius: 3px; }}
-
-    /* ═══════════════════════════════════════
-       DESKTOP / WALLPAPER
-    ═══════════════════════════════════════ */
-    .desktop {{
+      --ub-aubergine: #2C001E;
+      --ub-aubergine-dark: #1f0015;
+      --ub-orange: #E95420;
+      --ub-orange-hover: #c4461a;
+      --ub-panel: #3D3D3D;
+      --ub-border: rgba(255,255,255,0.1);
+      --ub-text: #FFFFFF;
+      --ub-muted: #b0b0b0;
+      --ub-input: #2b2b2b;
+      --panel-radius: 8px;
+    }}
+
+    * {{ box-sizing: border-box; font-family: 'Ubuntu', sans-serif; }}
+
+    body {{
+      margin: 0; padding: 0;
+      background: var(--ub-aubergine);
+      color: var(--ub-text);
       min-height: 100vh;
-      background:
-        radial-gradient(ellipse at 20% 50%, rgba(233,84,32,0.08) 0%, transparent 60%),
-        radial-gradient(ellipse at 80% 20%, rgba(107,32,80,0.15) 0%, transparent 50%),
-        linear-gradient(160deg, #1a001a 0%, #200520 40%, #0d000d 100%);
-      display: flex;
-      align-items: flex-start;
-      justify-content: center;
-      padding: 30px 16px 40px;
-    }}
-
-    /* ═══════════════════════════════════════
-       TERMINAL WINDOW
-    ═══════════════════════════════════════ */
-    .terminal-window {{
-      width: 100%;
-      max-width: 1380px;
-      background: var(--term-bg);
-      border: 1px solid var(--term-border);
-      border-radius: 12px;
-      box-shadow:
-        0 0 0 1px rgba(107,32,80,0.3),
-        0 32px 80px rgba(0,0,0,0.7),
-        0 0 60px rgba(233,84,32,0.06),
-        inset 0 1px 0 rgba(255,255,255,0.05);
-      overflow: hidden;
       display: flex;
       flex-direction: column;
     }}
 
-    /* ── title bar ── */
-    .term-titlebar {{
-      height: 40px;
-      background: var(--term-chrome);
-      border-bottom: 1px solid var(--term-border);
-      display: flex;
-      align-items: center;
-      padding: 0 14px;
-      gap: 0;
-      flex-shrink: 0;
-      user-select: none;
-    }}
-
-    .term-buttons {{
-      display: flex;
-      gap: 8px;
-      align-items: center;
-    }}
-
-    .term-btn {{
-      width: 13px;
-      height: 13px;
-      border-radius: 50%;
-      border: none;
-      cursor: pointer;
-      display: flex;
-      align-items: center;
-      justify-content: center;
-      font-size: 8px;
-      color: transparent;
-      transition: color 0.15s;
-      flex-shrink: 0;
-    }}
-
-    .term-btn:hover {{ color: rgba(0,0,0,0.6); }}
-    .term-btn-close   {{ background: #e95420; box-shadow: 0 0 6px rgba(233,84,32,0.6); }}
-    .term-btn-min     {{ background: #f0c060; box-shadow: 0 0 6px rgba(240,192,96,0.5); }}
-    .term-btn-max     {{ background: #4caf50; box-shadow: 0 0 6px rgba(76,175,80,0.5); }}
-
-    .term-title {{
-      flex: 1;
+    .topbar {{
+      background: var(--ub-aubergine-dark);
+      padding: 16px 24px;
+      border-bottom: 1px solid var(--ub-border);
       text-align: center;
-      font-size: 13px;
-      font-weight: 700;
-      color: var(--text-dim);
-      letter-spacing: 0.04em;
-    }}
-
-    .term-title span {{ color: var(--orange2); }}
-
-    .term-status-bar {{
-      display: flex;
-      align-items: center;
-      gap: 10px;
-      margin-left: auto;
-    }}
-
-    .term-badge {{
-      font-size: 11px;
-      padding: 2px 8px;
-      border-radius: 3px;
-      font-weight: 700;
-      letter-spacing: 0.05em;
-    }}
-
-    .badge-device {{
-      background: rgba(76,175,80,0.2);
-      border: 1px solid rgba(76,175,80,0.4);
-      color: #8bc34a;
-    }}
-
-    .badge-status {{
-      background: var(--orange-soft);
-      border: 1px solid var(--orange-border);
-      color: var(--orange2);
-      transition: all 0.3s;
-    }}
-
-    .badge-status.running {{
-      background: rgba(233,84,32,0.3);
-      color: var(--orange);
-      animation: blink-badge 1s ease-in-out infinite;
-    }}
-
-    @keyframes blink-badge {{ 0%,100% {{ opacity:1; }} 50% {{ opacity:0.5; }} }}
-
-    /* ── tab bar ── */
-    .term-tabbar {{
-      height: 34px;
-      background: var(--term-chrome);
-      border-bottom: 1px solid var(--term-border);
-      display: flex;
-      align-items: flex-end;
-      padding: 0 14px;
-      gap: 2px;
-      flex-shrink: 0;
-    }}
-
-    .term-tab {{
-      height: 28px;
-      padding: 0 16px;
-      background: var(--term-bg);
-      border: 1px solid var(--term-border);
-      border-bottom: none;
-      border-radius: 6px 6px 0 0;
-      font-size: 12px;
       font-weight: 700;
-      color: var(--text-dim);
-      display: flex;
-      align-items: center;
-      gap: 6px;
-      cursor: pointer;
-      font-family: var(--mono);
+      letter-spacing: 0.5px;
+      color: var(--ub-orange);
     }}
 
-    .term-tab.active {{ color: var(--orange2); border-color: var(--term-border2); }}
-
-    .tab-dot {{
-      width: 7px;
-      height: 7px;
-      border-radius: 50%;
-      background: var(--text-muted);
-    }}
-
-    .term-tab.active .tab-dot {{ background: var(--orange); box-shadow: 0 0 5px var(--orange); }}
-
-    /* ── body layout ── */
-    .term-body {{
-      display: flex;
+    .container {{
+      max-width: 1300px;
+      margin: 0 auto;
+      padding: 30px 20px;
       flex: 1;
-      min-height: 0;
-      overflow: hidden;
-    }}
-
-    /* ── sidebar ── */
-    .term-sidebar {{
-      width: 380px;
-      min-width: 340px;
-      background: var(--term-bg3);
-      border-right: 1px solid var(--term-border);
-      display: flex;
-      flex-direction: column;
-      overflow: hidden;
+      width: 100%;
     }}
 
-    .sidebar-section {{
-      border-bottom: 1px solid var(--term-border);
-      flex-shrink: 0;
+    .header-text {{
+      text-align: center;
+      margin-bottom: 30px;
     }}
+    .header-text h1 {{ margin: 0 0 10px 0; font-size: 2.2rem; }}
+    .header-text p {{ color: var(--ub-muted); margin: 0; }}
 
-    .sidebar-section-header {{
-      height: 34px;
-      background: rgba(0,0,0,0.25);
-      display: flex;
-      align-items: center;
-      padding: 0 12px;
-      gap: 8px;
-      font-size: 11px;
-      font-weight: 700;
-      color: var(--text-muted);
-      letter-spacing: 0.1em;
-      text-transform: uppercase;
+    /* EQUAL HEIGHT PANELS */
+    .layout {{
+      display: grid;
+      grid-template-columns: 400px 1fr;
+      gap: 24px;
+      align-items: stretch; /* Forces equal height columns */
     }}
 
-    .sec-icon {{ color: var(--orange); font-size: 13px; }}
-
-    .sidebar-section-body {{
-      padding: 12px;
+    .panel {{
+      background: var(--ub-panel);
+      border-radius: var(--panel-radius);
+      box-shadow: 0 8px 24px rgba(0,0,0,0.2);
       display: flex;
       flex-direction: column;
-      gap: 10px;
-    }}
-
-    /* ── terminal prompt lines ── */
-    .term-line {{
-      display: flex;
-      align-items: flex-start;
-      gap: 0;
-      font-size: 13px;
-      line-height: 1.55;
+      overflow: hidden;
     }}
 
-    .term-prompt {{
-      flex-shrink: 0;
-      white-space: nowrap;
+    .panel-header {{
+      padding: 16px 20px;
+      background: rgba(0,0,0,0.2);
+      border-bottom: 1px solid var(--ub-border);
+      font-weight: 500;
+      font-size: 1.1rem;
     }}
 
-    .prompt-user  {{ color: var(--prompt-color); font-weight: 700; }}
-    .prompt-at    {{ color: var(--text-muted); }}
-    .prompt-host  {{ color: var(--cyan2); font-weight: 700; }}
-    .prompt-colon {{ color: var(--text-muted); }}
-    .prompt-path  {{ color: var(--cyan); }}
-    .prompt-dollar {{ color: var(--white); margin: 0 6px; }}
-
-    /* ── form inputs styled as terminal ── */
-    .term-input-wrap {{
+    .panel-body {{ 
+      padding: 20px; 
+      flex: 1; /* Fills remaining vertical space */
       display: flex;
       flex-direction: column;
-      gap: 4px;
-    }}
-
-    .term-label {{
-      font-size: 11px;
-      font-weight: 700;
-      color: var(--text-muted);
-      letter-spacing: 0.08em;
-      text-transform: uppercase;
-      padding-left: 2px;
-    }}
-
-    .term-field {{
-      position: relative;
-      display: flex;
-      align-items: center;
-    }}
-
-    .term-field-prompt {{
-      position: absolute;
-      left: 10px;
-      font-size: 13px;
-      color: var(--prompt-color);
-      font-weight: 700;
-      pointer-events: none;
-      z-index: 1;
-      white-space: nowrap;
-    }}
-
-    textarea.term-input,
-    input.term-input {{
-      width: 100%;
-      background: rgba(0,0,0,0.45);
-      border: 1px solid var(--term-border);
-      border-radius: 4px;
-      color: var(--white);
-      font-family: var(--mono);
-      font-size: 13px;
-      outline: none;
-      transition: border-color 0.2s, box-shadow 0.2s;
-      caret-color: var(--cursor-color);
-    }}
-
-    textarea.term-input:focus,
-    input.term-input:focus {{
-      border-color: var(--orange);
-      box-shadow: 0 0 0 2px rgba(233,84,32,0.15), inset 0 0 8px rgba(0,0,0,0.3);
     }}
 
-    textarea.term-input {{
-      padding: 8px 10px 8px 38px;
-      min-height: 80px;
-      resize: vertical;
-      line-height: 1.5;
+    /* Input Forms */
+    .form-group {{ margin-bottom: 20px; }}
+    .label {{
+      display: block; font-weight: 500; font-size: 14px;
+      color: var(--ub-muted); margin-bottom: 8px;
     }}
 
-    input.term-input {{
-      padding: 6px 10px 6px 38px;
-      height: 32px;
+    .textarea, .input {{
+      width: 100%; background: var(--ub-input); border: 1px solid var(--ub-border);
+      color: var(--ub-text); padding: 12px; border-radius: 4px; outline: none; font-size: 14px;
     }}
+    .textarea:focus, .input:focus {{ border-color: var(--ub-orange); }}
+    .textarea {{ min-height: 100px; resize: vertical; }}
 
-    .term-input::placeholder {{ color: var(--text-muted); }}
-
-    /* ── upload zone ── */
+    /* Upload Zone */
     .upload-zone {{
-      background: rgba(0,0,0,0.35);
-      border: 1px dashed var(--term-border2);
-      border-radius: 4px;
-      cursor: pointer;
-      transition: border-color 0.2s, background 0.2s;
-      overflow: hidden;
+      background: var(--ub-input); border: 1px dashed var(--ub-muted);
+      border-radius: 4px; padding: 16px; text-align: center; cursor: pointer; transition: border-color 0.2s;
     }}
-
     .upload-zone:hover, .upload-zone.dragover {{
-      border-color: var(--orange);
-      background: var(--orange-soft);
+      border-color: var(--ub-orange); background: rgba(233,84,32,0.05);
     }}
-
     .upload-zone input[type="file"] {{ display: none; }}
-
-    .upload-placeholder {{
-      padding: 18px 12px;
-      display: flex;
-      flex-direction: column;
-      align-items: center;
-      gap: 8px;
-      text-align: center;
-      background: transparent;
-      border: none;
-      width: 100%;
-      cursor: pointer;
-      color: var(--text);
-      font-family: var(--mono);
-    }}
-
-    .upload-ascii {{
-      font-size: 11px;
-      color: var(--term-border2);
-      line-height: 1.3;
-      white-space: pre;
-    }}
-
-    .upload-cmd {{
-      font-size: 12px;
-      color: var(--orange2);
-      font-weight: 700;
-    }}
-
-    .upload-sub {{
-      font-size: 11px;
-      color: var(--text-muted);
-    }}
-
-    /* ── preview grid ── */
-    .preview-header {{
-      display: flex;
-      align-items: center;
-      justify-content: space-between;
-      padding: 0 2px 6px;
-    }}
-
-    .preview-count {{
-      font-size: 12px;
-      color: var(--green2);
-      font-weight: 700;
-    }}
-
-    .preview-count::before {{ content: '$ ls -la images/ → '; color: var(--text-muted); }}
-
-    .preview-clear-btn {{
-      font-size: 11px;
-      font-family: var(--mono);
-      color: var(--red);
-      background: transparent;
-      border: 1px solid rgba(192,57,43,0.4);
-      border-radius: 3px;
-      padding: 2px 7px;
-      cursor: pointer;
-      transition: all 0.2s;
-    }}
-
-    .preview-clear-btn:hover {{
-      background: rgba(192,57,43,0.15);
-      border-color: var(--red);
-    }}
-
+    
     .preview-grid {{
-      display: grid;
-      grid-template-columns: repeat(3, 1fr);
-      gap: 6px;
+      display: grid; grid-template-columns: repeat(auto-fill, minmax(70px, 1fr));
+      gap: 10px; margin-top: 10px;
     }}
-
     .thumb {{
-      position: relative;
-      aspect-ratio: 1/1;
-      border-radius: 4px;
-      overflow: hidden;
-      border: 1px solid var(--term-border);
-      background: #111;
-    }}
-
-    .thumb img {{
-      width: 100%;
-      height: 100%;
-      object-fit: cover;
-      display: block;
-    }}
-
-    .thumb-overlay {{
-      position: absolute;
-      inset: 0;
-      background: rgba(0,0,0,0);
-      transition: background 0.2s;
-      display: flex;
-      align-items: flex-start;
-      justify-content: flex-end;
-      padding: 4px;
+      position: relative; aspect-ratio: 1;
+      border-radius: 4px; overflow: hidden; border: 1px solid var(--ub-border);
     }}
-
-    .thumb:hover .thumb-overlay {{ background: rgba(0,0,0,0.5); }}
-
+    .thumb img {{ width: 100%; height: 100%; object-fit: cover; display: block; }}
     .thumb-remove {{
-      width: 20px;
-      height: 20px;
-      border-radius: 50%;
-      background: rgba(233,84,32,0.9);
-      border: none;
-      color: white;
-      font-size: 14px;
-      line-height: 1;
-      cursor: pointer;
-      display: flex;
-      align-items: center;
-      justify-content: center;
-      opacity: 0;
-      transition: opacity 0.2s;
-      font-family: var(--mono);
-    }}
-
-    .thumb:hover .thumb-remove {{ opacity: 1; }}
-
-    .thumb-num {{
-      position: absolute;
-      bottom: 4px;
-      left: 4px;
-      width: 17px;
-      height: 17px;
-      border-radius: 50%;
-      background: var(--orange);
-      color: white;
-      font-size: 9px;
-      font-weight: 700;
-      display: flex;
-      align-items: center;
-      justify-content: center;
-    }}
-
-    .add-more-tile {{
-      aspect-ratio: 1/1;
-      border-radius: 4px;
-      border: 1px dashed var(--term-border2);
-      background: rgba(0,0,0,0.2);
-      display: flex;
-      align-items: center;
-      justify-content: center;
-      cursor: pointer;
-      color: var(--text-muted);
-      font-size: 22px;
-      font-family: var(--mono);
+      position: absolute; top: 4px; right: 4px;
+      background: rgba(0,0,0,0.7); color: white;
+      border: none; border-radius: 50%; width: 20px; height: 20px;
+      display: flex; align-items: center; justify-content: center;
+      cursor: pointer; font-size: 12px;
+    }}
+    
+    /* Plus Box for adding additional images */
+    .add-more-box {{
+      position: relative; aspect-ratio: 1; border-radius: 4px;
+      border: 1px dashed var(--ub-muted); background: rgba(255,255,255,0.03);
+      display: flex; align-items: center; justify-content: center;
+      font-size: 26px; color: var(--ub-muted); cursor: pointer;
       transition: all 0.2s;
     }}
-
-    .add-more-tile:hover {{
-      border-color: var(--orange);
-      color: var(--orange);
-      background: var(--orange-soft);
-    }}
-
-    /* ── slider rows ── */
-    .slider-group {{
-      display: grid;
-      grid-template-columns: 1fr 1fr;
-      gap: 8px;
-    }}
-
-    .slider-wrap {{
-      display: flex;
-      flex-direction: column;
-      gap: 4px;
-    }}
-
-    .slider-label-row {{
-      display: flex;
-      justify-content: space-between;
-      font-size: 11px;
-      color: var(--text-muted);
-    }}
-
-    .slider-val {{ color: var(--orange2); font-weight: 700; }}
-
-    input[type="range"] {{
-      -webkit-appearance: none;
-      width: 100%;
-      height: 3px;
-      background: rgba(255,255,255,0.1);
-      border-radius: 2px;
-      outline: none;
-    }}
-
-    input[type="range"]::-webkit-slider-thumb {{
-      -webkit-appearance: none;
-      width: 14px;
-      height: 14px;
-      border-radius: 50%;
-      background: var(--orange);
-      cursor: pointer;
-      box-shadow: 0 0 6px rgba(233,84,32,0.6);
-    }}
-
-    .checkbox-line {{
-      display: flex;
-      align-items: center;
-      gap: 8px;
-      font-size: 12px;
-      color: var(--text-dim);
+    .add-more-box:hover {{
+      border-color: var(--ub-orange); color: var(--ub-orange);
+      background: rgba(233,84,32,0.1);
     }}
 
-    .checkbox-line input {{
-      width: 14px;
-      height: 14px;
-      accent-color: var(--orange);
+    /* Buttons */
+    .btn {{
+      width: 100%; padding: 14px; border: none; border-radius: 4px;
+      font-size: 16px; font-weight: 700; cursor: pointer;
+      transition: opacity 0.2s, background 0.2s;
     }}
-
-    /* ── advanced toggle ── */
-    .adv-toggle {{
-      width: 100%;
-      background: rgba(0,0,0,0.3);
-      border: 1px solid var(--term-border);
-      border-radius: 4px;
-      color: var(--text-dim);
-      font-family: var(--mono);
-      font-size: 12px;
-      font-weight: 700;
-      padding: 7px 10px;
-      cursor: pointer;
-      display: flex;
-      align-items: center;
-      justify-content: space-between;
-      transition: border-color 0.2s;
-      letter-spacing: 0.03em;
+    .btn-primary {{
+      background: var(--ub-orange); color: white;
+      box-shadow: 0 4px 12px rgba(233,84,32,0.3);
+      margin-top: auto; /* Pushes button to bottom of the form panel */
     }}
+    .btn-primary:hover {{ background: var(--ub-orange-hover); }}
+    .btn:disabled {{ opacity: 0.6; cursor: not-allowed; }}
 
-    .adv-toggle::before {{ content: '$ '; color: var(--prompt-color); }}
-    .adv-toggle:hover {{ border-color: var(--orange); }}
-
-    .adv-body {{
-      display: none;
-      flex-direction: column;
-      gap: 10px;
-      padding: 10px;
-      background: rgba(0,0,0,0.25);
-      border: 1px solid var(--term-border);
-      border-top: none;
-      border-radius: 0 0 4px 4px;
+    /* Advanced Accordion */
+    .advanced-toggle {{
+      width: 100%; background: none; border: none; color: var(--ub-orange);
+      text-align: left; padding: 10px 0; font-weight: 500; cursor: pointer;
+      display: flex; justify-content: space-between; font-size: 14px;
     }}
+    .advanced-body {{ display: none; padding-top: 10px; padding-bottom: 20px; }}
+    .advanced-body.open {{ display: block; }}
+    .grid-2 {{ display: grid; grid-template-columns: 1fr 1fr; gap: 15px; }}
 
-    .adv-body.open {{ display: flex; }}
-
-    /* ── run button ── */
-    .run-btn {{
-      width: 100%;
-      height: 42px;
-      background: linear-gradient(135deg, #e95420, #c7461a);
-      border: 1px solid rgba(233,84,32,0.7);
-      border-radius: 4px;
-      color: white;
-      font-family: var(--mono);
-      font-size: 14px;
-      font-weight: 700;
-      cursor: pointer;
-      letter-spacing: 0.05em;
-      display: flex;
-      align-items: center;
-      justify-content: center;
-      gap: 10px;
-      transition: all 0.2s;
-      box-shadow: 0 2px 12px rgba(233,84,32,0.4), inset 0 1px 0 rgba(255,255,255,0.1);
-      text-shadow: 0 1px 2px rgba(0,0,0,0.4);
-    }}
-
-    .run-btn:hover {{
-      background: linear-gradient(135deg, #f47045, #e95420);
-      box-shadow: 0 4px 20px rgba(233,84,32,0.55);
-      transform: translateY(-1px);
-    }}
-
-    .run-btn:active {{ transform: translateY(0); }}
-    .run-btn:disabled {{ background: #3a1a10; box-shadow: none; cursor: not-allowed; transform: none; color: #7a5040; }}
-
-    /* ── sidebar scrollable area ── */
-    .sidebar-scroll {{
-      flex: 1;
-      overflow-y: auto;
-      display: flex;
-      flex-direction: column;
-    }}
-
-    /* ── main area ── */
-    .term-main {{
-      flex: 1;
-      min-width: 0;
-      display: flex;
-      flex-direction: column;
-      overflow: hidden;
-      background: var(--term-bg);
-    }}
-
-    /* ── output header ── */
-    .output-header {{
-      height: 38px;
-      background: rgba(0,0,0,0.3);
-      border-bottom: 1px solid var(--term-border);
-      display: flex;
-      align-items: center;
-      padding: 0 16px;
-      gap: 12px;
-      flex-shrink: 0;
-    }}
-
-    .out-label {{
-      font-size: 12px;
-      font-weight: 700;
-      color: var(--text-muted);
-      letter-spacing: 0.08em;
-      text-transform: uppercase;
-    }}
-
-    .out-decoder-badges {{
-      display: flex;
-      gap: 8px;
-    }}
-
-    .decoder-badge {{
-      font-size: 11px;
-      padding: 2px 8px;
-      border-radius: 3px;
-      font-weight: 700;
-      letter-spacing: 0.05em;
-    }}
-
-    .decoder-std {{
-      background: rgba(23,165,137,0.2);
-      border: 1px solid rgba(23,165,137,0.4);
-      color: var(--cyan2);
-    }}
-
-    .decoder-small {{
-      background: rgba(107,32,80,0.35);
-      border: 1px solid var(--term-border2);
-      color: #d080b0;
-    }}
-
-    .out-seed-display {{
-      margin-left: auto;
-      font-size: 12px;
-      color: var(--text-muted);
-    }}
-
-    .out-seed-display span {{ color: var(--orange2); font-weight: 700; }}
-
-    /* ── compare stage ── */
-    .compare-stage {{
-      flex: 1;
-      position: relative;
-      background: #0a0005;
-      display: flex;
-      align-items: center;
-      justify-content: center;
-      overflow: hidden;
-      min-height: 0;
-    }}
-
-    /* empty state */
-    .term-empty {{
-      display: flex;
-      flex-direction: column;
-      align-items: center;
-      gap: 16px;
-      color: var(--text-muted);
-      text-align: center;
-      padding: 40px 24px;
-    }}
-
-    .term-empty-ascii {{
-      font-size: 11px;
-      line-height: 1.4;
-      color: var(--term-border2);
-      white-space: pre;
-      opacity: 0.7;
-    }}
-
-    .term-empty-msg {{
-      font-size: 14px;
-      color: var(--text-dim);
-    }}
-
-    .term-empty-msg span {{ color: var(--orange2); }}
-
-    .term-empty-hint {{
-      font-size: 12px;
-      color: var(--text-muted);
-    }}
-
-    /* ── compare widget ── */
-    .compare-widget {{
-      display: none;
+    /* SLIDER CONTAINER */
+    .slider-stage {{
       position: relative;
       width: 100%;
-      height: 100%;
-      user-select: none;
-    }}
-
-    .compare-widget.visible {{ display: block; }}
-
-    .compare-base {{
-      position: absolute;
-      inset: 0;
-      overflow: hidden;
-    }}
-
-    .compare-base img {{
-      width: 100%;
-      height: 100%;
-      object-fit: contain;
-      display: block;
-    }}
-
-    .compare-layer {{
-      position: absolute;
-      top: 0;
-      left: 0;
-      height: 100%;
+      flex: 1; /* Stretches inside panel-body */
+      min-height: 550px;
+      background: #111;
+      border-radius: 4px;
       overflow: hidden;
-      will-change: width;
-    }}
-
-    .compare-layer img {{
-      width: 100%;
-      height: 100%;
-      object-fit: contain;
-      display: block;
-      position: absolute;
-      top: 0;
-      left: 0;
-    }}
-
-    /* handle */
-    .c-handle {{
-      position: absolute;
-      top: 0;
-      height: 100%;
-      width: 3px;
-      background: rgba(255,255,255,0.9);
-      cursor: ew-resize;
-      z-index: 10;
-      transform: translateX(-50%);
-      box-shadow: 0 0 10px rgba(0,0,0,0.5);
-    }}
-
-    .c-handle-grip {{
-      position: absolute;
-      top: 50%;
-      left: 50%;
-      transform: translate(-50%, -50%);
-      width: 40px;
-      height: 40px;
-      border-radius: 50%;
-      background: white;
-      border: 3px solid rgba(233,84,32,0.8);
-      box-shadow: 0 2px 12px rgba(0,0,0,0.5), 0 0 0 4px rgba(233,84,32,0.2);
       display: flex;
       align-items: center;
       justify-content: center;
-      font-size: 13px;
-      color: var(--orange);
-      font-weight: 700;
-      font-family: var(--mono);
-    }}
-
-    /* compare labels */
-    .c-label {{
-      position: absolute;
-      top: 14px;
-      z-index: 8;
-      font-size: 12px;
-      font-weight: 700;
-      font-family: var(--mono);
-      padding: 4px 10px;
-      border-radius: 3px;
-      letter-spacing: 0.05em;
-      pointer-events: none;
-    }}
-
-    .c-label-left {{
-      left: 14px;
-      background: rgba(23,165,137,0.85);
-      border: 1px solid rgba(118,215,196,0.5);
-      color: white;
-    }}
-
-    .c-label-right {{
-      right: 14px;
-      background: rgba(107,32,80,0.85);
-      border: 1px solid var(--term-border2);
-      color: #d080b0;
-    }}
-
-    /* ── loader ── */
-    .loader-overlay {{
-      position: absolute;
-      inset: 0;
-      background: rgba(10,0,5,0.82);
-      backdrop-filter: blur(10px);
-      -webkit-backdrop-filter: blur(10px);
-      display: none;
-      align-items: center;
-      justify-content: center;
-      flex-direction: column;
-      gap: 18px;
-      z-index: 30;
-    }}
-
-    .loader-overlay.active {{ display: flex; }}
-
-    .loader-terminal {{
-      background: rgba(0,0,0,0.7);
-      border: 1px solid var(--term-border2);
-      border-radius: 8px;
-      padding: 20px 28px;
-      min-width: 320px;
-      display: flex;
-      flex-direction: column;
-      gap: 12px;
-    }}
-
-    .loader-title {{
-      font-size: 13px;
-      font-weight: 700;
-      color: var(--orange2);
-      letter-spacing: 0.05em;
     }}
-
-    .loader-title::before {{ content: '$ '; color: var(--prompt-color); }}
-
-    .loader-lines {{
-      display: flex;
-      flex-direction: column;
-      gap: 6px;
-    }}
-
-    .loader-line {{
-      font-size: 12px;
-      color: var(--text-muted);
-      display: flex;
-      align-items: center;
-      gap: 8px;
-    }}
-
-    .loader-line.active {{ color: var(--orange2); }}
-    .loader-line.done   {{ color: var(--green2); }}
-
-    .loader-spinner {{
-      width: 12px;
-      height: 12px;
-      border-radius: 50%;
-      border: 2px solid rgba(233,84,32,0.3);
-      border-top-color: var(--orange);
-      animation: spin 0.8s linear infinite;
-      flex-shrink: 0;
-    }}
-
-    .loader-check {{ color: var(--green2); font-size: 13px; flex-shrink: 0; }}
-    .loader-dot   {{ color: var(--text-muted); font-size: 13px; flex-shrink: 0; }}
-
-    .progress-bar-wrap {{
-      height: 3px;
-      background: rgba(255,255,255,0.08);
-      border-radius: 2px;
-      overflow: hidden;
-      margin-top: 4px;
-    }}
-
-    .progress-bar {{
-      height: 100%;
-      background: linear-gradient(90deg, var(--orange), var(--orange2));
-      width: 0%;
-      transition: width 0.4s ease;
-      border-radius: 2px;
+    .slider-empty {{ color: var(--ub-muted); text-align: center; }}
+    
+    .slider-img {{
+      position: absolute; top: 0; left: 0;
+      width: 100%; height: 100%; object-fit: contain;
+      display: none; user-select: none; -webkit-user-drag: none;
     }}
+    #imgSmall {{ clip-path: inset(0 50% 0 0); }}
 
-    @keyframes spin {{ to {{ transform: rotate(360deg); }} }}
-
-    /* ── output footer ── */
-    .output-footer {{
-      height: 40px;
-      background: rgba(0,0,0,0.3);
-      border-top: 1px solid var(--term-border);
-      display: flex;
-      align-items: center;
-      padding: 0 14px;
-      gap: 8px;
-      flex-shrink: 0;
+    .slider-handle {{
+      position: absolute; left: 50%; top: 0; bottom: 0;
+      width: 4px; background: var(--ub-orange);
+      cursor: ew-resize; display: none; z-index: 10;
     }}
-
-    .dl-btn {{
-      display: inline-flex;
-      align-items: center;
-      gap: 5px;
-      padding: 4px 10px;
-      background: rgba(0,0,0,0.3);
-      border: 1px solid var(--term-border);
-      border-radius: 3px;
-      color: var(--text-dim);
-      font-family: var(--mono);
-      font-size: 11px;
-      font-weight: 700;
-      text-decoration: none;
-      cursor: pointer;
-      transition: all 0.2s;
-      letter-spacing: 0.03em;
+    .slider-handle::after {{
+      content: '◀ ▶'; position: absolute; top: 50%; left: 50%;
+      transform: translate(-50%, -50%); width: 40px; height: 30px;
+      background: var(--ub-orange); color: white; border-radius: 15px;
+      display: flex; align-items: center; justify-content: center;
+      font-size: 10px; font-weight: bold; box-shadow: 0 2px 6px rgba(0,0,0,0.5);
     }}
 
-    .dl-btn::before {{ content: '↓ '; color: var(--green2); }}
-    .dl-btn:hover {{ border-color: var(--orange); color: var(--orange2); }}
-    .dl-btn.hidden {{ display: none; }}
-
-    .footer-hint {{
-      margin-left: auto;
-      font-size: 11px;
-      color: var(--text-muted);
+    .slider-labels {{
+      position: absolute; top: 15px; left: 15px; right: 15px;
+      display: none; justify-content: space-between; pointer-events: none; z-index: 5;
     }}
-
-    /* ═══════════════════════════════════════
-       EXAMPLES SECTION (below terminal)
-    ═══════════════════════════════════════ */
-    .examples-window {{
-      width: 100%;
-      max-width: 1380px;
-      margin-top: 20px;
-      background: var(--term-bg3);
-      border: 1px solid var(--term-border);
-      border-radius: 10px;
-      overflow: hidden;
-      box-shadow: 0 8px 32px rgba(0,0,0,0.5);
+    .badge {{
+      background: rgba(0,0,0,0.6); color: white; padding: 6px 12px;
+      border-radius: 20px; font-size: 13px; backdrop-filter: blur(4px);
     }}
 
-    .ex-titlebar {{
-      height: 36px;
-      background: var(--term-chrome);
-      border-bottom: 1px solid var(--term-border);
-      display: flex;
-      align-items: center;
-      padding: 0 14px;
-      gap: 10px;
+    .loader {{
+      position: absolute; inset: 0; background: rgba(0,0,0,0.7);
+      display: none; flex-direction: column; align-items: center; justify-content: center; z-index: 20;
     }}
-
-    .ex-title {{
-      font-size: 12px;
-      font-weight: 700;
-      color: var(--text-dim);
-      letter-spacing: 0.05em;
+    .spinner {{
+      width: 40px; height: 40px; border: 4px solid rgba(255,255,255,0.2);
+      border-top-color: var(--ub-orange); border-radius: 50%;
+      animation: spin 1s linear infinite; margin-bottom: 15px;
     }}
 
-    .ex-title::before {{ content: '$ ls examples/ '; color: var(--prompt-color); }}
-
+    /* Examples */
+    .examples-section {{ margin-top: 40px; }}
+    .examples-section h3 {{ border-bottom: 1px solid var(--ub-border); padding-bottom: 10px; margin-bottom: 20px; }}
     .examples-grid {{
-      padding: 16px;
-      display: grid;
-      grid-template-columns: repeat(auto-fill, minmax(240px, 1fr));
-      gap: 12px;
-    }}
-
-    .example-card {{
-      background: rgba(0,0,0,0.35);
-      border: 1px solid var(--term-border);
-      border-radius: 6px;
-      overflow: hidden;
-      cursor: pointer;
-      transition: border-color 0.2s, transform 0.15s, box-shadow 0.2s;
-    }}
-
-    .example-card:hover {{
-      border-color: var(--orange);
-      transform: translateY(-2px);
-      box-shadow: 0 6px 20px rgba(0,0,0,0.4), 0 0 12px rgba(233,84,32,0.12);
-    }}
-
-    .ex-img-wrap {{
-      position: relative;
-      aspect-ratio: 16/9;
-      overflow: hidden;
-      background: #111;
-    }}
-
-    .ex-img-wrap img {{
-      width: 100%;
-      height: 100%;
-      object-fit: cover;
-      display: block;
-      transition: transform 0.3s;
-    }}
-
-    .example-card:hover .ex-img-wrap img {{ transform: scale(1.06); }}
-
-    .ex-file-badge {{
-      position: absolute;
-      top: 6px;
-      left: 6px;
-      font-size: 10px;
-      font-weight: 700;
-      font-family: var(--mono);
-      padding: 2px 6px;
-      background: rgba(0,0,0,0.7);
-      border: 1px solid var(--term-border);
-      border-radius: 3px;
-      color: var(--orange2);
-    }}
-
-    .ex-body {{
-      padding: 10px 12px;
-      display: flex;
-      flex-direction: column;
-      gap: 8px;
-    }}
-
-    .ex-prompt {{
-      font-size: 12px;
-      color: var(--text-dim);
-      line-height: 1.5;
-      display: -webkit-box;
-      -webkit-line-clamp: 2;
-      -webkit-box-orient: vertical;
-      overflow: hidden;
-    }}
-
-    .ex-prompt::before {{ content: '→ '; color: var(--orange); }}
-
-    .ex-use-btn {{
-      width: 100%;
-      padding: 5px;
-      background: rgba(233,84,32,0.12);
-      border: 1px solid var(--orange-border);
-      border-radius: 3px;
-      color: var(--orange2);
-      font-size: 11px;
-      font-weight: 700;
-      font-family: var(--mono);
-      cursor: pointer;
-      transition: background 0.2s;
-      letter-spacing: 0.04em;
+      display: grid; grid-template-columns: repeat(auto-fill, minmax(220px, 1fr)); gap: 20px;
     }}
-
-    .ex-use-btn::before {{ content: '$ use '; color: var(--prompt-color); }}
-    .ex-use-btn:hover {{ background: rgba(233,84,32,0.22); }}
-
-    /* ── toast ── */
-    .toast-wrap {{
-      position: fixed;
-      bottom: 20px;
-      right: 20px;
-      z-index: 9999;
-      display: flex;
-      flex-direction: column;
-      gap: 8px;
+    .ex-card {{
+      background: var(--ub-panel); border-radius: 4px; overflow: hidden;
+      cursor: pointer; transition: transform 0.2s, box-shadow 0.2s;
+      display: flex; flex-direction: column;
     }}
-
-    .toast {{
-      display: flex;
-      align-items: flex-start;
-      gap: 10px;
-      padding: 10px 14px;
-      background: var(--term-chrome);
-      border: 1px solid var(--term-border2);
-      border-radius: 6px;
-      box-shadow: 0 8px 24px rgba(0,0,0,0.6);
-      min-width: 240px;
-      max-width: 360px;
-      animation: slideUp 0.25s ease;
-      font-family: var(--mono);
+    .ex-card:hover {{ transform: translateY(-3px); box-shadow: 0 6px 16px rgba(0,0,0,0.3); }}
+    
+    .multi-img-preview {{
+      display: flex; width: 100%; aspect-ratio: 1; border-bottom: 1px solid var(--ub-border);
     }}
-
-    .toast-err  {{ border-color: rgba(192,57,43,0.6); }}
-    .toast-ok   {{ border-color: rgba(76,175,80,0.5); }}
-
-    .toast-icon {{ font-size: 14px; flex-shrink: 0; margin-top: 1px; }}
-    .toast-msg  {{ font-size: 12px; color: var(--text); flex: 1; line-height: 1.5; }}
-
-    .toast-close {{
-      background: transparent;
-      border: none;
-      color: var(--text-muted);
-      font-size: 16px;
-      cursor: pointer;
-      line-height: 1;
-      font-family: var(--mono);
+    .multi-img-preview img {{
+      width: 50%; height: 100%; object-fit: cover;
     }}
-
-    @keyframes slideUp {{
-      from {{ transform: translateY(10px); opacity: 0; }}
-      to   {{ transform: translateY(0);    opacity: 1; }}
+    .multi-img-preview img:first-child {{ border-right: 1px solid var(--ub-border); }}
+    .single-img-preview img {{
+      width: 100%; aspect-ratio: 1; object-fit: cover; display: block; border-bottom: 1px solid var(--ub-border);
     }}
+    
+    .ex-card p {{ padding: 12px; margin: 0; font-size: 13px; color: var(--ub-muted); line-height: 1.4; }}
 
-    /* ── responsive ── */
-    @media (max-width: 960px) {{
-      .term-body {{ flex-direction: column; }}
-      .term-sidebar {{ width: 100%; min-width: 0; border-right: none; border-bottom: 1px solid var(--term-border); max-height: 60vh; overflow-y: auto; }}
-      .compare-stage {{ min-height: 400px; }}
-    }}
+    @keyframes spin {{ to {{ transform: rotate(360deg); }} }}
 
-    @media (max-width: 600px) {{
-      .desktop {{ padding: 10px 8px 20px; }}
-      .term-title {{ font-size: 11px; }}
-      .slider-group {{ grid-template-columns: 1fr; }}
-      .examples-grid {{ grid-template-columns: 1fr 1fr; }}
+    @media (max-width: 900px) {{
+      .layout {{ grid-template-columns: 1fr; }}
+      .slider-stage {{ min-height: 400px; }}
     }}
   </style>
 </head>
 <body>
 
-  <div class="toast-wrap" id="toastWrap"></div>
-
-  <div class="desktop">
-
-    <!-- ═══ TERMINAL WINDOW ═══ -->
-    <div style="width:100%;max-width:1380px;display:flex;flex-direction:column;gap:0;">
-
-    <div class="terminal-window">
-
-      <!-- title bar -->
-      <div class="term-titlebar">
-        <div class="term-buttons">
-          <button class="term-btn term-btn-close"  title="Close">✕</button>
-          <button class="term-btn term-btn-min"    title="Minimize">−</button>
-          <button class="term-btn term-btn-max"    title="Maximize">+</button>
-        </div>
-        <div class="term-title">flux2@ubuntu: <span>~/decoder-comparator</span> — bash</div>
-        <div class="term-status-bar">
-          <span class="term-badge badge-device">⬥ {DEVICE_LABEL}</span>
-          <span class="term-badge badge-status" id="globalStatus">● idle</span>
-        </div>
-      </div>
+  <div class="topbar">Flux.2-4B VAE Decoder Comparator</div>
 
-      <!-- tab bar -->
-      <div class="term-tabbar">
-        <div class="term-tab active"><span class="tab-dot"></span>comparator.py</div>
-        <div class="term-tab"><span class="tab-dot"></span>config.yaml</div>
-        <div class="term-tab"><span class="tab-dot"></span>results/</div>
-      </div>
+  <div class="container">
+    <div class="header-text">
+      <h1>Standard vs. Small Decoder</h1>
+      <p>Upload an image, enter a prompt, and use the slider to compare outputs in real-time.</p>
+    </div>
 
-      <!-- body -->
-      <div class="term-body">
+    <div class="layout">
+      <div class="panel">
+        <div class="panel-header">Settings</div>
+        <div class="panel-body">
+          <div class="form-group">
+            <label class="label">Input Images (Optional)</label>
+            <div class="upload-zone" id="dropZone">
+              <input type="file" id="fileInput" multiple accept="image/*" />
+              <div id="uploadText" style="padding: 10px 0;">Click or Drag & Drop images here</div>
+              <div class="preview-grid" id="previewGrid"></div>
+            </div>
+          </div>
 
-        <!-- ══ SIDEBAR ══ -->
-        <div class="term-sidebar">
-          <div class="sidebar-scroll">
+          <div class="form-group">
+            <label class="label">Prompt</label>
+            <textarea id="promptInput" class="textarea" placeholder="Describe the edit or generation..."></textarea>
+          </div>
 
-            <!-- UPLOAD -->
-            <div class="sidebar-section">
-              <div class="sidebar-section-header">
-                <span class="sec-icon">⬆</span> upload images
-              </div>
-              <div class="sidebar-section-body">
-                <div class="term-line" style="margin-bottom:2px;">
-                  <span class="term-prompt">
-                    <span class="prompt-user">flux2</span><span class="prompt-at">@</span><span class="prompt-host">ubuntu</span><span class="prompt-colon">:</span><span class="prompt-path">~</span><span class="prompt-dollar">$</span>
-                  </span>
-                  <span style="font-size:12px;color:var(--cyan2);">open --upload-files</span>
-                </div>
-                <div class="upload-zone" id="uploadZone">
-                  <input id="fileInput" type="file" accept="image/*" multiple />
-                  <button class="upload-placeholder" id="uploadPlaceholder" type="button">
-                    <pre class="upload-ascii">┌──────────────────┐
-│  ↑  drag & drop  │
-│     images here  │
-└──────────────────┘</pre>
-                    <span class="upload-cmd">$ open images/*</span>
-                    <span class="upload-sub">PNG · JPG · WEBP · multiple OK</span>
-                  </button>
-                </div>
-
-                <div id="previewContainer" style="display:none;">
-                  <div class="preview-header">
-                    <span class="preview-count" id="previewCount">0 files</span>
-                    <button class="preview-clear-btn" id="clearImagesBtn" type="button">rm -rf</button>
-                  </div>
-                  <div class="preview-grid" id="previewGrid"></div>
-                </div>
-
-                <span style="font-size:11px;color:var(--text-muted);">
-                  # first image auto-fits width &amp; height
-                </span>
+          <button class="advanced-toggle" id="advToggle">
+            <span>Advanced Settings</span> <span id="advIcon" style="font-weight: 700; font-size:18px;">+</span>
+          </button>
+          
+          <div class="advanced-body" id="advBody">
+            <div class="grid-2">
+              <div class="form-group">
+                <label class="label">Seed</label>
+                <input type="number" id="seed" class="input" value="0">
               </div>
-            </div>
-
-            <!-- PROMPT -->
-            <div class="sidebar-section">
-              <div class="sidebar-section-header">
-                <span class="sec-icon">✎</span> prompt
+              <div class="form-group">
+                <label class="label">Steps</label>
+                <input type="number" id="steps" class="input" value="4">
               </div>
-              <div class="sidebar-section-body">
-                <div class="term-line" style="margin-bottom:4px;">
-                  <span class="term-prompt">
-                    <span class="prompt-user">flux2</span><span class="prompt-at">@</span><span class="prompt-host">ubuntu</span><span class="prompt-colon">:</span><span class="prompt-path">~</span><span class="prompt-dollar">$</span>
-                  </span>
-                  <span style="font-size:12px;color:var(--cyan2);">python run.py --prompt</span>
-                </div>
-                <div class="term-field">
-                  <span class="term-field-prompt">&gt;</span>
-                  <textarea id="prompt" class="term-input" placeholder="describe the edit or generation...&#10;e.g. 'change weather to stormy'"></textarea>
-                </div>
-                <span style="font-size:11px;color:var(--text-muted);"># ctrl+enter to run</span>
+              <div class="form-group">
+                <label class="label">Width</label>
+                <input type="number" id="width" class="input" value="1024" step="8">
               </div>
-            </div>
-
-            <!-- ADVANCED -->
-            <div class="sidebar-section">
-              <div class="sidebar-section-header">
-                <span class="sec-icon">⚙</span> parameters
+              <div class="form-group">
+                <label class="label">Height</label>
+                <input type="number" id="height" class="input" value="1024" step="8">
               </div>
-              <div class="sidebar-section-body">
-                <button class="adv-toggle" id="advancedToggle" type="button">
-                  <span>show --advanced-settings</span>
-                  <span id="chevronIcon">▼</span>
-                </button>
-                <div class="adv-body" id="advBody">
-                  <div class="slider-group">
-                    <div class="slider-wrap">
-                      <div class="slider-label-row">
-                        <span>--seed</span>
-                        <span class="slider-val" id="seedVal">0</span>
-                      </div>
-                      <input type="range" id="seed" min="0" max="{MAX_SEED}" step="1" value="0"
-                             oninput="document.getElementById('seedVal').textContent=this.value"/>
-                    </div>
-                    <div class="slider-wrap">
-                      <div class="slider-label-row">
-                        <span>--steps</span>
-                        <span class="slider-val" id="stepsVal">4</span>
-                      </div>
-                      <input type="range" id="steps" min="1" max="20" step="1" value="4"
-                             oninput="document.getElementById('stepsVal').textContent=this.value"/>
-                    </div>
-                    <div class="slider-wrap">
-                      <div class="slider-label-row">
-                        <span>--width</span>
-                        <span class="slider-val" id="widthVal">1024</span>
-                      </div>
-                      <input type="range" id="width" min="256" max="{MAX_IMAGE_SIZE}" step="8" value="1024"
-                             oninput="document.getElementById('widthVal').textContent=this.value"/>
-                    </div>
-                    <div class="slider-wrap">
-                      <div class="slider-label-row">
-                        <span>--height</span>
-                        <span class="slider-val" id="heightVal">1024</span>
-                      </div>
-                      <input type="range" id="height" min="256" max="{MAX_IMAGE_SIZE}" step="8" value="1024"
-                             oninput="document.getElementById('heightVal').textContent=this.value"/>
-                    </div>
-                    <div class="slider-wrap">
-                      <div class="slider-label-row">
-                        <span>--guidance</span>
-                        <span class="slider-val" id="guidanceVal">1.0</span>
-                      </div>
-                      <input type="range" id="guidance" min="0" max="10" step="0.1" value="1.0"
-                             oninput="document.getElementById('guidanceVal').textContent=parseFloat(this.value).toFixed(1)"/>
-                    </div>
-                  </div>
-                  <div class="checkbox-line">
-                    <input type="checkbox" id="randomizeSeed" checked/>
-                    <label for="randomizeSeed">--randomize-seed</label>
-                  </div>
-                </div>
+              <div class="form-group" style="grid-column: span 2;">
+                <label class="label">Guidance Scale</label>
+                <input type="number" id="guidance" class="input" value="1.0" step="0.1">
               </div>
-            </div>
-
-            <!-- RUN -->
-            <div class="sidebar-section" style="border-bottom:none;">
-              <div class="sidebar-section-body">
-                <div class="term-line" style="margin-bottom:4px;">
-                  <span class="term-prompt">
-                    <span class="prompt-user">flux2</span><span class="prompt-at">@</span><span class="prompt-host">ubuntu</span><span class="prompt-colon">:</span><span class="prompt-path">~</span><span class="prompt-dollar">$</span>
-                  </span>
-                  <span style="font-size:12px;color:var(--cyan2);">python run.py --parallel-decode</span>
-                </div>
-                <button class="run-btn" id="runBtn" type="button">
-                  <svg width="16" height="16" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2.5"><polygon points="5 3 19 12 5 21 5 3"/></svg>
-                  ./run_comparison.sh
-                </button>
+              <div class="form-group" style="grid-column: span 2;">
+                <label style="display:flex; align-items:center; gap:8px; font-size:14px; cursor:pointer;">
+                  <input type="checkbox" id="randomize" checked> Randomize Seed
+                </label>
               </div>
             </div>
-
-          </div><!-- /sidebar-scroll -->
-        </div><!-- /sidebar -->
-
-        <!-- ══ MAIN ══ -->
-        <div class="term-main">
-
-          <!-- output header -->
-          <div class="output-header">
-            <span class="out-label">output</span>
-            <div class="out-decoder-badges">
-              <span class="decoder-badge decoder-std">▶ standard-vae</span>
-              <span class="decoder-badge decoder-small">▶ small-decoder</span>
-            </div>
-            <div class="out-seed-display">seed: <span id="usedSeed">—</span></div>
           </div>
 
-          <!-- compare stage -->
-          <div class="compare-stage" id="compareStage">
-
-            <!-- empty -->
-            <div class="term-empty" id="emptyState">
-              <pre class="term-empty-ascii">
-  ┌─────────────────────────────────────┐
-  │                                     │
-  │   ░░░░░░░░░░░░░░░░░░░░░░░░░░░░░░   │
-  │   ░░                           ░░   │
-  │   ░░     awaiting output...    ░░   │
-  │   ░░                           ░░   │
-  │   ░░░░░░░░░░░░░░░░░░░░░░░░░░░░░░   │
-  │                                     │
-  └─────────────────────────────────────┘</pre>
-              <div class="term-empty-msg">
-                <span>$ run_comparison.sh</span> not yet executed
-              </div>
-              <div class="term-empty-hint"># drag the slider to compare standard vs small decoder</div>
-            </div>
-
-            <!-- compare widget -->
-            <div class="compare-widget" id="compareWidget">
-              <!-- base: small decoder (right side) -->
-              <div class="compare-base">
-                <img id="imgSmall" alt="Small Decoder"/>
-              </div>
-
-              <!-- layer: standard decoder (left side) -->
-              <div class="compare-layer" id="compareLayer">
-                <img id="imgStandard" alt="Standard Decoder"/>
-              </div>
-
-              <!-- labels -->
-              <span class="c-label c-label-left">standard-vae</span>
-              <span class="c-label c-label-right">small-decoder</span>
+          <button class="btn btn-primary" id="runBtn">Run Comparison</button>
+        </div>
+      </div>
 
-              <!-- handle -->
-              <div class="c-handle" id="cHandle">
-                <div class="c-handle-grip">◀▶</div>
-              </div>
+      <div class="panel">
+        <div class="panel-header">Comparison View</div>
+        <div class="panel-body" style="padding:0;">
+          <div class="slider-stage" id="sliderStage">
+            <div class="slider-empty" id="sliderEmpty">
+              <svg width="48" height="48" fill="none" stroke="currentColor" stroke-width="1.5" viewBox="0 0 24 24" style="margin-bottom:10px; opacity:0.5;">
+                <path d="M4 16l4.586-4.586a2 2 0 012.828 0L16 16m-2-2l1.586-1.586a2 2 0 012.828 0L20 14m-6-6h.01M6 20h12a2 2 0 002-2V6a2 2 0 00-2-2H6a2 2 0 00-2 2v12a2 2 0 002 2z"></path>
+              </svg>
+              <div>Results will appear here</div>
             </div>
 
-            <!-- loader -->
-            <div class="loader-overlay" id="loaderOverlay">
-              <div class="loader-terminal">
-                <div class="loader-title">run_comparison.sh --parallel</div>
-                <div class="loader-lines">
-                  <div class="loader-line active" id="ll1">
-                    <div class="loader-spinner"></div>
-                    <span>loading pipeline_standard...</span>
-                  </div>
-                  <div class="loader-line active" id="ll2">
-                    <div class="loader-spinner"></div>
-                    <span>loading pipeline_small_decoder...</span>
-                  </div>
-                  <div class="loader-line" id="ll3">
-                    <span class="loader-dot">○</span>
-                    <span>decoding latents...</span>
-                  </div>
-                  <div class="loader-line" id="ll4">
-                    <span class="loader-dot">○</span>
-                    <span>saving outputs...</span>
-                  </div>
-                </div>
-                <div class="progress-bar-wrap">
-                  <div class="progress-bar" id="progressBar"></div>
-                </div>
-              </div>
+            <img id="imgStd" class="slider-img" alt="Standard Decoder" />
+            <img id="imgSmall" class="slider-img" alt="Small Decoder" />
+            
+            <div class="slider-labels" id="sliderLabels">
+              <div class="badge">Standard Decoder</div>
+              <div class="badge">Small Decoder</div>
             </div>
 
-          </div><!-- /compare-stage -->
+            <div class="slider-handle" id="sliderHandle"></div>
 
-          <!-- footer -->
-          <div class="output-footer">
-            <a class="dl-btn hidden" id="dlStandard" download="standard_decoder.png">standard_vae.png</a>
-            <a class="dl-btn hidden" id="dlSmall"    download="small_decoder.png">small_decoder.png</a>
-            <span class="footer-hint"># drag handle ◀▶ to compare • ctrl+s to download</span>
+            <div class="loader" id="loader">
+              <div class="spinner"></div>
+              <div style="font-weight: 500;">Running both models...</div>
+            </div>
           </div>
-
-        </div><!-- /main -->
-      </div><!-- /body -->
-    </div><!-- /terminal-window -->
-
-    <!-- ═══ EXAMPLES ═══ -->
-    <div class="examples-window">
-      <div class="ex-titlebar">
-        <div class="ex-title">4 examples found</div>
+        </div>
       </div>
-      <div class="examples-grid" id="examplesGrid"></div>
     </div>
 
-    </div><!-- /wrapper -->
-  </div><!-- /desktop -->
+    <div class="examples-section">
+      <h3>Examples</h3>
+      <div class="examples-grid" id="examplesGrid"></div>
+    </div>
+  </div>
 
   <script>
     const examples = {examples_json};
-
-    /* ── state ── */
-    const S = {{ files: [], advOpen: false, stdUrl: null, smallUrl: null }};
-
-    /* ── refs ── */
-    const uploadZone      = document.getElementById('uploadZone');
-    const fileInput       = document.getElementById('fileInput');
-    const uploadPH        = document.getElementById('uploadPlaceholder');
-    const previewContainer= document.getElementById('previewContainer');
-    const previewGrid     = document.getElementById('previewGrid');
-    const previewCount    = document.getElementById('previewCount');
-    const clearImagesBtn  = document.getElementById('clearImagesBtn');
-
-    const promptEl        = document.getElementById('prompt');
-    const seedEl          = document.getElementById('seed');
-    const stepsEl         = document.getElementById('steps');
-    const widthEl         = document.getElementById('width');
-    const heightEl        = document.getElementById('height');
-    const guidanceEl      = document.getElementById('guidance');
-    const randomizeSeedEl = document.getElementById('randomizeSeed');
-
-    const advancedToggle  = document.getElementById('advancedToggle');
-    const advBody         = document.getElementById('advBody');
-    const chevronIcon     = document.getElementById('chevronIcon');
-
-    const runBtn          = document.getElementById('runBtn');
-
-    const compareStage    = document.getElementById('compareStage');
-    const emptyState      = document.getElementById('emptyState');
-    const compareWidget   = document.getElementById('compareWidget');
-    const imgSmall        = document.getElementById('imgSmall');
-    const imgStandard     = document.getElementById('imgStandard');
-    const compareLayer    = document.getElementById('compareLayer');
-    const cHandle         = document.getElementById('cHandle');
-
-    const loaderOverlay   = document.getElementById('loaderOverlay');
-    const progressBar     = document.getElementById('progressBar');
-
-    const dlStandard      = document.getElementById('dlStandard');
-    const dlSmall         = document.getElementById('dlSmall');
-    const usedSeed        = document.getElementById('usedSeed');
-
-    const toastWrap       = document.getElementById('toastWrap');
-    const examplesGrid    = document.getElementById('examplesGrid');
-    const globalStatus    = document.getElementById('globalStatus');
-
-    /* ── toast ── */
-    function toast(msg, type='info') {{
-      const el = document.createElement('div');
-      el.className = 'toast' + (type==='error' ? ' toast-err' : type==='success' ? ' toast-ok' : '');
-      const icon = type==='error' ? '✗' : type==='success' ? '✓' : '●';
-      el.innerHTML = `<span class="toast-icon" style="color:${{type==='error'?'var(--red)':type==='success'?'var(--green2)':'var(--orange2)'}};">${{icon}}</span><span class="toast-msg">${{msg}}</span><button class="toast-close" onclick="this.parentElement.remove()">×</button>`;
-      toastWrap.appendChild(el);
-      setTimeout(() => el.remove(), 5000);
-    }}
-
-    /* ── status ── */
-    function setRunning(running) {{
-      globalStatus.textContent = running ? '● running' : '● idle';
-      globalStatus.classList.toggle('running', running);
-      runBtn.disabled = running;
-    }}
-
-    /* ── advanced ── */
-    advancedToggle.addEventListener('click', () => {{
-      S.advOpen = !S.advOpen;
-      advBody.classList.toggle('open', S.advOpen);
-      chevronIcon.textContent = S.advOpen ? '▲' : '▼';
-    }});
-
-    /* ── file upload ── */
-    function addFiles(list) {{
-      const valid = Array.from(list).filter(f => f.type.startsWith('image/'));
-      if (!valid.length) {{ toast('No valid image files found.', 'error'); return; }}
-      S.files = [...S.files, ...valid];
-      renderPreviews();
-    }}
-
+    let filesState = [];
+
+    // UI Elements
+    const dropZone = document.getElementById('dropZone');
+    const fileInput = document.getElementById('fileInput');
+    const previewGrid = document.getElementById('previewGrid');
+    const uploadText = document.getElementById('uploadText');
+    const promptInput = document.getElementById('promptInput');
+    const runBtn = document.getElementById('runBtn');
+    
+    // Slider Elements
+    const sliderStage = document.getElementById('sliderStage');
+    const imgStd = document.getElementById('imgStd');
+    const imgSmall = document.getElementById('imgSmall');
+    const sliderHandle = document.getElementById('sliderHandle');
+    const sliderLabels = document.getElementById('sliderLabels');
+    const sliderEmpty = document.getElementById('sliderEmpty');
+    const loader = document.getElementById('loader');
+
+    // Advanced Toggle
+    document.getElementById('advToggle').onclick = function() {{
+      const body = document.getElementById('advBody');
+      body.classList.toggle('open');
+      document.getElementById('advIcon').innerText = body.classList.contains('open') ? '-' : '+';
+    }};
+
+    // --- File Upload Logic ---
     function renderPreviews() {{
       previewGrid.innerHTML = '';
-      if (!S.files.length) {{
-        previewContainer.style.display = 'none';
-        uploadPH.style.display = 'flex';
-        return;
-      }}
-      previewContainer.style.display = 'block';
-      uploadPH.style.display = 'none';
-      previewCount.textContent = S.files.length + ' file' + (S.files.length > 1 ? 's' : '');
-
-      S.files.forEach((file, idx) => {{
-        const wrap = document.createElement('div');
-        wrap.className = 'thumb';
-
-        const img = document.createElement('img');
-        img.src = URL.createObjectURL(file);
-        img.alt = file.name;
-
-        const ov = document.createElement('div');
-        ov.className = 'thumb-overlay';
-
-        const rm = document.createElement('button');
-        rm.type = 'button';
-        rm.className = 'thumb-remove';
-        rm.innerHTML = '×';
-        rm.addEventListener('click', (e) => {{
-          e.stopPropagation();
-          S.files.splice(idx, 1);
-          renderPreviews();
+      if(filesState.length > 0) {{
+        uploadText.style.display = 'none';
+        filesState.forEach((f, i) => {{
+          const div = document.createElement('div');
+          div.className = 'thumb';
+          const img = document.createElement('img');
+          img.src = URL.createObjectURL(f);
+          const btn = document.createElement('button');
+          btn.className = 'thumb-remove';
+          btn.innerText = '×';
+          btn.onclick = (e) => {{ e.stopPropagation(); filesState.splice(i, 1); renderPreviews(); }};
+          div.appendChild(img); div.appendChild(btn);
+          previewGrid.appendChild(div);
         }});
-        ov.appendChild(rm);
-
-        const num = document.createElement('div');
-        num.className = 'thumb-num';
-        num.textContent = idx + 1;
-
-        wrap.appendChild(img);
-        wrap.appendChild(ov);
-        wrap.appendChild(num);
-        previewGrid.appendChild(wrap);
-      }});
-
-      const addMore = document.createElement('div');
-      addMore.className = 'add-more-tile';
-      addMore.textContent = '+';
-      addMore.title = 'Add more images';
-      addMore.addEventListener('click', () => fileInput.click());
-      previewGrid.appendChild(addMore);
+        
+        // The dynamic "Add More" (+) Box
+        const addMore = document.createElement('div');
+        addMore.className = 'add-more-box';
+        addMore.innerText = '+';
+        addMore.title = "Add another image";
+        addMore.onclick = (e) => {{ e.stopPropagation(); fileInput.click(); }};
+        previewGrid.appendChild(addMore);
+        
+      }} else {{
+        uploadText.style.display = 'block';
+      }}
     }}
 
-    uploadPH.addEventListener('click', () => fileInput.click());
-    uploadZone.addEventListener('click', (e) => {{ if (e.target === uploadZone) fileInput.click(); }});
-    fileInput.addEventListener('change', (e) => {{ addFiles(e.target.files); fileInput.value = ''; }});
-    uploadZone.addEventListener('dragover', (e) => {{ e.preventDefault(); uploadZone.classList.add('dragover'); }});
-    uploadZone.addEventListener('dragleave', () => uploadZone.classList.remove('dragover'));
-    uploadZone.addEventListener('drop', (e) => {{
-      e.preventDefault();
-      uploadZone.classList.remove('dragover');
-      if (e.dataTransfer.files?.length) addFiles(e.dataTransfer.files);
+    dropZone.onclick = (e) => {{ 
+        if(e.target === dropZone || e.target === uploadText || e.target === previewGrid) fileInput.click(); 
+    }};
+    fileInput.onchange = (e) => {{ filesState.push(...Array.from(e.target.files)); renderPreviews(); fileInput.value=''; }};
+    dropZone.ondragover = (e) => {{ e.preventDefault(); dropZone.classList.add('dragover'); }};
+    dropZone.ondragleave = () => dropZone.classList.remove('dragover');
+    dropZone.ondrop = (e) => {{
+      e.preventDefault(); dropZone.classList.remove('dragover');
+      if(e.dataTransfer.files.length) {{ filesState.push(...Array.from(e.dataTransfer.files)); renderPreviews(); }}
+    }};
+
+    // --- Examples Logic ---
+    async function loadExample(urls, filenames, text) {{
+      try {{
+        filesState = [];
+        for(let i = 0; i < urls.length; i++) {{
+           const res = await fetch(urls[i]);
+           const blob = await res.blob();
+           filesState.push(new File([blob], filenames[i], {{type: blob.type || 'image/jpeg'}}));
+        }}
+        renderPreviews();
+        promptInput.value = text;
+        window.scrollTo({{top: 0, behavior: 'smooth'}});
+      }} catch (e) {{ alert('Failed to load example image.'); }}
+    }}
+
+    const exGrid = document.getElementById('examplesGrid');
+    examples.forEach(ex => {{
+      const card = document.createElement('div');
+      card.className = 'ex-card';
+      
+      let imgHtml = '';
+      if(ex.urls.length > 1) {{
+         imgHtml = `<div class="multi-img-preview">
+                       ${{ex.urls.map(u => `<img src="${{u}}">`).join('')}}
+                    </div>`;
+      }} else {{
+         imgHtml = `<div class="single-img-preview"><img src="${{ex.urls[0]}}"></div>`;
+      }}
+
+      card.innerHTML = `${{imgHtml}}<p>${{ex.prompt}}</p>`;
+      card.onclick = () => loadExample(ex.urls, ex.files, ex.prompt);
+      exGrid.appendChild(card);
     }});
-    clearImagesBtn.addEventListener('click', () => {{ S.files = []; renderPreviews(); }});
-
-    /* ── compare slider ── */
-    let dragging = false;
-    let sliderPct = 50;
-
-    function applyPct(pct) {{
-      sliderPct = Math.max(0, Math.min(100, pct));
-      cHandle.style.left = sliderPct + '%';
-      compareLayer.style.width = sliderPct + '%';
-      /* make the layer image fill the full stage width so clipping reveals it correctly */
-      imgStandard.style.width = (100 / (sliderPct / 100 || 0.001)) + '%';
+
+    // --- Image Slider Logic ---
+    let isDragging = false;
+    
+    function updateSlider(clientX) {{
+      const rect = sliderStage.getBoundingClientRect();
+      let pos = Math.max(0, Math.min(clientX - rect.left, rect.width));
+      let percent = (pos / rect.width) * 100;
+      
+      sliderHandle.style.left = percent + '%';
+      imgSmall.style.clipPath = `inset(0 ${{100 - percent}}% 0 0)`;
     }}
 
-    cHandle.addEventListener('mousedown', (e) => {{ dragging = true; e.preventDefault(); }});
+    sliderHandle.addEventListener('mousedown', () => isDragging = true);
+    window.addEventListener('mouseup', () => isDragging = false);
     window.addEventListener('mousemove', (e) => {{
-      if (!dragging) return;
-      const rect = compareWidget.getBoundingClientRect();
-      applyPct(((e.clientX - rect.left) / rect.width) * 100);
+      if (!isDragging) return;
+      updateSlider(e.clientX);
     }});
-    window.addEventListener('mouseup', () => dragging = false);
-
-    cHandle.addEventListener('touchstart', () => dragging = true, {{passive:true}});
+    
+    // Touch support for slider
+    sliderHandle.addEventListener('touchstart', () => isDragging = true);
+    window.addEventListener('touchend', () => isDragging = false);
     window.addEventListener('touchmove', (e) => {{
-      if (!dragging) return;
-      const t = e.touches[0];
-      const rect = compareWidget.getBoundingClientRect();
-      applyPct(((t.clientX - rect.left) / rect.width) * 100);
-    }}, {{passive:true}});
-    window.addEventListener('touchend', () => dragging = false);
-
-    /* ── loader animation ── */
-    let progressInterval = null;
-    function startLoader() {{
-      loaderOverlay.classList.add('active');
-      progressBar.style.width = '0%';
-
-      const ll1 = document.getElementById('ll1');
-      const ll2 = document.getElementById('ll2');
-      const ll3 = document.getElementById('ll3');
-      const ll4 = document.getElementById('ll4');
-
-      ll1.className = 'loader-line active';
-      ll2.className = 'loader-line active';
-      ll3.className = 'loader-line';
-      ll4.className = 'loader-line';
-
-      ll1.innerHTML = '<div class="loader-spinner"></div><span>loading pipeline_standard...</span>';
-      ll2.innerHTML = '<div class="loader-spinner"></div><span>loading pipeline_small_decoder...</span>';
-      ll3.innerHTML = '<span class="loader-dot">○</span><span>decoding latents...</span>';
-      ll4.innerHTML = '<span class="loader-dot">○</span><span>saving outputs...</span>';
-
-      let prog = 0;
-      progressInterval = setInterval(() => {{
-        prog = Math.min(prog + Math.random() * 3, 88);
-        progressBar.style.width = prog + '%';
-
-        if (prog > 30) {{
-          ll1.className = 'loader-line done';
-          ll1.innerHTML = '<span class="loader-check">✓</span><span>pipeline_standard ready</span>';
-        }}
-        if (prog > 50) {{
-          ll2.className = 'loader-line done';
-          ll2.innerHTML = '<span class="loader-check">✓</span><span>pipeline_small_decoder ready</span>';
-          ll3.className = 'loader-line active';
-          ll3.innerHTML = '<div class="loader-spinner"></div><span>decoding latents...</span>';
-        }}
-        if (prog > 72) {{
-          ll3.className = 'loader-line done';
-          ll3.innerHTML = '<span class="loader-check">✓</span><span>latents decoded</span>';
-          ll4.className = 'loader-line active';
-          ll4.innerHTML = '<div class="loader-spinner"></div><span>saving outputs...</span>';
-        }}
-      }}, 180);
-    }}
-
-    function stopLoader() {{
-      clearInterval(progressInterval);
-      progressBar.style.width = '100%';
-      setTimeout(() => loaderOverlay.classList.remove('active'), 350);
-    }}
-
-    /* ── show comparison ── */
-    function showComparison(stdUrl, smallUrl) {{
-      S.stdUrl   = stdUrl;
-      S.smallUrl = smallUrl;
-
-      imgStandard.onload = () => {{
-        emptyState.style.display   = 'none';
-        compareWidget.classList.add('visible');
-        applyPct(50);
-      }};
-
-      imgSmall.src    = smallUrl + '?t=' + Date.now();
-      imgStandard.src = stdUrl   + '?t=' + Date.now();
-
-      dlStandard.href = stdUrl;
-      dlStandard.classList.remove('hidden');
-      dlSmall.href    = smallUrl;
-      dlSmall.classList.remove('hidden');
-    }}
+      if (!isDragging) return;
+      updateSlider(e.touches[0].clientX);
+    }});
 
-    /* ── submit ── */
-    async function submitRun() {{
-      const prompt = promptEl.value.trim();
-      if (!prompt) {{ toast('$ error: prompt is required', 'error'); return; }}
+    // --- Form Submission ---
+    runBtn.onclick = async () => {{
+      const prompt = promptInput.value.trim();
+      if(!prompt) return alert("Enter a prompt");
 
       const fd = new FormData();
-      fd.append('prompt',          prompt);
-      fd.append('seed',            seedEl.value);
-      fd.append('randomize_seed',  String(randomizeSeedEl.checked));
-      fd.append('width',           widthEl.value);
-      fd.append('height',          heightEl.value);
-      fd.append('steps',           stepsEl.value);
-      fd.append('guidance_scale',  parseFloat(guidanceEl.value).toFixed(1));
-      S.files.forEach(f => fd.append('images', f));
-
-      setRunning(true);
-      startLoader();
+      fd.append('prompt', prompt);
+      fd.append('seed', document.getElementById('seed').value);
+      fd.append('randomize_seed', document.getElementById('randomize').checked);
+      fd.append('width', document.getElementById('width').value);
+      fd.append('height', document.getElementById('height').value);
+      fd.append('steps', document.getElementById('steps').value);
+      fd.append('guidance', document.getElementById('guidance').value);
+      
+      filesState.forEach(f => fd.append('images', f));
+
+      loader.style.display = 'flex';
+      runBtn.disabled = true;
 
       try {{
-        const res  = await fetch('/api/compare', {{ method: 'POST', body: fd }});
+        const res = await fetch('/api/compare', {{ method: 'POST', body: fd }});
         const data = await res.json();
-        if (!res.ok || !data.success) throw new Error(data.error || 'Process failed');
-
-        showComparison(data.standard.image_url, data.small.image_url);
-        usedSeed.textContent = String(data.seed);
-        toast('$ comparison complete — drag ◀▶ to compare', 'success');
-      }} catch(err) {{
-        toast('$ error: ' + (err.message || 'unexpected error'), 'error');
+        
+        if(data.success) {{
+          imgStd.src = data.std_url;
+          imgSmall.src = data.small_url;
+          
+          imgStd.onload = () => {{
+            sliderEmpty.style.display = 'none';
+            imgStd.style.display = 'block';
+            imgSmall.style.display = 'block';
+            sliderHandle.style.display = 'block';
+            sliderLabels.style.display = 'flex';
+            
+            const rect = sliderStage.getBoundingClientRect();
+            updateSlider(rect.left + rect.width / 2);
+          }};
+        }} else {{
+          alert('Error: ' + data.error);
+        }}
+      }} catch(e) {{
+        alert('Failed to connect to server.');
       }} finally {{
-        stopLoader();
-        setRunning(false);
+        loader.style.display = 'none';
+        runBtn.disabled = false;
       }}
-    }}
-
-    runBtn.addEventListener('click', submitRun);
-    promptEl.addEventListener('keydown', (e) => {{ if (e.ctrlKey && e.key === 'Enter') submitRun(); }});
-
-    /* ── examples ── */
-    async function fileFromUrl(url, name) {{
-      const res = await fetch(url);
-      if (!res.ok) throw new Error('Failed to fetch example: ' + url);
-      const blob = await res.blob();
-      return new File([blob], name, {{ type: blob.type || 'image/jpeg' }});
-    }}
-
-    function renderExamples() {{
-      examplesGrid.innerHTML = '';
-      if (!examples.length) {{
-        examplesGrid.innerHTML = '<div style="color:var(--text-muted);font-size:12px;padding:8px;"># no examples found in ./examples/</div>';
-        return;
-      }}
-      examples.forEach(item => {{
-        const card = document.createElement('div');
-        card.className = 'example-card';
-
-        const imgWrap = document.createElement('div');
-        imgWrap.className = 'ex-img-wrap';
-
-        const img = document.createElement('img');
-        img.src     = item.url;
-        img.alt     = item.file;
-        img.loading = 'lazy';
-
-        const badge = document.createElement('span');
-        badge.className   = 'ex-file-badge';
-        badge.textContent = item.file;
-
-        imgWrap.appendChild(img);
-        imgWrap.appendChild(badge);
-
-        const body = document.createElement('div');
-        body.className = 'ex-body';
-
-        const p = document.createElement('p');
-        p.className   = 'ex-prompt';
-        p.textContent = item.prompt;
-
-        const btn = document.createElement('button');
-        btn.type      = 'button';
-        btn.className = 'ex-use-btn';
-        btn.textContent = 'this-example';
-        btn.addEventListener('click', async (e) => {{
-          e.stopPropagation();
-          try {{
-            btn.textContent = 'loading...';
-            btn.disabled    = true;
-            const file = await fileFromUrl(item.url, item.file);
-            S.files = [file];
-            renderPreviews();
-            promptEl.value = item.prompt;
-            toast('$ example loaded: ' + item.file, 'success');
-          }} catch(err) {{
-            toast('$ error: ' + err.message, 'error');
-          }} finally {{
-            btn.textContent = 'this-example';
-            btn.disabled    = false;
-          }}
-        }});
-
-        body.appendChild(p);
-        body.appendChild(btn);
-        card.appendChild(imgWrap);
-        card.appendChild(body);
-        examplesGrid.appendChild(card);
-      }});
-    }}
-
-    /* ── init ── */
-    setRunning(false);
-    renderPreviews();
-    renderExamples();
+    }};
   </script>
 </body>
-</html>"""
-
+</html>
+"""
 
 app.launch()