studio with gradio server and workflow

README.md

@@ -4,12 +4,33 @@ emoji: 🖼️
 colorFrom: purple
 colorTo: indigo
 sdk: gradio
-sdk_version: 6.28.0
+sdk_version: 6.30.0
 python_version: "3.12"
 app_file: app.py
 pinned: false
 startup_duration_timeout: 1h
 short_description: Text-to-image and image editing, based on MiniMax-H3
+tags:
+  - ml-intern
 models:
   - PrunaAI/P-H3-Image
 ---
+
+<a href="https://hf.co/chat?mode=ml-intern"><img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/built-with-ml-intern-sm.svg" alt="Built with ML Intern" /></a>
+
+P-H3-Image on ZeroGPU (`xlarge`: a full RTX PRO 6000 Blackwell, 96 GB) — now behind two bespoke
+surfaces modeled on [briaai/Fibo-2.0-editing](https://huggingface.co/spaces/briaai/Fibo-2.0-editing),
+with the backend (`p_h3_image_fast.py`, prompt upsampler, warmup) unchanged:
+
+- **`/` — Studio** (`gr.Server` API + custom UI, Pruna design, dark and light themes): Generate
+  and Edit modes with prompt upsampling, aspect-ratio and resolution controls (4 / 8 / 16 MP
+  text-to-image, 4 / 8 MP edits at the source's aspect ratio), up to 8 reference images, seed
+  control, all of the original examples, before/after split view, run history with one-click
+  restore, and the upsampled prompt after each run. Both modes call the same `_render` as the
+  classic UI did.
+- **`/workflow` — Workflow canvas** (`gr.Workflow`): node-based generate/edit pipelines you can
+  rewire and chain — text-to-image into chained edits, or your own source image. To change the
+  starter graph, edit `workflow.json` in the repo.
+
+The classic two-tab form UI is replaced by the studio; `sdk_version` moves to 6.30.0 for
+`gr.Server`/`gr.Workflow`.

app.py

@@ -1,16 +1,24 @@
-"""P-H3-Image on ZeroGPU (`xlarge`: a full RTX PRO 6000 Blackwell, 96 GB).
+"""P-H3-Image Studio — gr.Server studio + gr.Workflow canvas.
 
-The pipeline is loaded once at startup, one component at a time (the host has ~97 GB), and slimmed to fit the slice
-with p_h3_image_fast.py (kernels only, no torch.compile): AdaLN tables instead of the transformer's AdaLN projections,
-FP8 linears everywhere, fused Triton DiT blocks, batched VAE tiles, and for the prompt upsampler a weight-only FP8 GEMV,
-decode attention and speculative decoding with its MTP head in one CUDA graph per step.
+The bespoke studio at / replaces the classic form UI; the backend is unchanged from
+PrunaAI/P-H3-Image: the pipeline is loaded once at startup, one component at a time (the host has
+~97 GB), and slimmed to fit the slice with p_h3_image_fast.py (kernels only, no torch.compile):
+AdaLN tables instead of the transformer's AdaLN projections, FP8 linears everywhere, fused Triton
+DiT blocks, batched VAE tiles, and for the prompt upsampler a weight-only FP8 GEMV, decode
+attention and speculative decoding with its MTP head in one CUDA graph per step. The studio's
+Generate/Edit calls and the workflow canvas at /workflow all run through the same _render.
+
+Server + workflow structure adapted from https://huggingface.co/spaces/briaai/Fibo-2.0-editing.
 """
 
+import base64
 import json
 import os
 import random
+import shutil
 import tempfile
 import time
+from urllib.parse import unquote, urlparse
 
 os.environ.setdefault("DIFFUSERS_TRUST_REMOTE_KERNELS", "true")
 os.environ.setdefault("PYTORCH_CUDA_ALLOC_CONF", "expandable_segments:True")
@@ -18,19 +26,23 @@ os.environ.setdefault("PYTORCH_CUDA_ALLOC_CONF", "expandable_segments:True")
 import gradio as gr
 import spaces
 import torch
+import uvicorn
+from fastapi.responses import FileResponse, HTMLResponse
+from gradio.utils import get_upload_folder
+from gradio_client import utils as client_utils
 from huggingface_hub import snapshot_download
 from PIL import Image, ImageOps
 from torchvision.io import encode_jpeg
 
 import p_h3_image_fast as fast
 from diffusers import ModularPipeline
-from examples import EXAMPLES
-from theme import CSS, HEAD, THEME, render_footer, render_header
 
 REPO, REVISION = "PrunaAI/P-H3-Image", "ee4a685de1fbd4337a50bcf9b3c407e4f8463d49"
+APP_DIR = os.path.dirname(os.path.abspath(__file__))
 # Canvases (width, height), multiples of 32: 4 and 8 MP as tested, 16 MP from the same ratios.
 RATIOS = {"1:1": 1.0, "4:3": 4 / 3, "3:4": 3 / 4, "3:2": 3 / 2, "2:3": 2 / 3, "16:9": 16 / 9, "9:16": 9 / 16}
 T2I_RESOLUTIONS, EDIT_RESOLUTIONS = ["4 MP", "8 MP", "16 MP"], ["4 MP", "8 MP"]
+MAX_SEED = 2**31 - 1
 
 
 def _canvas(ratio: float, megapixels: int) -> tuple[int, int]:
@@ -52,7 +64,7 @@ REFERENCE_BUDGET = 8 * 1024**2
 
 def load_pipeline():
     path = snapshot_download(REPO, revision=REVISION, max_workers=16)
-    with open(os.path.join(os.path.dirname(__file__), "adaln_calls.json")) as handle:
+    with open(os.path.join(APP_DIR, "adaln_calls.json")) as handle:
         calls = json.load(handle)
     pipe, mtp = fast.load_slim_pipeline(path, calls, log=lambda message: print(message, flush=True))
     # The same components without the prompt upsampler step, for requests that skip upsampling.
@@ -166,17 +178,6 @@ def _info(timing, size, seed):
     return f"{head} ({upsample:.1f} s prompt upsampling{rate}, {timing['total'] - upsample:.1f} s rendering)"
 
 
-def text_to_image(prompt, aspect, resolution, upsample):
-    if not prompt or not prompt.strip():
-        raise gr.Error("Write a prompt first.")
-    seed = random.randint(0, 2**31 - 1)
-    try:
-        path, size, upsampled, timing = _render("t2i", prompt.strip(), [], aspect, resolution, upsample, seed)
-    except RuntimeError as error:
-        raise gr.Error(str(error).split("\n")[0])
-    return path, upsampled, _info(timing, size, seed)
-
-
 def _fit(image: Image.Image, pixels: float) -> Image.Image:
     """Downscale to at most `pixels` (the pipeline keeps references at their own size, from 1 MP up)."""
     scale = (pixels / (image.width * image.height)) ** 0.5
@@ -185,63 +186,207 @@ def _fit(image: Image.Image, pixels: float) -> Image.Image:
     return image.resize((max(1, round(image.width * scale)), max(1, round(image.height * scale))), Image.LANCZOS)
 
 
-def edit(gallery, prompt, resolution, upsample):
-    images = []
-    for item in gallery or []:
-        item = item[0] if isinstance(item, (tuple, list)) else item
-        item = Image.open(item) if isinstance(item, str) else item
-        images.append(ImageOps.exif_transpose(item).convert("RGB"))
-    if not images:
-        raise gr.Error("Add at least one image; the first one is edited.")
-    if len(images) > MAX_IMAGES:
-        raise gr.Error(f"At most {MAX_IMAGES} images.")
-    if not prompt or not prompt.strip():
-        raise gr.Error("Describe the edit first.")
-    images = [_fit(image, REFERENCE_BUDGET / len(images)) for image in images]
-    seed = random.randint(0, 2**31 - 1)
+# --------------------------------------------------------------------------- helpers
+
+
+def _resolve_seed(seed, randomize) -> int:
+    return random.randint(0, MAX_SEED) if randomize else min(max(int(float(seed)), 0), MAX_SEED)
+
+
+def _check_choice(value, choices, name):
+    if value not in choices:
+        raise gr.Error(f"{name} must be one of: {', '.join(choices)}.")
+    return value
+
+
+def _as_data_uri(path: str) -> str:
+    """Inline a result image so the studio page never needs a file route."""
+    with open(path, "rb") as handle:
+        return "data:image/jpeg;base64," + base64.b64encode(handle.read()).decode()
+
+
+def _requested_path(value) -> str:
+    """Turn whatever the canvas/API client sends into a path."""
+    if isinstance(value, str):
+        return value
+    if not isinstance(value, dict):
+        raise gr.Error(f"Expected an image, got {type(value).__name__}")
+    if path := value.get("path"):
+        return path
+    url = value.get("url") or ""
+    if "/gradio_api/file=" in url:
+        return unquote(url.split("/gradio_api/file=", 1)[1])
+    if url.startswith(("http://", "https://")):
+        raise gr.Error("Upload the image instead of passing a URL.")
+    return unquote(urlparse(url).path)
+
+
+def _local_path(value) -> str:
+    """A path this process may open: only files inside gradio's upload folder,
+    where /gradio_api/upload puts uploads and _publish puts results."""
+    path = os.path.realpath(_requested_path(value))
+    folder = os.path.realpath(get_upload_folder())
+    if os.path.commonpath([path, folder]) != folder:
+        raise gr.Error("Upload the image first.")
+    return path
+
+
+def _open_rgb(value) -> Image.Image:
+    return ImageOps.exif_transpose(Image.open(_local_path(value))).convert("RGB")
+
+
+def _publish(path: str) -> str:
+    """Copy a result into gradio's upload folder, which it already serves, so the workflow canvas can render it."""
+    folder = get_upload_folder()
+    os.makedirs(folder, exist_ok=True)
+    out = os.path.join(folder, f"ph3_{os.urandom(8).hex()}.jpg")
+    shutil.copyfile(path, out)
+    return out
+
+
+def _as_media(path: str) -> dict:
+    """The dict shape the workflow canvas renders media from."""
+    return {
+        "path": path,
+        "url": f"/gradio_api/file={client_utils.encode_file_path(path)}",
+        "is_file": True,
+    }
+
+
+def _run(mode, prompt, images, aspect, resolution, upsample, seed):
+    """One _render call with the classic app's error surface: messages, not stack traces."""
     try:
-        path, size, upsampled, timing = _render("edit", prompt.strip(), images, None, resolution, upsample, seed)
+        return _render(mode, prompt, images, aspect, resolution, upsample, seed)
     except (RuntimeError, ValueError) as error:
         raise gr.Error(str(error).split("\n")[0])
-    return path, upsampled, _info(timing, size, seed)
-
-
-with gr.Blocks(title="P-H3-Image", fill_width=True) as demo:
-    render_header()
-    with gr.Tabs(elem_classes="main-tabs"):
-        with gr.Tab("Text to image"):
-            with gr.Row():
-                with gr.Column():
-                    t_prompt = gr.Textbox(label="Prompt", lines=4)
-                    with gr.Row():
-                        t_aspect = gr.Dropdown(list(RATIOS), value="4:3", label="Aspect ratio", filterable=False)
-                        t_resolution = gr.Radio(T2I_RESOLUTIONS, value="8 MP", label="Resolution")
-                    t_upsample = gr.Checkbox(value=True, label="Prompt upsampling")
-                    t_button = gr.Button("Generate", variant="primary")
-                with gr.Column():
-                    t_image = gr.Image(label="Result", type="filepath")
-                    t_info = gr.Markdown()
-                    with gr.Accordion("Upsampled prompt", open=False):
-                        t_upsampled = gr.Textbox(show_label=False, lines=10)
-            gr.Examples([[prompt, aspect] for _, aspect, prompt in EXAMPLES], inputs=[t_prompt, t_aspect],
-                        example_labels=[label for label, _, _ in EXAMPLES])
-            t_button.click(text_to_image, [t_prompt, t_aspect, t_resolution, t_upsample], [t_image, t_upsampled, t_info])
-        with gr.Tab("Edit"):
-            with gr.Row():
-                with gr.Column():
-                    e_images = gr.Gallery(label=f"Images (the first one is edited, up to {MAX_IMAGES})", type="pil",
-                                          columns=4, height=260, interactive=True)
-                    e_prompt = gr.Textbox(label="Edit", lines=3, placeholder="e.g. put the fox from <Picture 2> on the bench")
-                    e_resolution = gr.Radio(EDIT_RESOLUTIONS, value="8 MP", label="Resolution",
-                                            info="The first image's aspect ratio at this size.")
-                    e_upsample = gr.Checkbox(value=True, label="Prompt upsampling")
-                    e_button = gr.Button("Edit", variant="primary")
-                with gr.Column():
-                    e_image = gr.Image(label="Result", type="filepath")
-                    e_info = gr.Markdown()
-                    with gr.Accordion("Upsampled prompt", open=False):
-                        e_upsampled = gr.Textbox(show_label=False, lines=10)
-            e_button.click(edit, [e_images, e_prompt, e_resolution, e_upsample], [e_image, e_upsampled, e_info])
-    render_footer()
-
-demo.queue(max_size=20).launch(theme=THEME, css=CSS, head=HEAD)
+
+
+# ------------------------------------------------------------------ workflow fns
+
+
+def ph3_generate(prompt, aspect: str = "4:3", resolution: str = "8 MP", seed: str = "42") -> dict:
+    """Text-to-image with P-H3-Image (prompt upsampling on)."""
+    if not prompt or not str(prompt).strip():
+        raise gr.Error("Write a prompt first.")
+    _check_choice(aspect, RATIOS, "Aspect ratio")
+    _check_choice(resolution, T2I_RESOLUTIONS, "Resolution")
+    seed_value = min(max(int(float(seed)), 0), MAX_SEED)
+    path, _, _, _ = _run("t2i", str(prompt).strip(), [], aspect, resolution, True, seed_value)
+    return _as_media(_publish(path))
+
+
+def ph3_edit(image, prompt, seed: str = "42") -> dict:
+    """Edit an image with P-H3-Image (prompt upsampling on, 8 MP at the image's aspect ratio)."""
+    if not prompt or not str(prompt).strip():
+        raise gr.Error("Describe the edit first.")
+    source = _fit(_open_rgb(image), REFERENCE_BUDGET)
+    seed_value = min(max(int(float(seed)), 0), MAX_SEED)
+    path, _, _, _ = _run("edit", str(prompt).strip(), [source], None, "8 MP", True, seed_value)
+    return _as_media(_publish(path))
+
+
+# ------------------------------------------------------------------ studio server
+
+server = gr.Server(title="P-H3-Image Studio")
+
+
+@server.get("/", response_class=HTMLResponse)
+def _studio_page():
+    with open(os.path.join(APP_DIR, "studio.html"), encoding="utf-8") as handle:
+        return handle.read()
+
+
+@server.api(name="generate", concurrency_limit=1)
+def api_generate(
+    prompt: str,
+    aspect: str = "4:3",
+    resolution: str = "8 MP",
+    upsample: bool = True,
+    seed: float = 42,
+    randomize: bool = True,
+) -> dict:
+    """Text-to-image with P-H3-Image."""
+    if not prompt or not prompt.strip():
+        raise gr.Error("Write a prompt first.")
+    _check_choice(aspect, RATIOS, "Aspect ratio")
+    _check_choice(resolution, T2I_RESOLUTIONS, "Resolution")
+    seed_value = _resolve_seed(seed, randomize)
+    path, size, upsampled, timing = _run("t2i", prompt.strip(), [], aspect, resolution, bool(upsample), seed_value)
+    return {"image_data": _as_data_uri(path), "seed": seed_value, "prompt": upsampled,
+            "info": _info(timing, size, seed_value)}
+
+
+@server.api(name="edit", concurrency_limit=1)
+def api_edit(
+    source: str,
+    instruction: str,
+    ref2: str | None = None,
+    ref3: str | None = None,
+    ref4: str | None = None,
+    ref5: str | None = None,
+    ref6: str | None = None,
+    ref7: str | None = None,
+    ref8: str | None = None,
+    resolution: str = "8 MP",
+    upsample: bool = True,
+    seed: float = 42,
+    randomize: bool = True,
+) -> dict:
+    """Edit the source (+ up to 7 extra references) with P-H3-Image."""
+    if source is None:
+        raise gr.Error("Upload a source image first.")
+    if not instruction or not instruction.strip():
+        raise gr.Error("Describe the edit first.")
+    _check_choice(resolution, EDIT_RESOLUTIONS, "Resolution")
+    images = [_open_rgb(path) for path in (source, ref2, ref3, ref4, ref5, ref6, ref7, ref8) if path]
+    if len(images) > MAX_IMAGES:
+        raise gr.Error(f"At most {MAX_IMAGES} images.")
+    images = [_fit(image, REFERENCE_BUDGET / len(images)) for image in images]
+    seed_value = _resolve_seed(seed, randomize)
+    path, size, upsampled, timing = _run("edit", instruction.strip(), images, None, resolution, bool(upsample),
+                                         seed_value)
+    return {"image_data": _as_data_uri(path), "seed": seed_value, "prompt": upsampled,
+            "info": _info(timing, size, seed_value)}
+
+
+# The @server.api endpoints are deferred until Server.launch(), which also fires
+# the ZeroGPU registration hook (pack weights + startup report) through the
+# patched Blocks.launch path. Launch on a loopback port for registration, then
+# serve the same app via uvicorn.
+_port_env = os.environ.get("GRADIO_SERVER_PORT")
+os.environ["GRADIO_SERVER_PORT"] = "7861"
+try:
+    server.launch(
+        prevent_thread_lock=True,
+        server_name="127.0.0.1",
+        server_port=7861,
+        quiet=True,
+        show_error=False,
+        ssr_mode=False,
+    )
+finally:
+    if _port_env is None:
+        del os.environ["GRADIO_SERVER_PORT"]
+    else:
+        os.environ["GRADIO_SERVER_PORT"] = _port_env
+
+try:
+    server.close()
+except Exception:
+    pass
+
+# ------------------------------------------------------------------ workflow canvas
+# Mounted only after the registration launch above. Mounting first starts the
+# workflow's queue on the loopback server's event loop, and close() then strands
+# it: jobs are accepted on port 7860 but never run.
+
+workflow_demo = gr.Workflow(
+    graph="workflow.json",
+    bind={"ph3_generate": ph3_generate, "ph3_edit": ph3_edit},
+)
+workflow_demo.title = "P-H3-Image Workflow"
+
+server = gr.mount_gradio_app(server, workflow_demo, path="/workflow", ssr_mode=False)
+
+if __name__ == "__main__":
+    uvicorn.run(server, host="0.0.0.0", port=7860)

studio.html

@@ -0,0 +1,781 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+<meta charset="utf-8" />
+<meta name="viewport" content="width=device-width, initial-scale=1" />
+<title>P-H3-Image Studio</title>
+<link rel="preconnect" href="https://fonts.googleapis.com" />
+<link rel="preconnect" href="https://fonts.gstatic.com" crossorigin />
+<link href="https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&family=IBM+Plex+Mono:wght@400;500&display=swap" rel="stylesheet" />
+<style>
+  :root {
+    --bg: #120b1b; --panel: #171021; --panel2: #1d1429; --raised: #251b34;
+    --border: rgba(74,57,98,0.75); --hairline: rgba(255,255,255,0.08);
+    --text: #fafafa; --muted: #a3a3a3; --dim: #71717a;
+    --accent: #9334e9; --accent2: #8f35ff; --accent-soft: rgba(147,52,233,0.14);
+    --lavender: #d8b4fe; --link: #bf8bff;
+    --glow: radial-gradient(ellipse 60% 35% at 50% -5%, rgba(147,52,233,0.18), transparent 55%);
+    --stage-base: #0e0a16; --stage-square: #171221; --thumb-bg: #120c1c;
+    --radius: 10px;
+  }
+  body.light {
+    --bg: #f6f3fa; --panel: #ffffff; --panel2: #f6f3fa; --raised: #f3eef8;
+    --border: rgba(126,34,206,0.16); --hairline: rgba(24,24,27,0.08);
+    --text: #18181b; --muted: #71717a; --dim: #a1a1aa;
+    --accent: #9334e9; --accent2: #7c22ce; --accent-soft: rgba(147,52,233,0.10);
+    --lavender: #7e22ce; --link: #6d28d9;
+    --glow: radial-gradient(ellipse 60% 35% at 50% -5%, rgba(147,52,233,0.10), transparent 55%);
+    --stage-base: #e9e4f2; --stage-square: #f8f5fc; --thumb-bg: #f3eef8;
+  }
+  * { margin: 0; padding: 0; box-sizing: border-box; }
+  [hidden] { display: none !important; }
+  body {
+    font-family: "Inter", ui-sans-serif, system-ui, sans-serif;
+    background: var(--bg); background-image: var(--glow); background-repeat: no-repeat;
+    background-attachment: fixed;
+    color: var(--text); height: 100vh;
+    display: flex; flex-direction: column; overflow: hidden;
+    font-size: 13px; -webkit-font-smoothing: antialiased;
+  }
+  .topbar {
+    display: flex; align-items: center; gap: 14px; padding: 8px 16px;
+    background: var(--panel); border-bottom: 1px solid var(--hairline); flex-shrink: 0;
+  }
+  .brand { font-weight: 700; font-size: 0.95rem; white-space: nowrap; letter-spacing: -0.02em; display: flex; align-items: center; }
+  .brand img { height: 26px; margin-right: 10px; }
+  .studio-badge {
+    font-size: 0.6rem; font-weight: 650; letter-spacing: 0.12em; text-transform: uppercase;
+    color: var(--lavender); border: 1px solid var(--border); border-radius: 999px;
+    padding: 2px 8px; margin-left: 10px; background: var(--accent-soft);
+  }
+  .mode-tabs { display: flex; gap: 2px; background: var(--panel2); border: 1px solid var(--border); border-radius: 9px; padding: 3px; }
+  .mode-tabs button {
+    border: none; border-radius: 7px; padding: 6px 16px; font-weight: 550; font-size: 0.82rem;
+    background: transparent; color: var(--muted); cursor: pointer; transition: color 0.15s, background 0.15s;
+    font-family: inherit;
+  }
+  .mode-tabs button:hover { color: var(--text); }
+  .mode-tabs button.active { background: var(--raised); color: var(--text); box-shadow: 0 1px 3px rgba(0,0,0,0.45); }
+  body.light .mode-tabs button.active { box-shadow: 0 1px 3px rgba(88,28,135,0.15); }
+  .spacer { flex: 1; }
+  .ml-badge { color: var(--muted); text-decoration: none; font-size: 0.8rem; font-weight: 500; transition: color 0.15s; white-space: nowrap; }
+  .ml-badge:hover { color: var(--text); }
+  .ml-badge-img { display: inline-flex; align-items: center; }
+  .ml-badge-img img { height: 20px; display: block; }
+  .ml-badge-img:hover { opacity: 0.85; }
+  .ml-cta {
+    color: var(--lavender); text-decoration: none; font-size: 0.8rem; font-weight: 600;
+    padding: 5px 13px; border: 1px solid var(--accent); border-radius: 999px;
+    background: var(--accent-soft); transition: filter 0.15s; white-space: nowrap;
+  }
+  .ml-cta:hover { filter: brightness(1.2); }
+  .theme-toggle {
+    width: 34px; height: 34px; border: 1px solid var(--border); border-radius: 9px;
+    background: transparent; color: var(--muted); cursor: pointer; flex-shrink: 0;
+    display: inline-flex; align-items: center; justify-content: center;
+    transition: color 0.15s, background 0.15s;
+  }
+  .theme-toggle:hover { color: var(--text); background: var(--panel2); }
+  .theme-toggle svg { width: 17px; height: 17px; }
+  .theme-toggle .theme-icon-moon { display: none; }
+  .theme-toggle[data-mode="light"] .theme-icon-sun { display: none; }
+  .theme-toggle[data-mode="light"] .theme-icon-moon { display: block; }
+  .main { display: flex; flex: 1; min-height: 0; }
+  .rail {
+    width: 52px; flex-shrink: 0; display: flex; flex-direction: column; align-items: center;
+    gap: 6px; padding: 10px 0; background: var(--panel); border-right: 1px solid var(--hairline);
+  }
+  .rail button {
+    width: 38px; height: 38px; border: 1px solid transparent; border-radius: 8px;
+    background: transparent; color: var(--muted); cursor: pointer;
+    display: flex; align-items: center; justify-content: center;
+    transition: color 0.15s, background 0.15s;
+  }
+  .rail button:hover { background: var(--panel2); color: var(--text); }
+  .rail button.active { color: var(--accent2); background: var(--accent-soft); }
+  .viewport { flex: 1; min-width: 0; display: flex; flex-direction: column; overflow: auto; position: relative; }
+  .workflow-view { position: absolute; inset: 0; padding: 12px; }
+  .workflow-view iframe { width: 100%; height: 100%; border: none; border-radius: 10px; background: var(--stage-base); }
+  .doc-grid { flex: 1; min-height: 0; display: flex; gap: 16px; width: 100%; align-items: stretch; justify-content: center; padding: 20px 20px 10px; }
+  .doc {
+    flex: 1; min-width: 0; max-width: 640px; display: flex; flex-direction: column;
+    background: var(--panel); border: 1px solid var(--border); border-radius: 10px; overflow: hidden;
+  }
+  .doc figcaption {
+    padding: 8px 12px; font-size: 0.62rem; font-weight: 650; letter-spacing: 0.12em;
+    color: var(--dim); border-bottom: 1px solid var(--hairline); text-transform: uppercase;
+    display: flex; align-items: center;
+  }
+  .doc .stage {
+    position: relative; flex: 1; min-height: 0; display: flex; align-items: center; justify-content: center;
+    overflow: hidden; background-color: var(--stage-base);
+    background-image:
+      linear-gradient(45deg, var(--stage-square) 25%, transparent 25%),
+      linear-gradient(-45deg, var(--stage-square) 25%, transparent 25%),
+      linear-gradient(45deg, transparent 75%, var(--stage-square) 75%),
+      linear-gradient(-45deg, transparent 75%, var(--stage-square) 75%);
+    background-size: 20px 20px;
+    background-position: 0 0, 0 10px, 10px -10px, -10px 0;
+  }
+  .doc img { max-width: 100%; max-height: 100%; object-fit: contain; display: block; }
+  .doc img.zoomed { max-width: none; max-height: none; }
+  .doc .stage.scrollable { overflow: auto; justify-content: flex-start; align-items: flex-start; }
+  .drop-hint { color: var(--dim); font-size: 0.85rem; text-align: center; padding: 20px; }
+  .drop-hint b { color: var(--muted); display: block; margin-bottom: 6px; }
+  .zoom-btn {
+    margin-left: auto; border: 1px solid var(--border); background: transparent; color: var(--muted);
+    border-radius: 5px; font-size: 0.62rem; padding: 2px 8px; cursor: pointer; letter-spacing: 0.08em;
+    font-family: inherit;
+  }
+  .zoom-btn:hover { color: var(--text); border-color: var(--accent); }
+  .doc-bar {
+    display: flex; align-items: center; gap: 10px; padding: 6px 12px;
+    border-top: 1px solid var(--hairline); background: var(--panel);
+    font-size: 0.72rem; color: var(--muted); flex-shrink: 0;
+  }
+  .doc-bar button {
+    border: 1px solid var(--border); background: transparent; color: var(--muted);
+    border-radius: 5px; padding: 3px 10px; font-size: 0.68rem; cursor: pointer; letter-spacing: 0.06em;
+    font-family: inherit;
+  }
+  .doc-bar button:hover, .doc-bar button.active { color: var(--text); border-color: var(--accent); }
+  .doc-bar button.active { background: var(--accent-soft); color: var(--lavender); }
+  .split-wrap { position: absolute; inset: 0; overflow: hidden; }
+  .split-wrap img { position: absolute; inset: 0; width: 100%; height: 100%; object-fit: contain; }
+  .split-divider {
+    position: absolute; top: 0; bottom: 0; width: 3px; background: var(--accent);
+    cursor: ew-resize; z-index: 5; box-shadow: 0 0 8px rgba(147,52,233,0.6);
+  }
+  .split-divider::after {
+    content: "◂ ▸"; position: absolute; top: 50%; left: 50%; transform: translate(-50%,-50%);
+    background: var(--accent); color: white; border-radius: 999px; padding: 4px 7px;
+    font-size: 0.6rem; white-space: nowrap;
+  }
+  .history-strip {
+    display: flex; gap: 8px; padding: 8px 12px; overflow-x: auto; flex-shrink: 0;
+    background: var(--panel); border-top: 1px solid var(--hairline); align-items: center;
+  }
+  .history-strip .hlabel { font-size: 0.62rem; font-weight: 650; letter-spacing: 0.14em; color: var(--dim); text-transform: uppercase; white-space: nowrap; }
+  .history-chip {
+    width: 52px; height: 52px; border-radius: 6px; border: 1px solid var(--border);
+    overflow: hidden; cursor: pointer; flex-shrink: 0; position: relative;
+  }
+  .history-chip:hover { border-color: var(--accent); }
+  .history-chip img { width: 100%; height: 100%; object-fit: cover; }
+  .history-chip .hseed {
+    position: absolute; bottom: 0; left: 0; right: 0; background: rgba(0,0,0,0.65);
+    color: #ddd; font-size: 0.52rem; padding: 1px 3px; text-align: center;
+  }
+  body.light .history-chip .hseed { background: rgba(255,255,255,0.78); color: #3f3f46; }
+  .panels {
+    width: 320px; flex-shrink: 0; display: flex; flex-direction: column; gap: 10px;
+    padding: 12px; overflow-y: auto; background: var(--panel); border-left: 1px solid var(--hairline);
+  }
+  .panel { background: var(--panel2); border: 1px solid var(--border); border-radius: var(--radius); padding: 12px; }
+  .panel-title { font-size: 0.62rem; font-weight: 650; letter-spacing: 0.12em; color: var(--dim); text-transform: uppercase; margin-bottom: 10px; cursor: pointer; user-select: none; display: flex; align-items: center; justify-content: space-between; }
+  .panel .panel-title::after { content: "▾"; font-size: 0.7rem; color: var(--dim); }
+  .panel.collapsed .panel-title::after { content: "▸"; }
+  .panel.collapsed .panel-body { display: none; }
+  .panel label { display: block; font-size: 0.72rem; font-weight: 550; color: var(--muted); margin: 10px 0 4px; }
+  .panel label:first-of-type { margin-top: 0; }
+  textarea, input[type="text"], input[type="number"], select {
+    width: 100%; background: var(--raised); color: var(--text); border: 1px solid var(--border);
+    border-radius: 7px; padding: 8px 10px; font-family: inherit; font-size: 0.83rem; resize: vertical;
+    transition: border-color 0.15s, box-shadow 0.15s;
+  }
+  select { appearance: none; background-image: linear-gradient(45deg, transparent 50%, var(--muted) 50%), linear-gradient(135deg, var(--muted) 50%, transparent 50%); background-position: calc(100% - 16px) 50%, calc(100% - 11px) 50%; background-size: 5px 5px; background-repeat: no-repeat; }
+  textarea:focus, input:focus, select:focus { outline: none; border-color: var(--accent2); box-shadow: 0 0 0 3px var(--accent-soft); }
+  textarea.code { font-family: "IBM Plex Mono", ui-monospace, monospace; font-size: 0.72rem; }
+  textarea[readonly] { color: var(--muted); }
+  .row { display: flex; gap: 8px; align-items: center; }
+  .row > * { flex: 1; }
+  .switch { display: inline-flex; align-items: center; gap: 8px; cursor: pointer; }
+  .switch input { display: none; }
+  .switch .sw {
+    display: inline-block;
+    width: 34px; height: 20px; border-radius: 999px; background: var(--raised);
+    border: 1px solid var(--border); position: relative; flex-shrink: 0; transition: background 0.18s, border-color 0.18s;
+  }
+  .switch .sw::after {
+    content: ""; position: absolute; top: 2px; left: 2px; width: 14px; height: 14px;
+    border-radius: 50%; background: var(--muted); transition: transform 0.18s, background 0.18s;
+  }
+  .switch input:checked + .sw { background: var(--accent); border-color: var(--accent); }
+  .switch input:checked + .sw::after { transform: translateX(14px); background: #fff; }
+  .switch .sw-label { font-size: 0.75rem; color: var(--muted); }
+  .layer-row {
+    display: flex; align-items: center; gap: 10px; padding: 6px 8px;
+    border: 1px solid var(--border); border-radius: 6px; margin-top: 6px;
+    background: var(--raised);
+  }
+  .layer-row .thumb {
+    width: 40px; height: 40px; border-radius: 4px; object-fit: cover; flex-shrink: 0;
+    background: var(--thumb-bg); border: 1px solid var(--border);
+    display: flex; align-items: center; justify-content: center;
+    color: var(--dim); font-size: 0.65rem; text-align: center;
+  }
+  .layer-row .thumb img { width: 100%; height: 100%; object-fit: cover; border-radius: 3px; }
+  .layer-row .lname { flex: 1; font-size: 0.78rem; color: var(--text); }
+  .layer-row .lsub { font-size: 0.62rem; color: var(--dim); display: block; }
+  .ref-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 8px; margin-top: 8px; }
+  .ref-slot {
+    height: 64px; border: 1px dashed var(--border); border-radius: 6px;
+    display: flex; align-items: center; justify-content: center; cursor: pointer; overflow: hidden;
+    color: var(--dim); font-size: 0.68rem; background: var(--raised); text-align: center;
+  }
+  .ref-slot:hover { border-color: var(--accent); }
+  .ref-slot img { width: 100%; height: 100%; object-fit: cover; }
+  .hint { font-size: 0.68rem; color: var(--dim); margin-top: 8px; line-height: 1.4; }
+  #examples { display: flex; flex-direction: column; gap: 6px; max-height: 260px; overflow-y: auto; }
+  .ex-chip {
+    border: 1px solid var(--border); border-radius: 8px; cursor: pointer; flex-shrink: 0;
+    background: var(--raised); padding: 7px 10px; text-align: left;
+    display: flex; align-items: center; justify-content: space-between; gap: 8px;
+    transition: border-color 0.15s; font-family: inherit;
+  }
+  .ex-chip:hover { border-color: var(--accent); }
+  .ex-chip .ex-label { font-size: 0.75rem; color: var(--text); font-weight: 550; }
+  .ex-chip .ex-aspect { font-size: 0.62rem; color: var(--lavender); font-family: "IBM Plex Mono", ui-monospace, monospace; }
+  #run-btn {
+    margin-top: 4px; padding: 12px; border: 1px solid rgba(147,52,233,0.7); border-radius: 8px; cursor: pointer;
+    background: var(--accent); color: white; box-shadow: inset 0 1px rgba(255,255,255,0.2);
+    font-weight: 600; font-size: 0.9rem; letter-spacing: 0.01em; font-family: inherit;
+    transition: filter 0.15s, transform 0.05s;
+  }
+  #run-btn:hover { filter: brightness(1.12); }
+  #run-btn:active { transform: translateY(1px); }
+  #run-btn:disabled { opacity: 0.5; cursor: wait; }
+  .statusbar {
+    display: flex; align-items: center; gap: 16px; padding: 7px 16px; flex-shrink: 0;
+    background: var(--panel); border-top: 1px solid var(--hairline);
+    font-size: 0.75rem; color: var(--muted);
+  }
+  .statusbar .pulse { width: 8px; height: 8px; border-radius: 50%; background: #4fd1a5; }
+  .statusbar.busy .pulse { background: var(--accent2); animation: pulse 1s infinite alternate; }
+  @keyframes pulse { from { opacity: 0.4; } to { opacity: 1; } }
+  .foot-links { display: flex; gap: 12px; align-items: center; }
+  .foot-links a { color: var(--dim); text-decoration: none; font-size: 0.72rem; font-weight: 500; }
+  .foot-links a:hover { color: var(--lavender); }
+  #file-input { display: none; }
+  @media (max-width: 860px) {
+    html, body { height: auto !important; overflow: visible !important; }
+    body { display: block; }
+    .topbar { flex-wrap: wrap; gap: 8px; padding: 8px 12px; }
+    .brand { font-size: 0.88rem; }
+    .topbar .ml-badge, .topbar .ml-badge-img { display: none; }
+    .main { display: block; height: auto; min-height: 0; }
+    .rail {
+      flex-direction: row; width: 100%; justify-content: center; gap: 10px;
+      padding: 6px 10px; border-right: none; border-bottom: 1px solid var(--hairline);
+    }
+    .rail button { width: 44px; height: 44px; }
+    .viewport { display: block; padding: 10px; overflow: visible; }
+    .workflow-view { position: static; height: 78vh; padding: 0 0 10px; }
+    .doc-grid { flex-direction: column; align-items: stretch; height: auto; }
+    .doc { max-width: 100%; width: 100%; }
+    .doc .stage { min-height: 220px; }
+    .panels {
+      width: 100%; border-left: none; border-top: 1px solid var(--hairline);
+      padding: 10px 12px; overflow-y: visible;
+    }
+    .mode-tabs button { padding: 6px 13px; font-size: 0.8rem; }
+    .foot-links { display: none; }
+    .statusbar { position: sticky; bottom: 0; }
+  }
+</style>
+</head>
+<body>
+  <header class="topbar">
+    <div class="brand"><img src="https://huggingface.co/spaces/PrunaAI/P-H3-Image/resolve/main/pruna-logo.png" alt="Pruna" />P-H3-Image<span class="studio-badge">Studio</span></div>
+    <div class="mode-tabs">
+      <button id="mode-generate" class="active">Generate</button>
+      <button id="mode-edit">Edit</button>
+      <button id="mode-workflow">Workflow</button>
+    </div>
+    <div class="spacer"></div>
+    <a class="ml-badge-img" href="https://hf.co/chat?mode=ml-intern" target="_blank" rel="noopener">
+      <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/built-with-ml-intern-md.svg" alt="Built with ML Intern" />
+    </a>
+    <a class="ml-cta" href="https://hf.co/chat?mode=ml-intern" target="_blank" rel="noopener">Build your own app here →</a>
+    <a class="ml-badge" href="https://huggingface.co/PrunaAI/P-H3-Image" target="_blank" rel="noopener">Model card</a>
+    <a class="ml-badge" href="https://www.pruna.ai/" target="_blank" rel="noopener">pruna.ai</a>
+    <button type="button" class="theme-toggle" id="theme-toggle" data-mode="dark" aria-label="Switch to light mode" title="Switch to light mode">
+      <svg class="theme-icon-sun" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" aria-hidden="true">
+        <circle cx="12" cy="12" r="4"></circle>
+        <path d="M12 2v2M12 20v2M4.93 4.93l1.41 1.41M17.66 17.66l1.41 1.41M2 12h2M20 12h2M4.93 19.07l1.41-1.41M17.66 6.34l1.41-1.41"></path>
+      </svg>
+      <svg class="theme-icon-moon" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" aria-hidden="true">
+        <path d="M21 14.5A8.5 8.5 0 1 1 9.5 3 7 7 0 0 0 21 14.5z"></path>
+      </svg>
+    </button>
+  </header>
+
+  <div class="main">
+    <div class="rail">
+      <button id="tool-upload" title="Upload an image to edit"><svg width="19" height="19" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.7" stroke-linecap="round" stroke-linejoin="round"><path d="M21 15v4a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2v-4"/><polyline points="17 8 12 3 7 8"/><line x1="12" y1="3" x2="12" y2="15"/></svg></button>
+      <button id="tool-clear-doc" title="Clear document"><svg width="19" height="19" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.7" stroke-linecap="round" stroke-linejoin="round"><polyline points="3 6 5 6 21 6"/><path d="M19 6v14a2 2 0 0 1-2 2H7a2 2 0 0 1-2-2V6m3 0V4a2 2 0 0 1 2-2h4a2 2 0 0 1 2 2v2"/></svg></button>
+    </div>
+
+    <div class="viewport">
+      <div class="workflow-view" id="workflow-view" hidden>
+        <iframe src="/workflow/?ui=canvas" title="Workflow canvas"></iframe>
+      </div>
+      <div class="doc-grid">
+        <figure class="doc" id="doc-before">
+          <figcaption id="before-caption">Source</figcaption>
+          <div class="stage" id="before-stage">
+            <img id="source-img" alt="" hidden />
+            <div class="drop-hint" id="source-hint"><b>No source image</b>Use the upload tool or drop a file here</div>
+          </div>
+        </figure>
+        <figure class="doc" id="doc-after">
+          <figcaption id="after-caption">Result</figcaption>
+          <div class="stage" id="after-stage">
+            <img id="result-img" alt="" hidden />
+            <div class="split-wrap" id="split-wrap" hidden>
+              <img id="split-source" alt="" style="clip-path: inset(0 50% 0 0)" />
+              <div class="split-divider" id="split-divider" style="left:50%"></div>
+            </div>
+            <div class="drop-hint" id="result-hint"><b>Nothing yet</b>Run to see the result</div>
+          </div>
+          <div class="doc-bar">
+            <span id="result-meta">—</span>
+            <div class="spacer"></div>
+            <button id="split-toggle" hidden>SPLIT</button>
+          </div>
+        </figure>
+      </div>
+      <div class="history-strip" id="history-strip"><span class="hlabel">History</span></div>
+    </div>
+
+    <aside class="panels">
+      <section class="panel">
+        <div class="panel-title">Examples</div>
+        <div id="examples"></div>
+      </section>
+      <section class="panel">
+        <div class="panel-title">Prompt</div>
+        <textarea id="prompt" rows="6" placeholder="Describe the image…"></textarea>
+      </section>
+      <section class="panel">
+        <div class="panel-title">Composition</div>
+        <div id="aspect-row">
+          <label for="aspect">Aspect ratio</label>
+          <select id="aspect">
+            <option>1:1</option>
+            <option selected>4:3</option>
+            <option>3:4</option>
+            <option>3:2</option>
+            <option>2:3</option>
+            <option>16:9</option>
+            <option>9:16</option>
+          </select>
+        </div>
+        <label for="resolution">Resolution</label>
+        <select id="resolution"></select>
+        <div class="row" style="margin-top:10px">
+          <label class="switch"><input type="checkbox" id="upsample" checked /><span class="sw"></span><span class="sw-label">Prompt upsampling</span></label>
+        </div>
+      </section>
+      <section class="panel" id="layers-panel" hidden>
+        <div class="panel-title">Layers</div>
+        <div class="layer-row" id="layer-source">
+          <div class="thumb" id="layer-source-thumb">BG</div>
+          <div class="lname">Background<span class="lsub">source image — this one is edited</span></div>
+        </div>
+        <div class="ref-grid">
+          <div class="ref-slot" data-ref="2">+ Ref 2</div>
+          <div class="ref-slot" data-ref="3">+ Ref 3</div>
+          <div class="ref-slot" data-ref="4">+ Ref 4</div>
+          <div class="ref-slot" data-ref="5">+ Ref 5</div>
+          <div class="ref-slot" data-ref="6">+ Ref 6</div>
+          <div class="ref-slot" data-ref="7">+ Ref 7</div>
+          <div class="ref-slot" data-ref="8">+ Ref 8</div>
+        </div>
+        <div class="hint">Reference extra images in the prompt as &lt;Picture 2&gt; … &lt;Picture 8&gt; (e.g. "put the fox from &lt;Picture 2&gt; on the bench").</div>
+      </section>
+      <section class="panel">
+        <div class="panel-title">Advanced</div>
+        <div class="row">
+          <div><label for="seed">Seed</label><input type="number" id="seed" value="42" min="0" /></div>
+          <label class="switch"><input type="checkbox" id="randomize" checked /><span class="sw"></span><span class="sw-label">Randomize</span></label>
+        </div>
+      </section>
+      <section class="panel">
+        <div class="panel-title">Upsampled prompt</div>
+        <textarea id="upsampled" class="code" rows="5" readonly placeholder="Filled in after a run with prompt upsampling on."></textarea>
+      </section>
+      <button id="run-btn">Generate</button>
+    </aside>
+  </div>
+
+  <footer class="statusbar" id="statusbar">
+    <div class="pulse"></div>
+    <span id="status">Idle</span>
+    <div class="spacer"></div>
+    <span class="foot-links">
+      <a href="https://github.com/PrunaAI/pruna" target="_blank" rel="noopener">GitHub</a>
+      <a href="https://discord.gg/JFQmtFKCjd" target="_blank" rel="noopener">Discord</a>
+      <a href="https://x.com/PrunaAI" target="_blank" rel="noopener">X</a>
+    </span>
+    <span id="seed-out"></span>
+  </footer>
+
+  <input type="file" id="file-input" accept="image/*" />
+
+<script type="module">
+  import { client } from "https://cdn.jsdelivr.net/npm/@gradio/client@2.7.1/dist/index.min.js";
+
+  const $ = (id) => document.getElementById(id);
+  const state = { mode: "generate", sourcePath: null, refPaths: {}, app: null };
+  for (let i = 2; i <= 8; i++) state.refPaths[i] = null;
+  const docState = { source: null, result: null, history: [] };
+
+  const EXAMPLES = [
+    { label: "Negotiation in orbit", aspect: "16:9",
+      prompt: "Medium wide shot from a low angle across a long table, on a 35mm lens, with the window and far wall behind the figures. Five people sit in tense negotiation, leaning toward each other. Two humans in worn grey wool suits face three alien delegates. The first alien is tall and thin-limbed, with grey skin, long folded fingers and a robe of coarse woven fiber. The second is squat and heavy, its back plated with pale ridged shell, gripping a stylus in three thick fingers. The third has a dark mantle of fine filaments and wears a harness of dented brass clips. A cracked ceramic jug, scuffed data slates and curled paper printouts sit on the table. Earth fills the large curved window, a band of blue cloud over dark ocean. Ribbed steel girders frame the foreground. Cold sunlight rakes across the table while one amber lamp hangs overhead, casting soft shadows under the chins. Fine dust drifts in the light. Palette of slate, bone, rust and deep teal, natural film color, fine grain, shadows deep with detail kept." },
+    { label: "Astronaut in a flooded jungle", aspect: "16:9",
+      prompt: "A frame from a feature film: a wide establishing shot from a low angle, with a small astronaut standing waist-deep in a flooded jungle clearing, captured on a large-format digital cinema camera with a 35mm lens.\n\nThe astronaut wears a worn white spacesuit with grey mud on the knees, scuffed gloves, and a cracked visor that reflects the canopy. One hand grips a rusted machete while the other rests on a broken comms pack. The helmet is tilted toward a sound in the trees, caught mid-step as if listening for danger.\n\nBehind the figure, tangled roots and fallen trunks rise out of black water. A pale strangler fig wraps around a torn fragment of a landing hull. Farther back, ferns and mist-shrouded trunks fade into layers of blue-grey haze.\n\nThe light is a low, overcast sky above the canopy, a cold top light with a faint blue-white bounce off the water. A single battery lantern clipped to the pack casts a small warm pool across the suit.\n\nDrifting mist hangs between the trunks, and fine rain dimples the water surface. The palette is restrained: slate blue, steel grey, muted sage green and pale ivory, with film-like color and fine grain." },
+    { label: "Knight in a frozen forest", aspect: "3:2",
+      prompt: "A photograph by an experienced landscape photographer, shot at eye level from a low position with a 35mm lens at f/2.8, the knight placed in the middle distance on a narrow forest path.\n\nThe knight is a weathered man in his forties, his beard rimmed with frost and his brow tight as he scans the trees. His dented steel plate is dulled with grey scratches, a wool cloak hangs heavy and damp at his shoulders, and his right hand rests loosely on a sword hilt wrapped in worn leather. His breath drifts visibly in the cold air as he walks mid-stride, one boot sinking into the snow.\n\nBehind him, birch and pine trunks are coated in white frost, and fallen branches lie half-buried under drifts. Moonlight comes from high upper left, pale and cool, lifting the frost on the bark and the rim of his pauldron, while the forest floor falls into deep blue shadow where texture remains visible. The palette stays restrained: slate blue, grey-white snow, muted brown leather and tarnished steel, with neutral white balance and fine film grain." },
+    { label: "Anime sword duel", aspect: "16:9",
+      prompt: "A frame from a Japanese anime television series, hand-drawn cel animation over a painted background, widescreen composition. On a rocky mountaintop in a thunderstorm, a swordsman in a torn dark coat swings a massive broadsword that glows pale blue, its blade mid-arc and his teeth clenched. Opposite him, a slender fighter in a white scarf blurs into motion, leaving streaks of afterimage and a ring of cracked stone. Lightning forks across the sky behind them, rain slants through the frame, and bright white highlights gleam on wet rock." },
+    { label: "1970s sci-fi comic", aspect: "3:2",
+      prompt: "A gouache and ink illustration in the style of a 1970s European science-fiction comic, filling the whole picture edge to edge. A lone space traveler in a bulky white suit stands on a rocky ledge, helmet visor catching the faint light, gazing up at towering mushroom trees with broad caps and thick pale stalks. A dim violet sky stretches above, with a few pale spores drifting through the air. Flat violet and ochre color with visible brush texture and inked contour lines, shown in a wide landscape view from a low angle." },
+    { label: "Linen lookbook", aspect: "2:3",
+      prompt: "A lookbook photograph by an experienced fashion photographer: a woman in her late twenties stands full length in a pale sandy-linen dress, a loose sleeveless A-line with a square neckline, a gathered waist and a hem that falls just below the calf. The natural linen creases softly at the hips and lifts slightly as she shifts her weight onto one foot in a light breeze. Her chestnut hair is loosely tied back, she wears simple leather sandals and small gold hoops, and her expression is calm as she looks just past the camera. The location is a whitewashed plaster wall beside a weathered wooden door, with the shadow of a palm frond falling across the plaster. Late-afternoon sunlight from the left gives warm directional light with visible detail in the shadows. Shot at 85mm from a few steps back at standing eye height, shallow depth of field, true white balance, fine grain, restrained palette of cream, sand and sage. No text, no other objects." },
+    { label: "Risograph cyclist", aspect: "3:4",
+      prompt: "A two-color risograph print on cream paper, the artwork running edge to edge with no border or margin. A cyclist in a hooded jacket and helmet leans into a turn on a bicycle down a wide city avenue, seen from a low side angle. Behind them rise tall apartment blocks with rows of windows, a tram line curving into the distance, and a few parked cars. Only the cyclist, the buildings and the road are shown. The image uses fluorescent pink and deep blue inks, which overprint to a dark purple where they overlap. Visible halftone dots, grainy ink coverage, and slight misregistration between the two passes run across the whole sheet." },
+    { label: "Byzantine mosaic", aspect: "3:4",
+      prompt: "A Byzantine gold-ground mosaic panel, small glass and stone tesserae set in lime mortar, with the gold tiles catching light unevenly. A lighthouse keeper in a dark blue tunic and short cloak stands at the base of a tall white tower, one hand raised to shield his eyes as he looks out to sea. Behind him a lantern glows with a red-gold flame, and the water below is built in bands of teal and deep blue tesserae, with a small ship on the horizon. The gold ground fills the background, the figures are outlined in dark grey-brown tesserae, and the grout lines are slightly uneven. The mosaic fills the whole image edge to edge, with no frame or border." },
+    { label: "Kirchner street at night", aspect: "3:4",
+      prompt: "A German Expressionist oil painting on canvas in the manner of Ernst Ludwig Kirchner's Berlin street scenes. A narrow city street at night tilts sharply toward the viewer, its buildings leaning into angular, flattened planes. Gaunt figures in dark coats hurry past a gas lamp whose light is a thick yellow smear dragged across the wet paving. Heavy impasto brushstrokes and visible palette-knife edges, with raw cadmium yellow, cobalt blue and vermilion patches in the lit windows. The paint covers the canvas edge to edge, with no frame or border." },
+  ];
+
+  async function getApp() {
+    if (!state.app) state.app = await client(location.origin);
+    return state.app;
+  }
+
+  async function uploadFile(file) {
+    const form = new FormData();
+    form.append("files", file);
+    const resp = await fetch("/gradio_api/upload", { method: "POST", body: form });
+    if (!resp.ok) throw new Error(`Upload failed (${resp.status})`);
+    const paths = await resp.json();
+    return Array.isArray(paths) ? paths[0] : paths;
+  }
+
+  function setStatus(text, busy = false) {
+    $("status").textContent = text;
+    $("statusbar").classList.toggle("busy", busy);
+    $("run-btn").disabled = busy;
+  }
+
+  // ---------- theme ----------
+  const themeBtn = $("theme-toggle");
+  function applyTheme(mode) {
+    document.body.classList.toggle("light", mode === "light");
+    themeBtn.dataset.mode = mode;
+    const label = mode === "dark" ? "Switch to light mode" : "Switch to dark mode";
+    themeBtn.setAttribute("aria-label", label);
+    themeBtn.setAttribute("title", label);
+    try { localStorage.setItem("pruna-theme", mode); } catch (e) {}
+  }
+  themeBtn.onclick = () => applyTheme(document.body.classList.contains("light") ? "dark" : "light");
+  (function initTheme() {
+    let mode = "dark";
+    try { if (localStorage.getItem("pruna-theme") === "light") mode = "light"; } catch (e) {}
+    applyTheme(mode);
+  })();
+
+  function setResolutions(list) {
+    const sel = $("resolution");
+    const prev = sel.value;
+    sel.innerHTML = "";
+    for (const r of list) {
+      const opt = document.createElement("option");
+      opt.value = r;
+      opt.textContent = r;
+      sel.appendChild(opt);
+    }
+    sel.value = list.includes(prev) ? prev : "8 MP";
+  }
+
+  function setMode(mode) {
+    state.mode = mode;
+    for (const m of ["generate", "edit", "workflow"]) $(`mode-${m}`).classList.toggle("active", mode === m);
+    const workflow = mode === "workflow";
+    $("workflow-view").hidden = !workflow;
+    document.querySelector(".doc-grid").style.display = workflow ? "none" : "";
+    document.querySelector(".rail").style.display = workflow ? "none" : "";
+    document.querySelector(".panels").style.display = workflow ? "none" : "";
+    $("history-strip").style.display = workflow ? "none" : "";
+    if (!workflow) {
+      const edit = mode === "edit";
+      $("doc-before").style.display = edit ? "" : "none";
+      $("layers-panel").hidden = !edit;
+      $("aspect-row").style.display = edit ? "none" : "";
+      setResolutions(edit ? ["4 MP", "8 MP"] : ["4 MP", "8 MP", "16 MP"]);
+      $("run-btn").textContent = edit ? "Generate Edit" : "Generate";
+      $("prompt").placeholder = edit ? "e.g. put the fox from <Picture 2> on the bench" : "Describe the image…";
+    }
+  }
+  $("mode-generate").onclick = () => setMode("generate");
+  $("mode-edit").onclick = () => setMode("edit");
+  $("mode-workflow").onclick = () => setMode("workflow");
+
+  // ---------- uploads ----------
+  let uploadTarget = "source";
+  function pickFile(target) {
+    uploadTarget = target;
+    $("file-input").click();
+  }
+  $("tool-upload").onclick = () => { setMode("edit"); pickFile("source"); };
+  document.querySelectorAll(".ref-slot").forEach((slot) => {
+    slot.onclick = () => pickFile(slot.dataset.ref);
+  });
+  $("file-input").onchange = async (e) => {
+    const file = e.target.files[0];
+    e.target.value = "";
+    if (!file) return;
+    setStatus("Uploading…", true);
+    try {
+      const path = await uploadFile(file);
+      if (uploadTarget === "source") {
+        state.sourcePath = path;
+        showSource(file);
+      } else {
+        state.refPaths[uploadTarget] = path;
+        const slot = document.querySelector(`.ref-slot[data-ref="${uploadTarget}"]`);
+        slot.innerHTML = "";
+        const img = document.createElement("img");
+        img.src = URL.createObjectURL(file);
+        slot.appendChild(img);
+      }
+      setStatus("Idle");
+    } catch (err) {
+      setStatus(String(err));
+    }
+  };
+  $("before-stage").addEventListener("dragover", (e) => e.preventDefault());
+  $("before-stage").addEventListener("drop", async (e) => {
+    e.preventDefault();
+    const file = e.dataTransfer.files[0];
+    if (!file) return;
+    setStatus("Uploading…", true);
+    try {
+      state.sourcePath = await uploadFile(file);
+      showSource(file);
+      setStatus("Idle");
+    } catch (err) {
+      setStatus(String(err));
+    }
+  });
+
+  function showSource(file) {
+    const img = $("source-img");
+    img.src = URL.createObjectURL(file);
+    img.hidden = false;
+    $("source-hint").style.display = "none";
+    docState.source = img.src;
+    const st = $("layer-source-thumb");
+    st.innerHTML = "";
+    const stImg = document.createElement("img");
+    stImg.src = img.src;
+    st.appendChild(stImg);
+    setMode("edit");
+  }
+
+  $("tool-clear-doc").onclick = () => {
+    state.sourcePath = null;
+    for (let i = 2; i <= 8; i++) state.refPaths[i] = null;
+    docState.source = null;
+    docState.result = null;
+    $("source-img").hidden = true;
+    $("source-hint").style.display = "";
+    $("result-img").hidden = true;
+    $("result-hint").style.display = "";
+    $("result-meta").textContent = "—";
+    $("seed-out").textContent = "";
+    $("upsampled").value = "";
+    $("split-toggle").hidden = true;
+    $("split-wrap").hidden = true;
+    $("layer-source-thumb").textContent = "BG";
+    document.querySelectorAll(".ref-slot").forEach((slot) => {
+      slot.textContent = `+ Ref ${slot.dataset.ref}`;
+    });
+  };
+
+  // ---------- history ----------
+  function renderHistory() {
+    const strip = $("history-strip");
+    strip.innerHTML = '<span class="hlabel">History</span>';
+    docState.history.forEach((h) => {
+      const chip = document.createElement("div");
+      chip.className = "history-chip";
+      chip.title = h.prompt || "";
+      const img = document.createElement("img");
+      img.src = h.uri;
+      const tag = document.createElement("div");
+      tag.className = "hseed";
+      tag.textContent = h.seed ?? "";
+      chip.appendChild(img);
+      chip.appendChild(tag);
+      chip.onclick = () => {
+        setMode(h.mode || "generate");
+        $("prompt").value = h.prompt;
+        $("seed").value = h.seed ?? 42;
+        $("randomize").checked = false;
+        docState.result = h.uri;
+        $("result-img").src = h.uri;
+        $("result-img").hidden = false;
+        $("result-hint").style.display = "none";
+        $("split-toggle").hidden = !(h.mode === "edit" && !$("source-img").hidden);
+      };
+      strip.appendChild(chip);
+    });
+  }
+  function pushHistory(uri, seed, prompt, mode) {
+    docState.history.unshift({ uri, seed, prompt, mode });
+    if (docState.history.length > 8) docState.history.pop();
+    renderHistory();
+  }
+
+  // ---------- before/after split view ----------
+  let splitOn = false;
+  $("split-toggle").onclick = () => {
+    if ($("result-img").hidden) return;
+    splitOn = !splitOn;
+    $("split-toggle").classList.toggle("active", splitOn);
+    $("split-wrap").hidden = !splitOn;
+    if (splitOn && !$("source-img").hidden) $("split-source").src = $("source-img").src;
+  };
+  (function () {
+    const divider = $("split-divider");
+    let dragging = false;
+    divider.addEventListener("pointerdown", (e) => { dragging = true; e.preventDefault(); });
+    window.addEventListener("pointermove", (e) => {
+      if (!dragging) return;
+      const stage = $("after-stage").getBoundingClientRect();
+      const pct = Math.min(97, Math.max(3, ((e.clientX - stage.left) / stage.width) * 100));
+      divider.style.left = pct + "%";
+      $("split-source").style.clipPath = `inset(0 ${100 - pct}% 0 0)`;
+    });
+    window.addEventListener("pointerup", () => { dragging = false; });
+  })();
+
+  // ---------- zoom ----------
+  document.querySelectorAll(".doc figcaption").forEach((caption) => {
+    const btn = document.createElement("button");
+    btn.className = "zoom-btn";
+    btn.textContent = "FIT";
+    btn.onclick = () => {
+      const img = caption.parentElement.querySelector("img");
+      const stage = caption.parentElement.querySelector(".stage");
+      const zoomed = img.classList.toggle("zoomed");
+      stage.classList.toggle("scrollable", zoomed);
+      btn.textContent = zoomed ? "1:1" : "FIT";
+    };
+    caption.appendChild(btn);
+  });
+
+  // ---------- collapsible panels ----------
+  document.querySelectorAll(".panel .panel-title").forEach((title) => {
+    const body = document.createElement("div");
+    body.className = "panel-body";
+    [...title.parentElement.children].forEach((child) => {
+      if (child !== title) body.appendChild(child);
+    });
+    title.parentElement.appendChild(body);
+    title.onclick = () => title.parentElement.classList.toggle("collapsed");
+  });
+
+  // ---------- examples ----------
+  const exRow = $("examples");
+  EXAMPLES.forEach((ex) => {
+    const chip = document.createElement("button");
+    chip.className = "ex-chip";
+    const label = document.createElement("span");
+    label.className = "ex-label";
+    label.textContent = ex.label;
+    const aspect = document.createElement("span");
+    aspect.className = "ex-aspect";
+    aspect.textContent = ex.aspect;
+    chip.appendChild(label);
+    chip.appendChild(aspect);
+    chip.onclick = () => {
+      setMode("generate");
+      $("prompt").value = ex.prompt;
+      $("aspect").value = ex.aspect;
+      setStatus("Example loaded — hit Generate");
+    };
+    exRow.appendChild(chip);
+  });
+
+  // ---------- result handling ----------
+  function unwrap(result) {
+    const data = Array.isArray(result?.data) ? result.data[0] : result?.data;
+    return data ?? {};
+  }
+  function showResult(result) {
+    const payload = unwrap(result);
+    const url = typeof payload.image_data === "string" ? payload.image_data : null;
+    if (!url) { setStatus("Run finished but no image came back (see console)."); return false; }
+    const img = $("result-img");
+    img.src = url;
+    img.hidden = false;
+    $("result-hint").style.display = "none";
+    docState.result = url;
+    if (payload.seed != null) $("seed-out").textContent = "seed " + payload.seed;
+    if (payload.prompt) $("upsampled").value = payload.prompt;
+    $("result-meta").textContent = payload.info || "Done";
+    $("split-toggle").hidden = !(state.mode === "edit" && !$("source-img").hidden);
+    pushHistory(url, payload.seed, $("prompt").value.trim(), state.mode);
+    return true;
+  }
+
+  $("run-btn").onclick = async () => {
+    const app = await getApp();
+    const prompt = $("prompt").value.trim();
+    const seed = parseInt($("seed").value, 10) || 0;
+    const randomize = $("randomize").checked;
+    const upsample = $("upsample").checked;
+    const resolution = $("resolution").value;
+    try {
+      let result;
+      if (state.mode === "edit") {
+        if (!state.sourcePath) { setStatus("Upload a source image first."); return; }
+        if (!prompt) { setStatus("Describe the edit first."); return; }
+        setStatus("Editing…", true);
+        result = await app.predict("/edit", {
+          source: state.sourcePath,
+          instruction: prompt,
+          ref2: state.refPaths[2],
+          ref3: state.refPaths[3],
+          ref4: state.refPaths[4],
+          ref5: state.refPaths[5],
+          ref6: state.refPaths[6],
+          ref7: state.refPaths[7],
+          ref8: state.refPaths[8],
+          resolution, upsample, seed, randomize,
+        });
+      } else {
+        if (!prompt) { setStatus("Write a prompt first."); return; }
+        setStatus(upsample ? "Upsampling the prompt, then rendering…" : "Rendering…", true);
+        result = await app.predict("/generate", {
+          prompt,
+          aspect: $("aspect").value,
+          resolution, upsample, seed, randomize,
+        });
+      }
+      if (showResult(result)) setStatus("Done");
+    } catch (err) {
+      setStatus(err?.message || String(err));
+    }
+  };
+
+  setMode("generate");
+</script>
+</body>
+</html>

workflow.json

@@ -0,0 +1,210 @@
+{
+  "schema_version": "2",
+  "name": "P-H3-Image Workflow",
+  "references": [
+    {
+      "id": "ref_source",
+      "label": "Source Image",
+      "role": "reference",
+      "asset_type": "image",
+      "inputs": [{"id": "in", "label": "Image", "type": "image"}],
+      "outputs": [{"id": "out", "label": "Image", "type": "image"}],
+      "x": 40,
+      "y": 40,
+      "width": 220,
+      "height": 130,
+      "data": {}
+    },
+    {
+      "id": "ref_edit",
+      "label": "Edit Instruction",
+      "role": "reference",
+      "asset_type": "text",
+      "inputs": [{"id": "in", "label": "Text", "type": "text"}],
+      "outputs": [{"id": "out", "label": "Text", "type": "text"}],
+      "x": 300,
+      "y": 40,
+      "width": 280,
+      "height": 130,
+      "data": {"out": "make it night, with warm light glowing in the windows"}
+    },
+    {
+      "id": "ref_edit_2",
+      "label": "Second Edit Instruction",
+      "role": "reference",
+      "asset_type": "text",
+      "inputs": [{"id": "in", "label": "Text", "type": "text"}],
+      "outputs": [{"id": "out", "label": "Text", "type": "text"}],
+      "x": 620,
+      "y": 250,
+      "width": 280,
+      "height": 130,
+      "data": {"out": "add a small boat with a lantern on the water"}
+    },
+    {
+      "id": "ref_seed",
+      "label": "Seed",
+      "role": "reference",
+      "asset_type": "text",
+      "inputs": [{"id": "in", "label": "Text", "type": "text"}],
+      "outputs": [{"id": "out", "label": "Text", "type": "text"}],
+      "x": 300,
+      "y": 190,
+      "width": 130,
+      "height": 90,
+      "data": {"out": "42"}
+    },
+    {
+      "id": "ref_prompt",
+      "label": "Prompt",
+      "role": "reference",
+      "asset_type": "text",
+      "inputs": [{"id": "in", "label": "Text", "type": "text"}],
+      "outputs": [{"id": "out", "label": "Text", "type": "text"}],
+      "x": 40,
+      "y": 460,
+      "width": 300,
+      "height": 160,
+      "data": {"out": "A lighthouse on a cliff at dusk, film still, cold light raking across the rocks, fine grain, restrained palette of slate and amber."}
+    },
+    {
+      "id": "ref_aspect",
+      "label": "Aspect Ratio",
+      "role": "reference",
+      "asset_type": "text",
+      "inputs": [{"id": "in", "label": "Text", "type": "text"}],
+      "outputs": [{"id": "out", "label": "Text", "type": "text"}],
+      "x": 40,
+      "y": 640,
+      "width": 140,
+      "height": 90,
+      "data": {"out": "4:3"}
+    },
+    {
+      "id": "ref_resolution",
+      "label": "Resolution",
+      "role": "reference",
+      "asset_type": "text",
+      "inputs": [{"id": "in", "label": "Text", "type": "text"}],
+      "outputs": [{"id": "out", "label": "Text", "type": "text"}],
+      "x": 200,
+      "y": 640,
+      "width": 140,
+      "height": 90,
+      "data": {"out": "8 MP"}
+    }
+  ],
+  "operators": [
+    {
+      "id": "op_edit",
+      "label": "P-H3 Edit",
+      "role": "operator",
+      "kind": "fn",
+      "fn": "ph3_edit",
+      "inputs": [
+        {"id": "image", "label": "Image", "type": "image", "required": true},
+        {"id": "prompt", "label": "Edit Instruction", "type": "text", "required": true},
+        {"id": "seed", "label": "Seed", "type": "text"}
+      ],
+      "outputs": [{"id": "out_0", "label": "Image", "type": "image", "output_index": 0}],
+      "x": 620,
+      "y": 60,
+      "width": 230,
+      "height": 150,
+      "data": {}
+    },
+    {
+      "id": "op_edit_2",
+      "label": "P-H3 Edit (chained)",
+      "role": "operator",
+      "kind": "fn",
+      "fn": "ph3_edit",
+      "inputs": [
+        {"id": "image", "label": "Image", "type": "image", "required": true},
+        {"id": "prompt", "label": "Edit Instruction", "type": "text", "required": true},
+        {"id": "seed", "label": "Seed", "type": "text"}
+      ],
+      "outputs": [{"id": "out_0", "label": "Image", "type": "image", "output_index": 0}],
+      "x": 960,
+      "y": 260,
+      "width": 230,
+      "height": 150,
+      "data": {}
+    },
+    {
+      "id": "op_generate",
+      "label": "P-H3 Generate",
+      "role": "operator",
+      "kind": "fn",
+      "fn": "ph3_generate",
+      "inputs": [
+        {"id": "prompt", "label": "Prompt", "type": "text", "required": true},
+        {"id": "aspect", "label": "Aspect Ratio", "type": "text"},
+        {"id": "resolution", "label": "Resolution", "type": "text"},
+        {"id": "seed", "label": "Seed", "type": "text"}
+      ],
+      "outputs": [{"id": "out_0", "label": "Image", "type": "image", "output_index": 0}],
+      "x": 420,
+      "y": 480,
+      "width": 230,
+      "height": 170,
+      "data": {}
+    }
+  ],
+  "subjects": [
+    {
+      "id": "sub_edited",
+      "label": "Edited Image",
+      "role": "subject",
+      "asset_type": "image",
+      "inputs": [{"id": "in", "label": "Image", "type": "image"}],
+      "outputs": [{"id": "out", "label": "Image", "type": "image"}],
+      "x": 900,
+      "y": 70,
+      "width": 230,
+      "height": 110,
+      "data": {}
+    },
+    {
+      "id": "sub_final",
+      "label": "Final Image",
+      "role": "subject",
+      "asset_type": "image",
+      "inputs": [{"id": "in", "label": "Image", "type": "image"}],
+      "outputs": [{"id": "out", "label": "Image", "type": "image"}],
+      "x": 1240,
+      "y": 270,
+      "width": 230,
+      "height": 110,
+      "data": {}
+    },
+    {
+      "id": "sub_generated",
+      "label": "Generated Image",
+      "role": "subject",
+      "asset_type": "image",
+      "inputs": [{"id": "in", "label": "Image", "type": "image"}],
+      "outputs": [{"id": "out", "label": "Image", "type": "image"}],
+      "x": 710,
+      "y": 490,
+      "width": 230,
+      "height": 110,
+      "data": {}
+    }
+  ],
+  "edges": [
+    {"id": "e_source", "from_node_id": "ref_source", "from_port_id": "out", "to_node_id": "op_edit", "to_port_id": "image", "type": "image"},
+    {"id": "e_edit", "from_node_id": "ref_edit", "from_port_id": "out", "to_node_id": "op_edit", "to_port_id": "prompt", "type": "text"},
+    {"id": "e_seed", "from_node_id": "ref_seed", "from_port_id": "out", "to_node_id": "op_edit", "to_port_id": "seed", "type": "text"},
+    {"id": "e_out", "from_node_id": "op_edit", "from_port_id": "out_0", "to_node_id": "sub_edited", "to_port_id": "in", "type": "image"},
+    {"id": "e_chain", "from_node_id": "op_edit", "from_port_id": "out_0", "to_node_id": "op_edit_2", "to_port_id": "image", "type": "image"},
+    {"id": "e_edit_2", "from_node_id": "ref_edit_2", "from_port_id": "out", "to_node_id": "op_edit_2", "to_port_id": "prompt", "type": "text"},
+    {"id": "e_seed_2", "from_node_id": "ref_seed", "from_port_id": "out", "to_node_id": "op_edit_2", "to_port_id": "seed", "type": "text"},
+    {"id": "e_out_2", "from_node_id": "op_edit_2", "from_port_id": "out_0", "to_node_id": "sub_final", "to_port_id": "in", "type": "image"},
+    {"id": "e_prompt", "from_node_id": "ref_prompt", "from_port_id": "out", "to_node_id": "op_generate", "to_port_id": "prompt", "type": "text"},
+    {"id": "e_aspect", "from_node_id": "ref_aspect", "from_port_id": "out", "to_node_id": "op_generate", "to_port_id": "aspect", "type": "text"},
+    {"id": "e_resolution", "from_node_id": "ref_resolution", "from_port_id": "out", "to_node_id": "op_generate", "to_port_id": "resolution", "type": "text"},
+    {"id": "e_seed_g", "from_node_id": "ref_seed", "from_port_id": "out", "to_node_id": "op_generate", "to_port_id": "seed", "type": "text"},
+    {"id": "e_gen", "from_node_id": "op_generate", "from_port_id": "out_0", "to_node_id": "sub_generated", "to_port_id": "in", "type": "image"}
+  ]
+}