fix

.agents/skills/add-model-01-prep/scripts/clone_reference_repo.py

@@ -10,7 +10,9 @@ from pathlib import Path
 
 
 def parse_args() -> argparse.Namespace:
-    parser = argparse.ArgumentParser(description="Clone a reference repo for FastVideo parity tests.")
+    parser = argparse.ArgumentParser(
+        description="Clone a reference repo for FastVideo parity tests."
+    )
     parser.add_argument("repo_url", help="Official reference repository URL")
     parser.add_argument("target_dir", help="Directory to clone into")
     parser.add_argument("--branch", help="Branch or tag to clone")
@@ -60,7 +62,9 @@ def gitignore_entry_for(target: Path) -> str:
     try:
         relative = resolved.relative_to(root)
     except ValueError as exc:
-        raise ValueError("--update-gitignore requires target_dir to be under the current directory") from exc
+        raise ValueError(
+            "--update-gitignore requires target_dir to be under the current directory"
+        ) from exc
 
     text = relative.as_posix().rstrip("/")
     return "/" + text + "/"

.agents/skills/add-model-01-prep/scripts/download_hf_weights.py

@@ -8,12 +8,14 @@ import os
 import sys
 from pathlib import Path
 
+
 HF_TOKEN_ENV_KEYS = ("HF_TOKEN", "HUGGINGFACE_HUB_TOKEN", "HF_API_KEY")
 
 
 def parse_args() -> argparse.Namespace:
     parser = argparse.ArgumentParser(
-        description="Download a HF model snapshot or selected files into a local directory.")
+        description="Download a HF model snapshot or selected files into a local directory."
+    )
     parser.add_argument("repo_id", help="HF repo id, for example Org/Model")
     parser.add_argument("local_dir", help="Destination directory")
     parser.add_argument("--repo-type", default="model", help="HF repo type (default: model)")

.agents/skills/add-model-01-prep/scripts/inspect_hf_layout.py

@@ -10,6 +10,7 @@ import sys
 from pathlib import Path
 from typing import Any
 
+
 HF_TOKEN_ENV_KEYS = ("HF_TOKEN", "HUGGINGFACE_HUB_TOKEN", "HF_API_KEY")
 RAW_WEIGHT_SUFFIXES = (".safetensors", ".pt", ".pth", ".ckpt", ".bin")
 KNOWN_COMPONENTS = {
@@ -33,7 +34,8 @@ KNOWN_COMPONENTS = {
 
 def parse_args() -> argparse.Namespace:
     parser = argparse.ArgumentParser(
-        description="Classify a HF repo or local directory as Diffusers, raw, custom, or unknown.")
+        description="Classify a HF repo or local directory as Diffusers, raw, custom, or unknown."
+    )
     parser.add_argument("source", help="HF repo id or local weights directory")
     parser.add_argument("--repo-type", default="model", help="HF repo type (default: model)")
     parser.add_argument("--revision", help="HF revision to inspect")
@@ -92,12 +94,14 @@ def load_remote_files(
 ) -> list[str]:
     from huggingface_hub import list_repo_files
 
-    return sorted(list_repo_files(
-        repo_id,
-        repo_type=repo_type,
-        revision=revision,
-        token=token,
-    ))
+    return sorted(
+        list_repo_files(
+            repo_id,
+            repo_type=repo_type,
+            revision=revision,
+            token=token,
+        )
+    )
 
 
 def load_remote_model_index(
@@ -211,24 +215,24 @@ def build_result(args: argparse.Namespace) -> dict[str, Any]:
         "components_seen": components,
         "file_count": len(files),
         "file_scan_truncated": truncated,
-        "files_sample": files[:args.sample_limit],
+        "files_sample": files[: args.sample_limit],
     }
 
 
 def print_human(result: dict[str, Any]) -> None:
     for key in (
-            "source",
-            "source_kind",
-            "repo_type",
-            "revision",
-            "token_env",
-            "source_layout",
-            "needs_conversion",
-            "model_index_class",
-            "model_index_diffusers_version",
-            "model_index_error",
-            "file_count",
-            "file_scan_truncated",
+        "source",
+        "source_kind",
+        "repo_type",
+        "revision",
+        "token_env",
+        "source_layout",
+        "needs_conversion",
+        "model_index_class",
+        "model_index_diffusers_version",
+        "model_index_error",
+        "file_count",
+        "file_scan_truncated",
     ):
         value = result.get(key)
         if value is not None:

.agents/skills/add-model-02-parity/templates/component_parity_test.py

@@ -18,6 +18,7 @@ import pytest
 import torch
 from torch.testing import assert_close
 
+
 os.environ.setdefault("MASTER_ADDR", "localhost")
 os.environ.setdefault("MASTER_PORT", "29519")
 os.environ.setdefault("DISABLE_SP", "1")
@@ -34,10 +35,15 @@ FASTVIDEO_CONFIG_CLASS = "<FastVideoConfig>"  # TODO.
 FASTVIDEO_MODEL_MODULE = "fastvideo.models.<bucket>.<module>"  # TODO.
 FASTVIDEO_MODEL_CLASS = "<FastVideoModel>"  # TODO.
 
-OFFICIAL_REF_DIR = Path(os.getenv("<FAMILY_UPPER>_OFFICIAL_REF_DIR", REPO_ROOT / "<ReferenceDir>"))
-LOCAL_WEIGHTS_DIR = Path(os.getenv("<FAMILY_UPPER>_LOCAL_WEIGHTS_DIR", REPO_ROOT / "official_weights" / FAMILY))
-CONVERTED_WEIGHTS_DIR = Path(os.getenv("<FAMILY_UPPER>_CONVERTED_WEIGHTS_DIR",
-                                       REPO_ROOT / "converted_weights" / FAMILY))
+OFFICIAL_REF_DIR = Path(
+    os.getenv("<FAMILY_UPPER>_OFFICIAL_REF_DIR", REPO_ROOT / "<ReferenceDir>")
+)
+LOCAL_WEIGHTS_DIR = Path(
+    os.getenv("<FAMILY_UPPER>_LOCAL_WEIGHTS_DIR", REPO_ROOT / "official_weights" / FAMILY)
+)
+CONVERTED_WEIGHTS_DIR = Path(
+    os.getenv("<FAMILY_UPPER>_CONVERTED_WEIGHTS_DIR", REPO_ROOT / "converted_weights" / FAMILY)
+)
 
 
 def _resolve_hf_token() -> str | None:
@@ -93,14 +99,18 @@ def _load_official_model(device: torch.device, dtype: torch.dtype) -> torch.nn.M
     model = OfficialClass()  # TODO: pass official config kwargs.
     state_dict = {}  # TODO: load official state dict from LOCAL_WEIGHTS_DIR.
     missing, unexpected = model.load_state_dict(state_dict, strict=True)
-    assert not missing and not unexpected, (f"official load mismatch missing={missing[:5]} unexpected={unexpected[:5]}")
+    assert not missing and not unexpected, (
+        f"official load mismatch missing={missing[:5]} unexpected={unexpected[:5]}"
+    )
     return model.to(device=device, dtype=dtype).eval()
 
 
 def _load_fastvideo_model(device: torch.device, dtype: torch.dtype) -> torch.nn.Module:
     """Load the FastVideo component with the same tensor content."""
     if not CONVERTED_WEIGHTS_DIR.exists() and not LOCAL_WEIGHTS_DIR.exists():
-        pytest.skip(f"No FastVideo loadable weights: {CONVERTED_WEIGHTS_DIR} or {LOCAL_WEIGHTS_DIR}")
+        pytest.skip(
+            f"No FastVideo loadable weights: {CONVERTED_WEIGHTS_DIR} or {LOCAL_WEIGHTS_DIR}"
+        )
 
     # TODO: replace with the bucket-specific FastVideo config/class/loader.
     # DiT examples:
@@ -117,7 +127,8 @@ def _load_fastvideo_model(device: torch.device, dtype: torch.dtype) -> torch.nn.
     state_dict = {}  # TODO: load converted or directly mapped state dict.
     missing, unexpected = model.load_state_dict(state_dict, strict=True)
     assert not missing and not unexpected, (
-        f"FastVideo load mismatch missing={missing[:5]} unexpected={unexpected[:5]}")
+        f"FastVideo load mismatch missing={missing[:5]} unexpected={unexpected[:5]}"
+    )
     return model.to(device=device, dtype=dtype).eval()
 
 
@@ -176,9 +187,11 @@ def test_component_parity():
 
     assert official_out.shape == fastvideo_out.shape
     diff = (official_out - fastvideo_out).abs()
-    print(f"official abs_mean={official_out.abs().mean().item():.6f} "
-          f"fastvideo abs_mean={fastvideo_out.abs().mean().item():.6f} "
-          f"diff_max={diff.max().item():.6f} diff_mean={diff.mean().item():.6f}")
+    print(
+        f"official abs_mean={official_out.abs().mean().item():.6f} "
+        f"fastvideo abs_mean={fastvideo_out.abs().mean().item():.6f} "
+        f"diff_max={diff.max().item():.6f} diff_mean={diff.mean().item():.6f}"
+    )
 
     # TODO: pick tolerance by scope:
     # - single block / same kernel: 1e-4

.agents/skills/add-model-07-conversion/templates/family_to_diffusers.py

@@ -27,6 +27,7 @@ try:
 except ImportError:  # pragma: no cover - optional local conversion dependency
     snapshot_download = None
 
+
 # TODO: fill with authoritative component prefixes for monolithic checkpoints.
 # Example: {"model.model.": "transformer", "pretransform.model.": "vae"}
 COMPONENT_PREFIXES: dict[str, str] = {}
@@ -46,7 +47,10 @@ SKIP_PATTERNS: tuple[str, ...] = ()
 
 
 def _hf_token() -> str | None:
-    return (os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACE_HUB_TOKEN") or os.environ.get("HF_API_KEY"))
+    return (
+        os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACE_HUB_TOKEN")
+        or os.environ.get("HF_API_KEY")
+    )
 
 
 def resolve_src(src: str, revision: str | None) -> Path:
@@ -91,10 +95,11 @@ def apply_mapping(key: str) -> str | None:
     return key
 
 
-def split_monolithic(state: dict[str, torch.Tensor], ) -> dict[str, OrderedDict[str, torch.Tensor]]:
+def split_monolithic(
+    state: dict[str, torch.Tensor],
+) -> dict[str, OrderedDict[str, torch.Tensor]]:
     components: dict[str, OrderedDict[str, torch.Tensor]] = {
-        name: OrderedDict()
-        for name in set(COMPONENT_PREFIXES.values())
+        name: OrderedDict() for name in set(COMPONENT_PREFIXES.values())
     }
     intentionally_skipped: list[str] = []
     unowned: list[str] = []
@@ -112,8 +117,10 @@ def split_monolithic(state: dict[str, torch.Tensor], ) -> dict[str, OrderedDict[
             unowned.append(key)
     if unowned:
         sample = ", ".join(unowned[:10])
-        raise ValueError(f"Unowned monolithic keys: {len(unowned)}. "
-                         f"Add COMPONENT_PREFIXES or SKIP_PATTERNS entries. Sample: {sample}")
+        raise ValueError(
+            f"Unowned monolithic keys: {len(unowned)}. "
+            f"Add COMPONENT_PREFIXES or SKIP_PATTERNS entries. Sample: {sample}"
+        )
     if intentionally_skipped:
         print(f"Intentionally skipped {len(intentionally_skipped)} keys")
     return {name: weights for name, weights in components.items() if weights}
@@ -136,12 +143,8 @@ def build_component_configs(_src_dir: Path) -> dict[str, dict[str, Any]]:
     # TODO: emit config content accepted by FastVideo loaders. Most components use
     # config.json; schedulers use scheduler_config.json.
     return {
-        "transformer": {
-            "_class_name": "<FastVideoTransformerClass>"
-        },
-        "vae": {
-            "_class_name": "<FastVideoVAEClass>"
-        },
+        "transformer": {"_class_name": "<FastVideoTransformerClass>"},
+        "vae": {"_class_name": "<FastVideoVAEClass>"},
     }
 
 
@@ -174,13 +177,19 @@ def build_model_index(
     }
     if revision:
         index["_fastvideo_converted_revision"] = revision
-    return {key: value for key, value in index.items() if key.startswith("_") or key in available_components}
+    return {
+        key: value
+        for key, value in index.items()
+        if key.startswith("_") or key in available_components
+    }
 
 
 def validate_component_configs(configs: dict[str, dict[str, Any]]) -> None:
     # TODO: instantiate each FastVideo config and call update_model_arch(...) or
     # update_model_config(...) with this JSON so unknown emitted keys fail here.
-    placeholder_configs = [name for name, config in configs.items() if "<" in json.dumps(config)]
+    placeholder_configs = [
+        name for name, config in configs.items() if "<" in json.dumps(config)
+    ]
     if placeholder_configs:
         raise ValueError(f"Replace config placeholders for: {placeholder_configs}")
 
@@ -192,7 +201,9 @@ def verify_conversion(
     del dst_dir, components
     # TODO: load each emitted stateful component through its production loader and
     # assert strict load, or document exact allowed missing/unexpected keys.
-    raise NotImplementedError("Implement production config validation and strict-load checks")
+    raise NotImplementedError(
+        "Implement production config validation and strict-load checks"
+    )
 
 
 def write_component(
@@ -205,7 +216,9 @@ def write_component(
     if component_dir.exists() and any(component_dir.iterdir()):
         shutil.rmtree(component_dir)
     component_dir.mkdir(parents=True, exist_ok=True)
-    save_file(dict(state), str(component_dir / "diffusion_pytorch_model.safetensors"))
+    save_file(
+        dict(state), str(component_dir / "diffusion_pytorch_model.safetensors")
+    )
     if config is not None:
         config_path = component_dir / config_filename(name)
         with config_path.open("w", encoding="utf-8") as f:
@@ -248,7 +261,9 @@ def convert(
 
     if layout in {"monolithic", "raw_official"}:
         # TODO: replace model.safetensors with the official monolithic file name.
-        components = split_monolithic(load_checkpoint(default_monolithic_checkpoint(src_path)))
+        components = split_monolithic(
+            load_checkpoint(default_monolithic_checkpoint(src_path))
+        )
     elif layout in {"separate_components", "mixed"}:
         if not src_path.is_dir():
             raise ValueError(f"{layout} layout requires a source directory: {src_path}")
@@ -256,7 +271,9 @@ def convert(
     else:
         raise ValueError(f"Unsupported template layout: {layout}")
 
-    copied = (copy_passthrough(src_path, dst_dir) if src_path.is_dir() else [])
+    copied = (
+        copy_passthrough(src_path, dst_dir) if src_path.is_dir() else []
+    )
     configs = build_component_configs(src_path if src_path.is_dir() else src_path.parent)
     validate_component_configs(configs)
     for name, state in components.items():
@@ -272,7 +289,9 @@ def convert(
 
 def main() -> None:
     parser = argparse.ArgumentParser(description=__doc__)
-    parser.add_argument("--src", required=True, help="HF repo id, local dir, or checkpoint path")
+    parser.add_argument(
+        "--src", required=True, help="HF repo id, local dir, or checkpoint path"
+    )
     parser.add_argument("--revision", help="HF branch, tag, or commit for repo sources")
     parser.add_argument(
         "--dst",

.agents/skills/add-model-08-trace/templates/block_trace_debug.py

@@ -24,8 +24,8 @@ from typing import Any
 
 import torch
 
-FAMILY: str = "<family>"  # e.g. "magi_human", "ltx2", "wan"
-COMPONENT: str = "<component>"  # e.g. "dit", "vae", "encoder"
+FAMILY: str = "<family>"       # e.g. "magi_human", "ltx2", "wan"
+COMPONENT: str = "<component>" # e.g. "dit", "vae", "encoder"
 DRILL_LAYER_ENV: str = "<FAMILY>_DEBUG_DRILL_LAYER"
 HYPOTHESIS_ENV: str = "<FAMILY>_DEBUG_PATCH_<HYPOTHESIS>"
 REL_THRESHOLD: float = 0.005  # 0.5% abs_mean drift flags a block as divergent
@@ -94,7 +94,6 @@ def _attach_block_hooks(
     handles: list[Any] = []
 
     def _hook(name: str):
-
         def fn(_module, _inputs, outputs):
             t = outputs[0] if isinstance(outputs, tuple) else outputs
             if not torch.is_tensor(t):
@@ -102,7 +101,6 @@ def _attach_block_hooks(
             log.append({"side": label, **_stat(name, t)})
             if tensors is not None:
                 tensors[name] = t.detach().float().cpu()
-
         return fn
 
     def _pre_hook(name: str):
@@ -116,7 +114,6 @@ def _attach_block_hooks(
             log.append({"side": label, **_stat(key, t)})
             if tensors is not None:
                 tensors[key] = t.detach().float().cpu()
-
         return fn
 
     # TODO: adapt attribute paths to your model. Remove adapter block if absent.
@@ -134,21 +131,43 @@ def _attach_block_hooks(
             # magi-human uses: attention, mlp.pre_norm, mlp.up_gate_proj,
             # mlp.down_proj (pre+post), mlp, attn_post_norm, mlp_post_norm.
             if hasattr(layer, "attention"):
-                handles.append(layer.attention.register_forward_hook(_hook(f"{tag}.attention")))
+                handles.append(
+                    layer.attention.register_forward_hook(_hook(f"{tag}.attention"))
+                )
             if hasattr(layer, "mlp"):
                 mlp = layer.mlp
                 if hasattr(mlp, "pre_norm"):
-                    handles.append(mlp.pre_norm.register_forward_hook(_hook(f"{tag}.mlp.pre_norm")))
+                    handles.append(
+                        mlp.pre_norm.register_forward_hook(_hook(f"{tag}.mlp.pre_norm"))
+                    )
                 if hasattr(mlp, "up_gate_proj"):
-                    handles.append(mlp.up_gate_proj.register_forward_hook(_hook(f"{tag}.mlp.up_gate_proj")))
+                    handles.append(
+                        mlp.up_gate_proj.register_forward_hook(
+                            _hook(f"{tag}.mlp.up_gate_proj")
+                        )
+                    )
                 if hasattr(mlp, "down_proj"):
-                    handles.append(mlp.down_proj.register_forward_pre_hook(_pre_hook(f"{tag}.mlp.down_proj")))
-                    handles.append(mlp.down_proj.register_forward_hook(_hook(f"{tag}.mlp.down_proj")))
+                    handles.append(
+                        mlp.down_proj.register_forward_pre_hook(
+                            _pre_hook(f"{tag}.mlp.down_proj")
+                        )
+                    )
+                    handles.append(
+                        mlp.down_proj.register_forward_hook(_hook(f"{tag}.mlp.down_proj"))
+                    )
                 handles.append(mlp.register_forward_hook(_hook(f"{tag}.mlp")))
             if hasattr(layer, "attn_post_norm"):
-                handles.append(layer.attn_post_norm.register_forward_hook(_hook(f"{tag}.attn_post_norm")))
+                handles.append(
+                    layer.attn_post_norm.register_forward_hook(
+                        _hook(f"{tag}.attn_post_norm")
+                    )
+                )
             if hasattr(layer, "mlp_post_norm"):
-                handles.append(layer.mlp_post_norm.register_forward_hook(_hook(f"{tag}.mlp_post_norm")))
+                handles.append(
+                    layer.mlp_post_norm.register_forward_hook(
+                        _hook(f"{tag}.mlp_post_norm")
+                    )
+                )
     return handles
 
 
@@ -174,9 +193,11 @@ def _write_log(entries: list[dict], path: Path) -> None:
     path.parent.mkdir(parents=True, exist_ok=True)
     with open(path, "w") as f:
         for e in entries:
-            f.write(f"{e['name']} {e['shape']} "
-                    f"{e['abs_mean']:.8f} {e['sum']:.4f} "
-                    f"{e['min']:.6f} {e['max']:.6f}\n")
+            f.write(
+                f"{e['name']} {e['shape']} "
+                f"{e['abs_mean']:.8f} {e['sum']:.4f} "
+                f"{e['min']:.6f} {e['max']:.6f}\n"
+            )
 
 
 def _sort_key(name: str, drill_layer: int) -> tuple:
@@ -184,14 +205,9 @@ def _sort_key(name: str, drill_layer: int) -> tuple:
         return (0, "")
     if name.startswith(f"L{drill_layer:02d}."):
         sub_order = {
-            "attention": 0,
-            "attn_post_norm": 1,
-            "mlp.pre_norm": 2,
-            "mlp.up_gate_proj": 3,
-            "mlp.down_proj<in>": 4,
-            "mlp.down_proj": 5,
-            "mlp": 6,
-            "mlp_post_norm": 7,
+            "attention": 0, "attn_post_norm": 1, "mlp.pre_norm": 2,
+            "mlp.up_gate_proj": 3, "mlp.down_proj<in>": 4,
+            "mlp.down_proj": 5, "mlp": 6, "mlp_post_norm": 7,
         }.get(name.split(".", 1)[1], 9)
         return (1, f"block[{drill_layer:02d}]", sub_order)
     if name.startswith("block["):
@@ -200,8 +216,10 @@ def _sort_key(name: str, drill_layer: int) -> tuple:
 
 
 def _print_table(by_name: dict[str, dict], drill_layer: int) -> int | None:
-    hdr = (f"{'name':<18} {'up_shape':<22} {'up_absmean':>12} {'fv_absmean':>12} "
-           f"{'absmean_diff':>14} {'rel%':>8} {'up_sum':>14} {'fv_sum':>14} {'sum_diff':>12}")
+    hdr = (
+        f"{'name':<18} {'up_shape':<22} {'up_absmean':>12} {'fv_absmean':>12} "
+        f"{'absmean_diff':>14} {'rel%':>8} {'up_sum':>14} {'fv_sum':>14} {'sum_diff':>12}"
+    )
     print(f"\n{hdr}\n{'-' * len(hdr)}")
     first_div: int | None = None
     for name in sorted(by_name.keys(), key=lambda n: _sort_key(n, drill_layer)):
@@ -217,9 +235,11 @@ def _print_table(by_name: dict[str, dict], drill_layer: int) -> int | None:
             flag = " <<< DIVERGE"
             if first_div is None:
                 first_div = int(name[len("block["):-1])
-        print(f"{name:<18} {str(up['shape']):<22} {up['abs_mean']:>12.6f} "
-              f"{fv['abs_mean']:>12.6f} {am_diff:>14.6f} {am_rel * 100:>7.3f}% "
-              f"{up['sum']:>14.4f} {fv['sum']:>14.4f} {sum_diff:>12.4f}{flag}")
+        print(
+            f"{name:<18} {str(up['shape']):<22} {up['abs_mean']:>12.6f} "
+            f"{fv['abs_mean']:>12.6f} {am_diff:>14.6f} {am_rel * 100:>7.3f}% "
+            f"{up['sum']:>14.4f} {fv['sum']:>14.4f} {sum_diff:>12.4f}{flag}"
+        )
     return first_div
 
 
@@ -235,8 +255,10 @@ def _print_elementwise(up_t: dict[str, torch.Tensor], fv_t: dict[str, torch.Tens
             continue
         diff = (a - b).abs()
         rel = (diff.mean().item() / max(a.abs().mean().item(), 1e-9)) * 100
-        print(f"{name:<30} {str(tuple(a.shape)):<22} "
-              f"{diff.max().item():>12.6f} {diff.mean().item():>12.6f} {rel:>9.4f}%")
+        print(
+            f"{name:<30} {str(tuple(a.shape)):<22} "
+            f"{diff.max().item():>12.6f} {diff.mean().item():>12.6f} {rel:>9.4f}%"
+        )
 
 
 def main() -> None:

.agents/skills/add-model-09-pipeline/templates/pipeline_parity_test.py

@@ -43,10 +43,12 @@ def _add_official_to_path() -> Path:
 
 def _log_tensor_stats(label: str, tensor: torch.Tensor) -> None:
     value = tensor.detach().float()
-    print(f"[{_MODEL_FAMILY} PIPELINE] {label}: shape={tuple(tensor.shape)} "
-          f"dtype={tensor.dtype} device={tensor.device} "
-          f"min={value.min().item():.6f} max={value.max().item():.6f} "
-          f"mean={value.mean().item():.6f} std={value.std().item():.6f}")
+    print(
+        f"[{_MODEL_FAMILY} PIPELINE] {label}: shape={tuple(tensor.shape)} "
+        f"dtype={tensor.dtype} device={tensor.device} "
+        f"min={value.min().item():.6f} max={value.max().item():.6f} "
+        f"mean={value.mean().item():.6f} std={value.std().item():.6f}"
+    )
 
 
 def _extract_tensor(output: Any, key: str) -> torch.Tensor:
@@ -71,8 +73,10 @@ def _run_official_pipeline(
     device: torch.device,
 ) -> Any:
     del official_path, params, device
-    pytest.skip("TODO: import the official pipeline/factory, load official weights, "
-                "run with params, and return the comparison target.")
+    pytest.skip(
+        "TODO: import the official pipeline/factory, load official weights, "
+        "run with params, and return the comparison target."
+    )
 
 
 def _run_fastvideo_pipeline(model_path: Path, params: dict[str, Any]) -> Any:
@@ -142,6 +146,8 @@ def test_todo_model_family_pipeline_official_parity() -> None:
     assert official_tensor.shape == fastvideo_tensor.shape
 
     diff = (official_tensor - fastvideo_tensor).abs()
-    print(f"diff max={diff.max().item():.6f} "
-          f"mean={diff.mean().item():.6f} median={diff.median().item():.6f}")
+    print(
+        f"diff max={diff.max().item():.6f} "
+        f"mean={diff.mean().item():.6f} median={diff.median().item():.6f}"
+    )
     assert_close(fastvideo_tensor, official_tensor, atol=1e-2, rtol=1e-2)

.buildkite/pipeline.yml

@@ -68,17 +68,6 @@ steps:
             limit: 2
       agents:
         queue: "default"
-    - label: ":vertical_traffic_light: Golden-Gate Tests"
-      if: build.env("TEST_SCOPE") == "direct" && build.env("TEST_TYPE") == "golden_gate"
-      command: "timeout 90m .buildkite/scripts/pr_test.sh"
-      retry:
-        automatic:
-          - exit_status: 128
-            limit: 3
-          - exit_status: -1
-            limit: 2
-      agents:
-        queue: "default"
     - label: ":microscope: Unit Tests"
       if: build.env("TEST_SCOPE") == "direct" && build.env("TEST_TYPE") == "unit_test"
       command: "timeout 90m .buildkite/scripts/pr_test.sh"
@@ -415,7 +404,7 @@ steps:
                 - "pyproject.toml"
                 - "docker/Dockerfile"
               config:
-                command: "timeout 25m .buildkite/scripts/pr_test.sh"
+                command: "timeout 15m .buildkite/scripts/pr_test.sh"
                 label: ":test_tube: Training Tests"
                 env:
                   - TEST_TYPE=training
@@ -426,7 +415,7 @@ steps:
                 - "pyproject.toml"
                 - "docker/Dockerfile"
               config:
-                command: "timeout 25m .buildkite/scripts/pr_test.sh"
+                command: "timeout 15m .buildkite/scripts/pr_test.sh"
                 label: ":test_tube: Distillation DMD Tests"
                 env:
                   - TEST_TYPE=distillation_dmd

.buildkite/scripts/pr_test.sh

@@ -187,10 +187,6 @@ case "$TEST_TYPE" in
         log "Running transformer tests..."
         MODAL_COMMAND="$MODAL_ENV HF_API_KEY=$HF_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_transformer_tests"
         ;;
-    "golden_gate")
-        log "Running golden-gate tests..."
-        MODAL_COMMAND="$MODAL_ENV HF_API_KEY=$HF_API_KEY python3 -m modal run $MODAL_TEST_FILE::run_golden_gate_tests"
-        ;;
     "ssim")
         log "Running SSIM tests..."
         SSIM_BOOTSTRAP_ARGS=$(ssim_bootstrap_args)

.github/workflows/ci-macos-mlx.yml

@@ -1,149 +0,0 @@
-name: macOS MLX Smoke
-
-on:
-  pull_request:
-    branches: [main]
-    paths:
-      - ".github/workflows/ci-macos-mlx.yml"
-      - "fastvideo/mlx_runtime/**"
-      - "fastvideo/tests/mlx/**"
-      - "fastvideo/tests/platforms/test_mps_vsa_error.py"
-      - "fastvideo/platforms/mps.py"
-      - "fastvideo/platforms/__init__.py"
-      - "fastvideo/__init__.py"
-      - "examples/inference/basic/mlx_*.py"
-      - "fastvideo/benchmarks/mlx_*.py"
-      - "pyproject.toml"
-  workflow_dispatch:
-
-permissions:
-  contents: read
-
-concurrency:
-  group: macos-mlx-${{ github.ref }}
-  cancel-in-progress: true
-
-jobs:
-  mlx-smoke:
-    if: github.event_name == 'workflow_dispatch' || github.event.pull_request.draft != true
-    runs-on: macos-15
-    timeout-minutes: 25
-    env:
-      FASTVIDEO_ATTENTION_BACKEND: TORCH_SDPA
-      TOKENIZERS_PARALLELISM: "false"
-      MASTER_ADDR: localhost
-      MASTER_PORT: "29513"
-    steps:
-      - uses: actions/checkout@v4
-
-      - uses: actions/setup-python@v5
-        with:
-          python-version: "3.12"
-          cache: pip
-
-      - uses: astral-sh/setup-uv@v3
-
-      - name: Install lightweight MLX smoke dependencies
-        run: |
-          uv pip install --system \
-            --index-url https://download.pytorch.org/whl/cpu \
-            torch==2.11.0 torchvision torchaudio
-          uv pip install --system \
-            pytest numpy scipy pillow imageio einops cloudpickle filelock \
-            PyYAML diffusers huggingface_hub remote-pdb safetensors loguru mlx \
-            "ftfy>=6.3.1" "opencv-python>=4.10.0.84" psutil "transformers>=5.0.0"
-
-      - name: Show Apple runtime
-        run: |
-          python - <<'PY'
-          import platform
-          import mlx.core as mx
-          import torch
-
-          print("machine:", platform.machine())
-          print("processor:", platform.processor())
-          print("mlx default device:", mx.default_device())
-          memory_size = mx.metal.device_info().get("memory_size") if mx.metal.is_available() else "metal unavailable"
-          print("mlx memory_size:", memory_size)
-          print("torch:", torch.__version__)
-          print("torch mps available:", torch.backends.mps.is_available())
-          PY
-
-      - name: Run MLX smoke tests
-        run: |
-          python -m pytest \
-            fastvideo/tests/mlx/test_dmd_sampling.py \
-            fastvideo/tests/mlx/test_memory_limits.py \
-            fastvideo/tests/mlx/test_quant_capability.py \
-            fastvideo/tests/mlx/test_mlx_dit_parity.py \
-            fastvideo/tests/mlx/test_mlx_compile_parity.py \
-            fastvideo/tests/mlx/test_mlx_checkpoint.py \
-            fastvideo/tests/mlx/test_mlx_fastwan_benchmark.py \
-            fastvideo/tests/mlx/test_taehv_decode.py \
-            fastvideo/tests/mlx/test_frame_upsample.py \
-            fastvideo/tests/mlx/test_mlx_fast_spatial.py \
-            fastvideo/tests/mlx/test_mlx_refine.py \
-            fastvideo/tests/mlx/test_mlx_prompt_to_video_decode.py \
-            fastvideo/tests/mlx/test_mlx_wan22_prompt_cache_fingerprint.py \
-            fastvideo/tests/mlx/test_wan22_sample.py \
-            fastvideo/tests/mlx/test_windowed_attention.py \
-            fastvideo/tests/mlx/test_mlx_rife_interpolation.py::test_rife_download_unavailable_has_specific_error \
-            fastvideo/tests/mlx/test_mlx_rife_interpolation.py::test_rife_backend_regression_is_not_skip_eligible \
-            fastvideo/tests/platforms/test_mps_vsa_error.py \
-            -q
-
-  # Same tests on MLX's CPU backend. Hosted macOS runners are scarce and
-  # slower to schedule; this Linux job gives fast PR signal on the identical
-  # graph (the parity tests were designed to be backend-agnostic), while the
-  # macOS job above stays the source of truth for Metal behavior.
-  mlx-smoke-linux-cpu:
-    if: github.event_name == 'workflow_dispatch' || github.event.pull_request.draft != true
-    runs-on: ubuntu-latest
-    timeout-minutes: 20
-    env:
-      FASTVIDEO_ATTENTION_BACKEND: TORCH_SDPA
-      TOKENIZERS_PARALLELISM: "false"
-      MASTER_ADDR: localhost
-      MASTER_PORT: "29513"
-    steps:
-      - uses: actions/checkout@v4
-
-      - uses: actions/setup-python@v5
-        with:
-          python-version: "3.12"
-          cache: pip
-
-      - uses: astral-sh/setup-uv@v3
-
-      - name: Install lightweight MLX smoke dependencies (CPU backend)
-        run: |
-          uv pip install --system \
-            --index-url https://download.pytorch.org/whl/cpu \
-            torch==2.11.0 torchvision torchaudio
-          uv pip install --system \
-            pytest numpy scipy pillow imageio einops cloudpickle filelock \
-            PyYAML diffusers huggingface_hub remote-pdb safetensors loguru "mlx[cpu]" \
-            "ftfy>=6.3.1" "opencv-python>=4.10.0.84" psutil "transformers>=5.0.0"
-
-      - name: Run MLX smoke tests (CPU backend)
-        run: |
-          python -m pytest \
-            fastvideo/tests/mlx/test_dmd_sampling.py \
-            fastvideo/tests/mlx/test_memory_limits.py \
-            fastvideo/tests/mlx/test_quant_capability.py \
-            fastvideo/tests/mlx/test_mlx_dit_parity.py \
-            fastvideo/tests/mlx/test_mlx_compile_parity.py \
-            fastvideo/tests/mlx/test_mlx_checkpoint.py \
-            fastvideo/tests/mlx/test_mlx_fastwan_benchmark.py \
-            fastvideo/tests/mlx/test_taehv_decode.py \
-            fastvideo/tests/mlx/test_frame_upsample.py \
-            fastvideo/tests/mlx/test_mlx_fast_spatial.py \
-            fastvideo/tests/mlx/test_mlx_refine.py \
-            fastvideo/tests/mlx/test_mlx_prompt_to_video_decode.py \
-            fastvideo/tests/mlx/test_mlx_wan22_prompt_cache_fingerprint.py \
-            fastvideo/tests/mlx/test_wan22_sample.py \
-            fastvideo/tests/mlx/test_windowed_attention.py \
-            fastvideo/tests/mlx/test_mlx_rife_interpolation.py::test_rife_download_unavailable_has_specific_error \
-            fastvideo/tests/mlx/test_mlx_rife_interpolation.py::test_rife_backend_regression_is_not_skip_eligible \
-            fastvideo/tests/platforms/test_mps_vsa_error.py \
-            -q

.github/workflows/ci-slash-commands.yml

@@ -129,7 +129,7 @@ jobs:
           set -euo pipefail
           TEST_NAME=$(echo "$COMMENT" | grep -oP '(?<=/test\s)\S+' | head -1 || true)
 
-          VALID="encoder vae transformer kernel unit dreamverse ssim golden-gate training lora-inference lora-training lora-extraction distillation self-forcing vsa vmoba performance api train-framework eval full fastcheck pre-commit"
+          VALID="encoder vae transformer kernel unit dreamverse ssim training lora-inference lora-training lora-extraction distillation self-forcing vsa vmoba performance api train-framework eval full fastcheck pre-commit"
           if [ -z "$TEST_NAME" ] || ! echo "$VALID" | grep -qw "$TEST_NAME"; then
             echo "Unknown test: '$TEST_NAME'. Valid: $VALID"
             exit 1
@@ -138,7 +138,7 @@ jobs:
           declare -A MAP=(
             [encoder]=encoder [vae]=vae [transformer]=transformer
             [kernel]=kernel_tests [unit]=unit_test [dreamverse]=dreamverse_app
-            [ssim]=ssim [golden-gate]=golden_gate [training]=training
+            [ssim]=ssim [training]=training
             [lora-inference]=inference_lora [lora-training]=training_lora
             [lora-extraction]=lora_extraction
             [distillation]=distillation_dmd [self-forcing]=self_forcing

.github/workflows/infra-build-image.yml

@@ -13,23 +13,16 @@ on:
         required: false
         default: false
         type: boolean
-  # Auto-rebuild the CUDA images when a repository-controlled image input
-  # changes on main. This includes the trusted SM89 kernel artifact's source,
-  # metadata/key helper, ABI dependency metadata, and build orchestration.
-  # Dreamverse (apps/dreamverse/docker/Dockerfile) and the ROCm Dockerfile stay
-  # manual-dispatch only.
+  # Auto-rebuild the CUDA images when their Dockerfile changes on main. The CUDA
+  # matrix is the only lane that builds from docker/Dockerfile, so a path-scoped
+  # push trigger is a sufficient change detector on its own -- no separate
+  # detect-changes/paths-filter job is needed now that there is a single
+  # in-scope Dockerfile. Dreamverse (apps/dreamverse/docker/Dockerfile) and the
+  # rocm Dockerfile stay manual-dispatch only.
   push:
     branches: [main]
     paths:
-      - '.dockerignore'
-      - '.github/workflows/_template-build-image.yml'
-      - '.github/workflows/infra-build-image.yml'
-      - '.gitmodules'
       - 'docker/Dockerfile'
-      - 'docker/uv-excludes'
-      - 'fastvideo-kernel/**'
-      - 'fastvideo/tests/modal/kernel_build_cache.py'
-      - 'pyproject.toml'
 
 
 permissions:
@@ -57,7 +50,7 @@ jobs:
   # 2.8.3 comes from the architecture-specific prebuilt releases.
   build-cuda-images:
     # Runs on a manual dispatch when build_cuda_matrix is set, or automatically
-    # on an in-scope main push (inputs are null on push). The
+    # on a push that changed docker/Dockerfile (inputs are null on push). The
     # repository guard keeps fork syncs from auto-building; manual dispatch
     # still works in forks.
     if: ${{ (github.event_name == 'push' && github.repository == 'hao-ai-lab/FastVideo') || github.event.inputs.build_cuda_matrix == 'true' }}

.github/workflows/infra-docs.yml

@@ -6,7 +6,6 @@ on:
     paths:
       - 'docs/**'
       - 'examples/**'
-      - 'scripts/inference/**'
       - 'mkdocs.yml'
       - 'requirements-mkdocs.in'
       - 'requirements-mkdocs.txt'
@@ -17,7 +16,6 @@ on:
     paths:
       - 'docs/**'
       - 'examples/**'
-      - 'scripts/inference/**'
       - 'mkdocs.yml'
       - 'requirements-mkdocs.in'
       - 'requirements-mkdocs.txt'

.gitignore

@@ -6,7 +6,7 @@ results/
 wandb/
 *.ipynb
 *.jpg
-!examples/datasets/lingbotworld2/image.jpg
+!examples/dataset/lingbotworld2/image.jpg
 *.safetensors
 *.mp4
 *.png
@@ -23,7 +23,6 @@ Miniconda3-latest-Linux-x86_64.sh
 *validation/
 data/
 outputs/
-outputs_audio/
 outputs_video
 checkpoints/
 sbatch.sh
@@ -76,7 +75,6 @@ docs/distillation/examples/
 *.pkl
 
 # Reference videos (negations must come after the catch-all on line below)
-!fastvideo/tests/nightly/reference_video_*.mp4
 
 # Static images
 !docs/assets/images/**/*.png

.pre-commit-config.yaml

@@ -22,7 +22,6 @@ repos:
   hooks:
   - id: yapf
     args: [--in-place, --verbose]
-    language_version: python3.12
     additional_dependencies: [toml] # TODO: Remove when yapf is upgraded
 - repo: https://github.com/astral-sh/ruff-pre-commit
   rev: v0.11.12

README.md

@@ -9,7 +9,6 @@
 **FastVideo is a unified post-training and real-time inference framework for accelerated video generation.**
 
 ## NEWS
-- `2026/08/19`: FastVideo now supports MLX on Apple Silicon with [FastMetal-QAD](https://huggingface.co/collections/FastVideo/fastmetal), a family of 1.3B, 5B, and 14B models optimized for Mac—follow the [Apple Silicon guide](https://hao-ai-lab.github.io/FastVideo/getting_started/installation/mps/) and read the [Blog](https://haoailab.com/blogs/fastmetal/).
 - `2026/06/23`: Release FastWan-QAD: 5s of Video generated in 1.8s E2E. See the [FastWan-QAD models](https://huggingface.co/FastVideo/FastWan-QAD-FP8-1.3B), [Attn-QAT training guide](https://haoailab.com/FastVideo/training/attn_qat/), and [blog](https://haoailab.com/blogs/fastwan-qad/).
 - `2026/03/17`: Release demo: Into the Dreamverse: Vibe Directing in FastVideo, check out the [Blog](https://haoailab.com/blogs/dreamverse/).
 - `2026/03/13`: Release demo: Create a 5s 1080p Video in 4.5s with FastVideo on a Single GPU, check out the [Blog](https://haoailab.com/blogs/fastvideo_realtime_1080p/).
@@ -34,7 +33,7 @@ FastVideo has the following features:
   - [Sparse distillation](https://hao-ai-lab.github.io/blogs/fastvideo_post_training/) to achieve >50x denoising speedup
   - Scalable training with FSDP2, sequence parallelism, and selective activation checkpointing.
   - Causal distillation through Self-Forcing
-  - See this [page](https://hao-ai-lab.github.io/FastVideo/training/overview/) for the supported training workflows, and the [support matrix](https://hao-ai-lab.github.io/FastVideo/inference/support_matrix/) for supported models.
+  - See this [page](https://hao-ai-lab.github.io/FastVideo/training/overview/) for full list of supported models and recipes.
 - State-of-the-art performance optimizations for inference
   - Sequence Parallelism for distributed inference
   - Multiple state-of-the-art attention backends
@@ -63,11 +62,6 @@ UV_TORCH_BACKEND=cu126 uv pip install fastvideo
 Use `UV_TORCH_BACKEND=cu130` on CUDA 13. Apple silicon users should follow the
 [MPS installation guide](https://hao-ai-lab.github.io/FastVideo/getting_started/installation/mps/).
 
-> **On an Apple Silicon Mac?** FastVideo runs FastWan text-to-video natively
-> through an MLX runtime — a 5-second 480p clip generated locally, no cloud,
-> no discrete GPU. Install with `uv pip install -e '.[mlx]'` and follow the
-> [Apple Silicon guide](https://hao-ai-lab.github.io/FastVideo/getting_started/installation/mps/).
-
 Please see our [docs](https://hao-ai-lab.github.io/FastVideo/getting_started/installation/) for more detailed installation instructions.
 
 > **On an NVIDIA DGX Spark (GB10 / ARM64 + CUDA 13)?** There's no prebuilt ARM wheel for the FastVideo CUDA kernel, so it's an editable from-source install (`UV_TORCH_BACKEND=cu130 uv pip install -e .`, which compiles that kernel for you) rather than `UV_TORCH_BACKEND=cu130 uv pip install fastvideo`. A compatible prebuilt ARM64 FlashAttention wheel is available separately. Follow the [DGX Spark install guide](https://hao-ai-lab.github.io/FastVideo/getting_started/installation/spark/).

apps/dreamverse/dreamverse/tests/conftest.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import sys
 from pathlib import Path
 
+
 TESTS_DIR = Path(__file__).resolve().parent
 DREAMVERSE_PACKAGE_DIR = TESTS_DIR.parent
 DREAMVERSE_APP_DIR = DREAMVERSE_PACKAGE_DIR.parent

apps/dreamverse/dreamverse/tests/test_config.py

@@ -5,6 +5,7 @@ from pathlib import Path
 
 import pytest
 
+
 SERVER_DIR = Path(__file__).resolve().parents[1]
 
 
@@ -52,7 +53,9 @@ def test_config_defaults_to_cerebras_with_parallel_groq_fallback_stage(monkeypat
     module = _load_config_module()
 
     assert module.PROMPT_PROVIDER == "cerebras"
-    assert module.PROMPT_PROVIDER_RUNTIME_STAGES == (("cerebras", "groq"), )
+    assert module.PROMPT_PROVIDER_RUNTIME_STAGES == (
+        ("cerebras", "groq"),
+    )
     assert module.PROMPT_PROVIDER_PRIORITY == (
         "cerebras",
         "groq",
@@ -83,7 +86,9 @@ def test_config_ignores_legacy_groq_primary_override(monkeypatch):
     module = _load_config_module()
 
     assert module.PROMPT_PROVIDER == "cerebras"
-    assert module.PROMPT_PROVIDER_RUNTIME_STAGES == (("cerebras", "groq"), )
+    assert module.PROMPT_PROVIDER_RUNTIME_STAGES == (
+        ("cerebras", "groq"),
+    )
     assert module.PROMPT_PROVIDER_PRIORITY == (
         "cerebras",
         "groq",
@@ -101,17 +106,24 @@ def test_config_uses_local_overlay_paths_when_devtools_enabled(monkeypatch, tmp_
 
     assert module.DEVTOOLS_ENABLED is True
     assert module.FRONTEND_ROOT.as_posix().endswith("apps/dreamverse/web")
-    assert module.PROMPT_ENHANCE_SYSTEM_PROMPT_PATH.endswith("dreamverse/prompts.local/next_segment_system_prompt.md")
+    assert module.PROMPT_ENHANCE_SYSTEM_PROMPT_PATH.endswith(
+        "dreamverse/prompts.local/next_segment_system_prompt.md"
+    )
     assert module.PROMPT_ENHANCE_SYSTEM_PROMPT_FALLBACK_PATH.endswith(
-        "dreamverse/prompts/next_segment_system_prompt.md")
+        "dreamverse/prompts/next_segment_system_prompt.md"
+    )
     assert module.PROMPT_REWRITE_USER_SYSTEM_PROMPT_PATH.endswith(
-        "dreamverse/prompts.local/rewrite_user_system_prompt.md")
+        "dreamverse/prompts.local/rewrite_user_system_prompt.md"
+    )
     assert module.PROMPT_REWRITE_USER_SYSTEM_PROMPT_FALLBACK_PATH.endswith(
-        "dreamverse/prompts/rewrite_user_system_prompt.md")
+        "dreamverse/prompts/rewrite_user_system_prompt.md"
+    )
     assert module.CURATED_PRESETS_FILE_PATH.endswith(
-        "apps/dreamverse/web/prompts.local/selected_ltx2_continuation_story_presets.json")
+        "apps/dreamverse/web/prompts.local/selected_ltx2_continuation_story_presets.json"
+    )
     assert module.CURATED_PRESETS_FALLBACK_FILE_PATH.endswith(
-        "apps/dreamverse/web/prompts/selected_ltx2_continuation_story_presets.json")
+        "apps/dreamverse/web/prompts/selected_ltx2_continuation_story_presets.json"
+    )
     assert module.FRONTEND_STATIC_DIR_CANDIDATES[:2] == (
         str(module.FRONTEND_ROOT / "out"),
         str(module.FRONTEND_ROOT / "dist"),

apps/dreamverse/dreamverse/tests/test_entrypoints.py

@@ -9,7 +9,6 @@ from fastapi.testclient import TestClient
 import fastvideo.entrypoints.streaming as streaming_entrypoints
 import pytest
 
-
 def _install_stack03_import_stubs(monkeypatch):
     """Keep entrypoint tests focused while later-stack runtime modules are absent."""
     if not hasattr(streaming_entrypoints, "build_health_router"):
@@ -18,7 +17,6 @@ def _install_stack03_import_stubs(monkeypatch):
     gpu_pool_stub = types.ModuleType("dreamverse.gpu_pool")
 
     class GPUPool:
-
         def __init__(self, _gpu_ids):
             pass
 
@@ -51,7 +49,6 @@ def _install_stack03_import_stubs(monkeypatch):
     controller_stub = types.ModuleType("dreamverse.session.controller")
 
     class SessionController:
-
         def __init__(self, **_kwargs):
             pass
 
@@ -79,11 +76,13 @@ def _run_cli(module, monkeypatch, argv: list[str]) -> list[dict[str, object]]:
     uvicorn_stub = types.ModuleType("uvicorn")
 
     def run(app, host: str, port: int) -> None:
-        calls.append({
-            "app": app,
-            "host": host,
-            "port": port,
-        })
+        calls.append(
+            {
+                "app": app,
+                "host": host,
+                "port": port,
+            }
+        )
 
     uvicorn_stub.run = run
     monkeypatch.setitem(sys.modules, "uvicorn", uvicorn_stub)
@@ -100,11 +99,13 @@ def test_server_cli_defaults_to_local_web_port(monkeypatch):
     server_main = _import_server_main(monkeypatch)
     calls = _run_cli(server_main, monkeypatch, ["dreamverse-server"])
 
-    assert calls == [{
-        "app": server_main.app,
-        "host": "0.0.0.0",
-        "port": 8009,
-    }]
+    assert calls == [
+        {
+            "app": server_main.app,
+            "host": "0.0.0.0",
+            "port": 8009,
+        }
+    ]
 
 
 def test_server_cli_allows_explicit_host_and_port(monkeypatch):
@@ -115,11 +116,13 @@ def test_server_cli_allows_explicit_host_and_port(monkeypatch):
         ["dreamverse-server", "--host", "127.0.0.1", "--port", "8123"],
     )
 
-    assert calls == [{
-        "app": server_main.app,
-        "host": "127.0.0.1",
-        "port": 8123,
-    }]
+    assert calls == [
+        {
+            "app": server_main.app,
+            "host": "127.0.0.1",
+            "port": 8123,
+        }
+    ]
 
 
 def test_server_does_not_expose_backend_source_as_static_assets(monkeypatch):
@@ -139,11 +142,13 @@ def test_mock_server_cli_defaults_to_local_web_port(monkeypatch):
         ["dreamverse-mock-server"],
     )
 
-    assert calls == [{
-        "app": mock_server.app,
-        "host": "0.0.0.0",
-        "port": 8009,
-    }]
+    assert calls == [
+        {
+            "app": mock_server.app,
+            "host": "0.0.0.0",
+            "port": 8009,
+        }
+    ]
 
 
 def test_mock_server_cli_updates_latency(monkeypatch):
@@ -156,11 +161,13 @@ def test_mock_server_cli_updates_latency(monkeypatch):
             ["dreamverse-mock-server", "--latency", "321", "--port", "8111"],
         )
 
-        assert calls == [{
-            "app": mock_server.app,
-            "host": "0.0.0.0",
-            "port": 8111,
-        }]
+        assert calls == [
+            {
+                "app": mock_server.app,
+                "host": "0.0.0.0",
+                "port": 8111,
+            }
+        ]
         assert mock_server.LATENCY_MS == 321
     finally:
         mock_server.LATENCY_MS = old_latency_ms

apps/dreamverse/dreamverse/tests/test_gpu_pool.py

@@ -7,6 +7,7 @@ from types import SimpleNamespace
 
 import pytest
 
+
 import dreamverse.gpu_pool as gpu_pool
 
 
@@ -84,7 +85,9 @@ def test_send_command_raises_on_worker_death():
     cmd_q = ctx.Queue()
     resp_q = ctx.Queue()
 
-    proc = ctx.Process(target=_child_consume_and_exit, args=(cmd_q, resp_q))
+    proc = ctx.Process(
+        target=_child_consume_and_exit, args=(cmd_q, resp_q)
+    )
     proc.start()
 
     # Wait for the spawn child to fully boot.  Allow generous time —

apps/dreamverse/dreamverse/tests/test_import_contract.py

@@ -7,7 +7,7 @@ ALLOWED_PREFIXES = (
     "fastvideo.entrypoints.video_generator",
     "fastvideo.configs",
 )
-ALLOWED_EXACT = ("fastvideo", )
+ALLOWED_EXACT = ("fastvideo",)
 FORBIDDEN_PREFIXES = (
     "fastvideo.pipelines",
     "fastvideo.models",
@@ -38,13 +38,19 @@ def test_dreamverse_server_imports_only_public_fastvideo_surfaces() -> None:
         except SyntaxError as task_exc:
             raise AssertionError(f"Failed to parse {path}") from task_exc
         for node in ast.walk(tree):
-            names = ([a.name for a in node.names] if isinstance(node, ast.Import) else
-                     [node.module] if isinstance(node, ast.ImportFrom) and node.module else [])
+            names = (
+                [a.name for a in node.names] if isinstance(node, ast.Import)
+                else [node.module] if isinstance(node, ast.ImportFrom) and node.module
+                else []
+            )
             for name in names:
                 if not name:
                     continue
                 rel_path = str(path.relative_to(root))
-                if (name.startswith(FORBIDDEN_PREFIXES) and (rel_path, name) not in ALLOWED_INTERNAL_IMPORTS):
+                if (
+                    name.startswith(FORBIDDEN_PREFIXES)
+                    and (rel_path, name) not in ALLOWED_INTERNAL_IMPORTS
+                ):
                     bad.append((str(path.relative_to(root)), getattr(node, "lineno", 0), name))
 
     assert bad == [], f"Forbidden internal imports: {bad}"

apps/dreamverse/dreamverse/tests/test_mock_server.py

@@ -6,6 +6,7 @@ import os
 
 from fastapi import WebSocketDisconnect
 
+
 os.environ.setdefault("CEREBRAS_API_KEY", "dummy")
 os.environ.setdefault("GROQ_API_KEY", "dummy")
 
@@ -13,7 +14,6 @@ import dreamverse.mock_server as mock_server
 
 
 class _FakeWebSocket:
-
     def __init__(self, messages: list[tuple[float, dict[str, object]]]):
         self._messages = messages
         self._index = 0
@@ -49,34 +49,34 @@ def test_mock_server_matches_current_single5s_protocol():
         mock_server.MOCK_SEGMENT_BYTES = b"mock-fmp4-bytes"
         mock_server.LATENCY_MS = 1
 
-        ws = _FakeWebSocket([
-            (
-                0.0,
-                {
-                    "type": "session_init_v2",
-                    "preset_id": "simple_prompt_1",
-                    "curated_prompts": ["selected prompt"],
-                    "single_clip_mode": True,
-                    "enhancement_enabled": False,
-                    "auto_extension_enabled": False,
-                    "loop_generation_enabled": False,
-                },
-            ),
-            (
-                0.01,
-                {
-                    "type": "simple_generate",
-                    "preset_id": "simple_custom_prompt",
-                    "prompt_id": "simple_custom_prompt",
-                    "prompt": "custom prompt",
-                    "enhancement_enabled": True,
-                    "initial_image": None,
-                },
-            ),
-            (0.20, {
-                "type": "leave"
-            }),
-        ])
+        ws = _FakeWebSocket(
+            [
+                (
+                    0.0,
+                    {
+                        "type": "session_init_v2",
+                        "preset_id": "simple_prompt_1",
+                        "curated_prompts": ["selected prompt"],
+                        "single_clip_mode": True,
+                        "enhancement_enabled": False,
+                        "auto_extension_enabled": False,
+                        "loop_generation_enabled": False,
+                    },
+                ),
+                (
+                    0.01,
+                    {
+                        "type": "simple_generate",
+                        "preset_id": "simple_custom_prompt",
+                        "prompt_id": "simple_custom_prompt",
+                        "prompt": "custom prompt",
+                        "enhancement_enabled": True,
+                        "initial_image": None,
+                    },
+                ),
+                (0.20, {"type": "leave"}),
+            ]
+        )
 
         asyncio.run(mock_server.websocket_endpoint(ws))
 
@@ -92,14 +92,24 @@ def test_mock_server_matches_current_single5s_protocol():
         assert message_types.count("ltx2_stream_complete") == 2
         assert "prompt_sources_blocked" not in message_types
 
-        segment_start_events = [payload for payload in ws.sent_json if payload["type"] == "ltx2_segment_start"]
-        gpu_assigned_event = next(payload for payload in ws.sent_json if payload["type"] == "gpu_assigned")
+        segment_start_events = [
+            payload
+            for payload in ws.sent_json
+            if payload["type"] == "ltx2_segment_start"
+        ]
+        gpu_assigned_event = next(
+            payload for payload in ws.sent_json if payload["type"] == "gpu_assigned"
+        )
         assert gpu_assigned_event["session_timeout"] == mock_server.SESSION_TIMEOUT_SECONDS
         assert [payload["segment_idx"] for payload in segment_start_events] == [1, 1]
         assert segment_start_events[0]["prompt"] == "selected prompt"
         assert segment_start_events[1]["prompt"] == "custom prompt"
 
-        step_complete_events = [payload for payload in ws.sent_json if payload["type"] == "step_complete"]
+        step_complete_events = [
+            payload
+            for payload in ws.sent_json
+            if payload["type"] == "step_complete"
+        ]
         assert len(step_complete_events) == 2
         assert step_complete_events[0]["latency_ms"] == {
             "total": 121.0,
@@ -124,29 +134,29 @@ def test_mock_server_regular_cap_waits_for_rewrite_rollout():
         mock_server.LATENCY_MS = 1
         mock_server.GENERATION_SEGMENT_CAP = 1
 
-        ws = _FakeWebSocket([
-            (
-                0.0,
-                {
-                    "type": "session_init_v2",
-                    "preset_id": "test_preset",
-                    "curated_prompts": ["segment one"],
-                    "enhancement_enabled": True,
-                    "auto_extension_enabled": False,
-                    "loop_generation_enabled": False,
-                },
-            ),
-            (
-                0.02,
-                {
-                    "type": "rewrite_seed_prompts",
-                    "rewrite_instruction": "start a new rollout",
-                },
-            ),
-            (0.20, {
-                "type": "leave"
-            }),
-        ])
+        ws = _FakeWebSocket(
+            [
+                (
+                    0.0,
+                    {
+                        "type": "session_init_v2",
+                        "preset_id": "test_preset",
+                        "curated_prompts": ["segment one"],
+                        "enhancement_enabled": True,
+                        "auto_extension_enabled": False,
+                        "loop_generation_enabled": False,
+                    },
+                ),
+                (
+                    0.02,
+                    {
+                        "type": "rewrite_seed_prompts",
+                        "rewrite_instruction": "start a new rollout",
+                    },
+                ),
+                (0.20, {"type": "leave"}),
+            ]
+        )
 
         asyncio.run(mock_server.websocket_endpoint(ws))
 
@@ -156,7 +166,11 @@ def test_mock_server_regular_cap_waits_for_rewrite_rollout():
         assert "generation_cap_reached" not in message_types
         assert "prompt_sources_blocked" not in message_types
 
-        segment_start_events = [payload for payload in ws.sent_json if payload["type"] == "ltx2_segment_start"]
+        segment_start_events = [
+            payload
+            for payload in ws.sent_json
+            if payload["type"] == "ltx2_segment_start"
+        ]
         assert [payload["segment_idx"] for payload in segment_start_events] == [1, 1]
         assert segment_start_events[0]["prompt"] == "segment one"
         assert segment_start_events[1]["prompt"] == "segment one [start a new rollout]"
@@ -173,40 +187,54 @@ def test_mock_server_rewrite_during_active_segment_restarts_from_first_rewritten
         mock_server.MOCK_SEGMENT_BYTES = b"mock-fmp4-bytes"
         mock_server.LATENCY_MS = 100
 
-        ws = _FakeWebSocket([
-            (
-                0.0,
-                {
-                    "type": "session_init_v2",
-                    "preset_id": "test_preset",
-                    "curated_prompts": ["segment one", "segment two"],
-                    "enhancement_enabled": True,
-                    "auto_extension_enabled": False,
-                    "loop_generation_enabled": False,
-                },
-            ),
-            (
-                0.02,
-                {
-                    "type": "rewrite_seed_prompts",
-                    "rewrite_instruction": "restart from rewrite",
-                },
-            ),
-            (0.40, {
-                "type": "leave"
-            }),
-        ])
+        ws = _FakeWebSocket(
+            [
+                (
+                    0.0,
+                    {
+                        "type": "session_init_v2",
+                        "preset_id": "test_preset",
+                        "curated_prompts": ["segment one", "segment two"],
+                        "enhancement_enabled": True,
+                        "auto_extension_enabled": False,
+                        "loop_generation_enabled": False,
+                    },
+                ),
+                (
+                    0.02,
+                    {
+                        "type": "rewrite_seed_prompts",
+                        "rewrite_instruction": "restart from rewrite",
+                    },
+                ),
+                (0.40, {"type": "leave"}),
+            ]
+        )
 
         asyncio.run(mock_server.websocket_endpoint(ws))
 
-        segment_start_events = [payload for payload in ws.sent_json if payload["type"] == "ltx2_segment_start"]
+        segment_start_events = [
+            payload
+            for payload in ws.sent_json
+            if payload["type"] == "ltx2_segment_start"
+        ]
         assert [payload["prompt"] for payload in segment_start_events[:2]] == [
             "segment one",
             "segment one [restart from rewrite]",
         ]
-        assert all(payload["prompt"] != "segment two" for payload in segment_start_events[1:])
-        reset_events = [payload for payload in ws.sent_json if payload.get("type") == "seed_prompts_reset_applied"]
-        assert any(payload.get("reason") == "rewrite_during_generation" for payload in reset_events)
+        assert all(
+            payload["prompt"] != "segment two"
+            for payload in segment_start_events[1:]
+        )
+        reset_events = [
+            payload
+            for payload in ws.sent_json
+            if payload.get("type") == "seed_prompts_reset_applied"
+        ]
+        assert any(
+            payload.get("reason") == "rewrite_during_generation"
+            for payload in reset_events
+        )
     finally:
         mock_server.MOCK_SEGMENT_BYTES = old_segment_bytes
         mock_server.LATENCY_MS = old_latency_ms
@@ -219,24 +247,24 @@ def test_mock_server_supports_initial_custom_rollout_prompt():
         mock_server.MOCK_SEGMENT_BYTES = b"mock-fmp4-bytes"
         mock_server.LATENCY_MS = 1
 
-        ws = _FakeWebSocket([
-            (
-                0.0,
-                {
-                    "type": "session_init_v2",
-                    "preset_id": "custom_editable",
-                    "preset_label": "Custom rollout",
-                    "curated_prompts": [],
-                    "initial_rollout_prompt": "A moonbase corridor thriller with flooding",
-                    "enhancement_enabled": True,
-                    "auto_extension_enabled": False,
-                    "loop_generation_enabled": False,
-                },
-            ),
-            (0.20, {
-                "type": "leave"
-            }),
-        ])
+        ws = _FakeWebSocket(
+            [
+                (
+                    0.0,
+                    {
+                        "type": "session_init_v2",
+                        "preset_id": "custom_editable",
+                        "preset_label": "Custom rollout",
+                        "curated_prompts": [],
+                        "initial_rollout_prompt": "A moonbase corridor thriller with flooding",
+                        "enhancement_enabled": True,
+                        "auto_extension_enabled": False,
+                        "loop_generation_enabled": False,
+                    },
+                ),
+                (0.20, {"type": "leave"}),
+            ]
+        )
 
         asyncio.run(mock_server.websocket_endpoint(ws))
 
@@ -248,9 +276,15 @@ def test_mock_server_supports_initial_custom_rollout_prompt():
         assert "ltx2_stream_start" in message_types
         assert "prompt_sources_blocked" not in message_types
 
-        segment_start_events = [payload for payload in ws.sent_json if payload["type"] == "ltx2_segment_start"]
+        segment_start_events = [
+            payload
+            for payload in ws.sent_json
+            if payload["type"] == "ltx2_segment_start"
+        ]
         assert segment_start_events
-        assert segment_start_events[0]["prompt"] == ("A moonbase corridor thriller with flooding [segment 1]")
+        assert segment_start_events[0]["prompt"] == (
+            "A moonbase corridor thriller with flooding [segment 1]"
+        )
     finally:
         mock_server.MOCK_SEGMENT_BYTES = old_segment_bytes
         mock_server.LATENCY_MS = old_latency_ms
@@ -263,37 +297,35 @@ def test_mock_server_can_start_new_project_without_reconnecting():
         mock_server.MOCK_SEGMENT_BYTES = b"mock-fmp4-bytes"
         mock_server.LATENCY_MS = 40
 
-        ws = _FakeWebSocket([
-            (
-                0.0,
-                {
-                    "type": "session_init_v2",
-                    "preset_id": "test_preset",
-                    "curated_prompts": ["segment one"],
-                    "enhancement_enabled": True,
-                    "auto_extension_enabled": False,
-                    "loop_generation_enabled": False,
-                },
-            ),
-            (0.02, {
-                "type": "end_project_keep_session"
-            }),
-            (
-                0.20,
-                {
-                    "type": "project_init_v1",
-                    "preset_id": "test_preset_2",
-                    "preset_label": "Test Preset 2",
-                    "curated_prompts": ["segment two"],
-                    "enhancement_enabled": True,
-                    "auto_extension_enabled": False,
-                    "loop_generation_enabled": False,
-                },
-            ),
-            (0.40, {
-                "type": "leave"
-            }),
-        ])
+        ws = _FakeWebSocket(
+            [
+                (
+                    0.0,
+                    {
+                        "type": "session_init_v2",
+                        "preset_id": "test_preset",
+                        "curated_prompts": ["segment one"],
+                        "enhancement_enabled": True,
+                        "auto_extension_enabled": False,
+                        "loop_generation_enabled": False,
+                    },
+                ),
+                (0.02, {"type": "end_project_keep_session"}),
+                (
+                    0.20,
+                    {
+                        "type": "project_init_v1",
+                        "preset_id": "test_preset_2",
+                        "preset_label": "Test Preset 2",
+                        "curated_prompts": ["segment two"],
+                        "enhancement_enabled": True,
+                        "auto_extension_enabled": False,
+                        "loop_generation_enabled": False,
+                    },
+                ),
+                (0.40, {"type": "leave"}),
+            ]
+        )
 
         asyncio.run(mock_server.websocket_endpoint(ws))
 
@@ -304,11 +336,16 @@ def test_mock_server_can_start_new_project_without_reconnecting():
 
         project_idle_index = message_types.index("project_idle")
         stream_start_indexes = [
-            index for index, message_type in enumerate(message_types) if message_type == "ltx2_stream_start"
+            index for index, message_type in enumerate(message_types)
+            if message_type == "ltx2_stream_start"
         ]
         assert stream_start_indexes[0] < project_idle_index < stream_start_indexes[1]
 
-        segment_start_events = [payload for payload in ws.sent_json if payload["type"] == "ltx2_segment_start"]
+        segment_start_events = [
+            payload
+            for payload in ws.sent_json
+            if payload["type"] == "ltx2_segment_start"
+        ]
         assert [payload["prompt"] for payload in segment_start_events[:2]] == [
             "segment one",
             "segment two",

apps/dreamverse/dreamverse/tests/test_prompt_enhancer.py

@@ -6,6 +6,7 @@ import os
 import re
 import time
 
+
 os.environ.setdefault("CEREBRAS_API_KEY", "dummy")
 os.environ.setdefault("GROQ_API_KEY", "dummy")
 
@@ -21,7 +22,6 @@ from dreamverse.prompt_enhancer import (
 
 
 class _FakeResponse:
-
     def __init__(self, payload: dict):
         self._payload = payload
 
@@ -30,7 +30,6 @@ class _FakeResponse:
 
 
 class _FakeSyncCompletions:
-
     def __init__(self, payload: dict):
         self._payload = payload
 
@@ -39,7 +38,6 @@ class _FakeSyncCompletions:
 
 
 class _FakeSyncClient:
-
     def __init__(self, payload: dict):
         self.chat = type(
             "_FakeChat",
@@ -49,7 +47,6 @@ class _FakeSyncClient:
 
 
 class _DelayedSyncCompletions:
-
     def __init__(self, payload: dict, delay_s: float = 0.0, exc: Exception | None = None):
         self._payload = payload
         self._delay_s = delay_s
@@ -64,26 +61,29 @@ class _DelayedSyncCompletions:
 
 
 class _DelayedSyncClient:
-
     def __init__(self, payload: dict, delay_s: float = 0.0, exc: Exception | None = None):
         self.chat = type(
             "_FakeChat",
             (),
-            {"completions": _DelayedSyncCompletions(
-                payload,
-                delay_s=delay_s,
-                exc=exc,
-            )},
+            {
+                "completions": _DelayedSyncCompletions(
+                    payload,
+                    delay_s=delay_s,
+                    exc=exc,
+                )
+            },
         )()
 
 
 def _chat_payload_with_content(content: str) -> dict:
     return {
-        "choices": [{
-            "message": {
-                "content": content,
+        "choices": [
+            {
+                "message": {
+                    "content": content,
+                }
             }
-        }]
+        ]
     }
 
 
@@ -172,7 +172,6 @@ def _build_staged_enhancer(
 
 
 class _FakeOpenAIClient:
-
     def __init__(self, **kwargs):
         self.kwargs = kwargs
         self.chat = type(
@@ -183,7 +182,6 @@ class _FakeOpenAIClient:
 
 
 class _FakeCerebrasClient:
-
     def __init__(self, **kwargs):
         self.kwargs = kwargs
         self.chat = type(
@@ -194,12 +192,16 @@ class _FakeCerebrasClient:
 
 
 def test_parse_json_response_accepts_fenced_json_with_prose():
-    parsed = _parse_json_response("Here is the rewrite:\n```json\n{\"segment_prompts\":[\"A\",\"B\"]}\n```\nThanks.")
+    parsed = _parse_json_response(
+        "Here is the rewrite:\n```json\n{\"segment_prompts\":[\"A\",\"B\"]}\n```\nThanks."
+    )
     assert parsed == {"segment_prompts": ["A", "B"]}
 
 
 def test_parse_json_response_extracts_first_embedded_object():
-    parsed = _parse_json_response("Model output:\n{\"segment_prompts\":[\"A\",\"B\"]}\n(complete)")
+    parsed = _parse_json_response(
+        "Model output:\n{\"segment_prompts\":[\"A\",\"B\"]}\n(complete)"
+    )
     assert parsed == {"segment_prompts": ["A", "B"]}
 
 
@@ -266,12 +268,16 @@ def test_build_client_supports_groq_provider(monkeypatch):
 
 def test_rewrite_prompt_sequence_accepts_segment_prompts_output():
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     result = asyncio.run(
         enhancer.rewrite_prompt_sequence(
             ["prompt one", "prompt two"],
             rewrite_instruction="make it cinematic",
-        ))
+        )
+    )
     assert result.fallback_used is False
     assert result.error is None
     assert result.rollout_id == "preset_a"
@@ -280,12 +286,15 @@ def test_rewrite_prompt_sequence_accepts_segment_prompts_output():
 
 
 def test_rewrite_prompt_sequence_accepts_legacy_rewritten_prompts_output():
-    enhancer = _build_test_enhancer(_chat_payload_with_content('{"rewritten_prompts":["A","B"]}'))
+    enhancer = _build_test_enhancer(
+        _chat_payload_with_content('{"rewritten_prompts":["A","B"]}')
+    )
     result = asyncio.run(
         enhancer.rewrite_prompt_sequence(
             ["prompt one", "prompt two"],
             rewrite_instruction="make it cinematic",
-        ))
+        )
+    )
     assert result.fallback_used is False
     assert result.error is None
     assert result.rollout_id == "current_rollout"
@@ -294,14 +303,19 @@ def test_rewrite_prompt_sequence_accepts_legacy_rewritten_prompts_output():
 
 
 def test_rewrite_prompt_sequence_accepts_segment_dicts_without_top_level_rollout_metadata():
-    enhancer = _build_test_enhancer(_chat_payload_with_content('{"segments":[{"prompt":"A"},{"text":"B"}]}'))
+    enhancer = _build_test_enhancer(
+        _chat_payload_with_content(
+            '{"segments":[{"prompt":"A"},{"text":"B"}]}'
+        )
+    )
     result = asyncio.run(
         enhancer.rewrite_prompt_sequence(
             ["prompt one", "prompt two"],
             preset_id="preset_a",
             preset_label="Preset A",
             rewrite_instruction="make it cinematic",
-        ))
+        )
+    )
     assert result.fallback_used is False
     assert result.error is None
     assert result.rollout_id == "preset_a"
@@ -315,12 +329,14 @@ def test_rewrite_prompt_sequence_accepts_numbered_prose_output():
             "The user is asking for a cinematic rewrite.\n\n"
             "1. A dog bounds across the moon's dusty surface, kicking up silver regolith as it chases a rabbit beneath the black sky.\n"
             "2. The rabbit darts around a crater rim while the dog lunges after it, Earth glowing blue in the distance.\n"
-        ))
+        )
+    )
     result = asyncio.run(
         enhancer.rewrite_prompt_sequence(
             ["prompt one", "prompt two"],
             rewrite_instruction="make it cinematic",
-        ))
+        )
+    )
     assert result.fallback_used is False
     assert result.error is None
     assert result.rollout_id == "current_rollout"
@@ -339,10 +355,12 @@ def test_enhance_prompt_prefers_cerebras_before_groq_fallback():
         groq_delay_s=0.01,
     )
 
-    result = asyncio.run(enhancer.enhance_prompt(
-        "A rainy alley at night",
-        mode="single_clip",
-    ))
+    result = asyncio.run(
+        enhancer.enhance_prompt(
+            "A rainy alley at night",
+            mode="single_clip",
+        )
+    )
 
     assert result.fallback_used is False
     assert result.error is None
@@ -363,10 +381,12 @@ def test_enhance_prompt_uses_groq_when_cerebras_fails():
         groq_delay_s=0.01,
     )
 
-    result = asyncio.run(enhancer.enhance_prompt(
-        "A rainy alley at night",
-        mode="single_clip",
-    ))
+    result = asyncio.run(
+        enhancer.enhance_prompt(
+            "A rainy alley at night",
+            mode="single_clip",
+        )
+    )
 
     assert result.fallback_used is False
     assert result.error is None
@@ -388,11 +408,13 @@ def test_enhance_prompt_can_use_groq_when_cerebras_times_out():
     enhancer.http_timeout_ms = 50
     enhancer.default_timeout_ms = 50
 
-    result = asyncio.run(enhancer.enhance_prompt(
-        "A rainy alley at night",
-        mode="single_clip",
-        timeout_ms=50,
-    ))
+    result = asyncio.run(
+        enhancer.enhance_prompt(
+            "A rainy alley at night",
+            mode="single_clip",
+            timeout_ms=50,
+        )
+    )
 
     assert result.fallback_used is False
     assert result.error is None
@@ -412,10 +434,12 @@ def test_enhance_prompt_can_use_cerebras_when_it_returns_first():
         groq_delay_s=0.08,
     )
 
-    result = asyncio.run(enhancer.enhance_prompt(
-        "A rainy alley at night",
-        mode="single_clip",
-    ))
+    result = asyncio.run(
+        enhancer.enhance_prompt(
+            "A rainy alley at night",
+            mode="single_clip",
+        )
+    )
 
     assert result.fallback_used is False
     assert result.error is None
@@ -429,12 +453,15 @@ def test_enhance_prompt_can_use_cerebras_when_it_returns_first():
 
 
 def test_rewrite_prompt_sequence_keeps_raw_output_on_parse_error():
-    enhancer = _build_test_enhancer(_chat_payload_with_content("I cannot comply with JSON right now."))
+    enhancer = _build_test_enhancer(
+        _chat_payload_with_content("I cannot comply with JSON right now.")
+    )
     result = asyncio.run(
         enhancer.rewrite_prompt_sequence(
             ["prompt one", "prompt two"],
             rewrite_instruction="make it cinematic",
-        ))
+        )
+    )
     assert result.fallback_used is True
     assert "No JSON object found in assistant response." in (result.error or "")
     assert result.raw_response_text == "I cannot comply with JSON right now."
@@ -446,7 +473,9 @@ def test_rewrite_prompt_sequence_keeps_raw_output_on_parse_error():
 def test_rewrite_prompt_sequence_uses_current_rollout_payload_shape():
     enhancer = _build_test_enhancer(
         _chat_payload_with_content(
-            '{"id":"rewritten_rollout","label":"Rewritten Rollout","segment_prompts":["A","B"]}'))
+            '{"id":"rewritten_rollout","label":"Rewritten Rollout","segment_prompts":["A","B"]}'
+        )
+    )
     captured = {
         "body": None,
         "timeout_seconds": None,
@@ -457,7 +486,8 @@ def test_rewrite_prompt_sequence_uses_current_rollout_payload_shape():
         captured["timeout_seconds"] = timeout_seconds
         return (
             _chat_payload_with_content(
-                '{"id":"rewritten_rollout","label":"Rewritten Rollout","segment_prompts":["A","B"]}'),
+                '{"id":"rewritten_rollout","label":"Rewritten Rollout","segment_prompts":["A","B"]}'
+            ),
             '{"id":"rewritten_rollout","label":"Rewritten Rollout","segment_prompts":["A","B"]}',
         )
 
@@ -472,7 +502,8 @@ def test_rewrite_prompt_sequence_uses_current_rollout_payload_shape():
             rewrite_model="gpt-test",
             rewrite_temperature=0.2,
             timeout_ms=800,
-        ))
+        )
+    )
 
     assert result.fallback_used is False
     assert captured["body"]["messages"][0] == {
@@ -481,12 +512,12 @@ def test_rewrite_prompt_sequence_uses_current_rollout_payload_shape():
     }
     assert captured["body"]["messages"][1]["role"] == "user"
     assert prompt_enhancer_module.json.loads(captured["body"]["messages"][1]["content"]) == {
-        "mode":
-        "edit_existing_rollout",
-        "request": ("Rewrite all segment prompts with improved continuity and cinematic detail. "
-                    "Keep count and ordering identical."),
-        "user_instruction":
-        "make it cinematic",
+        "mode": "edit_existing_rollout",
+        "request": (
+            "Rewrite all segment prompts with improved continuity and cinematic detail. "
+            "Keep count and ordering identical."
+        ),
+        "user_instruction": "make it cinematic",
         "current_rollout": {
             "id": "preset_a",
             "label": "Preset A",
@@ -497,8 +528,11 @@ def test_rewrite_prompt_sequence_uses_current_rollout_payload_shape():
 
 def test_rewrite_prompt_sequence_supports_new_rollout_mode():
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"custom_editable","label":"Custom rollout","segment_prompts":['
-                                   '"A","B","C","D","E","F"]}'))
+        _chat_payload_with_content(
+            '{"id":"custom_editable","label":"Custom rollout","segment_prompts":['
+            '"A","B","C","D","E","F"]}'
+        )
+    )
     captured = {
         "body": None,
     }
@@ -507,8 +541,10 @@ def test_rewrite_prompt_sequence_supports_new_rollout_mode():
         del timeout_seconds
         captured["body"] = body
         return (
-            _chat_payload_with_content('{"id":"custom_editable","label":"Custom rollout","segment_prompts":['
-                                       '"A","B","C","D","E","F"]}'),
+            _chat_payload_with_content(
+                '{"id":"custom_editable","label":"Custom rollout","segment_prompts":['
+                '"A","B","C","D","E","F"]}'
+            ),
             '{"id":"custom_editable","label":"Custom rollout","segment_prompts":['
             '"A","B","C","D","E","F"]}',
         )
@@ -524,29 +560,30 @@ def test_rewrite_prompt_sequence_supports_new_rollout_mode():
             rewrite_model="gpt-test",
             rewrite_temperature=0.2,
             timeout_ms=800,
-        ))
+        )
+    )
 
     assert result.fallback_used is False
     assert result.prompts == ["A", "B", "C", "D", "E", "F"]
     assert prompt_enhancer_module.json.loads(captured["body"]["messages"][1]["content"]) == {
-        "mode":
-        "new_rollout",
-        "request": ("Rewrite all segment prompts with improved continuity and cinematic detail. "
-                    "Keep count and ordering identical."),
-        "user_instruction":
-        "A moonbase corridor thriller with flooding and red alarms",
-        "desired_segment_count":
-        6,
-        "rollout_id_hint":
-        "custom_editable",
-        "rollout_label_hint":
-        "Custom rollout",
+        "mode": "new_rollout",
+        "request": (
+            "Rewrite all segment prompts with improved continuity and cinematic detail. "
+            "Keep count and ordering identical."
+        ),
+        "user_instruction": "A moonbase corridor thriller with flooding and red alarms",
+        "desired_segment_count": 6,
+        "rollout_id_hint": "custom_editable",
+        "rollout_label_hint": "Custom rollout",
     }
 
 
 def test_rewrite_prompt_sequence_uses_session_override_system_prompt():
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     enhancer.rewrite_all_system_prompt = "shared system prompt"
     captured = {
         "body": None,
@@ -556,7 +593,9 @@ def test_rewrite_prompt_sequence_uses_session_override_system_prompt():
         del timeout_seconds
         captured["body"] = body
         return (
-            _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'),
+            _chat_payload_with_content(
+                '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+            ),
             '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}',
         )
 
@@ -570,7 +609,8 @@ def test_rewrite_prompt_sequence_uses_session_override_system_prompt():
             rewrite_instruction="make it cinematic",
             rewrite_model="gpt-test",
             system_prompt_override="session specific system prompt",
-        ))
+        )
+    )
 
     assert result.fallback_used is False
     assert captured["body"]["messages"][0] == {
@@ -581,7 +621,10 @@ def test_rewrite_prompt_sequence_uses_session_override_system_prompt():
 
 def test_resolve_rewrite_new_rollout_system_prompt_uses_dedicated_prompt():
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     enhancer.rewrite_all_system_prompt = "shared rewrite system prompt"
     enhancer.rewrite_user_system_prompt = "new rollout rewrite system prompt"
 
@@ -592,17 +635,24 @@ def test_resolve_rewrite_new_rollout_system_prompt_uses_dedicated_prompt():
 
 def test_resolve_rewrite_new_rollout_system_prompt_prefers_override():
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     enhancer.rewrite_all_system_prompt = "shared rewrite system prompt"
     enhancer.rewrite_user_system_prompt = "new rollout rewrite system prompt"
 
-    resolved = enhancer.resolve_rewrite_new_rollout_system_prompt("session specific system prompt")
+    resolved = enhancer.resolve_rewrite_new_rollout_system_prompt(
+        "session specific system prompt"
+    )
 
     assert resolved == "session specific system prompt"
 
 
 def test_generate_auto_prompt_uses_selected_model():
-    enhancer = _build_test_enhancer(_chat_payload_with_content('{"next_prompt":"Auto next"}'))
+    enhancer = _build_test_enhancer(
+        _chat_payload_with_content('{"next_prompt":"Auto next"}')
+    )
     enhancer.auto_system_prompt = "auto system prompt"
     enhancer.rewrite_model_options = ["gpt-test", "gpt-alt"]
     enhancer.rewrite_default_model = "gpt-test"
@@ -630,7 +680,8 @@ def test_generate_auto_prompt_uses_selected_model():
             next_segment_idx=2,
             model="gpt-alt",
             timeout_ms=800,
-        ))
+        )
+    )
     assert result.fallback_used is False
     assert result.error is None
     assert result.prompt == "Auto next"
@@ -639,7 +690,9 @@ def test_generate_auto_prompt_uses_selected_model():
 
 
 def test_enhance_prompt_uses_selected_model():
-    enhancer = _build_test_enhancer(_chat_payload_with_content('{"next_prompt":"Enhanced next"}'))
+    enhancer = _build_test_enhancer(
+        _chat_payload_with_content('{"next_prompt":"Enhanced next"}')
+    )
     enhancer.enhance_system_prompt = "enhance system prompt"
     enhancer.auto_system_prompt = "auto system prompt"
     enhancer.rewrite_model_options = ["gpt-test", "gpt-alt"]
@@ -669,7 +722,8 @@ def test_enhance_prompt_uses_selected_model():
             next_segment_idx=2,
             model="gpt-alt",
             timeout_ms=800,
-        ))
+        )
+    )
     assert result.fallback_used is False
     assert result.error is None
     assert result.prompt == "Enhanced next"
@@ -678,7 +732,9 @@ def test_enhance_prompt_uses_selected_model():
 
 
 def test_enhance_prompt_single_clip_uses_auto_extension_prompt_and_prompt_field():
-    enhancer = _build_test_enhancer(_chat_payload_with_content('{"prompt":"Extended single clip"}'))
+    enhancer = _build_test_enhancer(
+        _chat_payload_with_content('{"prompt":"Extended single clip"}')
+    )
     enhancer.enhance_system_prompt = "enhance system prompt"
     enhancer.auto_system_prompt = "auto system prompt"
     enhancer.rewrite_model_options = ["gpt-test", "gpt-alt"]
@@ -708,12 +764,14 @@ def test_enhance_prompt_single_clip_uses_auto_extension_prompt_and_prompt_field(
 
     enhancer._request_content = _fake_request_content  # type: ignore[attr-defined]
 
-    result = asyncio.run(enhancer.enhance_prompt(
-        "short 5s idea",
-        mode="single_clip",
-        model="gpt-alt",
-        timeout_ms=800,
-    ))
+    result = asyncio.run(
+        enhancer.enhance_prompt(
+            "short 5s idea",
+            mode="single_clip",
+            model="gpt-alt",
+            timeout_ms=800,
+        )
+    )
     assert result.fallback_used is False
     assert result.error is None
     assert result.prompt == "Extended single clip"
@@ -726,15 +784,17 @@ def test_enhance_prompt_single_clip_uses_auto_extension_prompt_and_prompt_field(
             "single 5-second LTX-2.3 video clip. Respond with "
             'valid JSON only as {"prompt": "..."}.'  # noqa: E501
         ),
-        "user_prompt":
-        "short 5s idea",
+        "user_prompt": "short 5s idea",
     }
 
 
 def test_enhance_prompt_single_clip_rejects_plain_text_response():
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content("Medium shot of a woman by a rainy cafe window as she lifts her "
-                                   "phone, exhales softly, and the camera makes a slow push in."))
+        _chat_payload_with_content(
+            "Medium shot of a woman by a rainy cafe window as she lifts her "
+            "phone, exhales softly, and the camera makes a slow push in."
+        )
+    )
     enhancer.auto_system_prompt = "auto system prompt"
 
     result = asyncio.run(
@@ -743,14 +803,17 @@ def test_enhance_prompt_single_clip_rejects_plain_text_response():
             mode="single_clip",
             model="gpt-test",
             timeout_ms=800,
-        ))
+        )
+    )
     assert result.fallback_used is True
     assert "No JSON object found in assistant response." in result.error
     assert result.prompt == ""
 
 
 def test_enhance_prompt_single_clip_rejects_segment_prompts_json():
-    enhancer = _build_test_enhancer(_chat_payload_with_content('{"segment_prompts":["A","B"]}'))
+    enhancer = _build_test_enhancer(
+        _chat_payload_with_content('{"segment_prompts":["A","B"]}')
+    )
     enhancer.auto_system_prompt = "auto system prompt"
     enhancer.rewrite_model_options = ["gpt-test"]
     enhancer.rewrite_default_model = "gpt-test"
@@ -761,14 +824,17 @@ def test_enhance_prompt_single_clip_rejects_segment_prompts_json():
             mode="single_clip",
             model="gpt-test",
             timeout_ms=800,
-        ))
+        )
+    )
     assert result.fallback_used is True
     assert result.prompt == ""
     assert "Missing prompt string." in (result.error or "")
 
 
 def test_enhance_prompt_requires_json_and_does_not_fallback_to_raw_text():
-    enhancer = _build_test_enhancer(_chat_payload_with_content("A cinematic continuation with slow dolly movement."))
+    enhancer = _build_test_enhancer(
+        _chat_payload_with_content("A cinematic continuation with slow dolly movement.")
+    )
     enhancer.enhance_system_prompt = "enhance system prompt"
     enhancer.rewrite_model_options = ["gpt-test"]
     enhancer.rewrite_default_model = "gpt-test"
@@ -780,14 +846,17 @@ def test_enhance_prompt_requires_json_and_does_not_fallback_to_raw_text():
             next_segment_idx=2,
             model="gpt-test",
             timeout_ms=800,
-        ))
+        )
+    )
     assert result.fallback_used is True
     assert result.prompt == ""
     assert "No JSON object found in assistant response." in (result.error or "")
 
 
 def test_generate_auto_prompt_requires_json_and_does_not_fallback_to_raw_text():
-    enhancer = _build_test_enhancer(_chat_payload_with_content("A calm, grounded continuation with subtle motion."))
+    enhancer = _build_test_enhancer(
+        _chat_payload_with_content("A calm, grounded continuation with subtle motion.")
+    )
     enhancer.auto_system_prompt = "auto system prompt"
     enhancer.rewrite_model_options = ["gpt-test"]
     enhancer.rewrite_default_model = "gpt-test"
@@ -798,30 +867,34 @@ def test_generate_auto_prompt_requires_json_and_does_not_fallback_to_raw_text():
             next_segment_idx=2,
             model="gpt-test",
             timeout_ms=800,
-        ))
+        )
+    )
     assert result.fallback_used is True
     assert result.prompt == ""
     assert "No JSON object found in assistant response." in (result.error or "")
 
 
 def test_rewrite_prompt_sequence_includes_raw_json_when_content_empty():
-    enhancer = _build_test_enhancer({
-        "choices": [{
-            "finish_reason": "length",
-            "message": {
-                "content": [],
-                "refusal": None,
-            },
-        }],
-        "usage": {
-            "completion_tokens": 0
-        },
-    })
+    enhancer = _build_test_enhancer(
+        {
+            "choices": [
+                {
+                    "finish_reason": "length",
+                    "message": {
+                        "content": [],
+                        "refusal": None,
+                    },
+                }
+            ],
+            "usage": {"completion_tokens": 0},
+        }
+    )
     result = asyncio.run(
         enhancer.rewrite_prompt_sequence(
             ["prompt one", "prompt two"],
             rewrite_instruction="make it cinematic",
-        ))
+        )
+    )
     assert result.fallback_used is True
     assert "No rewrite segment prompts found in assistant response." in (result.error or "")
     assert isinstance(result.raw_response_text, str)
@@ -830,7 +903,10 @@ def test_rewrite_prompt_sequence_includes_raw_json_when_content_empty():
 
 def test_get_rewrite_model_config_returns_fixed_defaults():
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     enhancer.rewrite_default_model = "gpt-oss-120b"
     enhancer.rewrite_model_options = ["gpt-oss-120b"]
 
@@ -842,7 +918,10 @@ def test_get_rewrite_model_config_returns_fixed_defaults():
 
 def test_get_prompt_config_includes_auto_extension_prompt():
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     enhancer.enhance_system_prompt_path = "/tmp/next.md"
     enhancer.auto_system_prompt_path = "/tmp/auto.md"
     enhancer.rewrite_all_system_prompt_path = "/tmp/rewrite.md"
@@ -869,14 +948,19 @@ def test_get_prompt_config_includes_auto_extension_prompt():
 
 def test_get_prompt_config_reports_loaded_fallback_prompt_path(tmp_path):
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     rewrite_fallback_path = tmp_path / "rewrite_window_system_prompt.md"
     rewrite_fallback_path.write_text("rewrite prompt\n", encoding="utf-8")
     next_path = tmp_path / "next.md"
     next_path.write_text("next prompt\n", encoding="utf-8")
     auto_path = tmp_path / "auto.md"
     auto_path.write_text("auto prompt\n", encoding="utf-8")
-    enhancer.rewrite_all_system_prompt_path = str(tmp_path / "prompts.local" / "rewrite_window_system_prompt.md")
+    enhancer.rewrite_all_system_prompt_path = str(
+        tmp_path / "prompts.local" / "rewrite_window_system_prompt.md"
+    )
     enhancer.rewrite_all_system_prompt_fallback_path = str(rewrite_fallback_path)
     enhancer.enhance_system_prompt_path = str(next_path)
     enhancer.auto_system_prompt_path = str(auto_path)
@@ -889,9 +973,14 @@ def test_get_prompt_config_reports_loaded_fallback_prompt_path(tmp_path):
     assert config["rewrite_window_system_prompt_path"] == str(rewrite_fallback_path)
 
 
-def test_reload_system_prompts_falls_back_to_rewrite_window_when_user_prompt_empty(tmp_path, ):
+def test_reload_system_prompts_falls_back_to_rewrite_window_when_user_prompt_empty(
+    tmp_path,
+):
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     next_path = tmp_path / "next.md"
     auto_path = tmp_path / "auto.md"
     rewrite_path = tmp_path / "rewrite_window_system_prompt.md"
@@ -918,7 +1007,10 @@ def test_reload_system_prompts_falls_back_to_rewrite_window_when_user_prompt_emp
 
 def test_save_prompt_config_updates_auto_extension_prompt(tmp_path):
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     next_path = tmp_path / "next.md"
     auto_path = tmp_path / "auto.md"
     rewrite_path = tmp_path / "rewrite.md"
@@ -929,7 +1021,9 @@ def test_save_prompt_config_updates_auto_extension_prompt(tmp_path):
     enhancer.auto_system_prompt_path = str(auto_path)
     enhancer.rewrite_all_system_prompt_path = str(rewrite_path)
 
-    config = enhancer.save_prompt_config(auto_extension_system_prompt="auto updated", )
+    config = enhancer.save_prompt_config(
+        auto_extension_system_prompt="auto updated",
+    )
 
     assert auto_path.read_text(encoding="utf-8").strip() == "auto updated"
     assert config["auto_extension_system_prompt"] == "auto updated"
@@ -937,7 +1031,10 @@ def test_save_prompt_config_updates_auto_extension_prompt(tmp_path):
 
 def test_save_prompt_config_updates_rewrite_user_prompt(tmp_path):
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     next_path = tmp_path / "next.md"
     auto_path = tmp_path / "auto.md"
     rewrite_path = tmp_path / "rewrite.md"
@@ -955,7 +1052,9 @@ def test_save_prompt_config_updates_rewrite_user_prompt(tmp_path):
     enhancer.rewrite_all_system_prompt_fallback_path = None
     enhancer.rewrite_user_system_prompt_fallback_path = None
 
-    config = enhancer.save_prompt_config(rewrite_user_system_prompt="rewrite user updated", )
+    config = enhancer.save_prompt_config(
+        rewrite_user_system_prompt="rewrite user updated",
+    )
 
     assert rewrite_user_path.read_text(encoding="utf-8").strip() == "rewrite user updated"
     assert config["rewrite_user_system_prompt"] == "rewrite user updated"
@@ -963,7 +1062,10 @@ def test_save_prompt_config_updates_rewrite_user_prompt(tmp_path):
 
 def test_save_prompt_config_updates_rewrite_model(tmp_path):
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     next_path = tmp_path / "next.md"
     auto_path = tmp_path / "auto.md"
     rewrite_path = tmp_path / "rewrite.md"
@@ -979,7 +1081,9 @@ def test_save_prompt_config_updates_rewrite_model(tmp_path):
     enhancer.rewrite_default_model = "gpt-test"
     enhancer.rewrite_model_options = ["gpt-test", "gpt-alt"]
 
-    config = enhancer.save_prompt_config(rewrite_model="gpt-alt", )
+    config = enhancer.save_prompt_config(
+        rewrite_model="gpt-alt",
+    )
 
     assert enhancer.rewrite_default_model == "gpt-alt"
     assert config["rewrite_model"] == "gpt-alt"
@@ -988,7 +1092,10 @@ def test_save_prompt_config_updates_rewrite_model(tmp_path):
 
 def test_save_prompt_config_updates_rewrite_temperature(tmp_path):
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     next_path = tmp_path / "next.md"
     auto_path = tmp_path / "auto.md"
     rewrite_path = tmp_path / "rewrite.md"
@@ -1002,7 +1109,9 @@ def test_save_prompt_config_updates_rewrite_temperature(tmp_path):
     enhancer.auto_system_prompt_fallback_path = None
     enhancer.rewrite_all_system_prompt_fallback_path = None
 
-    config = enhancer.save_prompt_config(rewrite_temperature=1.3, )
+    config = enhancer.save_prompt_config(
+        rewrite_temperature=1.3,
+    )
 
     assert enhancer.rewrite_default_temperature == 1.3
     assert config["rewrite_temperature"] == 1.3
@@ -1010,7 +1119,10 @@ def test_save_prompt_config_updates_rewrite_temperature(tmp_path):
 
 def test_save_prompt_config_creates_versioned_backup_for_existing_prompt(tmp_path):
     enhancer = _build_test_enhancer(
-        _chat_payload_with_content('{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'))
+        _chat_payload_with_content(
+            '{"id":"preset_a","label":"Preset A","segment_prompts":["A","B"]}'
+        )
+    )
     next_path = tmp_path / "next.md"
     auto_path = tmp_path / "auto.md"
     rewrite_path = tmp_path / "rewrite_window_system_prompt.md"
@@ -1024,9 +1136,13 @@ def test_save_prompt_config_creates_versioned_backup_for_existing_prompt(tmp_pat
     enhancer.auto_system_prompt_fallback_path = None
     enhancer.rewrite_all_system_prompt_fallback_path = None
 
-    enhancer.save_prompt_config(rewrite_window_system_prompt="rewrite updated", )
+    enhancer.save_prompt_config(
+        rewrite_window_system_prompt="rewrite updated",
+    )
 
-    backup_paths = sorted(tmp_path.glob("rewrite_window_system_prompt.*.bak.md"))
+    backup_paths = sorted(
+        tmp_path.glob("rewrite_window_system_prompt.*.bak.md")
+    )
 
     assert rewrite_path.read_text(encoding="utf-8").strip() == "rewrite updated"
     assert len(backup_paths) == 1

apps/dreamverse/dreamverse/tests/test_realtime_stress.py

@@ -27,8 +27,13 @@ try:
 except ModuleNotFoundError:
     websockets = None  # type: ignore[assignment]
 
-DEFAULT_PRESET_FILE = (Path(__file__).resolve().parents[2] / "web" / "prompts" /
-                       "selected_ltx2_continuation_story_presets.json")
+
+DEFAULT_PRESET_FILE = (
+    Path(__file__).resolve().parents[2]
+    / "web"
+    / "prompts"
+    / "selected_ltx2_continuation_story_presets.json"
+)
 
 
 def utc_now_iso() -> str:
@@ -60,7 +65,10 @@ def safe_percentile(values: list[float], percentile: float) -> float | None:
     if lower == upper:
         return sorted_values[lower]
     fraction = rank - lower
-    return (sorted_values[lower] + (sorted_values[upper] - sorted_values[lower]) * fraction)
+    return (
+        sorted_values[lower]
+        + (sorted_values[upper] - sorted_values[lower]) * fraction
+    )
 
 
 def summarize_series(values: list[float]) -> dict[str, float | int | None]:
@@ -137,16 +145,24 @@ def load_curated_prompts(
     selected_id = str(selected.get("id", "")).strip() or "unknown_preset"
     raw_prompts = selected.get("segment_prompts", [])
     if not isinstance(raw_prompts, list):
-        raise ValueError(f"Preset {selected_id} has invalid segment_prompts (must be list).")
+        raise ValueError(
+            f"Preset {selected_id} has invalid segment_prompts (must be list)."
+        )
 
-    prompts = [str(prompt).strip() for prompt in raw_prompts if isinstance(prompt, str) and str(prompt).strip()]
+    prompts = [
+        str(prompt).strip()
+        for prompt in raw_prompts
+        if isinstance(prompt, str) and str(prompt).strip()
+    ]
     if not prompts:
         raise ValueError(f"Preset {selected_id} has no non-empty prompts.")
 
     limited = prompts[:curated_limit]
     if not limited:
-        raise ValueError(f"curated_limit={curated_limit} produced no prompts for preset "
-                         f"{selected_id}.")
+        raise ValueError(
+            f"curated_limit={curated_limit} produced no prompts for preset "
+            f"{selected_id}."
+        )
     return selected_id, limited, len(prompts)
 
 
@@ -208,11 +224,11 @@ async def run_single_session(
 
     try:
         async with websockets.connect(
-                url,
-                max_size=None,
-                ping_interval=None,
-                open_timeout=connect_timeout_s,
-                close_timeout=2.0,
+            url,
+            max_size=None,
+            ping_interval=None,
+            open_timeout=connect_timeout_s,
+            close_timeout=2.0,
         ) as ws:
             connect_finish_monotonic = time.monotonic()
             session_data["connect_finish_ts_utc"] = utc_now_iso()
@@ -233,7 +249,9 @@ async def run_single_session(
                 timeout_remaining = session_timeout_s - elapsed_s
                 if timeout_remaining <= 0:
                     session_data["status"] = "timeout"
-                    session_data["error"] = (f"Session timed out after {session_timeout_s:.1f}s.")
+                    session_data["error"] = (
+                        f"Session timed out after {session_timeout_s:.1f}s."
+                    )
                     break
 
                 recv_start_epoch = time.time()
@@ -247,7 +265,9 @@ async def run_single_session(
                     )
                 except asyncio.TimeoutError:
                     session_data["status"] = "timeout"
-                    session_data["error"] = ("Timed out waiting for websocket message.")
+                    session_data["error"] = (
+                        "Timed out waiting for websocket message."
+                    )
                     break
                 except Exception as exc:
                     session_data["status"] = "failed"
@@ -268,16 +288,20 @@ async def run_single_session(
 
                     chunk_gap_ms: float | None = None
                     if last_chunk_finish_monotonic is not None:
-                        chunk_gap_ms = (recv_finish_monotonic - last_chunk_finish_monotonic) * 1000.0
+                        chunk_gap_ms = (
+                            recv_finish_monotonic - last_chunk_finish_monotonic
+                        ) * 1000.0
 
-                    session_data["chunks"].append({
-                        "segment_idx": current_segment_idx,
-                        "chunk_idx": session_data["total_chunks"],
-                        "size_bytes": len(message),
-                        "chunk_start_ts_utc": recv_start_iso,
-                        "chunk_finish_ts_utc": recv_finish_iso,
-                        "chunk_gap_ms": chunk_gap_ms,
-                    })
+                    session_data["chunks"].append(
+                        {
+                            "segment_idx": current_segment_idx,
+                            "chunk_idx": session_data["total_chunks"],
+                            "size_bytes": len(message),
+                            "chunk_start_ts_utc": recv_start_iso,
+                            "chunk_finish_ts_utc": recv_finish_iso,
+                            "chunk_gap_ms": chunk_gap_ms,
+                        }
+                    )
                     last_chunk_finish_monotonic = recv_finish_monotonic
                     last_chunk_finish_epoch = recv_finish_epoch
                     session_data["last_chunk_finish_ts_utc"] = recv_finish_iso
@@ -297,7 +321,9 @@ async def run_single_session(
                 if msg_type == "gpu_assigned":
                     session_data["gpu_assigned_ts_utc"] = recv_finish_iso
                     if connect_finish_monotonic is not None:
-                        session_data["queue_wait_ms"] = (recv_finish_monotonic - connect_finish_monotonic) * 1000.0
+                        session_data["queue_wait_ms"] = (
+                            recv_finish_monotonic - connect_finish_monotonic
+                        ) * 1000.0
                 elif msg_type == "ltx2_stream_start":
                     if initial_total_segments is None:
                         parsed_total = parse_int(data.get("total_segments"))
@@ -312,13 +338,20 @@ async def run_single_session(
                     session_data["media_segments_completed"] += 1
                     if first_media_segment_complete_epoch is None:
                         first_media_segment_complete_epoch = recv_finish_epoch
-                        session_data["first_media_segment_complete_ts_utc"] = recv_finish_iso
+                        session_data[
+                            "first_media_segment_complete_ts_utc"
+                        ] = recv_finish_iso
                 elif msg_type == "ltx2_segment_complete":
                     session_data["segments_completed"] += 1
                     seg_idx = parse_int(data.get("segment_idx"))
-                    if (initial_total_segments is not None and seg_idx is not None
-                            and seg_idx >= initial_total_segments):
-                        session_data["target_segment_complete_ts_utc"] = recv_finish_iso
+                    if (
+                        initial_total_segments is not None
+                        and seg_idx is not None
+                        and seg_idx >= initial_total_segments
+                    ):
+                        session_data[
+                            "target_segment_complete_ts_utc"
+                        ] = recv_finish_iso
                         await asyncio.sleep(post_complete_wait_s)
                         session_data["leave_sent_ts_utc"] = utc_now_iso()
                         try:
@@ -329,11 +362,15 @@ async def run_single_session(
                         break
                 elif msg_type == "session_timeout":
                     session_data["status"] = "timeout"
-                    session_data["error"] = str(data.get("message") or "Backend session timeout")
+                    session_data["error"] = str(
+                        data.get("message") or "Backend session timeout"
+                    )
                     break
                 elif msg_type == "error":
                     session_data["status"] = "failed"
-                    session_data["error"] = str(data.get("message") or "Backend error message")
+                    session_data["error"] = str(
+                        data.get("message") or "Backend error message"
+                    )
                     break
 
             if session_data["status"] == "failed" and session_data["error"] is None:
@@ -342,18 +379,29 @@ async def run_single_session(
         session_data["status"] = "failed"
         session_data["error"] = f"WebSocket connect/run failed: {exc}"
 
-    if (first_chunk_finish_epoch is not None and last_chunk_finish_epoch is not None
-            and session_data["total_chunk_bytes"] > 0):
+    if (
+        first_chunk_finish_epoch is not None
+        and last_chunk_finish_epoch is not None
+        and session_data["total_chunk_bytes"] > 0
+    ):
         duration_s = last_chunk_finish_epoch - first_chunk_finish_epoch
         if duration_s > 0:
-            session_data["session_goodput_mbps"] = (session_data["total_chunk_bytes"] * 8.0 / duration_s / 1_000_000.0)
+            session_data["session_goodput_mbps"] = (
+                session_data["total_chunk_bytes"] * 8.0 / duration_s / 1_000_000.0
+            )
 
-    if (first_chunk_finish_epoch is not None and first_media_segment_complete_epoch is not None):
-        session_data["first_chunk_before_first_media_complete"] = (first_chunk_finish_epoch
-                                                                   < first_media_segment_complete_epoch)
+    if (
+        first_chunk_finish_epoch is not None
+        and first_media_segment_complete_epoch is not None
+    ):
+        session_data["first_chunk_before_first_media_complete"] = (
+            first_chunk_finish_epoch < first_media_segment_complete_epoch
+        )
 
     session_data["close_ts_utc"] = utc_now_iso()
-    session_data["duration_ms"] = (time.monotonic() - session_start_monotonic) * 1000.0
+    session_data["duration_ms"] = (
+        time.monotonic() - session_start_monotonic
+    ) * 1000.0
     return session_data
 
 
@@ -364,11 +412,14 @@ async def run_worker_sessions(
     config: dict[str, Any],
 ) -> list[dict[str, Any]]:
     tasks = [
-        asyncio.create_task(run_single_session(
-            worker_id=worker_id,
-            worker_session_idx=idx,
-            config=config,
-        )) for idx in range(session_count)
+        asyncio.create_task(
+            run_single_session(
+                worker_id=worker_id,
+                worker_session_idx=idx,
+                config=config,
+            )
+        )
+        for idx in range(session_count)
     ]
     if not tasks:
         return []
@@ -386,23 +437,29 @@ def worker_entry(
     try:
         ready_queue.put({"worker_id": worker_id, "status": "ready"})
         start_event.wait()
-        sessions = asyncio.run(run_worker_sessions(
-            worker_id=worker_id,
-            session_count=session_count,
-            config=config,
-        ))
-        result_queue.put({
-            "worker_id": worker_id,
-            "status": "ok",
-            "sessions": sessions,
-        })
+        sessions = asyncio.run(
+            run_worker_sessions(
+                worker_id=worker_id,
+                session_count=session_count,
+                config=config,
+            )
+        )
+        result_queue.put(
+            {
+                "worker_id": worker_id,
+                "status": "ok",
+                "sessions": sessions,
+            }
+        )
     except Exception as exc:
-        result_queue.put({
-            "worker_id": worker_id,
-            "status": "error",
-            "error": str(exc),
-            "traceback": traceback.format_exc(),
-        })
+        result_queue.put(
+            {
+                "worker_id": worker_id,
+                "status": "error",
+                "error": str(exc),
+                "traceback": traceback.format_exc(),
+            }
+        )
 
 
 def build_summary(
@@ -460,22 +517,33 @@ def build_summary(
     if len(all_chunk_finish_epochs) >= 2 and total_chunk_bytes > 0:
         duration_s = max(all_chunk_finish_epochs) - min(all_chunk_finish_epochs)
         if duration_s > 0:
-            global_goodput_mbps = (total_chunk_bytes * 8.0 / duration_s / 1_000_000.0)
+            global_goodput_mbps = (
+                total_chunk_bytes * 8.0 / duration_s / 1_000_000.0
+            )
 
-    bucket_throughputs_mbps = [(bytes_count * 8.0) / 1_000_000.0 for _, bytes_count in sorted(bucket_bytes.items())]
+    bucket_throughputs_mbps = [
+        (bytes_count * 8.0) / 1_000_000.0
+        for _, bytes_count in sorted(bucket_bytes.items())
+    ]
     bucket_stats = summarize_series(bucket_throughputs_mbps)
 
-    chunk_gap_threshold_breaches = [value for value in chunk_gaps if value >= chunk_gap_threshold_ms]
+    chunk_gap_threshold_breaches = [
+        value for value in chunk_gaps if value >= chunk_gap_threshold_ms
+    ]
     non_success = len(sessions) - status_counts.get("success", 0)
 
     fail_reasons: list[str] = []
     if non_success > 0:
-        fail_reasons.append(f"{non_success} session(s) did not complete successfully.")
+        fail_reasons.append(
+            f"{non_success} session(s) did not complete successfully."
+        )
     if not chunk_gaps:
         fail_reasons.append("No chunk gap data collected.")
     if chunk_gap_threshold_breaches:
-        fail_reasons.append(f"{len(chunk_gap_threshold_breaches)} chunk gap(s) were >= "
-                            f"{chunk_gap_threshold_ms:.0f}ms.")
+        fail_reasons.append(
+            f"{len(chunk_gap_threshold_breaches)} chunk gap(s) were >= "
+            f"{chunk_gap_threshold_ms:.0f}ms."
+        )
 
     passed = len(fail_reasons) == 0
     progressive_ratio = None
@@ -486,18 +554,20 @@ def build_summary(
         "passed": passed,
         "fail_reasons": fail_reasons,
         "sessions": {
-            "total":
-            len(sessions),
-            "success":
-            status_counts.get("success", 0),
-            "failed":
-            status_counts.get("failed", 0),
-            "timeout":
-            status_counts.get("timeout", 0),
-            "protocol_error":
-            status_counts.get("protocol_error", 0),
-            "other": (len(sessions) - (status_counts.get("success", 0) + status_counts.get("failed", 0) +
-                                       status_counts.get("timeout", 0) + status_counts.get("protocol_error", 0))),
+            "total": len(sessions),
+            "success": status_counts.get("success", 0),
+            "failed": status_counts.get("failed", 0),
+            "timeout": status_counts.get("timeout", 0),
+            "protocol_error": status_counts.get("protocol_error", 0),
+            "other": (
+                len(sessions)
+                - (
+                    status_counts.get("success", 0)
+                    + status_counts.get("failed", 0)
+                    + status_counts.get("timeout", 0)
+                    + status_counts.get("protocol_error", 0)
+                )
+            ),
         },
         "chunk_gap_ms": {
             **chunk_gap_stats,
@@ -536,39 +606,51 @@ def print_summary(
     bucket_bw = bandwidth["bucketed_1s"]
 
     print("=== LTX2 Realtime Stress Test Summary ===")
-    print("Run: "
-          f"url={run_info['url']} clients={run_info['clients']} "
-          f"processes={run_info['processes']} "
-          f"preset={run_info['preset_id']} "
-          f"curated_limit={run_info['curated_limit']}")
-    print("Sessions: "
-          f"total={sessions['total']} success={sessions['success']} "
-          f"failed={sessions['failed']} timeout={sessions['timeout']} "
-          f"protocol_error={sessions['protocol_error']}")
-    print("Chunk gap ms: "
-          f"min={format_num(chunk_gap['min'])} "
-          f"p50={format_num(chunk_gap['p50'])} "
-          f"p95={format_num(chunk_gap['p95'])} "
-          f"p99={format_num(chunk_gap['p99'])} "
-          f"max={format_num(chunk_gap['max'])} "
-          f"threshold={format_num(chunk_gap['threshold_ms'])} "
-          f"breaches={chunk_gap['breach_count']}")
-    print("Queue wait ms: "
-          f"min={format_num(queue_wait['min'])} "
-          f"p50={format_num(queue_wait['p50'])} "
-          f"p95={format_num(queue_wait['p95'])} "
-          f"max={format_num(queue_wait['max'])}")
+    print(
+        "Run: "
+        f"url={run_info['url']} clients={run_info['clients']} "
+        f"processes={run_info['processes']} "
+        f"preset={run_info['preset_id']} "
+        f"curated_limit={run_info['curated_limit']}"
+    )
+    print(
+        "Sessions: "
+        f"total={sessions['total']} success={sessions['success']} "
+        f"failed={sessions['failed']} timeout={sessions['timeout']} "
+        f"protocol_error={sessions['protocol_error']}"
+    )
+    print(
+        "Chunk gap ms: "
+        f"min={format_num(chunk_gap['min'])} "
+        f"p50={format_num(chunk_gap['p50'])} "
+        f"p95={format_num(chunk_gap['p95'])} "
+        f"p99={format_num(chunk_gap['p99'])} "
+        f"max={format_num(chunk_gap['max'])} "
+        f"threshold={format_num(chunk_gap['threshold_ms'])} "
+        f"breaches={chunk_gap['breach_count']}"
+    )
+    print(
+        "Queue wait ms: "
+        f"min={format_num(queue_wait['min'])} "
+        f"p50={format_num(queue_wait['p50'])} "
+        f"p95={format_num(queue_wait['p95'])} "
+        f"max={format_num(queue_wait['max'])}"
+    )
     ratio = progressive["ratio"]
     ratio_text = "n/a" if ratio is None else f"{ratio * 100:.2f}%"
-    print("Progressive streaming: "
-          f"{progressive['success_sessions']}/"
-          f"{progressive['eligible_sessions']} ({ratio_text})")
-    print("Bandwidth Mbps: "
-          f"per_session_avg={format_num(per_session_bw['avg'])} "
-          f"per_session_p95={format_num(per_session_bw['p95'])} "
-          f"global={format_num(bandwidth['global_goodput_mbps'])} "
-          f"bucket_avg={format_num(bucket_bw['avg_mbps'])} "
-          f"bucket_peak={format_num(bucket_bw['peak_mbps'])}")
+    print(
+        "Progressive streaming: "
+        f"{progressive['success_sessions']}/"
+        f"{progressive['eligible_sessions']} ({ratio_text})"
+    )
+    print(
+        "Bandwidth Mbps: "
+        f"per_session_avg={format_num(per_session_bw['avg'])} "
+        f"per_session_p95={format_num(per_session_bw['p95'])} "
+        f"global={format_num(bandwidth['global_goodput_mbps'])} "
+        f"bucket_avg={format_num(bucket_bw['avg_mbps'])} "
+        f"bucket_peak={format_num(bucket_bw['peak_mbps'])}"
+    )
     print(f"VERDICT: {'PASS' if summary['passed'] else 'FAIL'}")
     if summary["fail_reasons"]:
         print("Fail reasons:")
@@ -588,8 +670,10 @@ def distribute_sessions(total_clients: int, process_count: int) -> list[int]:
 
 def run_stress(args: argparse.Namespace) -> tuple[dict[str, Any], int]:
     if websockets is None:
-        raise RuntimeError("Missing dependency: websockets. Install it before running this "
-                           "stress test.")
+        raise RuntimeError(
+            "Missing dependency: websockets. Install it before running this "
+            "stress test."
+        )
 
     preset_file = Path(args.preset_file).expanduser().resolve()
     selected_preset_id, curated_prompts, total_prompt_count = load_curated_prompts(
@@ -651,8 +735,13 @@ def run_stress(args: argparse.Namespace) -> tuple[dict[str, Any], int]:
 
         start_event.set()
 
-        result_deadline = (time.monotonic() + args.connect_timeout_s + args.session_timeout_s +
-                           args.post_complete_wait_s + 180.0)
+        result_deadline = (
+            time.monotonic()
+            + args.connect_timeout_s
+            + args.session_timeout_s
+            + args.post_complete_wait_s
+            + 180.0
+        )
         worker_results: list[dict[str, Any]] = []
         while len(worker_results) < len(processes):
             timeout_s = max(0.1, result_deadline - time.monotonic())
@@ -676,20 +765,24 @@ def run_stress(args: argparse.Namespace) -> tuple[dict[str, Any], int]:
             if result.get("status") == "ok":
                 sessions.extend(result.get("sessions", []))
             else:
-                worker_errors.append({
-                    "worker_id": result.get("worker_id"),
-                    "error": result.get("error"),
-                    "traceback": result.get("traceback"),
-                })
+                worker_errors.append(
+                    {
+                        "worker_id": result.get("worker_id"),
+                        "error": result.get("error"),
+                        "traceback": result.get("traceback"),
+                    }
+                )
 
         received_workers = {result.get("worker_id") for result in worker_results}
         expected_workers = set(range(len(processes)))
         missing_workers = sorted(expected_workers - received_workers)
         for worker_id in missing_workers:
-            worker_errors.append({
-                "worker_id": worker_id,
-                "error": "No worker result received.",
-            })
+            worker_errors.append(
+                {
+                    "worker_id": worker_id,
+                    "error": "No worker result received.",
+                }
+            )
 
         run_end_epoch = time.time()
         run_end_iso = iso_from_epoch(run_end_epoch)
@@ -702,8 +795,9 @@ def run_stress(args: argparse.Namespace) -> tuple[dict[str, Any], int]:
 
         if worker_errors:
             summary["passed"] = False
-            summary["fail_reasons"] = list(
-                summary["fail_reasons"]) + [f"{len(worker_errors)} worker error(s) occurred."]
+            summary["fail_reasons"] = list(summary["fail_reasons"]) + [
+                f"{len(worker_errors)} worker error(s) occurred."
+            ]
 
         output_payload = {
             "run_info": {
@@ -739,7 +833,9 @@ def run_stress(args: argparse.Namespace) -> tuple[dict[str, Any], int]:
 
 
 def parse_args() -> argparse.Namespace:
-    parser = argparse.ArgumentParser(description="Multiprocess realtime stress test for LTX2 streaming.", )
+    parser = argparse.ArgumentParser(
+        description="Multiprocess realtime stress test for LTX2 streaming.",
+    )
     parser.add_argument(
         "-u",
         "--url",

apps/dreamverse/dreamverse/tests/test_session_init_image.py

@@ -47,11 +47,13 @@ def test_persist_session_init_image_returns_none_when_missing_data():
 
 def test_persist_session_init_image_rejects_unsupported_mime():
     with pytest.raises(ValueError, match="PNG, JPEG, or WebP"):
-        persist_session_init_image({
-            "name": "frame.gif",
-            "mime_type": "image/gif",
-            "data_url": "data:image/gif;base64,R0lGODlhAQABAAAAACw=",
-        })
+        persist_session_init_image(
+            {
+                "name": "frame.gif",
+                "mime_type": "image/gif",
+                "data_url": "data:image/gif;base64,R0lGODlhAQABAAAAACw=",
+            }
+        )
 
 
 def test_persist_session_init_image_rejects_large_payload(monkeypatch):
@@ -64,8 +66,10 @@ def test_persist_session_init_image_rejects_large_payload(monkeypatch):
     monkeypatch.setattr(base64, "b64decode", fake_b64decode)
 
     with pytest.raises(ValueError, match="15 MB or smaller"):
-        persist_session_init_image({
-            "name": "frame.png",
-            "mime_type": "image/png",
-            "data_url": data_url,
-        })
+        persist_session_init_image(
+            {
+                "name": "frame.png",
+                "mime_type": "image/png",
+                "data_url": data_url,
+            }
+        )

apps/dreamverse/scripts/modal/modal_app.py

@@ -8,10 +8,12 @@ import modal
 
 IMAGE = os.environ.get("DREAMVERSE_IMAGE")
 if not IMAGE:
-    raise RuntimeError("DREAMVERSE_IMAGE is required. Set it to a published SHA-specific Dreamverse image, "
-                       "for example a dreamverse-backend-cuda13.0.0-sha-* tag or a "
-                       "dreamverse-ui-cuda13.0.0-sha-* tag if serving the static UI. "
-                       "CUDA 12 / cu126 images use the corresponding cuda12.6.3 tag.")
+    raise RuntimeError(
+        "DREAMVERSE_IMAGE is required. Set it to a published SHA-specific Dreamverse image, "
+        "for example a dreamverse-backend-cuda13.0.0-sha-* tag or a "
+        "dreamverse-ui-cuda13.0.0-sha-* tag if serving the static UI. "
+        "CUDA 12 / cu126 images use the corresponding cuda12.6.3 tag."
+    )
 
 # ``@modal.web_server`` invokes ``serve()`` directly and bypasses the image
 # ENTRYPOINT (``docker/docker_entrypoint.sh``).  That entrypoint normally
@@ -63,10 +65,14 @@ def serve():
     # ``or ""`` collapses ``None`` (unset) into an empty string, ``.strip()``
     # collapses whitespace-only values (e.g. ``"   "``) — both should be
     # treated as missing.
-    missing = [k for k in _REQUIRED_SECRET_KEYS if not (os.environ.get(k) or "").strip()]
+    missing = [
+        k for k in _REQUIRED_SECRET_KEYS
+        if not (os.environ.get(k) or "").strip()
+    ]
     if missing:
-        raise RuntimeError("dreamverse-api-keys secret is missing required entries: "
-                           f"{', '.join(missing)}. Add them with `modal secret create "
-                           "dreamverse-api-keys ... --force` and redeploy "
-                           "(see apps/dreamverse/scripts/modal/README.md).")
+        raise RuntimeError(
+            "dreamverse-api-keys secret is missing required entries: "
+            f"{', '.join(missing)}. Add them with `modal secret create "
+            "dreamverse-api-keys ... --force` and redeploy "
+            "(see apps/dreamverse/scripts/modal/README.md).")
     subprocess.Popen(["dreamverse-server", "--host", "0.0.0.0", "--port", "8009"])

apps/fastvideo_studio/e2e/create-job.spec.ts

@@ -13,9 +13,10 @@ test.describe('create inference job', () => {
   test('creates a T2V job and shows it in the queue', async ({ page }) => {
     await page.goto('/inference');
 
-    // The trigger opens a real menu on click, so this path works for touch,
-    // mouse, and keyboard users.
-    await page.getByRole('button', { name: /create job/i }).click();
+    // The "Create Job" button reveals a workload menu on hover; wait for the
+    // T2V item to become visible before clicking so the CSS hover transition
+    // can't race the click.
+    await page.getByRole('button', { name: /create job/i }).hover();
     const t2vItem = page.getByRole('menuitem', { name: /T2V/i });
     await expect(t2vItem).toBeVisible();
     await t2vItem.click();

apps/fastvideo_studio/e2e/gallery.spec.ts

@@ -4,7 +4,7 @@ import { API_BASE, skipWithoutMock } from './helpers';
 
 /**
  * Gallery page: the seeded completed inference job surfaces as a media tile
- * with playback controls or an explicit media-error fallback.
+ * (an <article> wrapping a <video>) captioned with its prompt.
  */
 test.describe('gallery', () => {
   skipWithoutMock();
@@ -30,15 +30,12 @@ test.describe('gallery', () => {
       page.getByRole('heading', { level: 1, name: 'Gallery' }),
     ).toBeVisible();
 
-    const tile = page.locator('article').filter({ hasText: completed!.prompt });
-    await expect(tile).toBeVisible();
-    await expect(
-      tile.locator('video').or(tile.getByText('Preview unavailable')),
-    ).toBeVisible();
+    // The completed job renders as an <article> containing a <video> tile.
+    const tile = page
+      .locator('article')
+      .filter({ has: page.locator('video') });
+    await expect(tile.first()).toBeVisible();
 
-    const video = tile.locator('video');
-    if (await video.isVisible()) {
-      await expect(video).toHaveAttribute('controls', '');
-    }
+    await expect(page.getByText(completed!.prompt)).toBeVisible();
   });
 });

apps/fastvideo_studio/e2e/shell.spec.ts

@@ -42,74 +42,6 @@ test.describe('app shell', () => {
       await expect(
         page.getByRole('heading', { level: 1, name: section.title }),
       ).toBeVisible();
-      await expect(page.getByRole('main')).toHaveCount(1);
     }
   });
-
-  test('keeps navigation and content usable at responsive breakpoints', async ({
-    page,
-  }) => {
-    for (const width of [320, 375, 414, 768]) {
-      await page.setViewportSize({ width, height: 800 });
-      await page.goto('/inference');
-
-      const main = page.getByRole('main');
-      await expect(main).toBeVisible();
-      await expect(
-        page.getByRole('button', { name: /Create Job/i }),
-      ).toBeVisible();
-
-      const initialBox = await main.boundingBox();
-      expect(initialBox?.x).toBe(width < 768 ? 0 : 220);
-      expect(initialBox?.width).toBe(width < 768 ? width : width - 220);
-
-      const navigation = page.getByRole('navigation', {
-        name: 'Primary navigation',
-      });
-      if (width < 768) {
-        await expect(
-          page.getByRole('button', { name: 'Open navigation' }),
-        ).toBeVisible();
-        await page.getByRole('button', { name: 'Open navigation' }).click();
-      }
-      await expect(navigation).toBeVisible();
-      await navigation.getByRole('link', { name: 'Datasets' }).click();
-
-      await expect(page).toHaveURL(/\/datasets$/);
-      expect(
-        await page.evaluate(
-          () => document.documentElement.scrollWidth <= window.innerWidth,
-        ),
-      ).toBe(true);
-    }
-  });
-
-  test('uses full-width detail drawers on mobile', async ({ page }) => {
-    await page.setViewportSize({ width: 320, height: 800 });
-    await page.goto('/inference');
-
-    await page
-      .locator('article button[aria-pressed="false"]')
-      .first()
-      .click();
-    const jobDrawer = page.getByRole('dialog', { name: 'Job details' });
-    await expect(jobDrawer).toBeVisible();
-    expect(await jobDrawer.boundingBox()).toMatchObject({ x: 0, width: 320 });
-    await jobDrawer.getByRole('button', { name: 'Close' }).click();
-
-    await page.goto('/datasets');
-    await page
-      .locator('article button[aria-pressed="false"]')
-      .first()
-      .click();
-
-    const datasetDrawer = page.getByRole('dialog', {
-      name: /dataset details$/,
-    });
-    await expect(datasetDrawer).toBeVisible();
-    expect(await datasetDrawer.boundingBox()).toMatchObject({
-      x: 0,
-      width: 320,
-    });
-  });
 });

apps/fastvideo_studio/package-lock.json

@@ -9,7 +9,6 @@
 			"version": "0.1.0",
 			"dependencies": {
 				"@radix-ui/react-dialog": "^1.1.0",
-				"@radix-ui/react-dropdown-menu": "^2.1.24",
 				"@radix-ui/react-label": "^2.1.8",
 				"@radix-ui/react-scroll-area": "^1.2.10",
 				"@radix-ui/react-select": "^2.2.6",
@@ -1937,183 +1936,6 @@
 				}
 			}
 		},
-		"node_modules/@radix-ui/react-dropdown-menu": {
-			"version": "2.1.24",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-dropdown-menu/-/react-dropdown-menu-2.1.24.tgz",
-			"integrity": "sha512-geq8l2rJkxvkXsT9RMgtUE3P8pITFpTsvYpbySi1IH4fZEABD/Gp85myayFgxk0ktljGMJnCbeFkyTusvSvv7g==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/primitive": "1.1.7",
-				"@radix-ui/react-compose-refs": "1.1.5",
-				"@radix-ui/react-context": "1.2.2",
-				"@radix-ui/react-id": "1.1.4",
-				"@radix-ui/react-menu": "2.1.24",
-				"@radix-ui/react-primitive": "2.1.10",
-				"@radix-ui/react-use-controllable-state": "1.2.6"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-dropdown-menu/node_modules/@radix-ui/primitive": {
-			"version": "1.1.7",
-			"resolved": "https://registry.npmjs.org/@radix-ui/primitive/-/primitive-1.1.7.tgz",
-			"integrity": "sha512-rqWnm76nYT8HoNNqEjpgJ7Pw/DrBj5iBTrmEPo6HTX5+VJyBNOqTdv4g89G63HuR5g0AaENoAcH7Is5fF2kZ8Q==",
-			"license": "MIT"
-		},
-		"node_modules/@radix-ui/react-dropdown-menu/node_modules/@radix-ui/react-compose-refs": {
-			"version": "1.1.5",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-compose-refs/-/react-compose-refs-1.1.5.tgz",
-			"integrity": "sha512-+48PbAAbq3didjJxa+OaWY2ZwgAKsNiRGyeHKszblZMQ+kcpd9pAaT11cMkGEie0vsOi3QdeTE6d5Fe3Gn61kA==",
-			"license": "MIT",
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-dropdown-menu/node_modules/@radix-ui/react-context": {
-			"version": "1.2.2",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-context/-/react-context-1.2.2.tgz",
-			"integrity": "sha512-RHCUGwKHDr0hDGg4X7ma4JG4/+12qxw8rkh5QKdDldlCvtja6nUx1Ef/8HVrJze81lEsgLQlqjzjGNHantgnQA==",
-			"license": "MIT",
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-dropdown-menu/node_modules/@radix-ui/react-id": {
-			"version": "1.1.4",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-id/-/react-id-1.1.4.tgz",
-			"integrity": "sha512-TMQp2llA+RYn7JcjnrMnz7wN4pcVttPZnRZo52PLQsoLVKzNlVwUeHmfePgTgRluXFvlD3GD5g5MOVVTJCO0qA==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-use-layout-effect": "1.1.4"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-dropdown-menu/node_modules/@radix-ui/react-primitive": {
-			"version": "2.1.10",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-primitive/-/react-primitive-2.1.10.tgz",
-			"integrity": "sha512-MucOnzh6hR5mid6VpkbglRAMYMjKLqRnGBbjXkzjK52fuQDd1qbkx78a5P40mkcnVXJdEVxm26E9OPAiUq7nBg==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-slot": "1.3.3"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-dropdown-menu/node_modules/@radix-ui/react-slot": {
-			"version": "1.3.3",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-slot/-/react-slot-1.3.3.tgz",
-			"integrity": "sha512-qx7oqnYbxnK9kYI9m317qmFmEgo6ywqWvbTogdj7cL9p3/yx4M48p7Rnw5z3H890cL/ow/EeWJsuTykeZVXP5Q==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-compose-refs": "1.1.5"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-dropdown-menu/node_modules/@radix-ui/react-use-controllable-state": {
-			"version": "1.2.6",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-use-controllable-state/-/react-use-controllable-state-1.2.6.tgz",
-			"integrity": "sha512-uEQJGT97ZA/TgP/Hydw47lHu+/vQj6z/0jA+WeTbK1o9Rx45GImjpD0tc3W5ad3D6XTSR6e1yEO0FvGq6WQfVQ==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/primitive": "1.1.7",
-				"@radix-ui/react-use-effect-event": "0.0.5",
-				"@radix-ui/react-use-layout-effect": "1.1.4"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-dropdown-menu/node_modules/@radix-ui/react-use-effect-event": {
-			"version": "0.0.5",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-use-effect-event/-/react-use-effect-event-0.0.5.tgz",
-			"integrity": "sha512-7cshFL8HGS/7HEiHH+9kL9HBwp2sa9yX18Knwek6KYWmXwM7pegMgta2AXMQKI+rq3JnfSj9x8wYqFMTdG1Jgg==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-use-layout-effect": "1.1.4"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-dropdown-menu/node_modules/@radix-ui/react-use-layout-effect": {
-			"version": "1.1.4",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-use-layout-effect/-/react-use-layout-effect-1.1.4.tgz",
-			"integrity": "sha512-K20DkRkUwDnxEYMBPcg3Y6voLkEy5p5QQmszZgLngKKiC7dzBR/aEuK3w1qlx2JWDUNH6FluahYdgR3BP+QbYw==",
-			"license": "MIT",
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
 		"node_modules/@radix-ui/react-focus-guards": {
 			"version": "1.1.4",
 			"resolved": "https://registry.npmjs.org/@radix-ui/react-focus-guards/-/react-focus-guards-1.1.4.tgz",
@@ -2195,494 +2017,6 @@
 				}
 			}
 		},
-		"node_modules/@radix-ui/react-menu": {
-			"version": "2.1.24",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-menu/-/react-menu-2.1.24.tgz",
-			"integrity": "sha512-uW7RVuU6Lp/ZtfeY4b3kL32zccgEWvPv1+cf17ubYzHa9cL8AHokmk36cG/XEiH/smbQvumnieXX9j/e9RqJWA==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/primitive": "1.1.7",
-				"@radix-ui/react-collection": "1.1.15",
-				"@radix-ui/react-compose-refs": "1.1.5",
-				"@radix-ui/react-context": "1.2.2",
-				"@radix-ui/react-direction": "1.1.4",
-				"@radix-ui/react-dismissable-layer": "1.1.19",
-				"@radix-ui/react-focus-guards": "1.1.6",
-				"@radix-ui/react-focus-scope": "1.1.16",
-				"@radix-ui/react-id": "1.1.4",
-				"@radix-ui/react-popper": "1.3.7",
-				"@radix-ui/react-portal": "1.1.17",
-				"@radix-ui/react-presence": "1.1.10",
-				"@radix-ui/react-primitive": "2.1.10",
-				"@radix-ui/react-roving-focus": "1.1.19",
-				"@radix-ui/react-slot": "1.3.3",
-				"@radix-ui/react-use-callback-ref": "1.1.4",
-				"aria-hidden": "^1.2.4",
-				"react-remove-scroll": "^2.7.2"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/primitive": {
-			"version": "1.1.7",
-			"resolved": "https://registry.npmjs.org/@radix-ui/primitive/-/primitive-1.1.7.tgz",
-			"integrity": "sha512-rqWnm76nYT8HoNNqEjpgJ7Pw/DrBj5iBTrmEPo6HTX5+VJyBNOqTdv4g89G63HuR5g0AaENoAcH7Is5fF2kZ8Q==",
-			"license": "MIT"
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-arrow": {
-			"version": "1.1.15",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-arrow/-/react-arrow-1.1.15.tgz",
-			"integrity": "sha512-v4zggRcjadnI+ClKDuijlQEW4tw3NoaeHc/PwpKnLoLLKNUG4InLegkstooLcRIUWCs+8L22dGURCVuFfOKfnA==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-primitive": "2.1.10"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-collection": {
-			"version": "1.1.15",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-collection/-/react-collection-1.1.15.tgz",
-			"integrity": "sha512-9W+B9NPF0NaaPh/1NJd3+KqsnlLqU9H7T2rvww+fp+T/evVXdNAyYcnfRQZFOjkR1ajQp3yORlqnI8soawLvNA==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-compose-refs": "1.1.5",
-				"@radix-ui/react-context": "1.2.2",
-				"@radix-ui/react-primitive": "2.1.10",
-				"@radix-ui/react-slot": "1.3.3"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-compose-refs": {
-			"version": "1.1.5",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-compose-refs/-/react-compose-refs-1.1.5.tgz",
-			"integrity": "sha512-+48PbAAbq3didjJxa+OaWY2ZwgAKsNiRGyeHKszblZMQ+kcpd9pAaT11cMkGEie0vsOi3QdeTE6d5Fe3Gn61kA==",
-			"license": "MIT",
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-context": {
-			"version": "1.2.2",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-context/-/react-context-1.2.2.tgz",
-			"integrity": "sha512-RHCUGwKHDr0hDGg4X7ma4JG4/+12qxw8rkh5QKdDldlCvtja6nUx1Ef/8HVrJze81lEsgLQlqjzjGNHantgnQA==",
-			"license": "MIT",
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-direction": {
-			"version": "1.1.4",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-direction/-/react-direction-1.1.4.tgz",
-			"integrity": "sha512-5pzg4FGQNpExhnhT2zlrP1wZFaYCd1K0nYWoFAdcYoYK868IEigqMX3B3f8yIoRlAhAeDWciLI6ZdCKHF9P4Vg==",
-			"license": "MIT",
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-dismissable-layer": {
-			"version": "1.1.19",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-dismissable-layer/-/react-dismissable-layer-1.1.19.tgz",
-			"integrity": "sha512-8g4pfOL9HoKKLWGiypT+dphVqjFfmcXO5GBnhsG6zI+lxAx/8feQpr+1LSN8Re3hiZ+XkLNS4O9ztK11/LzQ6w==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/primitive": "1.1.7",
-				"@radix-ui/react-compose-refs": "1.1.5",
-				"@radix-ui/react-primitive": "2.1.10",
-				"@radix-ui/react-use-callback-ref": "1.1.4",
-				"@radix-ui/react-use-effect-event": "0.0.5"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-focus-guards": {
-			"version": "1.1.6",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-focus-guards/-/react-focus-guards-1.1.6.tgz",
-			"integrity": "sha512-RNOJjfZMTyBM6xYmV3IVGXkPjIhcBAuv48POevAXwrGJhkWZ9p1rFoIS1JFooPuT193AZmRsCPhpoVJxx6OPoQ==",
-			"license": "MIT",
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-focus-scope": {
-			"version": "1.1.16",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-focus-scope/-/react-focus-scope-1.1.16.tgz",
-			"integrity": "sha512-wmRZ2WWLvmt6KHy2rNPOdPUjwq5xOHY02+m+udwJTn0aNIox/rkskAvJTyTLGhPK6KgrUjlJUJpgmx/+wFiFIQ==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-compose-refs": "1.1.5",
-				"@radix-ui/react-primitive": "2.1.10",
-				"@radix-ui/react-use-callback-ref": "1.1.4"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-id": {
-			"version": "1.1.4",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-id/-/react-id-1.1.4.tgz",
-			"integrity": "sha512-TMQp2llA+RYn7JcjnrMnz7wN4pcVttPZnRZo52PLQsoLVKzNlVwUeHmfePgTgRluXFvlD3GD5g5MOVVTJCO0qA==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-use-layout-effect": "1.1.4"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-popper": {
-			"version": "1.3.7",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-popper/-/react-popper-1.3.7.tgz",
-			"integrity": "sha512-UsJrrd7w4wuKKTdvd/DNERVlwSlUcyXzjhyDwBk+3aPOsCjOY6ZSbxuw8E6lZTjjfP8Cpd0J8VVkrYUWyGYXyg==",
-			"license": "MIT",
-			"dependencies": {
-				"@floating-ui/react-dom": "^2.0.0",
-				"@radix-ui/react-arrow": "1.1.15",
-				"@radix-ui/react-compose-refs": "1.1.5",
-				"@radix-ui/react-context": "1.2.2",
-				"@radix-ui/react-primitive": "2.1.10",
-				"@radix-ui/react-use-callback-ref": "1.1.4",
-				"@radix-ui/react-use-layout-effect": "1.1.4",
-				"@radix-ui/react-use-rect": "1.1.4",
-				"@radix-ui/react-use-size": "1.1.4",
-				"@radix-ui/rect": "1.1.3"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-portal": {
-			"version": "1.1.17",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-portal/-/react-portal-1.1.17.tgz",
-			"integrity": "sha512-vKQLcWypUnwZVvfV7UkGahH2g6ySe8M8R+zYBwPrv5byZ9QAW6cQVvNKo7GgmD+p8aYb6D9JBuvy8/WhOno2wQ==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-primitive": "2.1.10",
-				"@radix-ui/react-use-layout-effect": "1.1.4"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-presence": {
-			"version": "1.1.10",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-presence/-/react-presence-1.1.10.tgz",
-			"integrity": "sha512-3wyzCQ6+ubRA+D4uv9m95JYLXxmOHp05qjrkjeA7uKHHtjpPggQzc6DAb0URl7j67oR0K2foO4ip27TiX037Bw==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-use-layout-effect": "1.1.4"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-primitive": {
-			"version": "2.1.10",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-primitive/-/react-primitive-2.1.10.tgz",
-			"integrity": "sha512-MucOnzh6hR5mid6VpkbglRAMYMjKLqRnGBbjXkzjK52fuQDd1qbkx78a5P40mkcnVXJdEVxm26E9OPAiUq7nBg==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-slot": "1.3.3"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-roving-focus": {
-			"version": "1.1.19",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-roving-focus/-/react-roving-focus-1.1.19.tgz",
-			"integrity": "sha512-V9jI6hDjT7l3jsCQD9bLNvDLM3tH/gdbOTp7Tefp3hbbgCGQoK7tUvrWiRlcoBHIZ809ElXwNQwVo0B98LuTXQ==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/primitive": "1.1.7",
-				"@radix-ui/react-collection": "1.1.15",
-				"@radix-ui/react-compose-refs": "1.1.5",
-				"@radix-ui/react-context": "1.2.2",
-				"@radix-ui/react-direction": "1.1.4",
-				"@radix-ui/react-id": "1.1.4",
-				"@radix-ui/react-primitive": "2.1.10",
-				"@radix-ui/react-use-callback-ref": "1.1.4",
-				"@radix-ui/react-use-controllable-state": "1.2.6",
-				"@radix-ui/react-use-is-hydrated": "0.1.3",
-				"@radix-ui/react-use-layout-effect": "1.1.4"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"@types/react-dom": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc",
-				"react-dom": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				},
-				"@types/react-dom": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-slot": {
-			"version": "1.3.3",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-slot/-/react-slot-1.3.3.tgz",
-			"integrity": "sha512-qx7oqnYbxnK9kYI9m317qmFmEgo6ywqWvbTogdj7cL9p3/yx4M48p7Rnw5z3H890cL/ow/EeWJsuTykeZVXP5Q==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-compose-refs": "1.1.5"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-use-callback-ref": {
-			"version": "1.1.4",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-use-callback-ref/-/react-use-callback-ref-1.1.4.tgz",
-			"integrity": "sha512-R6OUY2e2fA6Yn6s+VSx5KBV6Nx8LQEhu+cz7LCej18rQ1HLyg9PSC9jP/ZNx0o6FAIK9c0F1kHylzSxKsdlkrQ==",
-			"license": "MIT",
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-use-controllable-state": {
-			"version": "1.2.6",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-use-controllable-state/-/react-use-controllable-state-1.2.6.tgz",
-			"integrity": "sha512-uEQJGT97ZA/TgP/Hydw47lHu+/vQj6z/0jA+WeTbK1o9Rx45GImjpD0tc3W5ad3D6XTSR6e1yEO0FvGq6WQfVQ==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/primitive": "1.1.7",
-				"@radix-ui/react-use-effect-event": "0.0.5",
-				"@radix-ui/react-use-layout-effect": "1.1.4"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-use-effect-event": {
-			"version": "0.0.5",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-use-effect-event/-/react-use-effect-event-0.0.5.tgz",
-			"integrity": "sha512-7cshFL8HGS/7HEiHH+9kL9HBwp2sa9yX18Knwek6KYWmXwM7pegMgta2AXMQKI+rq3JnfSj9x8wYqFMTdG1Jgg==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-use-layout-effect": "1.1.4"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-use-layout-effect": {
-			"version": "1.1.4",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-use-layout-effect/-/react-use-layout-effect-1.1.4.tgz",
-			"integrity": "sha512-K20DkRkUwDnxEYMBPcg3Y6voLkEy5p5QQmszZgLngKKiC7dzBR/aEuK3w1qlx2JWDUNH6FluahYdgR3BP+QbYw==",
-			"license": "MIT",
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-use-rect": {
-			"version": "1.1.4",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-use-rect/-/react-use-rect-1.1.4.tgz",
-			"integrity": "sha512-cSOCh6JlkmfjLyNcLiu2nB4v+nm+dkZ+Q5KHWk/soo4U7ZLiEQFKHK9/YmtBHjfCEaU43IBKQOc4/uJmCaiCTQ==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/rect": "1.1.3"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/react-use-size": {
-			"version": "1.1.4",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-use-size/-/react-use-size-1.1.4.tgz",
-			"integrity": "sha512-D3anSY15EJoxrihpsXI6SMrmmonnQtR2ni7arO+Lfdg3O95b9hNXxONk8jA5C8ANdF/h5HMAxejgs8PWJ6rlhw==",
-			"license": "MIT",
-			"dependencies": {
-				"@radix-ui/react-use-layout-effect": "1.1.4"
-			},
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
-		"node_modules/@radix-ui/react-menu/node_modules/@radix-ui/rect": {
-			"version": "1.1.3",
-			"resolved": "https://registry.npmjs.org/@radix-ui/rect/-/rect-1.1.3.tgz",
-			"integrity": "sha512-JtyZR+mqgBibTo8xea3B6ZRmzZiM/YeVBtUkas6zMuXjAlfIFIW2FgqeM9eLyvEaYX66vr6DJMK+4U6LV0KhNw==",
-			"license": "MIT"
-		},
 		"node_modules/@radix-ui/react-popper": {
 			"version": "1.3.2",
 			"resolved": "https://registry.npmjs.org/@radix-ui/react-popper/-/react-popper-1.3.2.tgz",
@@ -3076,21 +2410,6 @@
 				}
 			}
 		},
-		"node_modules/@radix-ui/react-use-is-hydrated": {
-			"version": "0.1.3",
-			"resolved": "https://registry.npmjs.org/@radix-ui/react-use-is-hydrated/-/react-use-is-hydrated-0.1.3.tgz",
-			"integrity": "sha512-umO/aJ+82CpOnhDZUTbILCQf7kU/g0iv+oGs/Q8jw7IkhWBzaEP4sA268PhFAJTFetbwp3ICc6ktpI4TqtxcIw==",
-			"license": "MIT",
-			"peerDependencies": {
-				"@types/react": "*",
-				"react": "^16.8 || ^17.0 || ^18.0 || ^19.0 || ^19.0.0-rc"
-			},
-			"peerDependenciesMeta": {
-				"@types/react": {
-					"optional": true
-				}
-			}
-		},
 		"node_modules/@radix-ui/react-use-layout-effect": {
 			"version": "1.1.2",
 			"resolved": "https://registry.npmjs.org/@radix-ui/react-use-layout-effect/-/react-use-layout-effect-1.1.2.tgz",

apps/fastvideo_studio/package.json

@@ -18,7 +18,6 @@
 	},
 	"dependencies": {
 		"@radix-ui/react-dialog": "^1.1.0",
-		"@radix-ui/react-dropdown-menu": "^2.1.24",
 		"@radix-ui/react-label": "^2.1.8",
 		"@radix-ui/react-scroll-area": "^1.2.10",
 		"@radix-ui/react-select": "^2.2.6",

apps/fastvideo_studio/src/app/datasets/page.test.tsx

@@ -1,64 +0,0 @@
-import { act, fireEvent, render, screen } from '@testing-library/react';
-import { describe, expect, it, vi } from 'vitest';
-
-import { HeaderActionsProvider } from '@/components/shell/HeaderActionsContext';
-import { getDatasets, type Dataset } from '@/lib/api';
-
-import DatasetsPage from './page';
-
-vi.mock('@/lib/api', () => ({
-  getDatasets: vi.fn(),
-}));
-
-vi.mock('@/components/datasets/AddDatasetButton', () => ({
-  default: () => null,
-}));
-
-vi.mock('@/components/datasets/CreateDatasetModal', () => ({
-  default: () => null,
-}));
-
-vi.mock('@/components/datasets/DatasetCard', () => ({
-  default: ({ dataset }: { dataset: Dataset }) => <div>{dataset.name}</div>,
-}));
-
-function renderPage() {
-  return render(
-    <HeaderActionsProvider>
-      <DatasetsPage />
-    </HeaderActionsProvider>,
-  );
-}
-
-describe('DatasetsPage', () => {
-  it('shows loading content before the initial request settles', async () => {
-    let resolveDatasets: (datasets: Dataset[]) => void = () => {};
-    vi.mocked(getDatasets).mockReturnValue(
-      new Promise<Dataset[]>((resolve) => {
-        resolveDatasets = resolve;
-      }),
-    );
-
-    renderPage();
-    expect(screen.getByLabelText('Loading datasets')).toBeInTheDocument();
-    expect(screen.queryByText('No datasets yet.')).not.toBeInTheDocument();
-
-    act(() => resolveDatasets([]));
-    expect(await screen.findByText('No datasets yet.')).toBeInTheDocument();
-  });
-
-  it('shows API failures separately from an empty list and retries', async () => {
-    vi.spyOn(console, 'error').mockImplementation(() => {});
-    vi.mocked(getDatasets).mockRejectedValueOnce(new Error('network down'));
-
-    renderPage();
-    expect(
-      await screen.findByText(/Could not load datasets from the Studio API/),
-    ).toBeInTheDocument();
-    expect(screen.queryByText('No datasets yet.')).not.toBeInTheDocument();
-
-    vi.mocked(getDatasets).mockResolvedValueOnce([]);
-    fireEvent.click(screen.getByRole('button', { name: 'Try Again' }));
-    expect(await screen.findByText('No datasets yet.')).toBeInTheDocument();
-  });
-});

apps/fastvideo_studio/src/app/datasets/page.tsx

@@ -1,14 +1,12 @@
 'use client';
 
 import * as React from 'react';
-import { AlertTriangle } from 'lucide-react';
 
 import AddDatasetButton from '@/components/datasets/AddDatasetButton';
 import CreateDatasetModal from '@/components/datasets/CreateDatasetModal';
 import DatasetCard from '@/components/datasets/DatasetCard';
 import { HeaderActions } from '@/components/shell/HeaderActionsContext';
 import { Card } from '@/components/ui/card';
-import { Button } from '@/components/ui/button';
 import { useStore } from '@/hooks/useStore';
 import { getDatasets } from '@/lib/api';
 import type { Dataset } from '@/lib/api';
@@ -23,28 +21,18 @@ import {
 
 export default function DatasetsPage() {
   const [datasets, setDatasets] = React.useState<Dataset[]>([]);
-  const [isInitialLoading, setIsInitialLoading] = React.useState(true);
   const [error, setError] = React.useState<string | null>(null);
   const { open } = useStore(createDatasetModalStore);
-  const fetchSequence = React.useRef(0);
 
   const fetchDatasets = React.useCallback(async () => {
-    const sequence = ++fetchSequence.current;
     try {
-      const next = await getDatasets();
-      if (sequence === fetchSequence.current) {
-        setDatasets(next);
-        setError(null);
-      }
+      setDatasets(await getDatasets());
+      setError(null);
     } catch (err) {
       console.error('Failed to fetch datasets:', err);
-      if (sequence === fetchSequence.current) {
-        setError(
-          'Could not load datasets from the Studio API. Check the server and try again.',
-        );
-      }
-    } finally {
-      if (sequence === fetchSequence.current) setIsInitialLoading(false);
+      // Distinguish an API outage from a genuinely empty list, so the user
+      // isn't told they have no datasets when the server is unreachable.
+      setError(err instanceof Error ? err.message : 'Failed to load datasets');
     }
   }, []);
 
@@ -62,67 +50,28 @@ export default function DatasetsPage() {
       <HeaderActions>
         <AddDatasetButton />
       </HeaderActions>
-      <div className="mx-auto flex w-full max-w-[850px] flex-col gap-6 px-4 pb-12">
+      <main className="mx-auto flex w-full max-w-[850px] flex-col gap-6 px-4 pb-12">
         <Card className="p-6">
-          <div aria-busy={isInitialLoading}>
-            {isInitialLoading ? (
-              <div
-                aria-label="Loading datasets"
-                className="flex flex-col gap-3 py-2"
-              >
-                {[0, 1, 2].map((item) => (
-                  <div
-                    key={item}
-                    className="h-24 animate-pulse rounded-lg border border-border bg-muted/50"
-                  />
-                ))}
-              </div>
-            ) : error && datasets.length === 0 ? (
-              <div
-                role="alert"
-                className="flex flex-col items-center gap-3 py-8 text-center"
-              >
-                <AlertTriangle
-                  className="size-6 text-destructive"
-                  aria-hidden
-                />
-                <p className="max-w-md text-sm text-muted-foreground">
-                  {error}
-                </p>
-                <Button type="button" variant="outline" onClick={fetchDatasets}>
-                  Try Again
-                </Button>
-              </div>
+          <div>
+            {error ? (
+              <p className="py-8 text-center text-destructive">{error}</p>
+            ) : datasets.length === 0 ? (
+              <p className="py-8 text-center text-muted-foreground">
+                No datasets yet.
+              </p>
             ) : (
-              <>
-                {error && (
-                  <p
-                    role="status"
-                    className="mb-3 rounded-lg border border-amber-500/50 bg-amber-500/10 px-3 py-2 text-sm text-foreground"
-                  >
-                    Dataset updates are temporarily unavailable. Showing the
-                    most recent results.
-                  </p>
-                )}
-                {datasets.length === 0 ? (
-                  <p className="py-8 text-center text-muted-foreground">
-                    No datasets yet.
-                  </p>
-                ) : (
-                  datasets.map((ds) => (
-                    <DatasetCard
-                      key={ds.id}
-                      dataset={ds}
-                      onUpdated={fetchDatasets}
-                      onSelect={() => handleSelectDataset(ds)}
-                    />
-                  ))
-                )}
-              </>
+              datasets.map((ds) => (
+                <DatasetCard
+                  key={ds.id}
+                  dataset={ds}
+                  onUpdated={fetchDatasets}
+                  onSelect={() => handleSelectDataset(ds)}
+                />
+              ))
             )}
           </div>
         </Card>
-      </div>
+      </main>
       <CreateDatasetModal
         isOpen={open}
         onClose={() => setCreateDatasetModalOpen(false)}

apps/fastvideo_studio/src/app/gallery/page.test.tsx

@@ -1,4 +1,4 @@
-import { fireEvent, render, screen } from '@testing-library/react';
+import { render, screen } from '@testing-library/react';
 import { describe, expect, it, vi } from 'vitest';
 
 import GalleryPage from './page';
@@ -43,22 +43,6 @@ describe('GalleryPage', () => {
     expect(getJobsList).toHaveBeenCalledWith('inference');
   });
 
-  it('provides video controls and a visible fallback when media fails', async () => {
-    vi.mocked(getJobsList).mockResolvedValue([makeJob()]);
-    renderGallery();
-
-    const video = await screen.findByLabelText(
-      'Generated video: a cat surfing a wave',
-    );
-    expect(video).toHaveAttribute('controls');
-
-    fireEvent.error(video);
-    expect(screen.getByText('Preview unavailable')).toBeInTheDocument();
-    expect(
-      screen.getByText('The generated file could not be loaded.'),
-    ).toBeInTheDocument();
-  });
-
   it('shows the empty state when no completed videos exist', async () => {
     vi.mocked(getJobsList).mockResolvedValue([
       makeJob({ status: 'running', output_path: null }),

apps/fastvideo_studio/src/app/gallery/page.tsx

@@ -1,9 +1,8 @@
 'use client';
 
-import { AlertTriangle, ImageOff, Loader2 } from 'lucide-react';
+import { Loader2 } from 'lucide-react';
 import { useEffect, useState } from 'react';
 
-import { Button } from '@/components/ui/button';
 import { Card } from '@/components/ui/card';
 import { getJobVideoUrl, getJobsList } from '@/lib/api';
 import type { Job } from '@/lib/types';
@@ -12,61 +11,11 @@ function isImage(job: Job): boolean {
   return job.output_path?.toLowerCase().endsWith('.png') ?? false;
 }
 
-function GalleryMedia({ job }: { job: Job }) {
-  const [failed, setFailed] = useState(false);
-
-  if (failed) {
-    return (
-      <div
-        role="status"
-        className="flex h-full flex-col items-center justify-center gap-2 px-4 text-center text-muted-foreground"
-      >
-        <ImageOff className="size-7" aria-hidden />
-        <span className="text-sm font-medium">Preview unavailable</span>
-        <span className="text-xs">
-          The generated file could not be loaded.
-        </span>
-      </div>
-    );
-  }
-
-  if (isImage(job)) {
-    return (
-      // eslint-disable-next-line @next/next/no-img-element
-      <img
-        src={getJobVideoUrl(job.id)}
-        alt={job.prompt}
-        className="block h-full w-full object-contain"
-        loading="lazy"
-        onError={() => setFailed(true)}
-      />
-    );
-  }
-
-  return (
-    <video
-      src={getJobVideoUrl(job.id)}
-      aria-label={
-        job.prompt ? `Generated video: ${job.prompt}` : 'Generated video'
-      }
-      className="block h-full w-full object-contain"
-      controls
-      muted
-      loop
-      playsInline
-      preload="metadata"
-      onError={() => setFailed(true)}
-    />
-  );
-}
-
 export default function GalleryPage() {
   const [jobs, setJobs] = useState<Job[]>([]);
   const [isLoading, setIsLoading] = useState(true);
   const [error, setError] = useState<string | null>(null);
 
-  const [reloadKey, setReloadKey] = useState(0);
-
   useEffect(() => {
     let cancelled = false;
     async function load() {
@@ -91,13 +40,7 @@ export default function GalleryPage() {
     return () => {
       cancelled = true;
     };
-  }, [reloadKey]);
-
-  function retry() {
-    setError(null);
-    setIsLoading(true);
-    setReloadKey((k) => k + 1);
-  }
+  }, []);
 
   const galleryJobs = jobs.filter(
     (j) =>
@@ -121,16 +64,7 @@ export default function GalleryPage() {
             <span>Loading gallery…</span>
           </div>
         ) : error ? (
-          <div
-            role="alert"
-            className="flex flex-col items-center gap-3 py-8 text-center"
-          >
-            <AlertTriangle className="size-6 text-destructive" aria-hidden />
-            <p className="max-w-md text-sm text-muted-foreground">{error}</p>
-            <Button type="button" variant="outline" onClick={retry}>
-              Try Again
-            </Button>
-          </div>
+          <p className="py-8 text-destructive">{error}</p>
         ) : galleryJobs.length === 0 ? (
           <p className="py-8 text-center text-muted-foreground">
             No completed videos yet
@@ -143,7 +77,24 @@ export default function GalleryPage() {
                 className="flex flex-col overflow-hidden rounded-lg border border-border bg-background"
               >
                 <div className="relative aspect-video overflow-hidden bg-muted">
-                  <GalleryMedia job={job} />
+                  {isImage(job) ? (
+                    // eslint-disable-next-line @next/next/no-img-element
+                    <img
+                      src={getJobVideoUrl(job.id)}
+                      alt={job.prompt}
+                      className="block h-full w-full object-contain"
+                      loading="lazy"
+                    />
+                  ) : (
+                    <video
+                      src={getJobVideoUrl(job.id)}
+                      className="block h-full w-full object-contain"
+                      muted
+                      loop
+                      playsInline
+                      preload="metadata"
+                    />
+                  )}
                 </div>
                 <p
                   className="line-clamp-3 border-t border-border px-4 py-3 text-sm text-muted-foreground"

apps/fastvideo_studio/src/app/globals.css

@@ -41,7 +41,7 @@
 
 	--border: #e2e8f0;
 	--input: #cbd5e1;
-	--ring: #1d4ed8;
+	--ring: #94a3b8;
 
 	--radius: 0.5rem;
 }
@@ -77,7 +77,7 @@
 
 	--border: #334155;
 	--input: #334155;
-	--ring: #7dd3fc;
+	--ring: #cbd5e1;
 }
 
 @theme inline {
@@ -125,7 +125,6 @@
 html,
 body {
 	min-height: 100%;
-	overflow-x: clip;
 }
 
 html {
@@ -164,22 +163,6 @@ a {
 	color: inherit;
 }
 
-:where(
-	a,
-	button,
-	input,
-	textarea,
-	select,
-	summary,
-	[role="button"],
-	[role="menuitem"],
-	[role="slider"],
-	[tabindex]
-):focus-visible {
-	outline: 3px solid var(--ring) !important;
-	outline-offset: 2px !important;
-}
-
 summary {
 	list-style: none;
 }

apps/fastvideo_studio/src/app/globals.test.ts

@@ -1,55 +0,0 @@
-import { readFileSync } from 'node:fs';
-import { join } from 'node:path';
-import { describe, expect, it } from 'vitest';
-
-const css = readFileSync(join(process.cwd(), 'src/app/globals.css'), 'utf8');
-
-function token(block: string, name: string): string {
-  const match = block.match(new RegExp(`--${name}:\\s*(#[0-9a-fA-F]{6})`));
-  if (!match) throw new Error(`Missing --${name} token`);
-  return match[1];
-}
-
-function luminance(hex: string): number {
-  const channels = hex
-    .slice(1)
-    .match(/.{2}/g)!
-    .map((channel) => parseInt(channel, 16) / 255)
-    .map((channel) =>
-      channel <= 0.04045
-        ? channel / 12.92
-        : ((channel + 0.055) / 1.055) ** 2.4,
-    );
-  return (
-    0.2126 * channels[0] + 0.7152 * channels[1] + 0.0722 * channels[2]
-  );
-}
-
-function contrast(first: string, second: string): number {
-  const firstLuminance = luminance(first);
-  const secondLuminance = luminance(second);
-  return (
-    (Math.max(firstLuminance, secondLuminance) + 0.05) /
-    (Math.min(firstLuminance, secondLuminance) + 0.05)
-  );
-}
-
-describe('global focus styles', () => {
-  it('keeps focus tokens above 3:1 against both page themes', () => {
-    const light = css.match(/:root\s*{([\s\S]*?)\n}/)?.[1] ?? '';
-    const dark = css.match(/\.dark\s*{([\s\S]*?)\n}/)?.[1] ?? '';
-
-    expect(contrast(token(light, 'ring'), token(light, 'background'))).toBeGreaterThanOrEqual(
-      3,
-    );
-    expect(contrast(token(dark, 'ring'), token(dark, 'background'))).toBeGreaterThanOrEqual(
-      3,
-    );
-  });
-
-  it('applies a non-animated three-pixel outline to focus-visible controls', () => {
-    expect(css).toContain('):focus-visible {');
-    expect(css).toContain('outline: 3px solid var(--ring) !important;');
-    expect(css).toContain('outline-offset: 2px !important;');
-  });
-});

apps/fastvideo_studio/src/app/gpus/page.tsx

@@ -4,8 +4,8 @@ import GpuGrid from '@/components/system/GpuGrid';
 
 export default function GpusPage() {
   return (
-    <div className="mx-auto flex w-full max-w-[1100px] flex-col gap-6 px-4 pb-12 pt-6">
+    <main className="mx-auto flex w-full max-w-[1100px] flex-col gap-6 px-4 pb-12 pt-6">
       <GpuGrid />
-    </div>
+    </main>
   );
 }

apps/fastvideo_studio/src/app/settings/page.test.tsx

@@ -52,20 +52,6 @@ describe('Settings page', () => {
     expect(updateOption).toHaveBeenCalledWith('numFrames', expect.any(Number));
   });
 
-  it('gives every slider an accessible name', () => {
-    renderPage();
-
-    const sliders = screen.getAllByRole('slider');
-    expect(sliders).toHaveLength(11);
-    for (const slider of sliders) {
-      expect(slider).toHaveAccessibleName();
-    }
-    expect(screen.getByRole('slider', { name: 'Frames' })).toBeInTheDocument();
-    expect(
-      screen.getByRole('slider', { name: 'Guidance Scale' }),
-    ).toBeInTheDocument();
-  });
-
   it('calls resetToDefaults when Reset to Defaults is clicked', () => {
     renderPage();
     fireEvent.click(screen.getByRole('button', { name: 'Reset to Defaults' }));

apps/fastvideo_studio/src/components/datasets/DatasetCard.test.tsx

@@ -25,16 +25,6 @@ beforeEach(() => {
 });
 
 describe('DatasetCard', () => {
-  it('keeps selection and delete buttons as semantic siblings', () => {
-    render(<DatasetCard dataset={dataset} onUpdated={() => {}} />);
-
-    const selectButton = screen.getByRole('button', { pressed: false });
-    const deleteButton = screen.getByRole('button', { name: 'Delete' });
-
-    expect(selectButton).toHaveTextContent('My Dataset');
-    expect(selectButton).not.toContainElement(deleteButton);
-  });
-
   it('renders the name, file count and human-readable size', () => {
     render(<DatasetCard dataset={dataset} onUpdated={() => {}} />);
     expect(screen.getByText('My Dataset')).toBeInTheDocument();
@@ -81,7 +71,7 @@ describe('DatasetCard', () => {
     expect(onSelect).not.toHaveBeenCalled();
   });
 
-  it('keeps the selection and delete actions separate', () => {
+  it('selects on keyboard activation of the card body but not of the Delete button', () => {
     const onSelect = vi.fn();
     render(
       <DatasetCard dataset={dataset} onUpdated={() => {}} onSelect={onSelect} />,
@@ -93,12 +83,8 @@ describe('DatasetCard', () => {
     });
     expect(onSelect).not.toHaveBeenCalled();
 
-    // Activating the dedicated selection button selects the dataset.
-    fireEvent.click(
-      screen.getByRole('button', {
-        name: /My Dataset.*3 files.*2.0 KB/,
-      }),
-    );
+    // Activating the card body itself does select.
+    fireEvent.keyDown(screen.getByText('My Dataset'), { key: 'Enter' });
     expect(onSelect).toHaveBeenCalledTimes(1);
   });
 

apps/fastvideo_studio/src/components/datasets/DatasetCard.tsx

@@ -55,33 +55,43 @@ export default function DatasetCard({
     }
   }
 
+  function handleKeyDown(e: React.KeyboardEvent) {
+    if ((e.target as HTMLElement).closest('button')) return;
+    if (e.key === 'Enter' || e.key === ' ') {
+      e.preventDefault();
+      onSelect();
+    }
+  }
+
   return (
-    <article
+    <div
       className={cn(
-        'mb-3 flex items-start gap-3 rounded-lg border border-border bg-background px-[1.15rem] py-4',
+        'mb-3 flex cursor-pointer flex-col gap-[0.6rem] rounded-lg border border-border bg-background px-[1.15rem] py-4',
         isSelected && 'border-accent-blue bg-accent-blue/5',
       )}
+      onClick={(e) => {
+        if ((e.target as HTMLElement).closest('button')) return;
+        onSelect();
+      }}
+      onKeyDown={handleKeyDown}
+      role="button"
+      tabIndex={0}
     >
-      <button
-        type="button"
-        aria-pressed={isSelected}
-        onClick={onSelect}
-        className="flex min-w-0 flex-1 cursor-pointer flex-col gap-[0.6rem] rounded-md text-left"
-      >
+      <div className="flex flex-wrap items-center justify-between gap-2">
         <span className="text-[0.95rem] font-semibold">{dataset.name}</span>
-        <span className="text-sm text-muted-foreground">
-          {fileCount} {fileCount === 1 ? 'file' : 'files'} · {sizeLabel}
-        </span>
-      </button>
-      <Button
-        type="button"
-        variant="destructive"
-        size="sm"
-        onClick={handleDelete}
-        disabled={isLoading}
-      >
-        Delete
-      </Button>
-    </article>
+        <Button
+          type="button"
+          variant="destructive"
+          size="sm"
+          onClick={handleDelete}
+          disabled={isLoading}
+        >
+          Delete
+        </Button>
+      </div>
+      <div className="text-sm text-muted-foreground">
+        {fileCount} {fileCount === 1 ? 'file' : 'files'} · {sizeLabel}
+      </div>
+    </div>
   );
 }

apps/fastvideo_studio/src/components/datasets/DatasetSidebar.test.tsx

@@ -6,9 +6,6 @@ import * as api from '@/lib/api';
 import type { Dataset } from '@/lib/api';
 
 vi.mock('@/lib/api');
-vi.mock('sonner', () => ({
-  toast: { error: vi.fn() },
-}));
 
 const mockedApi = vi.mocked(api);
 
@@ -30,27 +27,6 @@ beforeEach(() => {
 });
 
 describe('DatasetSidebar', () => {
-  it('fills the mobile viewport without reserving main-content width', async () => {
-    const onWidthChange = vi.fn();
-
-    render(
-      <DatasetSidebar
-        dataset={dataset}
-        isMobile
-        onClose={() => {}}
-        onWidthChange={onWidthChange}
-      />,
-    );
-
-    const drawer = screen.getByRole('dialog', {
-      name: 'My Dataset dataset details',
-    });
-    expect(drawer).toHaveStyle({ width: '100%', maxWidth: 'none' });
-    expect(drawer).toHaveAttribute('aria-modal', 'true');
-    expect(drawer).toHaveFocus();
-    expect(onWidthChange).toHaveBeenCalledWith(0);
-  });
-
   it('lists dataset files after loading', async () => {
     render(<DatasetSidebar dataset={dataset} onClose={() => {}} />);
 
@@ -67,16 +43,6 @@ describe('DatasetSidebar', () => {
     expect(mockedApi.getDatasetMediaUrl).toHaveBeenCalledWith('ds-1', 'b.mp4');
   });
 
-  it('shows a fallback when a dataset preview cannot load', async () => {
-    render(<DatasetSidebar dataset={dataset} onClose={() => {}} />);
-
-    const preview = await screen.findByLabelText('Preview of a.mp4');
-    fireEvent.error(preview);
-
-    expect(screen.getByText('Preview unavailable')).toBeInTheDocument();
-    expect(screen.queryByLabelText('Preview of a.mp4')).not.toBeInTheDocument();
-  });
-
   it('debounces caption save by 500ms', async () => {
     render(<DatasetSidebar dataset={dataset} onClose={() => {}} />);
     const textarea = await screen.findByDisplayValue('cap a');
@@ -133,39 +99,6 @@ describe('DatasetSidebar', () => {
     }
   });
 
-  it('shows a failed save and lets the user retry it', async () => {
-    vi.spyOn(console, 'error').mockImplementation(() => {});
-    mockedApi.updateDatasetCaption
-      .mockRejectedValueOnce(new Error('network down'))
-      .mockResolvedValueOnce(undefined);
-    render(<DatasetSidebar dataset={dataset} onClose={() => {}} />);
-    const textarea = await screen.findByDisplayValue('cap a');
-
-    vi.useFakeTimers();
-    try {
-      fireEvent.change(textarea, { target: { value: 'needs retry' } });
-      await act(async () => {
-        await vi.advanceTimersByTimeAsync(500);
-      });
-
-      expect(screen.getByText(/Not saved/)).toBeInTheDocument();
-      fireEvent.click(screen.getByRole('button', { name: 'Retry' }));
-      await act(async () => {
-        await Promise.resolve();
-      });
-
-      expect(mockedApi.updateDatasetCaption).toHaveBeenCalledTimes(2);
-      expect(mockedApi.updateDatasetCaption).toHaveBeenLastCalledWith(
-        'ds-1',
-        'a.mp4',
-        'needs retry',
-      );
-      expect(screen.getByText('Saved')).toBeInTheDocument();
-    } finally {
-      vi.useRealTimers();
-    }
-  });
-
   it('debounces per file: editing another caption does not cancel a pending save', async () => {
     render(<DatasetSidebar dataset={dataset} onClose={() => {}} />);
     await screen.findByDisplayValue('cap a');

apps/fastvideo_studio/src/components/datasets/DatasetSidebar.tsx

@@ -1,8 +1,7 @@
 'use client';
 
 import * as React from 'react';
-import { ImageOff, X } from 'lucide-react';
-import { toast } from 'sonner';
+import { X } from 'lucide-react';
 
 import DownloadCaptions from '@/components/datasets/DownloadCaptions';
 import { Textarea } from '@/components/ui/textarea';
@@ -14,14 +13,12 @@ import {
   type Dataset,
 } from '@/lib/api';
 import { cn } from '@/lib/utils';
-import { useDrawerFocus } from '@/hooks/useDrawerFocus';
 
 const SIDEBAR_MIN_WIDTH = 320;
 const SIDEBAR_MAX_WIDTH = 900;
 const INITIAL_PAGE_SIZE = 24;
 const PAGE_SIZE = 24;
 const SCROLL_THRESHOLD = 200;
-type CaptionSaveState = 'idle' | 'saving' | 'saved' | 'error';
 
 // Memoized so a caption keystroke re-renders only the edited card, not every
 // visible <video> in the grid (visibleCount grows unbounded with scrolling).
@@ -30,101 +27,54 @@ const DatasetFileCard = React.memo(function DatasetFileCard({
   mediaUrl,
   caption,
   thumbLoaded,
-  saveState,
   onCaptionChange,
-  onCaptionRetry,
   onThumbLoaded,
 }: {
   fileName: string;
   mediaUrl: string;
   caption: string;
   thumbLoaded: boolean;
-  saveState: CaptionSaveState;
   onCaptionChange: (fileName: string, value: string) => void;
-  onCaptionRetry: (fileName: string, value: string) => void;
   onThumbLoaded: (fileName: string) => void;
 }) {
-  const [mediaFailed, setMediaFailed] = React.useState(false);
-
-  React.useEffect(() => {
-    setMediaFailed(false);
-  }, [mediaUrl]);
-
   return (
     <div className="relative flex flex-col overflow-hidden rounded-lg border border-border bg-background">
-      {!thumbLoaded && !mediaFailed && (
+      {!thumbLoaded && (
         <div className="pointer-events-none absolute inset-0 flex items-center justify-center bg-background/70">
           <div className="h-6 w-6 animate-spin rounded-full border-2 border-muted-foreground/40 border-t-accent-blue" />
         </div>
       )}
-      {mediaFailed ? (
-        <div
-          role="status"
-          className="flex aspect-video w-full flex-col items-center justify-center gap-1 bg-muted px-2 text-center text-muted-foreground"
-        >
-          <ImageOff className="size-5" aria-hidden />
-          <span className="text-xs">Preview unavailable</span>
-        </div>
-      ) : (
-        // eslint-disable-next-line jsx-a11y/media-has-caption
-        <video
-          src={mediaUrl}
-          aria-label={`Preview of ${fileName}`}
-          className="aspect-video w-full bg-border object-cover"
-          muted
-          autoPlay
-          loop
-          playsInline
-          onLoadedData={() => onThumbLoaded(fileName)}
-          onError={() => {
-            setMediaFailed(true);
-            onThumbLoaded(fileName);
-          }}
-        />
-      )}
+      {/* eslint-disable-next-line jsx-a11y/media-has-caption */}
+      <video
+        src={mediaUrl}
+        className="aspect-video w-full bg-border object-cover"
+        muted
+        autoPlay
+        loop
+        playsInline
+        onLoadedData={() => onThumbLoaded(fileName)}
+        onError={() => onThumbLoaded(fileName)}
+      />
       <Textarea
-        aria-label={`Caption for ${fileName}`}
         value={caption}
         onChange={(e) => onCaptionChange(fileName, e.target.value)}
         placeholder="Caption"
         rows={2}
         className="min-h-[2.5rem] resize-y rounded-none border-0 bg-transparent p-1.5 text-xs shadow-none focus-visible:border-transparent focus-visible:ring-0"
       />
-      <div
-        aria-live="polite"
-        className="flex min-h-6 items-center px-1.5 pb-1 text-[0.7rem] text-muted-foreground"
-      >
-        {saveState === 'saving' && <span>Saving…</span>}
-        {saveState === 'saved' && <span>Saved</span>}
-        {saveState === 'error' && (
-          <span role="alert" className="text-destructive">
-            Not saved.{' '}
-            <button
-              type="button"
-              onClick={() => onCaptionRetry(fileName, caption)}
-              className="inline-flex min-h-11 items-center font-medium underline underline-offset-2"
-            >
-              Retry
-            </button>
-          </span>
-        )}
-      </div>
     </div>
   );
 });
 
 export default function DatasetSidebar({
   dataset,
-  isMobile = false,
   onClose,
   onWidthChange,
 }: {
   dataset: Dataset;
-  isMobile?: boolean;
   onClose: () => void;
   onWidthChange?: (w: number) => void;
 }) {
-  const drawerRef = useDrawerFocus<HTMLElement>(isMobile);
   const [width, setWidth] = React.useState(400);
   const [isDragging, setIsDragging] = React.useState(false);
   const [fileNames, setFileNames] = React.useState<string[]>([]);
@@ -134,21 +84,17 @@ export default function DatasetSidebar({
   const [thumbLoaded, setThumbLoaded] = React.useState<
     Record<string, boolean>
   >({});
-  const [captionSaveStates, setCaptionSaveStates] = React.useState<
-    Record<string, CaptionSaveState>
-  >({});
 
   // Pending debounced caption saves, keyed per file so editing one caption
   // can't cancel another file's pending save.
   const pendingSaves = React.useRef(
     new Map<string, { timer: ReturnType<typeof setTimeout>; save: () => void }>(),
   );
-  const captionVersions = React.useRef(new Map<string, number>());
   const scrollRef = React.useRef<HTMLDivElement>(null);
 
   React.useEffect(() => {
-    onWidthChange?.(isMobile ? 0 : width);
-  }, [isMobile, width, onWidthChange]);
+    onWidthChange?.(width);
+  }, [width, onWidthChange]);
 
   React.useEffect(() => {
     let cancelled = false;
@@ -160,8 +106,6 @@ export default function DatasetSidebar({
         setCaptions(data.captions);
         setVisibleCount(INITIAL_PAGE_SIZE);
         setThumbLoaded({});
-        setCaptionSaveStates({});
-        captionVersions.current.clear();
       })
       .catch((err) => console.error('Failed to load dataset files:', err))
       .finally(() => {
@@ -194,51 +138,23 @@ export default function DatasetSidebar({
   });
 
   const datasetId = dataset.id;
-  const persistCaption = React.useCallback(
-    (fileName: string, value: string, version: number) => {
-      setCaptionSaveStates((prev) => ({ ...prev, [fileName]: 'saving' }));
-      void updateDatasetCaption(datasetId, fileName, value)
-        .then(() => {
-          if (captionVersions.current.get(fileName) !== version) return;
-          setCaptionSaveStates((prev) => ({ ...prev, [fileName]: 'saved' }));
-        })
-        .catch((error) => {
-          if (captionVersions.current.get(fileName) !== version) return;
-          console.error('Failed to save caption:', error);
-          setCaptionSaveStates((prev) => ({ ...prev, [fileName]: 'error' }));
-          toast.error('Caption was not saved', {
-            description: `${fileName}: check the Studio API, then retry.`,
-          });
-        });
-    },
-    [datasetId],
-  );
-
   const handleCaptionChange = React.useCallback(
     (fileName: string, value: string) => {
       setCaptions((prev) => ({ ...prev, [fileName]: value }));
-      setCaptionSaveStates((prev) => ({ ...prev, [fileName]: 'idle' }));
       const pending = pendingSaves.current.get(fileName);
       if (pending) clearTimeout(pending.timer);
-      const version = (captionVersions.current.get(fileName) ?? 0) + 1;
-      captionVersions.current.set(fileName, version);
-      const save = () => persistCaption(fileName, value, version);
+      const save = () => {
+        updateDatasetCaption(datasetId, fileName, value).catch((err) =>
+          console.error('Failed to save caption:', err),
+        );
+      };
       const timer = setTimeout(() => {
         pendingSaves.current.delete(fileName);
         save();
       }, 500);
       pendingSaves.current.set(fileName, { timer, save });
     },
-    [persistCaption],
-  );
-
-  const handleCaptionRetry = React.useCallback(
-    (fileName: string, value: string) => {
-      const version = (captionVersions.current.get(fileName) ?? 0) + 1;
-      captionVersions.current.set(fileName, version);
-      persistCaption(fileName, value, version);
-    },
-    [persistCaption],
+    [datasetId],
   );
 
   function handleScroll() {
@@ -273,16 +189,8 @@ export default function DatasetSidebar({
 
   return (
     <aside
-      ref={drawerRef}
-      tabIndex={-1}
-      role="dialog"
-      aria-label={`${dataset.name} dataset details`}
-      aria-modal={isMobile || undefined}
-      className="fixed bottom-0 right-0 top-[var(--header-height)] z-50 flex max-h-[calc(100dvh-var(--header-height))] min-w-0 shrink-0 flex-col border-l border-border bg-card md:min-w-[320px]"
-      style={{
-        width: isMobile ? '100%' : width,
-        maxWidth: isMobile ? 'none' : SIDEBAR_MAX_WIDTH,
-      }}
+      className="fixed bottom-0 right-0 top-[var(--header-height)] z-50 flex max-h-[calc(100vh-var(--header-height))] min-w-[320px] shrink-0 flex-col border-l border-border bg-card"
+      style={{ width, maxWidth: SIDEBAR_MAX_WIDTH }}
     >
       <div className="flex shrink-0 items-center justify-between border-b border-border px-5 py-4">
         <h2 className="m-0 min-w-0 truncate text-base font-semibold text-foreground">
@@ -295,7 +203,7 @@ export default function DatasetSidebar({
             onClick={onClose}
             title="Close"
             aria-label="Close"
-            className="flex size-11 items-center justify-center rounded-lg text-muted-foreground transition-colors hover:bg-accent hover:text-foreground"
+            className="flex items-center justify-center rounded-lg p-1.5 text-muted-foreground transition-colors hover:bg-accent hover:text-foreground"
           >
             <X className="h-[18px] w-[18px]" />
           </button>
@@ -323,9 +231,7 @@ export default function DatasetSidebar({
                   mediaUrl={getDatasetMediaUrl(dataset.id, fileName)}
                   caption={captions[fileName] ?? ''}
                   thumbLoaded={!!thumbLoaded[fileName]}
-                  saveState={captionSaveStates[fileName] ?? 'idle'}
                   onCaptionChange={handleCaptionChange}
-                  onCaptionRetry={handleCaptionRetry}
                   onThumbLoaded={markThumbLoaded}
                 />
               ))}
@@ -334,14 +240,14 @@ export default function DatasetSidebar({
         </div>
       </div>
 
-      {!isMobile && <div
+      <div
         role="presentation"
         onMouseDown={onMouseDown}
         className={cn(
           'absolute bottom-0 left-0 top-0 z-[1] w-1.5 cursor-col-resize hover:bg-accent-blue/25',
           isDragging && 'bg-accent-blue/25',
         )}
-      />}
+      />
     </aside>
   );
 }

apps/fastvideo_studio/src/components/datasets/DownloadCaptions.tsx

@@ -6,7 +6,7 @@ import { Button } from '@/components/ui/button';
 import { downloadBlob } from '@/lib/utils';
 
 const MENU_ITEM =
-  'block min-h-11 w-full cursor-pointer px-4 py-2 text-left text-sm font-medium text-foreground transition-colors hover:bg-muted disabled:cursor-not-allowed disabled:opacity-50';
+  'block w-full cursor-pointer px-4 py-2 text-left text-sm font-medium text-foreground transition-colors hover:bg-muted disabled:cursor-not-allowed disabled:opacity-50';
 
 export default function DownloadCaptions({
   fileNames,

apps/fastvideo_studio/src/components/jobs/CreateJobButton.test.tsx

@@ -1,53 +0,0 @@
-import { render, screen } from '@testing-library/react';
-import userEvent from '@testing-library/user-event';
-import { describe, expect, it, vi } from 'vitest';
-
-import CreateJobButton from './CreateJobButton';
-
-vi.mock('./CreateJobModal', () => ({
-  default: ({
-    isOpen,
-    workloadType,
-  }: {
-    isOpen: boolean;
-    workloadType: string;
-  }) =>
-    isOpen ? (
-      <div role="dialog" data-workload-type={workloadType}>
-        Create job form
-      </div>
-    ) : null,
-}));
-
-describe('CreateJobButton', () => {
-  it('opens the workload menu on click and selects an item', async () => {
-    const user = userEvent.setup();
-    render(<CreateJobButton jobType="inference" />);
-
-    await user.click(screen.getByRole('button', { name: 'Create Job' }));
-    await user.click(screen.getByRole('menuitem', { name: /I2V/i }));
-
-    expect(screen.getByRole('dialog')).toHaveAttribute(
-      'data-workload-type',
-      'i2v',
-    );
-  });
-
-  it('opens and operates the workload menu from the keyboard', async () => {
-    const user = userEvent.setup();
-    render(<CreateJobButton jobType="inference" />);
-
-    const trigger = screen.getByRole('button', { name: 'Create Job' });
-    trigger.focus();
-    await user.keyboard('{Enter}');
-
-    const firstItem = await screen.findByRole('menuitem', { name: /T2V/i });
-    expect(firstItem).toHaveFocus();
-    await user.keyboard('{Enter}');
-
-    expect(screen.getByRole('dialog')).toHaveAttribute(
-      'data-workload-type',
-      't2v',
-    );
-  });
-});

apps/fastvideo_studio/src/components/jobs/CreateJobButton.tsx

@@ -2,7 +2,6 @@
 
 import * as React from 'react';
 import { ChevronDown } from 'lucide-react';
-import * as DropdownMenu from '@radix-ui/react-dropdown-menu';
 
 import CreateJobModal from '@/components/jobs/CreateJobModal';
 import { Button } from '@/components/ui/button';
@@ -34,35 +33,31 @@ export default function CreateJobButton({ jobType }: CreateJobButtonProps) {
 
   return (
     <>
-      <DropdownMenu.Root>
-        <DropdownMenu.Trigger asChild>
-          <Button type="button" className="gap-1.5">
-            Create Job
-            <ChevronDown className="size-3.5 opacity-85" aria-hidden />
-          </Button>
-        </DropdownMenu.Trigger>
-        <DropdownMenu.Portal>
-          <DropdownMenu.Content
-            align="end"
-            sideOffset={4}
-            collisionPadding={8}
-            className="z-[200] min-w-48 overflow-hidden rounded-lg border border-border bg-popover py-1 text-popover-foreground shadow-lg"
-          >
-            {options.map((opt) => (
-              <DropdownMenu.Item
-                key={opt.type}
-                onSelect={() => openModal(opt.type)}
-                className="flex min-h-11 cursor-pointer select-none flex-col justify-center px-4 py-2 text-left text-sm font-medium outline-none data-[highlighted]:bg-secondary"
-              >
-                {opt.label}
-                <span className="mt-0.5 block text-xs font-normal text-muted-foreground">
-                  {opt.desc}
-                </span>
-              </DropdownMenu.Item>
-            ))}
-          </DropdownMenu.Content>
-        </DropdownMenu.Portal>
-      </DropdownMenu.Root>
+      <div className="group relative inline-block">
+        <Button type="button" className="gap-1.5">
+          Create Job
+          <ChevronDown className="size-3.5 opacity-85" aria-hidden />
+        </Button>
+        <div
+          role="menu"
+          className="invisible absolute right-0 top-full z-[200] mt-1 min-w-full -translate-y-1 rounded-lg border border-border bg-popover py-1 opacity-0 shadow-lg transition-all duration-150 group-hover:visible group-hover:translate-y-0 group-hover:opacity-100"
+        >
+          {options.map((opt) => (
+            <button
+              key={opt.type}
+              type="button"
+              role="menuitem"
+              onClick={() => openModal(opt.type)}
+              className="block w-full whitespace-nowrap px-4 py-2 text-left text-sm font-medium text-popover-foreground transition-colors hover:bg-secondary"
+            >
+              {opt.label}
+              <span className="mt-0.5 block text-xs font-normal text-muted-foreground">
+                {opt.desc}
+              </span>
+            </button>
+          ))}
+        </div>
+      </div>
       <CreateJobModal
         isOpen={modalOpen}
         onClose={() => setModalOpen(false)}

apps/fastvideo_studio/src/components/jobs/CreateJobModal.test.tsx

@@ -50,57 +50,6 @@ function renderModal(
 }
 
 describe('CreateJobModal', () => {
-  it('shows a model loading error instead of an empty model list', async () => {
-    vi.spyOn(console, 'error').mockImplementation(() => {});
-    vi.mocked(getModels).mockRejectedValueOnce(new Error('network down'));
-
-    renderModal();
-
-    expect(
-      await screen.findByText(/Models could not be loaded/),
-    ).toBeInTheDocument();
-    expect(screen.getByLabelText('Model')).toHaveAttribute(
-      'aria-invalid',
-      'true',
-    );
-  });
-
-  it('keeps the form open and reports job creation failures', async () => {
-    vi.spyOn(console, 'error').mockImplementation(() => {});
-    vi.mocked(createJob).mockRejectedValueOnce(new Error('API rejected job'));
-    const user = userEvent.setup();
-    const { onClose, onSuccess } = renderModal();
-
-    await screen.findByRole('option', { name: 'Wan T2V (wan/t2v-1.3b)' });
-    await user.type(screen.getByLabelText('Prompt'), 'a careful test prompt');
-    await user.click(screen.getByRole('button', { name: 'Create Job' }));
-
-    expect(
-      await screen.findByText(/API rejected job.*then try again/),
-    ).toBeInTheDocument();
-    expect(onSuccess).not.toHaveBeenCalled();
-    expect(onClose).not.toHaveBeenCalled();
-  });
-
-  it('reports image upload failures next to the file input', async () => {
-    vi.spyOn(console, 'error').mockImplementation(() => {});
-    vi.mocked(uploadImage).mockRejectedValueOnce(new Error('Upload failed'));
-    const user = userEvent.setup();
-    renderModal({ workloadType: 'i2v' });
-
-    await screen.findByRole('option', { name: 'Wan T2V (wan/t2v-1.3b)' });
-    const input = screen.getByLabelText('Image');
-    await user.upload(
-      input,
-      new File(['image'], 'input.png', { type: 'image/png' }),
-    );
-
-    expect(
-      await screen.findByText(/Upload failed.*Choose the image again/),
-    ).toBeInTheDocument();
-    expect(input).toHaveAttribute('aria-invalid', 'true');
-  });
-
   it('renders the form fields for an inference job', async () => {
     renderModal();
 

apps/fastvideo_studio/src/components/jobs/CreateJobModal.tsx

@@ -103,17 +103,6 @@ export default function CreateJobModal({
   const [fakeScoreModelPath, setFakeScoreModelPath] = React.useState('');
   const [isSubmitting, setIsSubmitting] = React.useState(false);
   const [isLoadingModels, setIsLoadingModels] = React.useState(false);
-  const [isLoadingDatasets, setIsLoadingDatasets] = React.useState(false);
-  const [modelLoadError, setModelLoadError] = React.useState<string | null>(
-    null,
-  );
-  const [datasetLoadError, setDatasetLoadError] = React.useState<string | null>(
-    null,
-  );
-  const [imageUploadError, setImageUploadError] = React.useState<string | null>(
-    null,
-  );
-  const [submitError, setSubmitError] = React.useState<string | null>(null);
   const imageInputRef = React.useRef<HTMLInputElement>(null);
 
   // Seed field values from the persisted default options each time the modal
@@ -155,10 +144,6 @@ export default function CreateJobModal({
     setImageFileName('');
     setSelectedDatasetId('');
     setSelectedValidationDatasetId('');
-    setModelLoadError(null);
-    setDatasetLoadError(null);
-    setImageUploadError(null);
-    setSubmitError(null);
     if (workloadType === 'dmd_t2v') {
       setDmdUseVsa(false);
       setDmdVsaSparsity(0.8);
@@ -177,7 +162,6 @@ export default function CreateJobModal({
     // can't overwrite the current workload's model list/selection.
     let stale = false;
     setIsLoadingModels(true);
-    setModelLoadError(null);
     getModels(inferenceWorkload)
       .then((list) => {
         if (stale) return;
@@ -196,13 +180,7 @@ export default function CreateJobModal({
         }
       })
       .catch((e) => {
-        if (stale) return;
-        console.error('Failed to load models:', e);
-        setModels([]);
-        setModelId('');
-        setModelLoadError(
-          'Models could not be loaded. Check the Studio API and reopen this form to try again.',
-        );
+        if (!stale) console.error('Failed to load models:', e);
       })
       .finally(() => {
         if (!stale) setIsLoadingModels(false);
@@ -215,22 +193,11 @@ export default function CreateJobModal({
   // Training jobs need a dataset; load the ready datasets when relevant.
   React.useEffect(() => {
     if (isOpen && !isInference) {
-      setIsLoadingDatasets(true);
-      setDatasetLoadError(null);
       getDatasets()
         .then(setReadyDatasets)
-        .catch((error) => {
-          console.error('Failed to load datasets:', error);
-          setReadyDatasets([]);
-          setDatasetLoadError(
-            'Datasets could not be loaded. Check the Studio API and reopen this form to try again.',
-          );
-        })
-        .finally(() => setIsLoadingDatasets(false));
+        .catch(() => setReadyDatasets([]));
     } else {
       setReadyDatasets([]);
-      setIsLoadingDatasets(false);
-      setDatasetLoadError(null);
     }
   }, [isOpen, isInference]);
 
@@ -239,24 +206,16 @@ export default function CreateJobModal({
     if (!file) {
       setImagePath('');
       setImageFileName('');
-      setImageUploadError(null);
       return;
     }
     setIsUploadingImage(true);
     setImageFileName(file.name);
-    setImageUploadError(null);
     try {
       const { path } = await uploadImage(file);
       setImagePath(path);
-    } catch (error) {
-      console.error('Failed to upload image:', error);
+    } catch {
       setImagePath('');
       setImageFileName('');
-      setImageUploadError(
-        error instanceof Error
-          ? `${error.message}. Choose the image again to retry.`
-          : 'The image could not be uploaded. Choose it again to retry.',
-      );
     } finally {
       setIsUploadingImage(false);
     }
@@ -265,7 +224,6 @@ export default function CreateJobModal({
   function clearImage() {
     setImagePath('');
     setImageFileName('');
-    setImageUploadError(null);
     if (imageInputRef.current) imageInputRef.current.value = '';
   }
 
@@ -281,7 +239,6 @@ export default function CreateJobModal({
       workloadType === 'lora_t2v' ? 'lora' : jobType
     ) as JobType;
     setIsSubmitting(true);
-    setSubmitError(null);
     try {
       const payload: CreateJobRequest = {
         model_id: modelId,
@@ -339,11 +296,6 @@ export default function CreateJobModal({
       onClose();
     } catch (err) {
       console.error('Failed to create job:', err);
-      setSubmitError(
-        err instanceof Error
-          ? `${err.message}. Check the form and Studio API, then try again.`
-          : 'The job could not be created. Check the form and Studio API, then try again.',
-      );
     } finally {
       setIsSubmitting(false);
     }
@@ -391,11 +343,7 @@ export default function CreateJobModal({
               value={modelId}
               onChange={(e) => setModelId(e.target.value)}
               required
-              aria-describedby={
-                modelLoadError ? 'modal-model-error' : undefined
-              }
-              aria-invalid={modelLoadError ? true : undefined}
-              disabled={isSubmitting || isLoadingModels || !!modelLoadError}
+              disabled={isSubmitting || isLoadingModels}
             >
               <option value="" disabled>
                 {isLoadingModels
@@ -410,15 +358,6 @@ export default function CreateJobModal({
                 </option>
               ))}
             </NativeSelect>
-            {modelLoadError && (
-              <p
-                id="modal-model-error"
-                role="alert"
-                className="text-sm text-destructive"
-              >
-                {modelLoadError}
-              </p>
-            )}
           </FieldRow>
 
           {isInference && workloadType === 'i2v' && (
@@ -430,10 +369,6 @@ export default function CreateJobModal({
                 accept=".png,.jpg,.jpeg,.webp,.bmp"
                 onChange={handleImageChange}
                 disabled={isSubmitting || isUploadingImage}
-                aria-describedby={
-                  imageUploadError ? 'modal-image-error' : undefined
-                }
-                aria-invalid={imageUploadError ? true : undefined}
                 required
                 className="h-auto py-2 file:mr-3 file:cursor-pointer file:rounded-md file:border-0 file:bg-secondary file:px-2 file:py-1 file:text-sm file:text-secondary-foreground"
               />
@@ -450,15 +385,6 @@ export default function CreateJobModal({
                   </button>
                 </span>
               )}
-              {imageUploadError && (
-                <p
-                  id="modal-image-error"
-                  role="alert"
-                  className="text-sm text-destructive"
-                >
-                  {imageUploadError}
-                </p>
-              )}
             </FieldRow>
           )}
 
@@ -506,22 +432,12 @@ export default function CreateJobModal({
                     id="modal-dataset"
                     value={selectedDatasetId}
                     onChange={(e) => setSelectedDatasetId(e.target.value)}
-                    aria-describedby={
-                      datasetLoadError ? 'modal-dataset-error' : undefined
-                    }
-                    aria-invalid={datasetLoadError ? true : undefined}
-                    disabled={
-                      isSubmitting || isLoadingDatasets || !!datasetLoadError
-                    }
+                    disabled={isSubmitting}
                   >
                     <option value="" disabled>
-                      {isLoadingDatasets
-                        ? 'Loading datasets…'
-                        : datasetLoadError
-                          ? 'Datasets unavailable'
-                          : readyDatasets.length === 0
-                            ? 'No datasets (add in Datasets tab)'
-                            : 'Select a dataset…'}
+                      {readyDatasets.length === 0
+                        ? 'No datasets (add in Datasets tab)'
+                        : 'Select a dataset…'}
                     </option>
                     {readyDatasets.map((d) => (
                       <option key={d.id} value={d.id}>
@@ -529,15 +445,6 @@ export default function CreateJobModal({
                       </option>
                     ))}
                   </NativeSelect>
-                  {datasetLoadError && (
-                    <p
-                      id="modal-dataset-error"
-                      role="alert"
-                      className="text-sm text-destructive"
-                    >
-                      {datasetLoadError}
-                    </p>
-                  )}
                 </FieldRow>
                 <FieldRow
                   htmlFor="modal-validation-dataset"
@@ -550,9 +457,7 @@ export default function CreateJobModal({
                     onChange={(e) =>
                       setSelectedValidationDatasetId(e.target.value)
                     }
-                    disabled={
-                      isSubmitting || isLoadingDatasets || !!datasetLoadError
-                    }
+                    disabled={isSubmitting}
                   >
                     <option value="">None</option>
                     {readyDatasets.map((d) => (
@@ -906,22 +811,9 @@ export default function CreateJobModal({
             </details>
           )}
 
-          <div className="flex flex-col items-start gap-2">
-            {submitError && (
-              <p role="alert" className="text-sm text-destructive">
-                {submitError}
-              </p>
-            )}
-            <Button
-              type="submit"
-              disabled={
-                isSubmitting ||
-                isUploadingImage ||
-                !!modelLoadError ||
-                !!datasetLoadError
-              }
-            >
-              {isSubmitting ? 'Creating…' : 'Create Job'}
+          <div>
+            <Button type="submit" disabled={isSubmitting}>
+              {isSubmitting ? 'Creating...' : 'Create Job'}
             </Button>
           </div>
         </form>

apps/fastvideo_studio/src/components/jobs/JobCard.test.tsx

@@ -20,14 +20,6 @@ vi.mock('@/lib/api', () => ({
   downloadJobVideo: vi.fn(),
 }));
 
-vi.mock('@/lib/utils', async (importOriginal) => {
-  const actual = await importOriginal<typeof import('@/lib/utils')>();
-  return {
-    ...actual,
-    downloadBlob: vi.fn(),
-  };
-});
-
 const makeJob = (overrides: Partial<Job> = {}): Job =>
   makeBaseJob({
     model_id: 'Wan2.1-T2V',
@@ -54,16 +46,6 @@ beforeEach(() => {
 });
 
 describe('JobCard', () => {
-  it('keeps selection and job action buttons as semantic siblings', () => {
-    render(<JobCard job={makeJob()} />);
-
-    const selectButton = screen.getByRole('button', { pressed: false });
-    const deleteButton = screen.getByRole('button', { name: 'Delete' });
-
-    expect(selectButton).toHaveTextContent('Wan2.1-T2V');
-    expect(selectButton).not.toContainElement(deleteButton);
-  });
-
   it('renders the model, prompt, status and inference meta', () => {
     render(<JobCard job={makeJob()} />);
     expect(screen.getByText('Wan2.1-T2V')).toBeInTheDocument();

apps/fastvideo_studio/src/components/jobs/JobCard.tsx

@@ -119,10 +119,18 @@ export default function JobCard({ job, onJobUpdated }: JobCardProps) {
     }
   }
 
-  function handleSelectJob() {
+  function handleSelectJob(e: React.MouseEvent | React.KeyboardEvent) {
+    if ((e.target as HTMLElement).closest('button')) return;
     setActiveJobId(isSelected ? null : job.id);
   }
 
+  function handleKeyDown(e: React.KeyboardEvent) {
+    if (e.key === 'Enter' || e.key === ' ') {
+      e.preventDefault();
+      handleSelectJob(e);
+    }
+  }
+
   async function handleDownloadVideo(e: React.MouseEvent) {
     e.preventDefault();
     e.stopPropagation();
@@ -140,7 +148,11 @@ export default function JobCard({ job, onJobUpdated }: JobCardProps) {
   }
 
   return (
-    <article
+    <div
+      role="button"
+      tabIndex={0}
+      onClick={handleSelectJob}
+      onKeyDown={handleKeyDown}
       className={cn(
         'mb-3 flex cursor-pointer flex-col gap-2.5 rounded-lg border bg-background p-4 transition-colors last:mb-0',
         isSelected
@@ -148,42 +160,35 @@ export default function JobCard({ job, onJobUpdated }: JobCardProps) {
           : 'border-border hover:border-muted-foreground/40',
       )}
     >
-      <button
-        type="button"
-        aria-pressed={isSelected}
-        onClick={handleSelectJob}
-        className="flex w-full flex-col gap-2.5 rounded-md text-left"
-      >
-        <span className="flex flex-wrap items-center justify-between gap-2">
-          <span className="text-[0.95rem] font-semibold text-foreground">
-            {job.model_id}
-          </span>
-          <Badge variant={BADGE_VARIANTS[job.status] ?? 'secondary'}>
-            {job.status}
-          </Badge>
+      <div className="flex flex-wrap items-center justify-between gap-2">
+        <span className="text-[0.95rem] font-semibold text-foreground">
+          {job.model_id}
         </span>
-        <span className="max-w-full overflow-hidden text-ellipsis whitespace-nowrap text-sm text-muted-foreground">
-          {job.prompt}
-        </span>
-        <span className="flex flex-wrap items-center gap-4 text-xs text-muted-foreground">
-          {job.job_type === 'inference' ? (
-            <>
-              <span>{job.num_frames} frames</span>
-              <span>
-                {job.height}×{job.width}
-              </span>
-            </>
-          ) : (
-            <span>{job.workload_type?.replace(/_/g, ' ') ?? job.job_type}</span>
-          )}
-          {elapsedTime && (
-            <span className="inline-flex items-center gap-1">
-              <Timer className="size-3.5" aria-hidden />
-              {elapsedTime}
+        <Badge variant={BADGE_VARIANTS[job.status] ?? 'secondary'}>
+          {job.status}
+        </Badge>
+      </div>
+      <p className="max-w-full overflow-hidden text-ellipsis whitespace-nowrap text-sm text-muted-foreground">
+        {job.prompt}
+      </p>
+      <div className="flex flex-wrap items-center gap-4 text-xs text-muted-foreground">
+        {job.job_type === 'inference' ? (
+          <>
+            <span>{job.num_frames} frames</span>
+            <span>
+              {job.height}×{job.width}
             </span>
-          )}
-        </span>
-      </button>
+          </>
+        ) : (
+          <span>{job.workload_type?.replace(/_/g, ' ') ?? job.job_type}</span>
+        )}
+        {elapsedTime && (
+          <span className="inline-flex items-center gap-1">
+            <Timer className="size-3.5" aria-hidden />
+            {elapsedTime}
+          </span>
+        )}
+      </div>
       <div className="flex flex-wrap items-center gap-1.5">
         {job.status === 'running' ? (
           <Button
@@ -235,6 +240,6 @@ export default function JobCard({ job, onJobUpdated }: JobCardProps) {
           Delete
         </Button>
       </div>
-    </article>
+    </div>
   );
 }

apps/fastvideo_studio/src/components/jobs/JobDetailsSidebar.test.tsx

@@ -19,32 +19,6 @@ const makeJob = (overrides: Partial<Job> = {}): Job =>
   });
 
 describe('JobDetailsSidebar', () => {
-  it('fills the mobile viewport without reserving main-content width', async () => {
-    vi.mocked(getJobLogs).mockResolvedValue({
-      lines: [],
-      total: 0,
-      progress: 0,
-      progress_msg: '',
-      phase: '',
-    });
-    const onWidthChange = vi.fn();
-
-    render(
-      <JobDetailsSidebar
-        job={makeJob({ status: 'completed' })}
-        isMobile
-        onClose={vi.fn()}
-        onWidthChange={onWidthChange}
-      />,
-    );
-
-    const drawer = screen.getByRole('dialog', { name: 'Job details' });
-    expect(drawer).toHaveStyle({ width: '100%', maxWidth: 'none' });
-    expect(drawer).toHaveAttribute('aria-modal', 'true');
-    expect(drawer).toHaveFocus();
-    expect(onWidthChange).toHaveBeenCalledWith(0);
-  });
-
   it('renders log lines streamed from the job log poll', async () => {
     vi.mocked(getJobLogs).mockResolvedValue({
       lines: ['boot sequence started', 'loading model weights'],

apps/fastvideo_studio/src/components/jobs/JobDetailsSidebar.tsx

@@ -4,7 +4,6 @@ import * as React from 'react';
 import { X } from 'lucide-react';
 
 import { Button } from '@/components/ui/button';
-import { useDrawerFocus } from '@/hooks/useDrawerFocus';
 import { useResizable } from '@/hooks/useResizable';
 import { downloadJobLog, getJobLogs } from '@/lib/api';
 import type { Job } from '@/lib/types';
@@ -16,16 +15,13 @@ const POLL_INTERVAL_MS = 2000;
 
 export default function JobDetailsSidebar({
   job,
-  isMobile = false,
   onClose,
   onWidthChange,
 }: {
   job: Job;
-  isMobile?: boolean;
   onClose: () => void;
   onWidthChange?: (w: number) => void;
 }) {
-  const drawerRef = useDrawerFocus<HTMLElement>(isMobile);
   const [width, setWidth] = React.useState(360);
   const [isDragging, setIsDragging] = React.useState(false);
   const [isLoading, setIsLoading] = React.useState(false);
@@ -57,8 +53,8 @@ export default function JobDetailsSidebar({
   });
 
   React.useEffect(() => {
-    onWidthChange?.(isMobile ? 0 : width);
-  }, [isMobile, width, onWidthChange]);
+    onWidthChange?.(width);
+  }, [width, onWidthChange]);
 
   // Auto-scroll the console to the bottom whenever new lines land. Runs after
   // commit so scrollHeight reflects the freshly-rendered output.
@@ -141,16 +137,8 @@ export default function JobDetailsSidebar({
 
   return (
     <aside
-      ref={drawerRef}
-      tabIndex={-1}
-      role="dialog"
-      aria-label="Job details"
-      aria-modal={isMobile || undefined}
-      className="fixed bottom-0 right-0 top-[var(--header-height)] z-50 flex max-h-[calc(100dvh-var(--header-height))] min-w-0 shrink-0 flex-col border-l border-border bg-card md:min-w-[280px]"
-      style={{
-        width: isMobile ? '100%' : width,
-        maxWidth: isMobile ? 'none' : SIDEBAR_MAX_WIDTH,
-      }}
+      className="fixed bottom-0 right-0 top-[var(--header-height)] z-50 flex max-h-[calc(100vh-var(--header-height))] min-w-[280px] shrink-0 flex-col border-l border-border bg-card"
+      style={{ width, maxWidth: SIDEBAR_MAX_WIDTH }}
     >
       <div className="flex items-center justify-between border-b border-border px-5 py-4">
         <h2 className="m-0 text-base font-semibold text-foreground">
@@ -207,14 +195,14 @@ export default function JobDetailsSidebar({
         </pre>
       </div>
 
-      {!isMobile && <div
+      <div
         role="presentation"
         onMouseDown={onMouseDown}
         className={cn(
           'absolute bottom-0 left-0 top-0 z-[1] w-1.5 cursor-col-resize hover:bg-accent-blue/25',
           isDragging && 'bg-accent-blue/25',
         )}
-      />}
+      />
     </aside>
   );
 }

apps/fastvideo_studio/src/components/jobs/JobQueue.test.tsx

@@ -1,4 +1,4 @@
-import { act, fireEvent, render, screen, waitFor } from '@testing-library/react';
+import { act, render, screen, waitFor } from '@testing-library/react';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 
 import JobQueue from '@/components/jobs/JobQueue';
@@ -37,46 +37,6 @@ beforeEach(() => {
 });
 
 describe('JobQueue', () => {
-  it('shows a loading placeholder before the initial request settles', async () => {
-    let resolveJobs: (jobs: Job[]) => void = () => {};
-    vi.mocked(getJobsList).mockReturnValue(
-      new Promise<Job[]>((resolve) => {
-        resolveJobs = resolve;
-      }),
-    );
-
-    render(<JobQueue jobType="inference" />);
-
-    expect(screen.getByLabelText('Loading jobs')).toBeInTheDocument();
-    expect(
-      screen.queryByText('No inference jobs yet. Create one above.'),
-    ).not.toBeInTheDocument();
-
-    act(() => resolveJobs([]));
-    expect(
-      await screen.findByText('No inference jobs yet. Create one above.'),
-    ).toBeInTheDocument();
-  });
-
-  it('shows request failures separately from an empty queue and retries', async () => {
-    vi.spyOn(console, 'error').mockImplementation(() => {});
-    vi.mocked(getJobsList).mockRejectedValueOnce(new Error('network down'));
-    render(<JobQueue jobType="inference" />);
-
-    expect(
-      await screen.findByText(/Could not load jobs from the Studio API/),
-    ).toBeInTheDocument();
-    expect(
-      screen.queryByText('No inference jobs yet. Create one above.'),
-    ).not.toBeInTheDocument();
-
-    vi.mocked(getJobsList).mockResolvedValueOnce([]);
-    fireEvent.click(screen.getByRole('button', { name: 'Try Again' }));
-    expect(
-      await screen.findByText('No inference jobs yet. Create one above.'),
-    ).toBeInTheDocument();
-  });
-
   it('shows an empty placeholder and fetches for the single job type', async () => {
     render(<JobQueue jobType="inference" />);
     expect(

apps/fastvideo_studio/src/components/jobs/JobQueue.tsx

@@ -1,10 +1,8 @@
 'use client';
 
 import * as React from 'react';
-import { AlertTriangle } from 'lucide-react';
 
 import JobCard from '@/components/jobs/JobCard';
-import { Button } from '@/components/ui/button';
 import { useStore } from '@/hooks/useStore';
 import { getJobsList } from '@/lib/api';
 import type { Job, JobType } from '@/lib/types';
@@ -34,8 +32,6 @@ function jobsShallowEqual(a: Job | null, b: Job | null): boolean {
 
 export default function JobQueue({ jobType, jobTypesForList }: JobQueueProps) {
   const [jobs, setJobs] = React.useState<Job[]>([]);
-  const [isInitialLoading, setIsInitialLoading] = React.useState(true);
-  const [error, setError] = React.useState<string | null>(null);
   const { nonce } = useStore(jobsRefreshStore);
   const { activeJobId } = useStore(activeJobStore);
 
@@ -75,22 +71,11 @@ export default function JobQueue({ jobType, jobTypesForList }: JobQueueProps) {
               new Date(a.created_at ?? 0).getTime(),
           );
       }
-      if (seq === fetchSeq.current) {
-        setJobs(next);
-        setError(null);
-      }
+      if (seq === fetchSeq.current) setJobs(next);
     } catch (e) {
       console.error('Failed to fetch jobs:', e);
-      if (seq === fetchSeq.current) {
-        setError(
-          'Could not load jobs from the Studio API. Check the server and try again.',
-        );
-      }
     } finally {
-      if (seq === fetchSeq.current) {
-        inFlight.current = false;
-        setIsInitialLoading(false);
-      }
+      if (seq === fetchSeq.current) inFlight.current = false;
     }
   }, [typesKey]);
 
@@ -137,59 +122,20 @@ export default function JobQueue({ jobType, jobTypesForList }: JobQueueProps) {
   const multiType = typesToFetch.length > 1;
 
   return (
-    <div className="mx-auto flex w-full max-w-[850px] flex-col gap-6 px-4 pb-12">
+    <main className="mx-auto flex w-full max-w-[850px] flex-col gap-6 px-4 pb-12">
       <section className="p-6">
-        <div aria-busy={isInitialLoading}>
-          {isInitialLoading ? (
-            <div
-              aria-label="Loading jobs"
-              className="flex flex-col gap-3 py-2"
-            >
-              {[0, 1, 2].map((item) => (
-                <div
-                  key={item}
-                  className="h-32 animate-pulse rounded-lg border border-border bg-muted/50"
-                />
-              ))}
-            </div>
-          ) : error && jobs.length === 0 ? (
-            <div
-              role="alert"
-              className="flex flex-col items-center gap-3 py-8 text-center"
-            >
-              <AlertTriangle
-                className="size-6 text-destructive"
-                aria-hidden
-              />
-              <p className="max-w-md text-sm text-muted-foreground">{error}</p>
-              <Button type="button" variant="outline" onClick={fetchJobs}>
-                Try Again
-              </Button>
-            </div>
-          ) : (
-            <>
-              {error && (
-                <p
-                  role="status"
-                  className="mb-3 rounded-lg border border-amber-500/50 bg-amber-500/10 px-3 py-2 text-sm text-foreground"
-                >
-                  Job updates are temporarily unavailable. Showing the most
-                  recent results.
-                </p>
-              )}
-              {jobs.length === 0 ? (
+        <div>
+          {jobs.length === 0 ? (
             <p className="py-8 text-center text-muted-foreground">
               No {multiType ? 'jobs' : `${jobType} jobs`} yet. Create one above.
             </p>
-              ) : (
-                jobs.map((job) => (
-                  <JobCard key={job.id} job={job} onJobUpdated={fetchJobs} />
-                ))
-              )}
-            </>
+          ) : (
+            jobs.map((job) => (
+              <JobCard key={job.id} job={job} onJobUpdated={fetchJobs} />
+            ))
           )}
         </div>
       </section>
-    </div>
+    </main>
   );
 }

apps/fastvideo_studio/src/components/shell/AppShell.tsx

@@ -9,7 +9,6 @@ import { HeaderActionsProvider } from '@/components/shell/HeaderActionsContext';
 import PrimarySidebar from '@/components/shell/PrimarySidebar';
 import JobDetailsSidebar from '@/components/jobs/JobDetailsSidebar';
 import { Toaster } from '@/components/ui/sonner';
-import { useMediaQuery } from '@/hooks/useMediaQuery';
 import { useStore } from '@/hooks/useStore';
 import {
   activeDatasetStore,
@@ -24,80 +23,46 @@ export function AppShell({ children }: { children: React.ReactNode }) {
   const pathname = usePathname();
   const { activeJob } = useStore(activeJobStore);
   const { activeDataset } = useStore(activeDatasetStore);
-  const isMobile = useMediaQuery('(max-width: 767px)');
 
   const [primaryWidth, setPrimaryWidth] = React.useState(220);
   const [secondaryWidth, setSecondaryWidth] = React.useState(0);
-  const [primaryOpen, setPrimaryOpen] = React.useState(false);
 
   const jobSidebarOpen = JOB_ROUTES.includes(pathname) && activeJob != null;
   const datasetSidebarOpen =
     pathname === '/datasets' && activeDataset != null;
   const secondaryOpen = jobSidebarOpen || datasetSidebarOpen;
-  // Mobile detail drawers claim aria-modal, so everything behind them must
-  // actually be inert — the platform enforces what the ARIA claims.
-  const drawerModal = isMobile && secondaryOpen;
 
   React.useEffect(() => {
     initDefaultOptions();
   }, []);
 
-  React.useEffect(() => {
-    setPrimaryOpen(false);
-  }, [pathname]);
-
   React.useEffect(() => {
     function handleKeyDown(e: KeyboardEvent) {
-      if (e.key !== 'Escape' || document.querySelector('[data-modal]')) return;
-      if (primaryOpen) {
-        setPrimaryOpen(false);
-        return;
+      if (e.key === 'Escape' && !document.querySelector('[data-modal]')) {
+        if (activeJobStore.get().activeJob) setActiveJobId(null);
+        if (activeDatasetStore.get().activeDataset) setActiveDatasetId(null);
       }
-      if (activeJobStore.get().activeJob) setActiveJobId(null);
-      if (activeDatasetStore.get().activeDataset) setActiveDatasetId(null);
     }
     document.addEventListener('keydown', handleKeyDown);
     return () => document.removeEventListener('keydown', handleKeyDown);
-  }, [primaryOpen]);
+  }, []);
 
   return (
     <HeaderActionsProvider>
-      <div
-        style={{ display: 'contents' }}
-        inert={drawerModal ? true : undefined}
-      >
-        <Header
-          navigationOpen={primaryOpen}
-          onNavigationToggle={() => setPrimaryOpen((open) => !open)}
-        />
-      </div>
+      <Header />
       <div
         className="flex overflow-hidden"
         style={{
           marginTop: 'var(--header-height)',
-          height: 'calc(100dvh - var(--header-height))',
+          height: 'calc(100vh - var(--header-height))',
         }}
       >
-        <PrimarySidebar
-          isMobile={isMobile}
-          mobileOpen={primaryOpen}
-          onMobileClose={() => setPrimaryOpen(false)}
-          onWidthChange={setPrimaryWidth}
-        />
-        {primaryOpen && (
-          <button
-            type="button"
-            aria-label="Close navigation"
-            onClick={() => setPrimaryOpen(false)}
-            className="fixed inset-x-0 bottom-0 top-[var(--header-height)] z-40 bg-black/55 md:hidden"
-          />
-        )}
+        <PrimarySidebar onWidthChange={setPrimaryWidth} />
         <main
           className="flex min-w-0 flex-1 flex-col overflow-auto"
-          inert={drawerModal ? true : undefined}
           style={{
-            marginLeft: isMobile ? 0 : primaryWidth,
-            marginRight: isMobile || !secondaryOpen ? 0 : secondaryWidth,
+            marginLeft: primaryWidth,
+            marginRight: secondaryOpen ? secondaryWidth : 0,
           }}
         >
           {children}
@@ -105,7 +70,6 @@ export function AppShell({ children }: { children: React.ReactNode }) {
         {jobSidebarOpen && activeJob && (
           <JobDetailsSidebar
             job={activeJob}
-            isMobile={isMobile}
             onClose={() => setActiveJobId(null)}
             onWidthChange={setSecondaryWidth}
           />
@@ -113,7 +77,6 @@ export function AppShell({ children }: { children: React.ReactNode }) {
         {datasetSidebarOpen && activeDataset && (
           <DatasetSidebar
             dataset={activeDataset}
-            isMobile={isMobile}
             onClose={() => setActiveDatasetId(null)}
             onWidthChange={setSecondaryWidth}
           />

apps/fastvideo_studio/src/components/shell/Header.tsx

@@ -1,10 +1,8 @@
 'use client';
 
-import { Menu, X } from 'lucide-react';
 import { usePathname } from 'next/navigation';
 
 import { useHeaderActions } from '@/components/shell/HeaderActionsContext';
-import { Button } from '@/components/ui/button';
 import { ThemeToggle } from '@/components/ui/theme-toggle';
 
 const TAB_TITLES: Record<string, string> = {
@@ -17,47 +15,25 @@ const TAB_TITLES: Record<string, string> = {
   '/settings': 'Settings',
 };
 
-export default function Header({
-  navigationOpen,
-  onNavigationToggle,
-}: {
-  navigationOpen: boolean;
-  onNavigationToggle: () => void;
-}) {
+export default function Header() {
   const pathname = usePathname();
   const { actions } = useHeaderActions();
   const title = TAB_TITLES[pathname] ?? 'FastVideo';
 
   return (
-    <header className="fixed inset-x-0 top-0 z-[100] flex h-[var(--header-height)] items-center gap-2 border-b border-border bg-background/80 px-2 backdrop-blur sm:px-4 md:gap-6 md:px-6">
-      <Button
-        type="button"
-        variant="outline"
-        size="icon"
-        aria-label={navigationOpen ? 'Close navigation' : 'Open navigation'}
-        aria-controls="primary-navigation"
-        aria-expanded={navigationOpen}
-        onClick={onNavigationToggle}
-        className="shrink-0 md:hidden"
-      >
-        {navigationOpen ? (
-          <X className="size-5" aria-hidden />
-        ) : (
-          <Menu className="size-5" aria-hidden />
-        )}
-      </Button>
+    <header className="fixed inset-x-0 top-0 z-[100] flex h-[var(--header-height)] items-center gap-6 border-b border-border bg-background/80 px-6 backdrop-blur">
       {/* eslint-disable-next-line @next/next/no-img-element */}
       <img
         src="/logo.svg"
         alt="FastVideo Logo"
         width={100}
         height={42}
-        className="hidden h-[42px] w-[78px] shrink-0 object-contain min-[361px]:block md:w-[100px]"
+        className="block h-[42px] w-[100px]"
       />
-      <h1 className="sr-only m-0 flex-1 text-xl font-semibold tracking-tight md:not-sr-only">
+      <h1 className="m-0 flex-1 text-xl font-semibold tracking-tight">
         {title}
       </h1>
-      <div className="ml-auto flex min-w-0 items-center gap-2 md:gap-3">
+      <div className="flex items-center gap-3">
         {actions}
         <ThemeToggle />
       </div>

apps/fastvideo_studio/src/components/shell/PrimarySidebar.tsx

@@ -1,7 +1,6 @@
 'use client';
 
 import * as React from 'react';
-import { X } from 'lucide-react';
 import Link from 'next/link';
 import { usePathname } from 'next/navigation';
 
@@ -20,18 +19,12 @@ const JOB_ROUTES = [
 ] as const;
 
 const TAB_BASE =
-  'block min-h-11 px-5 py-[0.65rem] text-left text-sm text-muted-foreground transition-colors hover:bg-accent/60 hover:text-foreground';
+  'block px-5 py-[0.65rem] text-left text-sm text-muted-foreground transition-colors hover:bg-accent/60 hover:text-foreground';
 const TAB_ACTIVE = 'bg-accent-blue/10 font-medium text-accent-blue';
 
 export default function PrimarySidebar({
-  isMobile,
-  mobileOpen,
-  onMobileClose,
   onWidthChange,
 }: {
-  isMobile: boolean;
-  mobileOpen: boolean;
-  onMobileClose: () => void;
   onWidthChange?: (w: number) => void;
 }) {
   const pathname = usePathname();
@@ -45,8 +38,8 @@ export default function PrimarySidebar({
   const isJobsActive = JOB_ROUTES.some((r) => pathname === r.href);
 
   React.useEffect(() => {
-    onWidthChange?.(isMobile ? 0 : layoutWidth);
-  }, [isMobile, layoutWidth, onWidthChange]);
+    onWidthChange?.(layoutWidth);
+  }, [layoutWidth, onWidthChange]);
 
   React.useEffect(() => {
     if (JOB_ROUTES.some((r) => pathname === r.href)) {
@@ -65,37 +58,11 @@ export default function PrimarySidebar({
 
   return (
     <aside
-      id="primary-navigation"
-      aria-hidden={isMobile && !mobileOpen}
-      inert={isMobile && !mobileOpen ? true : undefined}
-      className={cn(
-        'fixed bottom-0 left-0 top-[var(--header-height)] z-50 flex max-h-[calc(100dvh-var(--header-height))] shrink-0 flex-col border-r border-border bg-card transition-transform duration-200 md:translate-x-0',
-        mobileOpen ? 'translate-x-0' : '-translate-x-full',
-      )}
-      style={{
-        width: isMobile
-          ? 'min(18rem, calc(100vw - 3rem))'
-          : effectiveWidth,
-      }}
+      className="fixed bottom-0 left-0 top-[var(--header-height)] z-50 flex max-h-[calc(100vh-var(--header-height))] shrink-0 flex-col border-r border-border bg-card"
+      style={{ width: effectiveWidth }}
     >
-      {isMobile && (
-        <div className="flex h-14 items-center justify-between border-b border-border px-4">
-          <span className="text-sm font-semibold">Navigation</span>
-          <button
-            type="button"
-            onClick={onMobileClose}
-            aria-label="Close navigation"
-            className="flex size-11 items-center justify-center rounded-lg text-muted-foreground hover:bg-accent hover:text-foreground"
-          >
-            <X className="size-5" aria-hidden />
-          </button>
-        </div>
-      )}
       {!isCollapsed && (
-        <nav
-          aria-label="Primary navigation"
-          className="flex flex-col overflow-y-auto py-2"
-        >
+        <nav className="flex flex-col py-2">
           <div className="flex flex-col">
             <button
               type="button"
@@ -128,8 +95,6 @@ export default function PrimarySidebar({
                   <Link
                     key={route.href}
                     href={route.href}
-                    aria-current={pathname === route.href ? 'page' : undefined}
-                    onClick={onMobileClose}
                     className={cn(
                       TAB_BASE,
                       'px-4 py-2 text-[0.85rem]',
@@ -144,32 +109,24 @@ export default function PrimarySidebar({
           </div>
           <Link
             href="/datasets"
-            aria-current={pathname === '/datasets' ? 'page' : undefined}
-            onClick={onMobileClose}
             className={cn(TAB_BASE, pathname === '/datasets' && TAB_ACTIVE)}
           >
             Datasets
           </Link>
           <Link
             href="/gallery"
-            aria-current={pathname === '/gallery' ? 'page' : undefined}
-            onClick={onMobileClose}
             className={cn(TAB_BASE, pathname === '/gallery' && TAB_ACTIVE)}
           >
             Gallery
           </Link>
           <Link
             href="/gpus"
-            aria-current={pathname === '/gpus' ? 'page' : undefined}
-            onClick={onMobileClose}
             className={cn(TAB_BASE, pathname === '/gpus' && TAB_ACTIVE)}
           >
             GPUs
           </Link>
           <Link
             href="/settings"
-            aria-current={pathname === '/settings' ? 'page' : undefined}
-            onClick={onMobileClose}
             className={cn(TAB_BASE, pathname === '/settings' && TAB_ACTIVE)}
           >
             Settings
@@ -177,7 +134,7 @@ export default function PrimarySidebar({
         </nav>
       )}
 
-      {!isMobile && <div
+      <div
         className={cn(
           'absolute bottom-0 p-2',
           isCollapsed ? '-right-[60px] top-0' : 'right-0',
@@ -188,7 +145,8 @@ export default function PrimarySidebar({
           onClick={() => setIsCollapsed((v) => !v)}
           title={isCollapsed ? 'Expand sidebar' : 'Collapse sidebar'}
           className={cn(
-            'flex size-11 items-center justify-center rounded-lg text-muted-foreground transition-colors hover:bg-accent hover:text-foreground',
+            'flex items-center justify-center rounded-lg text-muted-foreground transition-colors hover:bg-accent hover:text-foreground',
+            isCollapsed ? 'p-3' : 'p-2',
           )}
         >
           <svg
@@ -201,9 +159,9 @@ export default function PrimarySidebar({
             <path d={isCollapsed ? 'M9 18l6-6-6-6' : 'M15 18l-6-6 6-6'} />
           </svg>
         </button>
-      </div>}
+      </div>
 
-      {!isMobile && !isCollapsed && (
+      {!isCollapsed && (
         <div
           role="presentation"
           onMouseDown={onMouseDown}

apps/fastvideo_studio/src/components/system/GpuGrid.test.tsx

@@ -1,4 +1,4 @@
-import { act, render, screen } from '@testing-library/react';
+import { render, screen } from '@testing-library/react';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 
 import GpuGrid from './GpuGrid';
@@ -73,30 +73,4 @@ describe('GpuGrid', () => {
       await screen.findByText(/Could not reach the API server/),
     ).toBeInTheDocument();
   });
-
-  it('keeps the last snapshot visible and warns when a refresh fails', async () => {
-    vi.useFakeTimers();
-    try {
-      vi.mocked(getGpus)
-        .mockResolvedValueOnce(SNAPSHOT)
-        .mockRejectedValueOnce(new Error('network down'));
-
-      render(<GpuGrid />);
-      await act(async () => {
-        await vi.advanceTimersByTimeAsync(0);
-      });
-      expect(screen.getAllByText('NVIDIA B200')).toHaveLength(2);
-
-      await act(async () => {
-        await vi.advanceTimersByTimeAsync(3000);
-      });
-
-      expect(
-        screen.getByText(/values below may be stale/),
-      ).toBeInTheDocument();
-      expect(screen.getAllByText('NVIDIA B200')).toHaveLength(2);
-    } finally {
-      vi.useRealTimers();
-    }
-  });
 });

apps/fastvideo_studio/src/components/system/GpuGrid.tsx

@@ -1,9 +1,7 @@
 'use client';
 
 import * as React from 'react';
-import { AlertTriangle } from 'lucide-react';
 
-import { Button } from '@/components/ui/button';
 import { Card, CardContent } from '@/components/ui/card';
 import { getGpus, type GpuInfo, type GpuSnapshot } from '@/lib/api';
 import { cn } from '@/lib/utils';
@@ -99,8 +97,7 @@ function GpuCard({ gpu }: { gpu: GpuInfo }) {
 
 export default function GpuGrid() {
   const [snapshot, setSnapshot] = React.useState<GpuSnapshot | null>(null);
-  const [fetchError, setFetchError] = React.useState<string | null>(null);
-  const [retryToken, setRetryToken] = React.useState(0);
+  const [fetchError, setFetchError] = React.useState(false);
 
   React.useEffect(() => {
     let mounted = true;
@@ -113,14 +110,10 @@ export default function GpuGrid() {
         const next = await getGpus();
         if (mounted) {
           setSnapshot(next);
-          setFetchError(null);
+          setFetchError(false);
         }
       } catch {
-        if (mounted) {
-          setFetchError(
-            'GPU status could not be refreshed. The values below may be stale.',
-          );
-        }
+        if (mounted) setFetchError(true);
       } finally {
         inFlight = false;
       }
@@ -132,27 +125,14 @@ export default function GpuGrid() {
       mounted = false;
       clearInterval(interval);
     };
-  }, [retryToken]);
+  }, []);
 
   if (fetchError && !snapshot) {
     return (
-      <div
-        role="alert"
-        className="flex flex-col items-center gap-3 py-8 text-center"
-      >
-        <AlertTriangle className="size-6 text-destructive" aria-hidden />
-        <p className="text-muted-foreground">
-          Could not reach the API server. GPU status needs the Studio API server
-          running.
-        </p>
-        <Button
-          type="button"
-          variant="outline"
-          onClick={() => setRetryToken((token) => token + 1)}
-        >
-          Try Again
-        </Button>
-      </div>
+      <p className="py-8 text-center text-muted-foreground">
+        Could not reach the API server. GPU status needs the studio API server
+        running.
+      </p>
     );
   }
   if (!snapshot) {
@@ -170,22 +150,9 @@ export default function GpuGrid() {
   return (
     <div className="flex flex-col gap-4">
       {fetchError && (
-        <div
-          role="status"
-          aria-live="polite"
-          className="flex flex-wrap items-center gap-3 rounded-lg border border-amber-500/50 bg-amber-500/10 px-3 py-2 text-sm"
-        >
-          <AlertTriangle className="size-4 text-amber-600" aria-hidden />
-          <span className="min-w-0 flex-1">{fetchError}</span>
-          <Button
-            type="button"
-            variant="outline"
-            size="sm"
-            onClick={() => setRetryToken((token) => token + 1)}
-          >
-            Refresh Now
-          </Button>
-        </div>
+        <p className="rounded-md border border-amber-500/40 bg-amber-500/10 px-3 py-2 text-sm text-amber-600 dark:text-amber-400">
+          Lost contact with the API server — showing the last known values.
+        </p>
       )}
       <div className="grid gap-4 [grid-template-columns:repeat(auto-fill,minmax(280px,1fr))]">
         {snapshot.gpus.map((gpu) => (

apps/fastvideo_studio/src/components/ui/accessibility.test.tsx

@@ -1,48 +0,0 @@
-import { render, screen } from '@testing-library/react';
-import { describe, expect, it } from 'vitest';
-
-import { Button } from './button';
-import { Input } from './input';
-import { NativeSelect } from './native-select';
-import { Slider } from './slider';
-import { Switch } from './switch';
-
-describe('shared control accessibility', () => {
-  it('keeps button, input, and select targets at least 44px tall', () => {
-    render(
-      <>
-        <Button size="sm">Small action</Button>
-        <Input aria-label="Text value" />
-        <NativeSelect aria-label="Choice" defaultValue="one">
-          <option value="one">One</option>
-        </NativeSelect>
-      </>,
-    );
-
-    expect(screen.getByRole('button', { name: 'Small action' })).toHaveClass(
-      'h-11',
-    );
-    expect(screen.getByRole('textbox', { name: 'Text value' })).toHaveClass(
-      'h-11',
-    );
-    expect(screen.getByRole('combobox', { name: 'Choice' })).toHaveClass(
-      'h-11',
-    );
-  });
-
-  it('uses 44px switch and slider interaction surfaces', () => {
-    render(
-      <>
-        <Switch aria-label="Enabled" />
-        <Slider aria-label="Amount" defaultValue={[50]} />
-      </>,
-    );
-
-    expect(screen.getByRole('switch', { name: 'Enabled' })).toHaveClass(
-      'size-11',
-    );
-    expect(screen.getByRole('slider', { name: 'Amount' })).toHaveClass(
-      'size-11',
-    );
-  });
-});

apps/fastvideo_studio/src/components/ui/badge.tsx

@@ -29,12 +29,11 @@ const badgeVariants = cva(
 );
 
 export interface BadgeProps
-  extends React.HTMLAttributes<HTMLSpanElement>,
+  extends React.HTMLAttributes<HTMLDivElement>,
     VariantProps<typeof badgeVariants> {}
 
-// A span (phrasing content), so badges stay valid inside buttons and links.
 function Badge({ className, variant, ...props }: BadgeProps) {
-  return <span className={cn(badgeVariants({ variant }), className)} {...props} />;
+  return <div className={cn(badgeVariants({ variant }), className)} {...props} />;
 }
 
 export { Badge, badgeVariants };

apps/fastvideo_studio/src/components/ui/button.tsx

@@ -7,7 +7,7 @@ import { cva, type VariantProps } from "class-variance-authority";
 import { cn } from "@/lib/utils";
 
 const buttonVariants = cva(
-	"inline-flex items-center justify-center gap-2 whitespace-nowrap rounded-xl border !text-sm !font-semibold transition-colors duration-150 disabled:pointer-events-none disabled:cursor-not-allowed disabled:opacity-50",
+	"inline-flex items-center justify-center gap-2 whitespace-nowrap rounded-xl border !text-sm !font-semibold transition-colors duration-150 focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-sky-400/40 disabled:pointer-events-none disabled:cursor-not-allowed disabled:opacity-50",
 	{
 		variants: {
 			variant: {
@@ -18,11 +18,11 @@ const buttonVariants = cva(
 				destructive: "border-rose-500/60 bg-rose-600/90 text-white hover:bg-rose-500",
 			},
 			size: {
-				default: "h-11 px-4 py-2",
-				sm: "h-11 rounded-lg px-3 !text-xs",
-				lg: "h-12 px-5 !text-sm",
-				icon: "size-11",
-				"icon-sm": "size-11",
+				default: "h-10 px-4 py-2",
+				sm: "h-9 rounded-lg px-3 !text-xs",
+				lg: "h-11 px-5 !text-sm",
+				icon: "size-10",
+				"icon-sm": "size-8",
 			},
 		},
 		defaultVariants: {

apps/fastvideo_studio/src/components/ui/dialog.tsx

@@ -42,7 +42,7 @@ const DialogContent = React.forwardRef<
       {...props}
     >
       {children}
-      <DialogPrimitive.Close className="absolute right-2 top-2 flex size-11 items-center justify-center rounded-lg text-muted-foreground opacity-70 transition-opacity hover:bg-secondary hover:opacity-100 disabled:pointer-events-none sm:right-4 sm:top-4">
+      <DialogPrimitive.Close className="absolute right-4 top-4 rounded-lg p-1 text-muted-foreground opacity-70 transition-opacity hover:bg-secondary hover:opacity-100 focus:outline-none focus:ring-2 focus:ring-sky-400/40 disabled:pointer-events-none">
         <X className="h-4 w-4" />
         <span className="sr-only">Close</span>
       </DialogPrimitive.Close>

apps/fastvideo_studio/src/components/ui/input.tsx

@@ -9,7 +9,7 @@ const Input = React.forwardRef<HTMLInputElement, React.ComponentProps<"input">>(
 		<input
 			type={type}
 			className={cn(
-				"flex h-11 w-full rounded-xl border border-input bg-card/60 px-3 py-2 text-sm text-foreground shadow-sm backdrop-blur-md transition-colors placeholder:text-muted-foreground focus-visible:border-ring disabled:cursor-not-allowed disabled:opacity-50",
+				"flex h-10 w-full rounded-xl border border-input bg-card/60 px-3 py-2 text-sm text-foreground shadow-sm backdrop-blur-md transition-colors placeholder:text-muted-foreground focus-visible:border-sky-400/70 focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-sky-400/25 disabled:cursor-not-allowed disabled:opacity-50",
 				className,
 			)}
 			ref={ref}

apps/fastvideo_studio/src/components/ui/native-select.tsx

@@ -11,7 +11,7 @@ const NativeSelect = React.forwardRef<
   <select
     ref={ref}
     className={cn(
-      'flex h-11 w-full appearance-none rounded-xl border border-input bg-card px-3 py-2 text-sm text-foreground shadow-sm transition-colors focus-visible:border-ring disabled:cursor-not-allowed disabled:opacity-50',
+      'flex h-10 w-full appearance-none rounded-xl border border-input bg-card px-3 py-2 text-sm text-foreground shadow-sm transition-colors focus-visible:border-sky-400/70 focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-sky-400/25 disabled:cursor-not-allowed disabled:opacity-50',
       className,
     )}
     {...props}

apps/fastvideo_studio/src/components/ui/select.tsx

@@ -17,7 +17,7 @@ const SelectTrigger = React.forwardRef<
   <SelectPrimitive.Trigger
     ref={ref}
     className={cn(
-      'flex h-11 w-full items-center justify-between gap-2 rounded-xl border border-input bg-card px-3 py-2 text-sm text-foreground shadow-sm transition-colors placeholder:text-muted-foreground focus-visible:border-ring disabled:cursor-not-allowed disabled:opacity-50 [&>span]:line-clamp-1',
+      'flex h-10 w-full items-center justify-between gap-2 rounded-xl border border-input bg-card px-3 py-2 text-sm text-foreground shadow-sm outline-none transition-colors placeholder:text-muted-foreground focus:border-sky-400/70 focus:ring-2 focus:ring-sky-400/25 disabled:cursor-not-allowed disabled:opacity-50 [&>span]:line-clamp-1',
       className,
     )}
     {...props}
@@ -117,7 +117,7 @@ const SelectItem = React.forwardRef<
   <SelectPrimitive.Item
     ref={ref}
     className={cn(
-      'relative flex min-h-11 w-full cursor-default select-none items-center rounded-xl py-2 pl-8 pr-3 text-sm text-foreground outline-none data-[disabled]:pointer-events-none data-[disabled]:opacity-50 data-[highlighted]:bg-accent data-[highlighted]:text-accent-foreground',
+      'relative flex w-full cursor-default select-none items-center rounded-xl py-2 pl-8 pr-3 text-sm text-foreground outline-none data-[disabled]:pointer-events-none data-[disabled]:opacity-50 data-[highlighted]:bg-accent data-[highlighted]:text-accent-foreground',
       className,
     )}
     {...props}

apps/fastvideo_studio/src/components/ui/slider.tsx

@@ -8,39 +8,21 @@ import { cn } from '@/lib/utils';
 const Slider = React.forwardRef<
   React.ElementRef<typeof SliderPrimitive.Root>,
   React.ComponentPropsWithoutRef<typeof SliderPrimitive.Root>
->(
-  (
-    {
+>(({ className, ...props }, ref) => (
+  <SliderPrimitive.Root
+    ref={ref}
+    className={cn(
+      'relative flex w-full touch-none select-none items-center',
       className,
-      id,
-      'aria-label': ariaLabel,
-      'aria-labelledby': ariaLabelledBy,
-      'aria-describedby': ariaDescribedBy,
-      ...props
-    },
-    ref,
-  ) => (
-    <SliderPrimitive.Root
-      ref={ref}
-      className={cn(
-        'relative flex h-11 w-full touch-none select-none items-center',
-        className,
-      )}
-      {...props}
-    >
-      <SliderPrimitive.Track className="relative h-1.5 w-full grow overflow-hidden rounded-full bg-border">
-        <SliderPrimitive.Range className="absolute h-full bg-accent-blue" />
-      </SliderPrimitive.Track>
-      <SliderPrimitive.Thumb
-        id={id}
-        aria-label={ariaLabel}
-        aria-labelledby={ariaLabelledBy}
-        aria-describedby={ariaDescribedBy}
-        className="relative block size-11 rounded-full bg-transparent after:absolute after:left-1/2 after:top-1/2 after:size-4 after:-translate-x-1/2 after:-translate-y-1/2 after:rounded-full after:border-2 after:border-accent-blue after:bg-accent-blue after:shadow after:content-[''] hover:after:border-accent-blue/80 disabled:pointer-events-none disabled:opacity-50"
-      />
-    </SliderPrimitive.Root>
-  ),
-);
+    )}
+    {...props}
+  >
+    <SliderPrimitive.Track className="relative h-1.5 w-full grow overflow-hidden rounded-full bg-border">
+      <SliderPrimitive.Range className="absolute h-full bg-accent-blue" />
+    </SliderPrimitive.Track>
+    <SliderPrimitive.Thumb className="block h-4 w-4 rounded-full border-2 border-accent-blue bg-accent-blue shadow transition-colors hover:border-accent-blue/80 focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-sky-400/40 disabled:pointer-events-none disabled:opacity-50" />
+  </SliderPrimitive.Root>
+));
 Slider.displayName = SliderPrimitive.Root.displayName;
 
 export { Slider };

apps/fastvideo_studio/src/components/ui/switch.tsx

@@ -12,14 +12,14 @@ const Switch = React.forwardRef<
   <SwitchPrimitives.Root
     ref={ref}
     className={cn(
-      'peer relative inline-flex size-11 shrink-0 cursor-pointer items-center justify-center rounded-xl bg-transparent transition-colors before:absolute before:h-5 before:w-9 before:rounded-full before:border before:border-border before:bg-background data-[state=checked]:before:border-accent-blue data-[state=checked]:before:bg-accent-blue disabled:cursor-not-allowed disabled:opacity-50',
+      'peer inline-flex h-5 w-9 shrink-0 cursor-pointer items-center rounded-full border border-border transition-colors focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-sky-400/40 disabled:cursor-not-allowed disabled:opacity-50 data-[state=checked]:border-accent-blue data-[state=checked]:bg-accent-blue data-[state=unchecked]:bg-background',
       className,
     )}
     {...props}
   >
     <SwitchPrimitives.Thumb
       className={cn(
-        'pointer-events-none absolute left-1.5 top-[15px] z-[1] block h-3.5 w-3.5 rounded-full bg-muted-foreground shadow-lg ring-0 transition-transform data-[state=checked]:translate-x-4 data-[state=checked]:bg-white',
+        'pointer-events-none block h-3.5 w-3.5 rounded-full bg-muted-foreground shadow-lg ring-0 transition-transform data-[state=checked]:translate-x-4 data-[state=checked]:bg-white data-[state=unchecked]:translate-x-0.5',
       )}
     />
   </SwitchPrimitives.Root>

apps/fastvideo_studio/src/components/ui/tabs.tsx

@@ -29,7 +29,7 @@ const TabsTrigger = React.forwardRef<
   <TabsPrimitive.Trigger
     ref={ref}
     className={cn(
-      'inline-flex items-center justify-center whitespace-nowrap rounded-md px-3 py-1.5 text-sm font-medium text-muted-foreground transition-colors hover:text-foreground disabled:pointer-events-none disabled:opacity-50 data-[state=active]:bg-card data-[state=active]:text-foreground data-[state=active]:shadow-sm',
+      'inline-flex items-center justify-center whitespace-nowrap rounded-md px-3 py-1.5 text-sm font-medium text-muted-foreground transition-colors hover:text-foreground focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-sky-400/40 disabled:pointer-events-none disabled:opacity-50 data-[state=active]:bg-card data-[state=active]:text-foreground data-[state=active]:shadow-sm',
       className,
     )}
     {...props}

apps/fastvideo_studio/src/components/ui/textarea.tsx

@@ -8,7 +8,7 @@ const Textarea = React.forwardRef<HTMLTextAreaElement, React.ComponentProps<"tex
 	return (
 		<textarea
 			className={cn(
-				"flex min-h-24 w-full resize-y rounded-xl border border-input bg-card px-3 py-2 text-sm text-foreground shadow-sm transition-colors placeholder:text-muted-foreground focus-visible:border-ring disabled:cursor-not-allowed disabled:opacity-50",
+				"flex min-h-24 w-full resize-y rounded-xl border border-input bg-card px-3 py-2 text-sm text-foreground shadow-sm transition-colors placeholder:text-muted-foreground focus-visible:border-sky-400/70 focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-sky-400/25 disabled:cursor-not-allowed disabled:opacity-50",
 				className,
 			)}
 			ref={ref}

apps/fastvideo_studio/src/hooks/useDrawerFocus.ts

@@ -1,23 +0,0 @@
-'use client';
-
-import * as React from 'react';
-
-/**
- * Move focus into a drawer when it opens as a modal (mobile), and hand it
- * back to the previously focused element on close. Pairs with `inert` on
- * the background content — together they make `aria-modal` truthful.
- */
-export function useDrawerFocus<T extends HTMLElement>(active: boolean) {
-  const ref = React.useRef<T | null>(null);
-
-  React.useEffect(() => {
-    if (!active) return;
-    const previous = document.activeElement;
-    ref.current?.focus();
-    return () => {
-      if (previous instanceof HTMLElement) previous.focus();
-    };
-  }, [active]);
-
-  return ref;
-}

apps/fastvideo_studio/src/hooks/useMediaQuery.ts

@@ -1,18 +0,0 @@
-'use client';
-
-import * as React from 'react';
-
-export function useMediaQuery(query: string): boolean {
-  const [matches, setMatches] = React.useState(false);
-
-  React.useEffect(() => {
-    const mediaQuery = window.matchMedia(query);
-    const updateMatch = () => setMatches(mediaQuery.matches);
-
-    updateMatch();
-    mediaQuery.addEventListener('change', updateMatch);
-    return () => mediaQuery.removeEventListener('change', updateMatch);
-  }, [query]);
-
-  return matches;
-}

apps/fastvideo_studio/tests/test_gpu.py

@@ -74,8 +74,10 @@ def test_snapshot_shapes_devices(monkeypatch: pytest.MonkeyPatch) -> None:
     }
 
 
-def test_snapshot_tolerates_missing_sensors(monkeypatch: pytest.MonkeyPatch) -> None:
-    monkeypatch.setitem(sys.modules, "pynvml", _make_fake_pynvml(broken_sensors=True))
+def test_snapshot_tolerates_missing_sensors(
+        monkeypatch: pytest.MonkeyPatch) -> None:
+    monkeypatch.setitem(sys.modules, "pynvml",
+                        _make_fake_pynvml(broken_sensors=True))
     snap = gpu_mod.get_gpu_snapshot()
     assert snap["available"] is True
     g = snap["gpus"][0]
@@ -84,7 +86,8 @@ def test_snapshot_tolerates_missing_sensors(monkeypatch: pytest.MonkeyPatch) ->
     assert g["power_limit_watts"] is None
 
 
-def test_snapshot_reports_nvml_failure(monkeypatch: pytest.MonkeyPatch) -> None:
+def test_snapshot_reports_nvml_failure(
+        monkeypatch: pytest.MonkeyPatch) -> None:
     fake = _make_fake_pynvml()
     fake.nvmlInit = lambda: (_ for _ in ()).throw(_NVMLError("driver gone"))
     monkeypatch.setitem(sys.modules, "pynvml", fake)

apps/fastvideo_studio/tests/test_training_config.py

@@ -130,7 +130,8 @@ def test_dmd_builds_three_role_models_and_method_knobs() -> None:
 
 
 def test_dmd_vsa_maps_to_training_vsa_sparsity() -> None:
-    config = build_training_config(_job("dmd_t2v", dmd_use_vsa=True, dmd_vsa_sparsity=0.9), "out")
+    config = build_training_config(
+        _job("dmd_t2v", dmd_use_vsa=True, dmd_vsa_sparsity=0.9), "out")
     assert config["training"]["vsa"]["sparsity"] == 0.9
 
 
@@ -160,27 +161,31 @@ def test_validation_callback_only_when_file_given() -> None:
     without = build_training_config(_job("full_t2v"), "out")
     assert "validation" not in without["callbacks"]
 
-    with_file = build_training_config(_job("full_t2v", validation_dataset_file="val.json"), "out")
+    with_file = build_training_config(
+        _job("full_t2v", validation_dataset_file="val.json"), "out")
     validation = with_file["callbacks"]["validation"]
     assert validation["dataset_file"] == "val.json"
     assert validation["pipeline_target"].endswith(".WanPipeline")
     assert validation["sampling_steps"] == [50]
 
-    dmd = build_training_config(_job("dmd_t2v", validation_dataset_file="val.json"), "out")
+    dmd = build_training_config(
+        _job("dmd_t2v", validation_dataset_file="val.json"), "out")
     validation = dmd["callbacks"]["validation"]
     assert validation["pipeline_target"].endswith(".WanDMDPipeline")
     assert validation["sampling_steps"] == [3]
     assert validation["sampling_timesteps"] == [1000, 757, 522]
 
     # KD/ODE-init has no sampling-based validation pipeline.
-    ode = build_training_config(_job("ode_init", validation_dataset_file="val.json"), "out")
+    ode = build_training_config(
+        _job("ode_init", validation_dataset_file="val.json"), "out")
     assert "validation" not in ode["callbacks"]
 
 
 def test_ltx2_models_are_rejected() -> None:
     assert is_ltx2_model("Lightricks/LTX-2-19B")
     with pytest.raises(ValueError, match="LTX-2 training is not supported"):
-        build_training_config(_job("full_t2v", model_id="Lightricks/LTX-2-19B"), "out")
+        build_training_config(
+            _job("full_t2v", model_id="Lightricks/LTX-2-19B"), "out")
 
 
 def test_unknown_workload_is_rejected() -> None:
@@ -190,7 +195,8 @@ def test_unknown_workload_is_rejected() -> None:
 
 def test_invalid_denoising_steps_are_rejected() -> None:
     with pytest.raises(ValueError, match="Invalid DMD denoising steps"):
-        build_training_config(_job("dmd_t2v", dmd_denoising_steps="1000,abc"), "out")
+        build_training_config(
+            _job("dmd_t2v", dmd_denoising_steps="1000,abc"), "out")
 
 
 def test_training_env_has_no_backend_override() -> None:
@@ -205,7 +211,8 @@ def test_workloads_match_frontend_job_config() -> None:
     (src/lib/jobConfig.ts) — drift means creatable-but-unrunnable jobs."""
     import re
 
-    job_config = (Path(__file__).resolve().parents[1] / "src" / "lib" / "jobConfig.ts").read_text(encoding="utf-8")
+    job_config = (Path(__file__).resolve().parents[1] / "src" / "lib" /
+                  "jobConfig.ts").read_text(encoding="utf-8")
     all_types = set(re.findall(r'type:\s*"([^"]+)"', job_config))
     inference_types = {"t2v", "i2v", "t2i"}
     assert inference_types <= all_types, "jobConfig.ts parse failed"

apps/wantrack_control/README.md

@@ -0,0 +1,35 @@
+# Causal WanTrack Control
+
+This standalone prototype prepares one image and continuously generates causal
+WanTrack blocks. Handle updates received during block N are committed only at
+the next block boundary, so generated frames and control history are immutable.
+One GPU session may generate at a time. The SF checkpoint uses its fixed
+DMD four-step schedule (`method.dmd_denoising_steps`, default
+`[1000, 750, 500, 250]` with warp) without classifier-free guidance; the
+server does not accept client overrides for steps or guidance.
+
+Set a Diffusers-format causal WanTrack export and its Self-Forcing training
+YAML, then launch:
+
+```bash
+export WANTRACK_MODEL_DIR=/path/to/wantrack-causal-export
+export WANTRACK_YAML_PATH=/path/to/sf/config/run.yaml
+export WANTRACK_TAEHV_CHECKPOINT=/path/to/taew2_1.pth
+python -m apps.wantrack_control
+```
+
+Open `http://127.0.0.1:8010`. FFmpeg with `libx264` is required. Completed
+blocks are preserved under `WANTRACK_OUTPUT_DIR` (or the system temporary
+directory) and concatenated into a downloadable MP4 on Stop, disconnect, or a
+recoverable failure.
+
+The interactive preview uses the official TAEHV `StreamingTAEHV` decoder with
+the Wan 2.1 `taew2_1.pth` weights. The full Wan VAE remains loaded only because
+input preparation still needs its encoder.
+
+The `/ws` endpoint accepts `prepare`, `start`, `control_update`, and `stop`
+JSON messages. It emits phase-specific `progress`, `prepared`,
+`session_started`, `block_started`, `block_encoding`, `control_applied`,
+`media_init`, `media_segment_complete`, `stream_complete`, and terminal
+`error` events. Binary frames carry one fMP4 initialization section followed
+by ordered block fragments.

apps/wantrack_control/__init__.py

@@ -0,0 +1,5 @@
+"""Standalone causal WanTrack control prototype."""
+
+from apps.wantrack_control.server import create_app
+
+__all__ = ["create_app"]

apps/wantrack_control/__main__.py

@@ -0,0 +1,18 @@
+"""Launch the standalone WanTrack control server."""
+
+import os
+
+import uvicorn
+
+
+def main() -> None:
+    uvicorn.run(
+        "apps.wantrack_control.server:app",
+        host=os.getenv("WANTRACK_HOST", "127.0.0.1"),
+        port=int(os.getenv("WANTRACK_PORT", "8010")),
+        reload=False,
+    )
+
+
+if __name__ == "__main__":
+    main()

apps/wantrack_control/media.py

@@ -0,0 +1,236 @@
+"""Video-only fragmented-MP4 encoding and completed-prefix finalization."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+import os
+from pathlib import Path
+import shutil
+import subprocess
+from collections.abc import Iterable
+import uuid
+
+import numpy as np
+
+MEDIA_MIME = 'video/mp4; codecs="avc1.42E01E"'
+
+
+@dataclass(frozen=True, slots=True)
+class EncodedMediaSegment:
+    block_index: int
+    init_bytes: bytes
+    media_bytes: bytes
+    path: Path
+    mime: str = MEDIA_MIME
+
+
+def split_fmp4(data: bytes) -> tuple[bytes, bytes]:
+    """Split top-level fMP4 initialization boxes from moof/mdat media."""
+    offset = 0
+    first_fragment: int | None = None
+    while offset + 8 <= len(data):
+        size = int.from_bytes(data[offset:offset + 4], "big")
+        box_type = data[offset + 4:offset + 8]
+        header = 8
+        if size == 1:
+            if offset + 16 > len(data):
+                break
+            size = int.from_bytes(data[offset + 8:offset + 16], "big")
+            header = 16
+        elif size == 0:
+            size = len(data) - offset
+        if size < header or offset + size > len(data):
+            raise ValueError("invalid top-level MP4 box")
+        if box_type == b"moof":
+            first_fragment = offset
+            break
+        offset += size
+    if first_fragment is None:
+        raise ValueError("ffmpeg output did not contain an fMP4 moof box")
+    init_bytes = data[:first_fragment]
+    media_bytes = data[first_fragment:]
+    if b"ftyp" not in init_bytes or b"moov" not in init_bytes:
+        raise ValueError("ffmpeg output is missing fMP4 initialization boxes")
+    return init_bytes, media_bytes
+
+
+class FMP4BlockWriter:
+
+    def __init__(
+        self,
+        output_root: str | os.PathLike[str],
+        *,
+        fps: float,
+        ffmpeg_bin: str | None = None,
+    ) -> None:
+        self.fps = float(fps)
+        if self.fps <= 0:
+            raise ValueError("fps must be positive")
+        resolved_ffmpeg = ffmpeg_bin or shutil.which(os.getenv("WANTRACK_FFMPEG_BIN", "ffmpeg"))
+        if not resolved_ffmpeg:
+            raise RuntimeError("ffmpeg is required for WanTrack streaming")
+        self.ffmpeg_bin = resolved_ffmpeg
+        root = Path(output_root)
+        root.mkdir(parents=True, exist_ok=True)
+        self.session_id = uuid.uuid4().hex
+        self.session_dir = root / self.session_id
+        self.session_dir.mkdir()
+        self._block_paths: list[Path] = []
+
+    @property
+    def block_paths(self) -> tuple[Path, ...]:
+        return tuple(self._block_paths)
+
+    def encode_block(
+        self,
+        frames: np.ndarray,
+        block_index: int,
+    ) -> EncodedMediaSegment:
+        frames = np.asarray(frames)
+        if frames.ndim != 4 or frames.shape[-1] != 3:
+            raise ValueError("frames must have shape [T, H, W, 3]")
+        if frames.shape[0] <= 0:
+            raise ValueError("cannot encode an empty frame block")
+        frames = np.ascontiguousarray(frames, dtype=np.uint8)
+        height, width = int(frames.shape[1]), int(frames.shape[2])
+        gop = max(1, int(frames.shape[0]))
+        command = [
+            self.ffmpeg_bin,
+            "-hide_banner",
+            "-loglevel",
+            "error",
+            "-y",
+            "-f",
+            "rawvideo",
+            "-pix_fmt",
+            "rgb24",
+            "-s:v",
+            f"{width}x{height}",
+            "-r",
+            f"{self.fps:g}",
+            "-i",
+            "pipe:0",
+            "-an",
+            "-c:v",
+            "libx264",
+            "-preset",
+            "ultrafast",
+            "-tune",
+            "zerolatency",
+            "-profile:v",
+            "baseline",
+            "-pix_fmt",
+            "yuv420p",
+            "-g",
+            str(gop),
+            "-keyint_min",
+            str(gop),
+            "-sc_threshold",
+            "0",
+            "-movflags",
+            "+empty_moov+default_base_moof+frag_keyframe",
+            "-frag_duration",
+            str(max(1, round(1_000_000 * frames.shape[0] / self.fps))),
+            "-f",
+            "mp4",
+            "pipe:1",
+        ]
+        result = subprocess.run(
+            command,
+            input=frames.tobytes(),
+            capture_output=True,
+            check=False,
+        )
+        if result.returncode != 0 or not result.stdout:
+            stderr = result.stderr.decode("utf-8", errors="replace").strip()
+            raise RuntimeError(f"ffmpeg failed to encode WanTrack block {block_index}: "
+                               f"{stderr or f'exit {result.returncode}'}")
+        init_bytes, media_bytes = split_fmp4(result.stdout)
+        path = self.session_dir / f"block_{int(block_index):06d}.mp4"
+        path.write_bytes(result.stdout)
+        self._block_paths.append(path)
+        return EncodedMediaSegment(
+            block_index=int(block_index),
+            init_bytes=init_bytes,
+            media_bytes=media_bytes,
+            path=path,
+        )
+
+    def finalize(self) -> Path | None:
+        if not self._block_paths:
+            return None
+        output_path = self.session_dir / "wantrack_control.mp4"
+        concat_path = self.session_dir / "concat.txt"
+        concat_path.write_text(
+            "".join(f"file '{self._concat_escape(path)}'\n" for path in self._block_paths),
+            encoding="utf-8",
+        )
+        command = [
+            self.ffmpeg_bin,
+            "-hide_banner",
+            "-loglevel",
+            "error",
+            "-y",
+            "-f",
+            "concat",
+            "-safe",
+            "0",
+            "-i",
+            str(concat_path),
+            "-an",
+            "-c",
+            "copy",
+            "-movflags",
+            "+faststart",
+            str(output_path),
+        ]
+        result = subprocess.run(
+            command,
+            capture_output=True,
+            check=False,
+        )
+        if result.returncode != 0 or not output_path.is_file():
+            return self._finalize_reencode(output_path)
+        return output_path if output_path.stat().st_size > 0 else None
+
+    @staticmethod
+    def _concat_escape(path: Path) -> str:
+        return str(path.resolve()).replace("'", "'\\''")
+
+    def _finalize_reencode(self, output_path: Path) -> Path | None:
+        concat_path = self.session_dir / "concat.txt"
+        command = [
+            self.ffmpeg_bin,
+            "-hide_banner",
+            "-loglevel",
+            "error",
+            "-y",
+            "-f",
+            "concat",
+            "-safe",
+            "0",
+            "-i",
+            str(concat_path),
+            "-an",
+            "-c:v",
+            "libx264",
+            "-preset",
+            "fast",
+            "-pix_fmt",
+            "yuv420p",
+            "-movflags",
+            "+faststart",
+            str(output_path),
+        ]
+        result = subprocess.run(
+            command,
+            capture_output=True,
+            check=False,
+        )
+        if (result.returncode == 0 and output_path.is_file() and output_path.stat().st_size > 0):
+            return output_path
+        return None
+
+
+def total_size(paths: Iterable[Path]) -> int:
+    return sum(path.stat().st_size for path in paths if path.is_file())

apps/wantrack_control/server.py

@@ -0,0 +1,422 @@
+"""FastAPI/WebSocket server for causal WanTrack control."""
+
+from __future__ import annotations
+
+import asyncio
+import base64
+from contextlib import suppress
+import io
+import os
+from pathlib import Path
+import tempfile
+import time
+from typing import Any
+
+from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect
+from fastapi.responses import FileResponse
+from fastapi.staticfiles import StaticFiles
+from PIL import Image
+
+from apps.wantrack_control.media import FMP4BlockWriter
+
+_STATIC_DIR = Path(__file__).resolve().parent / "static"
+
+
+def _decode_image(value: str) -> bytes:
+    if not isinstance(value, str) or not value.strip():
+        raise ValueError("prepare.image must be a non-empty base64 string")
+    encoded = value.split(",", 1)[1] if value.startswith("data:") else value
+    try:
+        data = base64.b64decode(encoded, validate=True)
+    except Exception as exc:
+        raise ValueError("prepare.image is not valid base64") from exc
+    if not data:
+        raise ValueError("prepare.image decoded to no bytes")
+    return data
+
+
+def _encode_image(image: Image.Image) -> str:
+    output = io.BytesIO()
+    image.save(output, format="PNG")
+    encoded = base64.b64encode(output.getvalue()).decode("ascii")
+    return f"data:image/png;base64,{encoded}"
+
+
+def _block_value(block: Any, key: str, default: Any = None) -> Any:
+    if isinstance(block, dict):
+        return block.get(key, default)
+    return getattr(block, key, default)
+
+
+class _RuntimeProvider:
+
+    def __init__(self, runtime: Any | None) -> None:
+        self._runtime = runtime
+        self._lock = asyncio.Lock()
+
+    async def get(self) -> Any:
+        if self._runtime is not None:
+            return self._runtime
+        async with self._lock:
+            if self._runtime is None:
+                model_dir = os.getenv("WANTRACK_MODEL_DIR", "").strip()
+                yaml_path = os.getenv("WANTRACK_YAML_PATH", "").strip()
+                taehv_checkpoint = os.getenv("WANTRACK_TAEHV_CHECKPOINT", "").strip()
+                if not model_dir or not yaml_path or not taehv_checkpoint:
+                    raise RuntimeError("Set WANTRACK_MODEL_DIR, WANTRACK_YAML_PATH, and "
+                                       "WANTRACK_TAEHV_CHECKPOINT before preparing a session")
+                from fastvideo.train.models.wantrack.runtime import (
+                    WanTrackInferenceRuntime, )
+
+                self._runtime = await asyncio.to_thread(
+                    WanTrackInferenceRuntime.from_export,
+                    model_dir,
+                    yaml_path,
+                    taehv_checkpoint,
+                )
+        return self._runtime
+
+
+def create_app(
+    runtime: Any | None = None,
+    *,
+    output_dir: str | os.PathLike[str] | None = None,
+    writer_factory: Any = FMP4BlockWriter,
+) -> FastAPI:
+    app = FastAPI(title="Causal WanTrack Control")
+    app.state.runtime_provider = _RuntimeProvider(runtime)
+    app.state.active_generation = asyncio.Lock()
+    app.state.downloads = {}
+    if output_dir is None:
+        resolved_output_dir: str | os.PathLike[str] = os.getenv(
+            "WANTRACK_OUTPUT_DIR",
+            str(Path(tempfile.gettempdir()) / "wantrack_control"),
+        )
+    else:
+        resolved_output_dir = output_dir
+    app.state.output_dir = Path(resolved_output_dir)
+    app.state.writer_factory = writer_factory
+
+    app.mount("/static", StaticFiles(directory=_STATIC_DIR), name="static")
+
+    @app.get("/")
+    async def index() -> FileResponse:
+        return FileResponse(_STATIC_DIR / "index.html")
+
+    @app.get("/healthz")
+    async def healthz() -> dict[str, Any]:
+        return {
+            "status": "ok",
+            "active": app.state.active_generation.locked(),
+        }
+
+    @app.get("/downloads/{download_id}")
+    async def download(download_id: str) -> FileResponse:
+        path = app.state.downloads.get(download_id)
+        if path is None or not path.is_file():
+            raise HTTPException(status_code=404, detail="download not found")
+        return FileResponse(
+            path,
+            media_type="video/mp4",
+            filename="wantrack_control.mp4",
+        )
+
+    @app.websocket("/ws")
+    async def websocket_endpoint(websocket: WebSocket) -> None:
+        await websocket.accept()
+        send_lock = asyncio.Lock()
+        prepared: Any | None = None
+        session: Any | None = None
+        generation_task: asyncio.Task[None] | None = None
+        stop_event = asyncio.Event()
+        owns_generation_lock = False
+        connected = True
+
+        async def send_json(payload: dict[str, Any]) -> None:
+            nonlocal connected
+            if not connected:
+                return
+            async with send_lock:
+                try:
+                    await websocket.send_json(payload)
+                except Exception:
+                    connected = False
+                    raise
+
+        async def send_bytes(payload: bytes) -> None:
+            nonlocal connected
+            if not connected:
+                return
+            async with send_lock:
+                try:
+                    await websocket.send_bytes(payload)
+                except Exception:
+                    connected = False
+                    raise
+
+        async def run_generation(runtime_value: Any) -> None:
+            nonlocal owns_generation_lock, connected, generation_task
+            writer = app.state.writer_factory(
+                app.state.output_dir,
+                fps=float(getattr(runtime_value, "fps", 16.0)),
+            )
+            init_sent = False
+            last_applied_revision = 0
+            terminal_error: str | None = None
+            try:
+                while not stop_event.is_set():
+                    block_index = int(getattr(session, "block_index", 0))
+                    block_started_at = time.perf_counter()
+                    await send_json({
+                        "type": "block_started",
+                        "block_index": block_index,
+                        "num_inference_steps": len(
+                            getattr(runtime_value, "dmd_denoising_steps", [1000, 750, 500, 250])),
+                        "dmd_denoising_steps": list(
+                            getattr(runtime_value, "dmd_denoising_steps", [1000, 750, 500, 250])),
+                        "cfg_enabled": False,
+                    })
+                    block = await asyncio.to_thread(session.generate_next_block)
+                    generated_at = time.perf_counter()
+                    applied_revision = int(_block_value(block, "applied_revision", 0))
+                    if applied_revision > last_applied_revision:
+                        last_applied_revision = applied_revision
+                        await send_json({
+                            "type": "control_applied",
+                            "revision": applied_revision,
+                            "block_index": int(_block_value(block, "block_index", block_index)),
+                            "radius": float(_block_value(block, "radius", 0.0)),
+                            "active_handle_ids": list(_block_value(block, "active_handle_ids", ())),
+                        })
+                    frames = _block_value(block, "pixel_frames")
+                    await send_json({
+                        "type": "block_encoding",
+                        "block_index": block_index,
+                    })
+                    encoded = await asyncio.to_thread(
+                        writer.encode_block,
+                        frames,
+                        int(_block_value(block, "block_index", block_index)),
+                    )
+                    encoded_at = time.perf_counter()
+                    if not init_sent:
+                        await send_json({
+                            "type": "media_init",
+                            "mime": encoded.mime,
+                        })
+                        await send_bytes(encoded.init_bytes)
+                        init_sent = True
+                    await send_bytes(encoded.media_bytes)
+                    await send_json({
+                        "type": "media_segment_complete",
+                        "block_index": encoded.block_index,
+                        "bytes": len(encoded.media_bytes),
+                        "generation_ms": round((generated_at - block_started_at) * 1000),
+                        "encoding_ms": round((encoded_at - generated_at) * 1000),
+                    })
+            except Exception as exc:
+                terminal_error = str(exc) or type(exc).__name__
+            finally:
+                if session is not None:
+                    with suppress(Exception):
+                        await asyncio.to_thread(
+                            session.close,
+                            "error" if terminal_error else ("disconnect" if not connected else "stop"),
+                        )
+                final_path = await asyncio.to_thread(writer.finalize)
+                download_url = None
+                if final_path is not None:
+                    app.state.downloads[writer.session_id] = final_path
+                    download_url = f"/downloads/{writer.session_id}"
+                if owns_generation_lock:
+                    app.state.active_generation.release()
+                    owns_generation_lock = False
+                if terminal_error:
+                    if connected:
+                        with suppress(Exception):
+                            await send_json({
+                                "type": "error",
+                                "message": terminal_error,
+                                "download_url": download_url,
+                            })
+                elif connected:
+                    with suppress(Exception):
+                        await send_json({
+                            "type": "stream_complete",
+                            "blocks": len(writer.block_paths),
+                            "download_url": download_url,
+                        })
+                generation_task = None
+
+        async def handle_message(message: dict[str, Any]) -> None:
+            nonlocal prepared, session, generation_task
+            nonlocal owns_generation_lock
+            message_type = str(message.get("type", "")).strip()
+            if message_type == "prepare":
+                if generation_task is not None:
+                    raise ValueError("prepare is unavailable during generation")
+                prepare_started_at = time.perf_counter()
+                await send_json({
+                    "type": "progress",
+                    "phase": "loading_model",
+                    "message": "Loading Track-v0",
+                    "detail": "The first request loads the SF checkpoint onto the GPU.",
+                })
+                runtime_value = await app.state.runtime_provider.get()
+                await send_json({
+                    "type": "progress",
+                    "phase": "preparing_input",
+                    "message": "Preparing image and prompt",
+                    "detail": "Encoding text, the reference image, and its first-frame latent.",
+                })
+                image_bytes = _decode_image(message.get("image", ""))
+                prompt = str(message.get("prompt", "") or "")
+                prepared = await asyncio.to_thread(
+                    runtime_value.prepare,
+                    image_bytes,
+                    prompt,
+                )
+                processed_image = getattr(prepared, "image", None)
+                if not isinstance(processed_image, Image.Image):
+                    processed_image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
+                await send_json({
+                    "type": "prepared",
+                    "image": _encode_image(processed_image),
+                    "width": processed_image.width,
+                    "height": processed_image.height,
+                    "fps": float(getattr(runtime_value, "fps", 16.0)),
+                    "chunk_size": int(getattr(runtime_value, "chunk_size", 3)),
+                    "causal_recipe": getattr(runtime_value, "causal_recipe", {}),
+                    "decoder": str(getattr(runtime_value, "decoder_name", "unknown")),
+                    "prepare_ms": round((time.perf_counter() - prepare_started_at) * 1000),
+                })
+                return
+
+            if message_type == "start":
+                if prepared is None:
+                    raise ValueError("prepare must complete before start")
+                if generation_task is not None:
+                    raise ValueError("session is already generating")
+                handles = message.get("handles")
+                if not isinstance(handles, list) or not handles:
+                    raise ValueError("start requires at least one handle")
+                if app.state.active_generation.locked():
+                    raise RuntimeError("another WanTrack session is already generating")
+                start_started_at = time.perf_counter()
+                await send_json({
+                    "type": "progress",
+                    "phase": "starting_session",
+                    "message": "Starting causal session",
+                    "detail": "Initializing controls and the causal KV cache.",
+                })
+                await app.state.active_generation.acquire()
+                owns_generation_lock = True
+                runtime_value = await app.state.runtime_provider.get()
+                session = runtime_value.create_session()
+                dmd_steps = list(
+                    getattr(runtime_value, "dmd_denoising_steps", [1000, 750, 500, 250]))
+                sampling = {
+                    "seed": int(message.get("seed", 0)),
+                    "num_inference_steps": len(dmd_steps),
+                    "text_guidance_scale": 1.0,
+                    "motion_guidance_scale": 1.0,
+                    "motion_cfg": False,
+                }
+                try:
+                    await asyncio.to_thread(
+                        session.start,
+                        prepared,
+                        getattr(prepared, "prompt", str(message.get("prompt", "") or "")),
+                        handles,
+                        sampling,
+                        radius=float(message.get("radius", 0.15)),
+                    )
+                except Exception:
+                    app.state.active_generation.release()
+                    owns_generation_lock = False
+                    raise
+                await send_json({
+                    "type": "session_started",
+                    "fps": float(getattr(runtime_value, "fps", 16.0)),
+                    "chunk_size": int(getattr(runtime_value, "chunk_size", 3)),
+                    "num_inference_steps": len(dmd_steps),
+                    "dmd_denoising_steps": dmd_steps,
+                    "cfg_enabled": False,
+                    "causal_recipe": getattr(runtime_value, "causal_recipe", {}),
+                    "decoder": str(getattr(runtime_value, "decoder_name", "unknown")),
+                    "start_ms": round((time.perf_counter() - start_started_at) * 1000),
+                })
+                stop_event.clear()
+                generation_task = asyncio.create_task(run_generation(runtime_value))
+                return
+
+            if message_type == "control_update":
+                if session is None:
+                    raise ValueError("control_update requires a running session")
+                revision = int(message.get("revision", 0))
+                accepted = await asyncio.to_thread(
+                    session.apply_control_revision,
+                    revision,
+                    samples=message.get("samples"),
+                    add=message.get("add"),
+                    remove=message.get("remove"),
+                    handles=message.get("handles"),
+                    radius=message.get("radius"),
+                )
+                if not accepted:
+                    await send_json({
+                        "type": "control_applied",
+                        "revision": revision,
+                        "status": "ignored_stale",
+                    })
+                return
+
+            if message_type == "stop":
+                if generation_task is None:
+                    raise ValueError("stop requires a running session")
+                stop_event.set()
+                return
+            raise ValueError(f"unknown client message type: {message_type!r}")
+
+        try:
+            while True:
+                receive_task = asyncio.create_task(websocket.receive_json())
+                waiters: set[asyncio.Task[Any]] = {receive_task}
+                if generation_task is not None:
+                    waiters.add(generation_task)
+                done, _ = await asyncio.wait(
+                    waiters,
+                    return_when=asyncio.FIRST_COMPLETED,
+                )
+                if generation_task is not None and generation_task in done:
+                    receive_task.cancel()
+                    with suppress(asyncio.CancelledError):
+                        await receive_task
+                    await generation_task
+                    return
+                message = await receive_task
+                try:
+                    await handle_message(message)
+                except Exception as exc:
+                    await send_json({
+                        "type": "error",
+                        "message": str(exc) or type(exc).__name__,
+                    })
+                    if generation_task is not None:
+                        stop_event.set()
+        except WebSocketDisconnect:
+            connected = False
+        except Exception:
+            connected = False
+        finally:
+            stop_event.set()
+            if generation_task is not None:
+                with suppress(Exception):
+                    await generation_task
+            elif owns_generation_lock:
+                app.state.active_generation.release()
+
+    return app
+
+
+app = create_app()

apps/wantrack_control/static/app.js

@@ -0,0 +1,308 @@
+(() => {
+  const $ = (id) => document.getElementById(id);
+  const imageInput = $("image");
+  const promptInput = $("prompt");
+  const prepareButton = $("prepare");
+  const canvas = $("canvas");
+  const context = canvas.getContext("2d");
+  const addButton = $("add");
+  const removeButton = $("remove");
+  const gridInput = $("grid");
+  const radiusInput = $("radius");
+  const radiusOutput = document.querySelector(".radius output");
+  const startButton = $("start");
+  const stopButton = $("stop");
+  const video = $("video");
+  const status = $("status");
+  const statusLabel = $("status-label");
+  const statusDetail = $("status-detail");
+  const download = $("download");
+
+  let socket;
+  let preparedImage;
+  let handles = [];
+  let selectedId = null;
+  let addMode = false;
+  let dragging = false;
+  let generating = false;
+  let revision = 0;
+  let sessionStart = 0;
+  let mediaSource;
+  let sourceBuffer;
+  let mediaQueue = [];
+  let streamComplete = false;
+
+  function setStatus(value, detail = "", busy = false) {
+    statusLabel.textContent = value;
+    statusDetail.textContent = detail;
+    status.dataset.busy = String(busy);
+  }
+  function seconds(milliseconds) {
+    return `${(Number(milliseconds || 0) / 1000).toFixed(1)}s`;
+  }
+  function connect() {
+    const scheme = location.protocol === "https:" ? "wss" : "ws";
+    socket = new WebSocket(`${scheme}://${location.host}/ws`);
+    socket.binaryType = "arraybuffer";
+    socket.onopen = () => setStatus("Ready", "Choose an image to begin.");
+    socket.onclose = () => setStatus("Disconnected", "Refresh after the server reconnects.");
+    socket.onerror = () => setStatus("Connection error", "The control server is unreachable.");
+    socket.onmessage = async (event) => {
+      if (typeof event.data !== "string") {
+        mediaQueue.push(event.data);
+        flushMedia();
+        return;
+      }
+      const message = JSON.parse(event.data);
+      if (message.type === "progress") {
+        setStatus(message.message || "Working", message.detail || "", true);
+      } else if (message.type === "prepared") {
+        preparedImage = new Image();
+        preparedImage.onload = () => {
+          canvas.width = message.width;
+          canvas.height = message.height;
+          draw();
+        };
+        preparedImage.src = message.image;
+        handles = [];
+        selectedId = null;
+        addButton.disabled = false;
+        startButton.disabled = true;
+        prepareButton.disabled = false;
+        prepareButton.textContent = "Prepare again";
+        const recipe = message.causal_recipe || {};
+        setStatus(
+          `Prepared in ${seconds(message.prepare_ms)}`,
+          `${message.fps} FPS · ${message.decoder || "unknown decoder"} · ${recipe.rope_cache_policy || "unknown"} RoPE · local ${recipe.local_attn_size ?? "?"} · sink ${recipe.sink_size ?? "?"} · Add a handle, then press Start.`,
+        );
+      } else if (message.type === "session_started") {
+        generating = true;
+        sessionStart = performance.now();
+        startButton.disabled = true;
+        stopButton.disabled = false;
+        setStatus(
+          `Session started in ${seconds(message.start_ms)}`,
+          "4-step SF · CFG off · Building the first causal block.",
+          true,
+        );
+      } else if (message.type === "block_started") {
+        setStatus(
+          `Generating block ${message.block_index}`,
+          "DMD 4-step, single conditional branch. Controls update at the next block.",
+          true,
+        );
+      } else if (message.type === "block_encoding") {
+        setStatus(
+          `Encoding block ${message.block_index}`,
+          "The model is done; packaging frames for immediate playback.",
+          true,
+        );
+      } else if (message.type === "control_applied") {
+        if (message.status === "ignored_stale") {
+          setStatus(`Stale revision ${message.revision} ignored`, "Drag again to send a newer control update.");
+        } else {
+          setStatus(`Control ${message.revision} applied`, "The updated motion is active in this block.");
+        }
+      } else if (message.type === "media_init") {
+        setupMediaSource(message.mime);
+      } else if (message.type === "media_segment_complete") {
+        setStatus(
+          `Playing through block ${message.block_index}`,
+          `Generated in ${seconds(message.generation_ms)} · encoded in ${seconds(message.encoding_ms)} · drag handles for the next block.`,
+        );
+      } else if (message.type === "stream_complete") {
+        generating = false;
+        streamComplete = true;
+        stopButton.disabled = true;
+        startButton.disabled = false;
+        if (message.download_url) {
+          download.href = message.download_url;
+          download.hidden = false;
+        }
+        flushMedia();
+        setStatus(`Complete · ${message.blocks} blocks`, "The final MP4 is ready to download.");
+      } else if (message.type === "error") {
+        generating = false;
+        prepareButton.disabled = false;
+        stopButton.disabled = true;
+        if (message.download_url) {
+          download.href = message.download_url;
+          download.hidden = false;
+        }
+        setStatus("Request failed", message.message || "Unknown error");
+      }
+    };
+  }
+
+  function setupMediaSource(mime) {
+    mediaQueue = [];
+    streamComplete = false;
+    mediaSource = new MediaSource();
+    video.src = URL.createObjectURL(mediaSource);
+    mediaSource.addEventListener("sourceopen", () => {
+      sourceBuffer = mediaSource.addSourceBuffer(mime);
+      sourceBuffer.mode = "sequence";
+      sourceBuffer.addEventListener("updateend", flushMedia);
+      flushMedia();
+    }, { once: true });
+  }
+
+  function flushMedia() {
+    if (!sourceBuffer || sourceBuffer.updating) return;
+    if (mediaQueue.length) {
+      sourceBuffer.appendBuffer(mediaQueue.shift());
+      return;
+    }
+    if (streamComplete && mediaSource && mediaSource.readyState === "open") {
+      mediaSource.endOfStream();
+    }
+    video.play().catch(() => {});
+  }
+
+  function draw() {
+    context.clearRect(0, 0, canvas.width, canvas.height);
+    if (preparedImage) context.drawImage(preparedImage, 0, 0, canvas.width, canvas.height);
+    if (gridInput.checked) {
+      context.strokeStyle = "rgba(255,255,255,.12)";
+      context.lineWidth = 1;
+      for (let index = 0; index < 50; index += 1) {
+        const x = index * canvas.width / 49;
+        const y = index * canvas.height / 49;
+        context.beginPath(); context.moveTo(x, 0); context.lineTo(x, canvas.height); context.stroke();
+        context.beginPath(); context.moveTo(0, y); context.lineTo(canvas.width, y); context.stroke();
+      }
+    }
+    for (const handle of handles) {
+      const x = handle.x * canvas.width;
+      const y = handle.y * canvas.height;
+      context.beginPath();
+      context.arc(x, y, handle.id === selectedId ? 9 : 7, 0, Math.PI * 2);
+      context.fillStyle = handle.id === selectedId ? "#ffcf4a" : "#58c7ff";
+      context.fill();
+      context.strokeStyle = "#111";
+      context.lineWidth = 2;
+      context.stroke();
+    }
+    removeButton.disabled = !selectedId;
+    startButton.disabled = !preparedImage || handles.length === 0 || generating;
+  }
+
+  function canvasPoint(event) {
+    const rect = canvas.getBoundingClientRect();
+    return {
+      x: Math.min(1, Math.max(0, (event.clientX - rect.left) / rect.width)),
+      y: Math.min(1, Math.max(0, (event.clientY - rect.top) / rect.height)),
+    };
+  }
+  function nearest(point) {
+    let match = null;
+    let distance = Infinity;
+    for (const handle of handles) {
+      const value = Math.hypot(handle.x - point.x, handle.y - point.y);
+      if (value < distance && value < 18 / canvas.clientWidth) {
+        match = handle;
+        distance = value;
+      }
+    }
+    return match;
+  }
+  function sendControl(extra = {}) {
+    if (!generating) return;
+    revision += 1;
+    socket.send(JSON.stringify({ type: "control_update", revision, ...extra }));
+  }
+
+  canvas.addEventListener("pointerdown", (event) => {
+    if (!preparedImage) return;
+    const point = canvasPoint(event);
+    if (addMode) {
+      const handle = { id: crypto.randomUUID(), ...point };
+      handles.push(handle);
+      selectedId = handle.id;
+      addMode = false;
+      addButton.textContent = "Add handle";
+      sendControl({ add: [handle] });
+      draw();
+      return;
+    }
+    const handle = nearest(point);
+    selectedId = handle ? handle.id : null;
+    dragging = Boolean(handle);
+    if (dragging) canvas.setPointerCapture(event.pointerId);
+    draw();
+  });
+  canvas.addEventListener("pointermove", (event) => {
+    if (!dragging || !selectedId) return;
+    const point = canvasPoint(event);
+    const handle = handles.find((item) => item.id === selectedId);
+    if (!handle) return;
+    handle.x = point.x; handle.y = point.y;
+    sendControl({ samples: [{ id: handle.id, ...point, timestamp_ms: performance.now() - sessionStart }] });
+    draw();
+  });
+  canvas.addEventListener("pointerup", () => { dragging = false; });
+  addButton.addEventListener("click", () => {
+    addMode = !addMode;
+    addButton.textContent = addMode ? "Click canvas" : "Add handle";
+  });
+  removeButton.addEventListener("click", () => {
+    if (!selectedId) return;
+    const removed = selectedId;
+    handles = handles.filter((item) => item.id !== removed);
+    selectedId = null;
+    sendControl({ remove: [removed] });
+    draw();
+  });
+  gridInput.addEventListener("change", draw);
+  radiusInput.addEventListener("input", () => {
+    radiusOutput.textContent = Number(radiusInput.value).toFixed(2);
+    sendControl({ radius: Number(radiusInput.value) });
+  });
+
+  prepareButton.addEventListener("click", async () => {
+    const file = imageInput.files[0];
+    if (!file) {
+      setStatus("Choose an image", "Prepare needs a reference frame.");
+      return;
+    }
+    prepareButton.disabled = true;
+    prepareButton.textContent = "Preparing…";
+    setStatus("Reading image", "Using the browser's native file reader.", true);
+    try {
+      const image = await new Promise((resolve, reject) => {
+        const reader = new FileReader();
+        reader.onload = () => resolve(reader.result);
+        reader.onerror = () => reject(reader.error || new Error("Failed to read image"));
+        reader.readAsDataURL(file);
+      });
+      setStatus("Sending image", "The server will encode the prompt and reference frame next.", true);
+      socket.send(JSON.stringify({
+        type: "prepare",
+        image,
+        prompt: promptInput.value,
+      }));
+    } catch (error) {
+      prepareButton.disabled = false;
+      prepareButton.textContent = "Prepare";
+      setStatus("Could not read image", error.message || String(error));
+    }
+  });
+  startButton.addEventListener("click", () => {
+    revision = 0;
+    download.hidden = true;
+    startButton.disabled = true;
+    setStatus("Sending controls", "Starting the fixed DMD 4-step, CFG-free SF sampler.", true);
+    socket.send(JSON.stringify({
+      type: "start",
+      handles,
+      radius: Number(radiusInput.value),
+      seed: Number($("seed").value),
+    }));
+  });
+  stopButton.addEventListener("click", () => {
+    socket.send(JSON.stringify({ type: "stop" }));
+    stopButton.disabled = true;
+    setStatus("Finishing current block");
+  });
+  connect();
+})();

apps/wantrack_control/static/index.html

@@ -0,0 +1,56 @@
+<!doctype html>
+<html lang="en">
+<head>
+  <meta charset="utf-8">
+  <meta name="viewport" content="width=device-width,initial-scale=1">
+  <title>WanTrack Control</title>
+  <link rel="stylesheet" href="/static/styles.css">
+</head>
+<body>
+  <main>
+    <header>
+      <div>
+        <h1>WanTrack Control</h1>
+        <p>Drag points to steer the next causal block.</p>
+      </div>
+      <div id="status" role="status" aria-live="polite">
+        <span id="status-label">Disconnected</span>
+        <small id="status-detail">Waiting for the server connection.</small>
+      </div>
+    </header>
+
+    <section class="controls">
+      <label class="file">Image <input id="image" type="file" accept="image/*"></label>
+      <label class="prompt">Prompt <input id="prompt" type="text" placeholder="Optional scene description"></label>
+      <button id="prepare">Prepare</button>
+    </section>
+
+    <section class="workspace">
+      <div class="canvas-panel">
+        <div class="canvas-toolbar">
+          <button id="add" disabled>Add handle</button>
+          <button id="remove" disabled>Delete selected</button>
+          <label><input id="grid" type="checkbox"> Grid</label>
+        </div>
+        <canvas id="canvas" width="832" height="480"></canvas>
+        <label class="radius">Radius <input id="radius" type="range" min="0.03" max="0.45" step="0.01" value="0.15"><output>0.15</output></label>
+      </div>
+      <div class="video-panel">
+        <video id="video" muted autoplay playsinline controls></video>
+        <a id="download" hidden>Download completed MP4</a>
+      </div>
+    </section>
+
+    <section class="sampling">
+      <label>Seed <input id="seed" type="number" value="0" step="1"></label>
+      <div class="recipe">
+        <strong>SF checkpoint</strong>
+        <span>DMD [1000,750,500,250] · CFG off · TAEHV · relativistic RoPE · local 6 · sink 1</span>
+      </div>
+      <button id="start" disabled>Start</button>
+      <button id="stop" disabled>Stop</button>
+    </section>
+  </main>
+  <script src="/static/app.js"></script>
+</body>
+</html>

apps/wantrack_control/static/styles.css

@@ -0,0 +1,57 @@
+:root {
+  color-scheme: dark;
+  font-family: Inter, ui-sans-serif, system-ui, sans-serif;
+  background: #111315;
+  color: #edf0f2;
+}
+* { box-sizing: border-box; }
+body { margin: 0; }
+main { width: min(1180px, calc(100% - 32px)); margin: 24px auto; }
+header { display: flex; align-items: start; justify-content: space-between; gap: 16px; }
+h1 { margin: 0; font-size: 22px; }
+p { margin: 6px 0 20px; color: #929ba3; }
+#status {
+  display: grid; grid-template-columns: auto 1fr; gap: 2px 9px; min-width: 300px;
+  padding: 9px 12px; border-radius: 10px; background: #23282d; color: #dce2e7;
+}
+#status::before {
+  content: ""; grid-row: 1 / 3; align-self: center; width: 8px; height: 8px;
+  border-radius: 50%; background: #62d58b;
+}
+#status[data-busy="true"]::before {
+  width: 12px; height: 12px; border: 2px solid #65717a; border-top-color: #edf0f2;
+  background: transparent; animation: spin .8s linear infinite;
+}
+#status-label { font-size: 12px; font-weight: 700; }
+#status-detail { color: #9ea8b0; font-size: 11px; }
+@keyframes spin { to { transform: rotate(360deg); } }
+.controls, .sampling, .canvas-toolbar { display: flex; align-items: end; gap: 10px; flex-wrap: wrap; }
+label { color: #aeb6bd; font-size: 12px; }
+input[type="text"], input[type="number"], input[type="file"] {
+  display: block; margin-top: 5px; min-height: 36px; border: 1px solid #353b41;
+  border-radius: 7px; background: #1a1e22; color: inherit; padding: 7px 9px;
+}
+.prompt { flex: 1; }
+.prompt input { width: 100%; }
+button, #download {
+  border: 0; border-radius: 7px; background: #e7ebee; color: #111315;
+  min-height: 36px; padding: 8px 13px; font-weight: 650; cursor: pointer;
+}
+button:disabled { opacity: .38; cursor: default; }
+.workspace { display: grid; grid-template-columns: 1fr 1fr; gap: 16px; margin: 16px 0; }
+.canvas-panel, .video-panel { min-width: 0; border: 1px solid #2e3439; background: #181c20; border-radius: 10px; padding: 10px; }
+.canvas-toolbar { margin-bottom: 8px; }
+canvas, video { display: block; width: 100%; aspect-ratio: 832 / 480; object-fit: contain; background: #090a0b; border-radius: 6px; }
+canvas { touch-action: none; cursor: crosshair; }
+.radius { display: flex; align-items: center; gap: 9px; margin-top: 10px; }
+.radius input { flex: 1; }
+#download { display: inline-block; margin-top: 10px; text-decoration: none; }
+#download[hidden] { display: none; }
+.sampling input { width: 92px; }
+.recipe {
+  display: grid; gap: 2px; min-height: 36px; padding: 6px 10px;
+  border: 1px solid #353b41; border-radius: 7px; background: #1a1e22;
+}
+.recipe strong { color: #edf0f2; font-size: 12px; }
+.recipe span { color: #8f99a1; font-size: 11px; }
+@media (max-width: 800px) { .workspace { grid-template-columns: 1fr; } }

apps/wantrack_control/tests/test_protocol.py

@@ -0,0 +1,267 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from pathlib import Path
+import threading
+
+import numpy as np
+from PIL import Image
+from fastapi.testclient import TestClient
+
+from apps.wantrack_control.media import EncodedMediaSegment
+from apps.wantrack_control.server import create_app
+
+
+@dataclass
+class _Prepared:
+    image: Image.Image
+    prompt: str
+
+
+@dataclass
+class _Block:
+    block_index: int
+    pixel_frames: np.ndarray
+    applied_revision: int
+    radius: float = 0.15
+    active_handle_ids: tuple[str, ...] = ("h", )
+
+
+class _FakeSession:
+    def __init__(
+        self,
+        gates: list[threading.Event],
+        *,
+        fail_at: int | None = None,
+    ) -> None:
+        self.gates = gates
+        self.fail_at = fail_at
+        self.block_index = 0
+        self.pending_revision = 0
+        self.closed_reason = None
+
+    def start(self, image, prompt, handles, sampling, *, radius):
+        assert image.prompt == prompt
+        assert handles
+        assert sampling == {
+            "seed": 0,
+            "num_inference_steps": 4,
+            "text_guidance_scale": 1.0,
+            "motion_guidance_scale": 1.0,
+            "motion_cfg": False,
+        }
+        assert radius > 0
+
+    def apply_control_revision(self, revision, **kwargs):
+        del kwargs
+        if revision <= self.pending_revision:
+            return False
+        self.pending_revision = revision
+        return True
+
+    def generate_next_block(self):
+        index = self.block_index
+        applied = self.pending_revision
+        if index < len(self.gates):
+            assert self.gates[index].wait(timeout=5)
+        if self.fail_at == index:
+            raise RuntimeError("fake generation failure")
+        self.block_index += 1
+        return _Block(
+            block_index=index,
+            pixel_frames=np.zeros((2, 8, 8, 3), dtype=np.uint8),
+            applied_revision=applied,
+        )
+
+    def close(self, reason):
+        self.closed_reason = reason
+
+
+class _FakeRuntime:
+    fps = 16.0
+    chunk_size = 3
+    decoder_name = "TAEHV (taew2_1)"
+    causal_recipe = {
+        "local_attn_size": 6,
+        "sink_size": 1,
+        "rope_cache_policy": "relativistic",
+    }
+
+    def __init__(self, session):
+        self.session = session
+
+    def prepare(self, image_bytes, prompt):
+        assert image_bytes
+        return _Prepared(Image.new("RGB", (16, 12), "blue"), prompt)
+
+    def create_session(self):
+        return self.session
+
+
+class _FakeWriter:
+    def __init__(self, output_root, *, fps):
+        assert fps == 16.0
+        self.session_id = "fake-download"
+        self.session_dir = Path(output_root) / self.session_id
+        self.session_dir.mkdir(parents=True, exist_ok=True)
+        self.block_paths = []
+
+    def encode_block(self, frames, block_index):
+        assert frames.shape[-1] == 3
+        path = self.session_dir / f"block-{block_index}.mp4"
+        path.write_bytes(b"complete-block")
+        self.block_paths.append(path)
+        return EncodedMediaSegment(
+            block_index=block_index,
+            init_bytes=b"init",
+            media_bytes=f"media-{block_index}".encode(),
+            path=path,
+        )
+
+    def finalize(self):
+        if not self.block_paths:
+            return None
+        path = self.session_dir / "wantrack_control.mp4"
+        path.write_bytes(b"".join(item.read_bytes()
+                                 for item in self.block_paths))
+        return path
+
+
+def _json(websocket):
+    message = websocket.receive()
+    assert message["type"] == "websocket.send"
+    assert "text" in message
+    import json
+    return json.loads(message["text"])
+
+
+def _bytes(websocket):
+    message = websocket.receive()
+    assert message["type"] == "websocket.send"
+    return message["bytes"]
+
+
+def _prepare_and_start(websocket):
+    import base64
+    import io
+
+    image = io.BytesIO()
+    Image.new("RGB", (8, 8)).save(image, format="PNG")
+    websocket.send_json({
+        "type": "prepare",
+        "image": base64.b64encode(image.getvalue()).decode(),
+        "prompt": "",
+    })
+    assert _json(websocket)["phase"] == "loading_model"
+    assert _json(websocket)["phase"] == "preparing_input"
+    prepared = _json(websocket)
+    assert prepared["type"] == "prepared"
+    assert prepared["prepare_ms"] >= 0
+    assert prepared["causal_recipe"]["rope_cache_policy"] == "relativistic"
+    assert prepared["decoder"] == "TAEHV (taew2_1)"
+    websocket.send_json({
+        "type": "start",
+        "handles": [{
+            "id": "h",
+            "x": 0.5,
+            "y": 0.5,
+        }],
+        "radius": 0.15,
+        # Client overrides are ignored for the fixed SF recipe.
+        "steps": 99,
+        "text_guidance": 9.0,
+        "motion_guidance": 9.0,
+    })
+    assert _json(websocket)["phase"] == "starting_session"
+    started = _json(websocket)
+    assert started["type"] == "session_started"
+    assert started["num_inference_steps"] == 4
+    assert started["cfg_enabled"] is False
+    assert started["causal_recipe"]["local_attn_size"] == 6
+    assert started["decoder"] == "TAEHV (taew2_1)"
+
+
+def test_two_block_binary_order_future_update_and_stop(tmp_path):
+    gates = [threading.Event(), threading.Event()]
+    session = _FakeSession(gates)
+    app = create_app(
+        _FakeRuntime(session),
+        output_dir=tmp_path,
+        writer_factory=_FakeWriter,
+    )
+    with TestClient(app) as client:
+        with client.websocket_connect("/ws") as websocket:
+            _prepare_and_start(websocket)
+            started = _json(websocket)
+            assert started["type"] == "block_started"
+            assert started["block_index"] == 0
+            assert started["num_inference_steps"] == 4
+            assert started["cfg_enabled"] is False
+            websocket.send_json({
+                "type": "control_update",
+                "revision": 1,
+                "samples": [{
+                    "id": "h",
+                    "x": 0.7,
+                    "y": 0.5,
+                    "timestamp_ms": 10,
+                }],
+            })
+            websocket.send_json({
+                "type": "control_update",
+                "revision": 1,
+                "samples": [],
+            })
+            stale = _json(websocket)
+            assert stale["status"] == "ignored_stale"
+            gates[0].set()
+            assert _json(websocket)["type"] == "block_encoding"
+            assert _json(websocket)["type"] == "media_init"
+            assert _bytes(websocket) == b"init"
+            assert _bytes(websocket) == b"media-0"
+            assert _json(websocket)["type"] == "media_segment_complete"
+
+            assert _json(websocket)["type"] == "block_started"
+            websocket.send_json({"type": "stop"})
+            gates[1].set()
+            applied = _json(websocket)
+            assert applied["type"] == "control_applied"
+            assert applied["revision"] == 1
+            assert _json(websocket)["type"] == "block_encoding"
+            assert _bytes(websocket) == b"media-1"
+            assert _json(websocket)["type"] == "media_segment_complete"
+            complete = _json(websocket)
+            assert complete["type"] == "stream_complete"
+            assert complete["blocks"] == 2
+            assert complete["download_url"]
+        response = client.get(complete["download_url"])
+        assert response.status_code == 200
+        assert response.content
+        assert client.get("/healthz").json()["active"] is False
+
+
+def test_error_releases_lock_and_preserves_completed_prefix(tmp_path):
+    gates = [threading.Event(), threading.Event()]
+    gates[0].set()
+    gates[1].set()
+    session = _FakeSession(gates, fail_at=1)
+    app = create_app(
+        _FakeRuntime(session),
+        output_dir=tmp_path,
+        writer_factory=_FakeWriter,
+    )
+    with TestClient(app) as client:
+        with client.websocket_connect("/ws") as websocket:
+            _prepare_and_start(websocket)
+            assert _json(websocket)["type"] == "block_started"
+            assert _json(websocket)["type"] == "block_encoding"
+            assert _json(websocket)["type"] == "media_init"
+            assert _bytes(websocket) == b"init"
+            assert _bytes(websocket) == b"media-0"
+            assert _json(websocket)["type"] == "media_segment_complete"
+            assert _json(websocket)["type"] == "block_started"
+            error = _json(websocket)
+            assert error["type"] == "error"
+            assert error["download_url"]
+        assert client.get(error["download_url"]).content
+        assert client.get("/healthz").json()["active"] is False

docker/Dockerfile

@@ -68,7 +68,7 @@ ARG FLASH_ATTN_WHEEL_RELEASE_ARM64=https://github.com/mjun0812/flash-attention-p
 # cutlass-4.4 `cute.core.ThrMma` API, which crashes on the cutlass-dsl 4.5 that
 # flashinfer/quack pull in. After the wheel install we overlay this cutlass-4.5-safe
 # upstream cute (flash-attn-4) so the image runs FA4 instead of the FA2 fallback.
-ARG FA4_CUTE_REF=14c377950125c70b7a9dabf9c561fca53715ac7d
+ARG FA4_CUTE_REF=82d6441eec5d4dfec120153db2c0145ae855a083
 
 # Provided automatically by BuildKit/buildx (e.g. "amd64" / "arm64") and used to
 # select the prebuilt flash-attn wheel. Empty under a plain `docker build` without
@@ -186,40 +186,15 @@ RUN --mount=type=cache,target=/opt/uv/cache \
 
 COPY . .
 
-# Build immutable FastVideo kernel wheels for the published image. The requested
-# architecture remains installed for normal image users; amd64 images also carry
-# an SM89 artifact so the predominant L40S Modal lanes can reuse it exactly.
-ARG FASTVIDEO_KERNEL_PREBUILT_DIR=/opt/fastvideo-kernel-prebuilt
+# Install FastVideo Unified Kernel exactly once. build.sh initializes only its
+# CUTLASS/ThunderKittens submodules, then compiles for TORCH_CUDA_ARCH_LIST
+# (default Hopper sm_90a) without requiring a live GPU.
 RUN --mount=type=cache,target=/opt/uv/cache \
     source $HOME/.local/bin/env && \
     source /opt/venv/bin/activate && \
-    default_arch="${TORCH_CUDA_ARCH_LIST}" && \
-    default_wheel_dir="${FASTVIDEO_KERNEL_PREBUILT_DIR}/${default_arch}" && \
-    export TORCH_CUDA_ARCH_LIST="${default_arch}" && \
     cd fastvideo-kernel && \
-    CMAKE_ARGS= CMAKE_BUILD_PARALLEL_LEVEL=${CMAKE_BUILD_PARALLEL_LEVEL} \
-    ./build.sh --wheel-dir "${default_wheel_dir}" && \
-    cd /FastVideo && \
-    CMAKE_ARGS= python fastvideo/tests/modal/kernel_build_cache.py write-build-info \
-        --wheel-dir "${default_wheel_dir}" \
-        --output "${default_wheel_dir}/metadata.json" && \
-    if [[ "${TARGETARCH:-amd64}" == "amd64" && "${default_arch}" != "8.9" ]]; then \
-        export TORCH_CUDA_ARCH_LIST=8.9 && \
-        l40s_wheel_dir="${FASTVIDEO_KERNEL_PREBUILT_DIR}/8.9" && \
-        cd /FastVideo/fastvideo-kernel && \
-        CMAKE_ARGS= CMAKE_BUILD_PARALLEL_LEVEL=${CMAKE_BUILD_PARALLEL_LEVEL} \
-        ./build.sh --wheel-dir "${l40s_wheel_dir}" && \
-        cd /FastVideo && \
-        CMAKE_ARGS= python fastvideo/tests/modal/kernel_build_cache.py write-build-info \
-            --wheel-dir "${l40s_wheel_dir}" \
-            --output "${l40s_wheel_dir}/metadata.json"; \
-    fi && \
-    export TORCH_CUDA_ARCH_LIST="${default_arch}" && \
-    default_wheel="$(find "${default_wheel_dir}" -maxdepth 1 -type f \
-        \( -name 'fastvideo_kernel-*.whl' -o -name 'fastvideo-kernel-*.whl' \) \
-        | sort | tail -n 1)" && \
-    uv pip install "${default_wheel}" \
-        --reinstall-package fastvideo-kernel --no-deps
+    CMAKE_BUILD_PARALLEL_LEVEL=${CMAKE_BUILD_PARALLEL_LEVEL} \
+    TORCH_CUDA_ARCH_LIST=${TORCH_CUDA_ARCH_LIST} ./build.sh
 
 # Install FastVideo itself (editable) now that the source is present, and set up
 # shell configuration. Dependencies and the local kernel are already installed,

docs/assets/cookbook-recipes.json

@@ -1,52 +0,0 @@
-{
-  "recipes": [
-    {
-      "id": "fastwan21-t2v",
-      "task": "Text to video",
-      "label": "FastWan2.1 1.3B (distilled + VSA)",
-      "model": "FastVideo/FastWan2.1-T2V-1.3B-Diffusers",
-      "source": "scripts/inference/inference_wan_VSA_DMD_1_3B.yaml",
-      "command": "FASTVIDEO_ATTENTION_BACKEND=VIDEO_SPARSE_ATTN fastvideo generate --config scripts/inference/inference_wan_VSA_DMD_1_3B.yaml"
-    },
-    {
-      "id": "wan22-t2v",
-      "task": "Text to video",
-      "label": "Wan2.2 A14B",
-      "model": "Wan-AI/Wan2.2-T2V-A14B-Diffusers",
-      "source": "examples/inference/basic/basic_wan2_2.py",
-      "command": "python examples/inference/basic/basic_wan2_2.py"
-    },
-    {
-      "id": "wan21-i2v",
-      "task": "Image to video",
-      "label": "Wan2.1 14B 480P",
-      "model": "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers",
-      "source": "scripts/inference/inference_wan_i2v.yaml",
-      "command": "fastvideo generate --config scripts/inference/inference_wan_i2v.yaml"
-    },
-    {
-      "id": "turbowan22-i2v",
-      "task": "Image to video",
-      "label": "TurboWan2.2 A14B",
-      "model": "loayrashid/TurboWan2.2-I2V-A14B-Diffusers",
-      "source": "examples/inference/basic/basic_turbodiffusion_i2v.py",
-      "command": "python examples/inference/basic/basic_turbodiffusion_i2v.py"
-    },
-    {
-      "id": "wan22-ti2v",
-      "task": "Text or image to video",
-      "label": "Wan2.2 TI2V 5B",
-      "model": "Wan-AI/Wan2.2-TI2V-5B-Diffusers",
-      "source": "examples/inference/basic/basic_wan2_2_ti2v.py",
-      "command": "python examples/inference/basic/basic_wan2_2_ti2v.py"
-    },
-    {
-      "id": "matrix-game-2",
-      "task": "Interactive world",
-      "label": "Matrix Game 2.0",
-      "model": "FastVideo/Matrix-Game-2.0-Base-Distilled-Diffusers",
-      "source": "examples/inference/basic/basic_matrixgame2.py",
-      "command": "python examples/inference/basic/basic_matrixgame2.py"
-    }
-  ]
-}

docs/assets/cookbook.js

@@ -1,60 +0,0 @@
-(() => {
-  let recipesPromise;
-
-  const loadRecipes = (url) => {
-    recipesPromise ||= fetch(url).then((response) => {
-      if (!response.ok) throw new Error(`HTTP ${response.status}`);
-      return response.json();
-    });
-    return recipesPromise;
-  };
-
-  const init = () => {
-    document.querySelectorAll("[data-cookbook]").forEach(async (root) => {
-      if (root.dataset.initialized) return;
-      root.dataset.initialized = "true";
-
-      const select = root.querySelector("[data-cookbook-recipe]");
-      const model = root.querySelector("[data-cookbook-model]");
-      const source = root.querySelector("[data-cookbook-source]");
-      const command = root.querySelector("[data-cookbook-command]");
-      const status = root.querySelector("[data-cookbook-status]");
-
-      try {
-        const { recipes } = await loadRecipes(root.dataset.recipes);
-        const byId = new Map(recipes.map((recipe) => [recipe.id, recipe]));
-        const groups = new Map();
-
-        select.replaceChildren();
-        recipes.forEach((recipe) => {
-          if (!groups.has(recipe.task)) {
-            const group = document.createElement("optgroup");
-            group.label = recipe.task;
-            groups.set(recipe.task, group);
-            select.append(group);
-          }
-          groups.get(recipe.task).append(new Option(recipe.label, recipe.id));
-        });
-
-        const render = () => {
-          const recipe = byId.get(select.value);
-          model.textContent = recipe.model;
-          source.textContent = recipe.source;
-          source.href = `https://github.com/hao-ai-lab/FastVideo/blob/main/${recipe.source}`;
-          command.textContent = recipe.command;
-          status.textContent = `${recipe.label} selected.`;
-        };
-
-        select.addEventListener("change", render);
-        select.disabled = false;
-        render();
-      } catch (error) {
-        status.textContent = "Recipes could not be loaded. Use the examples link below.";
-        console.error("Failed to load FastVideo cookbook recipes", error);
-      }
-    });
-  };
-
-  if (window.document$) window.document$.subscribe(init);
-  else document.addEventListener("DOMContentLoaded", init);
-})();

docs/assets/custom.css

@@ -42,46 +42,6 @@ img {
     margin: 0 auto;
 }
 
-.cookbook-picker {
-    padding: 1rem;
-    border: 0.05rem solid var(--md-default-fg-color--lightest);
-    border-radius: 0.2rem;
-}
-
-.cookbook-picker select {
-    width: 100%;
-    padding: 0.6rem;
-    color: var(--md-default-fg-color);
-    background: var(--md-default-bg-color);
-    border: 0.05rem solid var(--md-default-fg-color--lighter);
-    border-radius: 0.2rem;
-}
-
-.cookbook-picker dl {
-    display: grid;
-    grid-template-columns: max-content 1fr;
-    gap: 0.25rem 1rem;
-}
-
-.cookbook-picker dt {
-    font-weight: 700;
-}
-
-.cookbook-picker dd {
-    margin: 0;
-    min-width: 0;
-    overflow-wrap: anywhere;
-}
-
-.cookbook-picker__status {
-    position: absolute;
-    width: 1px;
-    height: 1px;
-    overflow: hidden;
-    clip: rect(0, 0, 0, 0);
-    white-space: nowrap;
-}
-
 .md-typeset .copy-page-button.md-button {
     float: right;
     margin: 0 0 1rem 1rem;

docs/contributing/ci_architecture.md

@@ -269,26 +269,16 @@ The docs job:
 ### Docker Images
 
 `.github/workflows/infra-build-image.yml` supports manual `workflow_dispatch`
-runs and automatically rebuilds the CUDA matrix when a repository-controlled
-image input changes on `main` in the canonical repository. Those inputs include
-the CUDA Dockerfile and reusable workflow, dependency metadata, Docker context
-policy, `fastvideo-kernel/**`, and the kernel artifact metadata/key helper.
-Manual runs let maintainers choose which image families to build. The
-`fastvideo-dev` matrix builds Python 3.12 images for CUDA 12.6 and CUDA 13 on
-native `amd64` and `arm64` runners, then publishes one multi-platform manifest
-per CUDA version. CUDA 12.6 owns the `py3.12-latest` and global `latest` tags, as
-well as the explicit `py3.12-cuda12.6.3-latest` alias. CUDA 13 is published under
-the explicit `py3.12-cuda13.0.0-latest` tag. This publication policy does not
-change the unparameterized `docker/Dockerfile` build defaults, which remain CUDA
-13 and `cu130`.
-
-Published amd64 development images keep their configured Hopper kernel wheel
-installed and also carry an immutable SM89 wheel under
-`/opt/fastvideo-kernel-prebuilt`. Modal PR and SSIM jobs select the exact
-source, ABI, and GPU-architecture match from that directory, so L40S jobs reuse
-the trusted image artifact while kernel-changing PRs still build locally. Once
-a kernel or artifact-key change reaches `main`, the image workflow republishes
-the matching trusted artifact before later jobs consume the updated image tag.
+runs and automatically rebuilds the CUDA matrix when `docker/Dockerfile`
+changes on `main` in the canonical repository. Manual runs let maintainers
+choose which image families to build. The `fastvideo-dev` matrix builds Python
+3.12 images for CUDA 12.6 and CUDA 13 on native `amd64` and `arm64` runners,
+then publishes one multi-platform manifest per CUDA version. CUDA 12.6 owns the
+`py3.12-latest` and global `latest` tags, as well as the explicit
+`py3.12-cuda12.6.3-latest` alias. CUDA 13 is published under the explicit
+`py3.12-cuda13.0.0-latest` tag. This publication policy does not change the
+unparameterized `docker/Dockerfile` build defaults, which remain CUDA 13 and
+`cu130`.
 
 The optional Dreamverse matrix builds backend and UI images for CUDA 12.6 and
 CUDA 13 on `amd64`. Dreamverse remains `amd64`-only because its FA4 dependency

docs/cookbook/index.md

@@ -1,44 +0,0 @@
-# Inference Cookbook
-
-Choose a complete recipe maintained in the FastVideo repository. Each command
-runs its checked-in source directly, so coupled model, GPU, offload, and
-attention settings do not drift into unsupported combinations.
-
-The commands expect a local clone:
-
-```bash
-git clone https://github.com/hao-ai-lab/FastVideo.git
-cd FastVideo
-```
-
-<div class="cookbook-picker" data-cookbook data-recipes="../assets/cookbook-recipes.json">
-  <label for="cookbook-recipe"><strong>Recipe</strong></label>
-  <select id="cookbook-recipe" data-cookbook-recipe disabled>
-    <option>Loading recipes…</option>
-  </select>
-  <dl>
-    <dt>Model</dt>
-    <dd data-cookbook-model>Loading…</dd>
-    <dt>Source</dt>
-    <dd><a data-cookbook-source href="../inference/examples/basic/">Browse maintained examples</a></dd>
-  </dl>
-  <pre><code class="language-bash" data-cookbook-command>Loading…</code></pre>
-  <p class="cookbook-picker__status" role="status" aria-live="polite" data-cookbook-status></p>
-  <noscript>
-    JavaScript is needed for the recipe picker. Browse the
-    <a href="../inference/examples/examples_inference_index/">inference examples</a>
-    instead.
-  </noscript>
-</div>
-
-## Customize a recipe
-
-Start from the checked-in source, then change only the settings your model
-supports:
-
-- [Configuration](../inference/configuration.md) covers the Python and CLI
-  config surfaces.
-- [Optimizations](../inference/optimizations.md) covers attention backends,
-  compilation, and memory tradeoffs.
-- [Support matrix](../inference/support_matrix.md) lists supported models and
-  optimizations.

docs/design/apple_silicon_fast_mode.md

@@ -1,128 +0,0 @@
-# Fast mode (RIFE) — Apple Silicon
-
-`--fast` makes local generation ~2.7× faster by **generating fewer frames and
-interpolating the rest** with an Apple-Silicon-native RIFE model, instead of
-denoising every frame. Video-diffusion denoise is dominated by self-attention,
-which is O(tokens²); halving the frames cuts the token count ~2× and the denoise
-compute ~3.7×, so the wall-clock drops far more than 2×. RIFE (which estimates
-its own optical flow — no motion vectors needed) fills the dropped frames back
-in for ~1.4 s, and a light unsharp pass counters its softening.
-
-Measured on the 1.3B INT8 QAD model (fox, 480×832×81, M4): generate 41 + RIFE→81
-runs in ~35 s of denoise vs ~90 s full, at reconstruction MS-SSIM **0.97**.
-Reproduce with `python -m fastvideo.benchmarks.eval_metalfx_rife --mode int8`.
-
-> **Note:** Apple's *MetalFX* frame interpolation is **not** usable here — it
-> requires game-engine motion vectors + depth, which diffusion output lacks. We
-> use the video-native **`rife-mlx`** model instead (Metal-backed, torch-free).
-
-## Install
-
-```bash
-uv pip install -e ".[mlx]"   # RIFE ships vendored; this only needs MLX
-```
-
-## Use
-
-```bash
-python examples/inference/basic/mlx_wan_prompt_to_video.py \
-  --mlx-checkpoint <FastWan2.1-T2V-1.3B-INT8-QAD> \
-  --prompt "A red fox trotting through a snowy pine forest at golden hour, cinematic" \
-  --num-frames 81 --fast \
-  --output-path video_samples/fox_fast.mp4
-```
-
-`--num-frames` stays the *target* length; fast mode generates the smallest
-VAE-aligned keyframe count that RIFE can interpolate to that target.
-
-| Flag | Default | Meaning |
-|---|---|---|
-| `--fast` / `--no-fast` | off | enable fast mode |
-| `--fast-factor` | 2 | generate 1/factor of the frames (2 = half) |
-| `--fast-sharpen` | 0.6 | light unsharp strength to counter RIFE softness (0 disables) |
-
-Fast mode composes with everything else (`--mlx-quantization int8`,
-`--mlx-compile`, TAEHV vs `--decode-backend wan-vae`). Keep `--fast-factor` at 2
-for quality — larger temporal gaps are where RIFE starts inventing motion.
-
-## Spatial fast mode (`--fast-spatial`)
-
-The spatial twin of `--fast`: instead of dropping frames, drop pixels. Denoise
-*and decode* at `height/width // fast-spatial-scale`, then resample the decoded
-frames up to the requested size. Self-attention is O(tokens²), so halving each
-spatial axis cuts the token count 4× and the denoise time far more than that —
-measured on the 1.3B INT8 QAD model at 480×832×81, M4 Max: **86.1 s → 10.3 s**
-of denoise. It composes with `--fast`; both together run the same clip in
-**4.5 s** of denoise.
-
-```bash
-python examples/inference/basic/mlx_wan_prompt_to_video.py \
-  --prompt "A red fox trotting through a snowy pine forest at golden hour, cinematic" \
-  --height 480 --width 832 --num-frames 81 --fast-spatial \
-  --output-path video_samples/fox_fast_spatial.mp4
-```
-
-| Flag | Default | Meaning |
-|---|---|---|
-| `--fast-spatial` / `--no-fast-spatial` | off | enable spatial fast mode |
-| `--fast-spatial-scale` | 2 | denoise at 1/scale of each spatial axis |
-| `--fast-spatial-upsample-mode` | `lanczos` | pixel interpolation kernel (`lanczos`, `cubic`, `bilinear`, `nearest`) |
-| `--fast-spatial-sharpen` | 0.4 | light unsharp strength to counter resampling softness (0 disables) |
-
-### The upsample must happen in pixel space
-
-This is the one thing to get right. The obvious implementation — bilinearly
-upsample the finished latents and decode at the target size — **does not work**,
-and produces a distinctive failure: correct composition and silhouette under a
-smeared, hazy veil, with ringing along strong edges.
-
-A Wan latent cell is a *learned code* for an 8×8 (Wan2.1) or 16×16 (Wan2.2)
-pixel block, not a low-pass sample of the image. The average of two adjacent
-codes is not the code of the averaged blocks; it is a vector the decoder was
-never trained on. Measured on Wan2.1-1.3B at 480×832, a 2× bilinear latent
-upsample destroys **62%** of the latent's high-frequency energy while leaving
-its overall magnitude intact — exactly the signature of that veil. At Wan2.2-5B
-the same operation degrades to black or noise.
-
-Decoded RGB frames have no such problem: an image *is* a sampled 2-D signal, so
-Lanczos interpolation is the operation it was defined for. The result is soft —
-it carries stage-1's real detail budget and no more — but clean and coherent.
-
-`--refine` gets away with a latent-space upsample only because a second DMD pass
-re-denoises the hand-off; spatial fast mode passes the latent straight to the
-decoder, so it cannot.
-
-## Refine (`--refine`) stage-2 timesteps
-
-`--refine` hands stage 1 to stage 2 as `(1 - sigma) * upsampled + sigma * noise`,
-where `sigma` comes from the *first* stage-2 timestep. FastWan's DMD grid opens
-at `t=1000`, which is `sigma == 1` exactly — so a stage-2 grid that starts there
-weights the stage-1 result at zero and refine silently degrades into a plain
-full-resolution run at twice the cost.
-
-Left unset, `--refine-dmd-denoising-steps` now derives the stage-2 grid from the
-stage-1 one with leading full-noise steps dropped (`1000,757,522` → `757,522`).
-That keeps the pass on timesteps the distilled student was trained on while
-letting stage-1 structure through: hand-off `sigma = 0.757`, stage-1 weight
-`0.243`. Passing a grid that starts at full noise is now an error rather than a
-silently wasted pass.
-
-The run prints the resolved hand-off so it is visible:
-
-```
-[refine] stage-2 hand-off sigma=0.7568 (stage-1 weight 0.2432)
-```
-
-There is a trade-off in choosing that grid. Later start = more of the draft
-survives, but fewer stage-2 steps. On Wan2.1 the default `757,522` gives weight
-0.243 with two steps; `--refine-dmd-denoising-steps 522` gives weight 0.478 with
-one. `--refine-sigma` decouples the noise level from the timestep entirely — it
-logs a warning, because the DiT is then told a timestep that does not match the
-noise it receives.
-
-**Wan2.2-5B has a lower ceiling.** Its warped schedule maps `1000,757,522` to
-sigmas `1.000, 0.940, 0.845`, so the best available stage-1 weight is **0.060**
-(vs 0.243 at 1.3B). Un-warped (`--no-warp`) the same grid gives `1.000, 0.757,
-0.522` and a weight of 0.243 — but warping is what matches the FastVideo
-sampling schedule, so turning it off changes the timesteps the distilled student
-sees. Which is better at 5B is unresolved and needs a run on real 5B weights.

docs/design/inference_schema_parity_inventory.yaml

@@ -76,11 +76,6 @@ surfaces:
       lora_target_modules: "Legacy LoRA configuration surface pending dedicated component API."
       output_type: "Legacy output formatting surface pending GenerationResult cleanup."
       VSA_sparsity: "Model-specific inference optimization not yet represented in the typed public schema."
-      VSA_tile_size: "VSA-H3 tile geometry request; model-specific optimization not yet represented in the typed public schema."
-      vae_parallel_decode: "MiniMax-H3 sequence-parallel VAE decode opt-in; model-specific optimization not yet represented in the typed public schema."
-      vae_parallel_encode: "MiniMax-H3 sequence-parallel reference VAE encode opt-in; model-specific optimization not yet represented in the typed public schema."
-      vae_parallel_decode_strategy: "Chunk-transport collective for vae_parallel_decode; model-specific optimization not yet represented in the typed public schema."
-      attention_backend: "Process-wide default attention-backend request applied per component at load time; kernel-selection knob not yet represented in the typed public schema."
       moba_config_path: "Model-specific MoBA optimization surface not yet represented in the typed public schema."
       master_port: "Executor/bootstrap compatibility field; not part of the canonical inference schema."
       refine_transformer_path: "Generic stage-2 refine transformer override; no typed equivalent yet."
@@ -449,11 +444,7 @@ surfaces:
     moved:
       image_path: request.inputs.image_path
       pil_image: request.inputs.pil_image
-      last_image: request.inputs.last_image
-      references: request.inputs.references
       video_path: request.inputs.video_path
-      latents: request.inputs.latents
-      audio_latents: request.inputs.audio_latents
       mouse_cond: request.inputs.mouse_cond
       keyboard_cond: request.inputs.keyboard_cond
       grid_sizes: request.inputs.grid_sizes
@@ -533,6 +524,7 @@ surfaces:
       inpaint_mask: request.extensions.stable_audio.inpaint_mask
     internal_only:
       data_type: "Derived from the request shape and not a public input."
+      latents: "Pre-generated diffusion latents supplied by parity/debug harnesses; not a public input."
 
   sampling_param_extensions: {}
 

docs/generate_examples.py

@@ -2,7 +2,6 @@
 # adapted from vllm: https://github.com/vllm-project/vllm/blob/v0.7.3/docs/source/generate_examples.py
 
 import itertools
-import json
 import os
 import re
 from dataclasses import dataclass, field
@@ -20,40 +19,6 @@ GENERATED_DOC_PREFIXES = (
     "training/examples/",
     "distillation/examples/",
 )
-COOKBOOK_DATA = ROOT_DIR / "docs/assets/cookbook-recipes.json"
-COOKBOOK_SOURCE_ROOTS = (
-    ROOT_DIR / "examples/inference",
-    ROOT_DIR / "scripts/inference",
-)
-
-
-def validate_cookbook() -> None:
-    """Keep cookbook entries tied to checked-in runnable sources."""
-    recipes = json.loads(COOKBOOK_DATA.read_text(encoding="utf-8")).get("recipes")
-    if not isinstance(recipes, list) or not recipes:
-        raise ValueError(f"{COOKBOOK_DATA}: recipes must be a non-empty list")
-
-    seen: set[str] = set()
-    for recipe in recipes:
-        required = ("id", "task", "label", "model", "source", "command")
-        missing = {key for key in required if not recipe.get(key)}
-        if missing:
-            raise ValueError(f"Cookbook recipe is missing: {', '.join(sorted(missing))}")
-        if recipe["id"] in seen:
-            raise ValueError(f"Duplicate cookbook recipe id: {recipe['id']}")
-        seen.add(recipe["id"])
-
-        source = (ROOT_DIR / recipe["source"]).resolve()
-        if not any(source.is_relative_to(root.resolve()) for root in COOKBOOK_SOURCE_ROOTS):
-            raise ValueError(f"Cookbook source is outside an approved directory: {recipe['source']}")
-        if not source.is_file():
-            raise ValueError(f"Cookbook source does not exist: {recipe['source']}")
-
-        source_text = source.read_text(encoding="utf-8")
-        if recipe["model"] not in source_text:
-            raise ValueError(f"Cookbook model is not present in {recipe['source']}: {recipe['model']}")
-        if recipe["source"] not in recipe["command"]:
-            raise ValueError(f"Cookbook command does not invoke its source: {recipe['id']}")
 
 
 def fix_case(text: str) -> str:
@@ -571,7 +536,6 @@ def on_pre_build(config, **kwargs):
     MkDocs hook to generate examples before building the documentation.
     This function is called automatically by MkDocs' native hook system.
     """
-    validate_cookbook()
     print("Generating example documentation...")
     generate_examples(generate_main_index=True)
     print("Example documentation generated successfully!")
@@ -585,7 +549,6 @@ def on_page_context(context, page, **kwargs):
 
 
 if __name__ == "__main__":
-    validate_cookbook()
     print("Generating example documentation...")
     generate_examples(generate_main_index=True)
     print("Example documentation generated successfully!")

docs/getting_started/installation.md

@@ -5,7 +5,6 @@ FastVideo supports the following hardware platforms:
 
 - [NVIDIA CUDA](installation/gpu.md)
 - [NVIDIA DGX Spark / GB10 (ARM64 + CUDA 13)](installation/spark.md)
-  ([performance & tuning](installation/spark_performance.md))
 - [Apple silicon](installation/mps.md)
 
 ## Quick Installation
@@ -55,17 +54,14 @@ UV_TORCH_BACKEND=cu126 uv pip install -e .
 uv pip install flash-attn --no-build-isolation -v
 ```
 
-## Requirements
+## Hardware Requirements
 
-- **Python**: 3.10-3.12 is the tested and recommended range (the commands
-  above pin 3.12)
 - **NVIDIA GPUs**: CUDA 12.6+ with compute capability 7.0+
 - **Apple Silicon**: macOS 14.0+ with M1/M2/M3/M4 chips
 - **CPU**: x86_64 architecture (for CPU-only inference)
 
 ## Next Steps
 
-- [Quick Start](quick_start.md) - Generate your first video
-- [Inference Cookbook](../cookbook/index.md) - Choose a maintained recipe
+- [Quick Start Guide](quick_start.md) - Get started with your first video generation
 - [Configuration](../inference/configuration.md) - Learn about configuration options
-- [Examples](../inference/examples/examples_inference_index.md) - Explore scripts and notebooks
+- [Examples](../inference/examples/examples_inference_index.md) - Explore example scripts and notebooks

docs/getting_started/installation/gpu.md

@@ -134,4 +134,4 @@ If you're planning to contribute to FastVideo please see the following page:
 
 If you encounter any issues during installation, please open an issue on our [GitHub repository](https://github.com/hao-ai-lab/FastVideo).
 
-You can also join our [Slack community](https://join.slack.com/t/fastvideo/shared_invite/zt-3f4lao1uq-u~Ipx6Lt4J27AlD2y~IdLQ) for additional support.
+You can also join our [Slack community](https://join.slack.com/t/fastvideo/shared_invite/zt-38u6p1jqe-yDI1QJOCEnbtkLoaI5bjZQ) for additional support.

docs/getting_started/installation/mps.md

@@ -49,12 +49,10 @@ brew install ffmpeg
 
 ### Installation
 
-FastWan's native Apple Silicon runtime requires the `mlx` extra.
-
 #### With uv (recommended)
 
 ```bash
-uv pip install "fastvideo[mlx]"
+uv pip install fastvideo
 ```
 
 #### With Conda environment (alternative)
@@ -62,7 +60,7 @@ uv pip install "fastvideo[mlx]"
 `uv` works inside an active conda env too, so prefer `uv pip` for the actual install:
 
 ```bash
-uv pip install "fastvideo[mlx]"
+uv pip install fastvideo
 ```
 
 ### Installation from Source
@@ -78,13 +76,13 @@ git clone https://github.com/hao-ai-lab/FastVideo.git && cd FastVideo
 Basic installation:
 
 ```bash
-uv pip install -e ".[mlx]"
+uv pip install -e .
 ```
 
 Alternative with Conda environment:
 
 ```bash
-uv pip install -e ".[mlx]"
+uv pip install -e .
 ```
 
 ## Development Environment Setup
@@ -102,4 +100,4 @@ If you're planning to contribute to FastVideo please see the following page:
 
 If you encounter any issues during installation, please open an issue on our [GitHub repository](https://github.com/hao-ai-lab/FastVideo).
 
-You can also join our [Slack community](https://join.slack.com/t/fastvideo/shared_invite/zt-3f4lao1uq-u~Ipx6Lt4J27AlD2y~IdLQ) for additional support.
+You can also join our [Slack community](https://join.slack.com/t/fastvideo/shared_invite/zt-38u6p1jqe-yDI1QJOCEnbtkLoaI5bjZQ) for additional support.

docs/getting_started/installation/spark.md

@@ -134,16 +134,9 @@ uv pip install "https://github.com/mjun0812/flash-attention-prebuild-wheels/rele
 
 If you hit other issues, please open an issue on our
 [GitHub repository](https://github.com/hao-ai-lab/FastVideo). You can also join
-our [Slack community](https://join.slack.com/t/fastvideo/shared_invite/zt-3f4lao1uq-u~Ipx6Lt4J27AlD2y~IdLQ)
+our [Slack community](https://join.slack.com/t/fastvideo/shared_invite/zt-38u6p1jqe-yDI1QJOCEnbtkLoaI5bjZQ)
 for additional support.
 
-## Next: performance & tuning
-
-Installed and verified? See [DGX Spark: Performance & Tuning](spark_performance.md)
-for which models are practical on the GB10, what makes them faster, and what
-won't help on this hardware (and why) — so you don't spend a night tuning knobs
-that can't move here.
-
 ## Development Environment Setup
 
 If you're planning to contribute to FastVideo please see the

docs/getting_started/installation/spark_performance.md

@@ -1,195 +0,0 @@
-# DGX Spark (GB10): Performance & Tuning
-
-You have FastVideo [installed on a DGX Spark](spark.md) — this page is what to
-run next. It covers **which models are practical on the GB10, what actually
-makes them faster, and what won't help (and why)**, so you don't burn a night
-tuning knobs that can't move on this hardware.
-
-!!! tip "TL;DR"
-    - **Use distilled few-step models** (e.g. `FastVideo/FastWan2.1-T2V-1.3B-Diffusers`).
-      They run in ~40 s/video. Full-step models are 12–47 min on the GB10.
-    - On few-step models, **VAE decode is the bottleneck**, not attention — it's
-      bandwidth-bound on the Spark's unified memory.
-    - **bf16 VAE decode** is the real, lossless lever (FastVideo already turns it
-      on for Wan). **FlashAttention, linear quantization, and `torch.compile` of
-      the VAE give little or nothing here** — see the table below.
-    - Heavy runs can make the box unreachable — run generations with VAE tiling on
-      and `nice -n 19`. See [Running safely](#running-safely-dont-lock-the-box).
-
-## The hardware reality (this explains everything below)
-
-The GB10 pairs a Blackwell GPU (`sm_121`) with **128 GB of unified LPDDR5X memory
-(~270 GB/s) shared between CPU and GPU**. That bandwidth is roughly **10× below a
-datacenter GPU's HBM**. Two consequences drive every tuning decision:
-
-1. **Memory-bandwidth-bound stages hurt disproportionately.** VAE decode moves a
-   lot of data and becomes the dominant cost on short (few-step) generations.
-2. **Compute-bound stages scale with step count.** Full-step diffusion (50+
-   steps) is denoise-bound and simply takes a long time here.
-
-## Use distilled few-step models
-
-The single biggest lever on the GB10 is **model choice**. A 3-step distilled
-model is ~18× faster than the full-step version of the same architecture:
-
-| Model | Steps | Time / video | Bottleneck |
-|---|---|---|---|
-| FastWan2.1-T2V-1.3B (distilled) | 3 | **~40 s** | VAE decode |
-| Wan2.1-T2V-1.3B (full-step) | 50 | ~12 min | denoise |
-| Cosmos-Predict2.5-2B (full-step) | 51 | ~47 min | denoise |
-| LTX2.3-distilled (+audio) | 8 | ~6 min | mixed |
-
-The bottleneck flips from decode to denoise at around **4 steps**. Below that,
-you're paying mostly for VAE decode; above it, mostly for the denoising loop.
-
-!!! note "Few-step timings are noisy — measure in-process"
-    On a 3-step run, one-time per-process startup (Triton autotune, allocator
-    warmup) dominates and never amortizes, so single-run totals wobble ~±30%.
-    Compare levers **back-to-back in one process or as medians**, never as two
-    separate single runs. The [reproduction script](#reproduce-these-numbers)
-    does this for you.
-
-## bf16 VAE decode — the real lever (already on for Wan)
-
-Because few-step generation is decode-bound, VAE decode precision is where the
-time is. Decoding in **bf16 instead of fp32 is essentially lossless** (MS-SSIM
-~0.9999 vs fp32 on the identical latent) and ~1.14× faster — worth roughly
-5–7% end-to-end on a decode-bound few-step model.
-
-**FastVideo already defaults Wan's decode to bf16** (`vae_decode_precision="bf16"`,
-with encode kept at fp32), so for the recommended Wan/FastWan models there's
-nothing to set. If you run a model that still defaults to an fp32 decode, set the
-decode-only override yourself:
-
-```python
-from fastvideo.configs.pipelines.base import PipelineConfig
-
-pipeline_config = PipelineConfig.from_pretrained(model_id)
-pipeline_config.vae_decode_precision = "bf16"   # decode-only; leaves encode precision alone
-```
-
-Decode is output-only, so lowering its precision is safe. (Encode seeds the
-denoising trajectory for I2V/causal models, so that stays at the pipeline's
-default — don't lower `vae_precision` blindly for those.)
-
-## Memory: one unified 128 GB pool
-
-The GB10 has **no separate VRAM** — CPU and GPU share one 128 GB LPDDR5X pool
-(~118 GB usable). Two practical consequences:
-
-- **`nvidia-smi` reports memory as `[N/A]`** on the GB10, and the system "used"
-  figure conflates CPU + GPU + cache, so it's only a soft upper bound — treat the
-  whole 128 GB as one shared budget. For a per-run figure, use FastVideo's own
-  `peak_memory_mb` (reported on the generation result and by the performance
-  benchmark), which is measured inside the worker that runs the model.
-- **The 128 GB is a *working-set* ceiling, not storage** — the model cache lives
-  on the NVMe (3.7 TB, ample). What has to fit in 128 GB is the weights,
-  activations, and KV cache — and, critically, the **VAE decode buffers**, which
-  is why tiling matters
-  (an untiled high-res decode can spike the pool into swap and lock the box).
-
-The recommended few-step models are comfortable here: their weights are small
-(1.3–2 B) and few-step generation keeps activations modest — a Wan2.1-1.3B
-few-step generation peaks at **~8.4 GB** (measured), a small fraction of the pool.
-The pressure comes from **decode resolution/frames**, not the model — a
-1080p×121-frame untiled decode is what pushes the pool toward its ceiling, which
-is why VAE tiling stays
-on by default.
-
-## What helps vs. what doesn't on the GB10
-
-The honest summary — most "obvious" GPU optimizations don't move the needle on
-this hardware, for reasons specific to it:
-
-| Lever | Effect on the GB10 | Use it? |
-|---|---|---|
-| Distilled few-step model | ~18× vs full-step | ✅ **the primary lever** |
-| bf16 VAE decode | ~1.14×, lossless; ~5–7% e2e on few-step | ✅ default for Wan |
-| VSA (video sparse attention) | works out of the box (Triton kernel auto-selects on `sm_121`) | ✅ automatic |
-| Building FlashAttention | **no speedup** — Torch SDPA already hits an efficient flash kernel on `sm_121`, and FA2 ties it | ❌ not worth building |
-| `torch.compile` of the VAE decode | recompile storm (per-frame varying shapes) → ~1.1× | ❌ dead end |
-| Linear (fp8 / nvfp4) quantization on long-sequence models (e.g. Cosmos) | ~nothing — see below | ❌ wrong lever here |
-| FP4 attention (`ATTN_QAT_INFER`) | works on `sm_121` (runtime allowlist landed in #1647; kernel build is #1598); helps, but needs a QAT-trained checkpoint | ⚠️ opt-in — see below |
-| FP4 linear on short-sequence models (LTX2) | up to −24% denoise at 1080p (#1594) | ⚠️ model/resolution-dependent |
-
-### Why linear quantization is the wrong lever on long-sequence models
-
-Quantizing the linear (GEMM) layers is a natural first instinct, but on a
-long-sequence video model it buys almost nothing on the GB10. A video-DiT denoise
-step is dominated by **O(N²) attention** at these sequence lengths (tens of
-thousands of tokens); the linear layers are a small single-digit fraction of the
-work. Quantizing them faster leaves the attention-bound total essentially
-unchanged — measured at ~1% on Cosmos-2.5, i.e. noise, and full-step CFG models
-also lose quality to per-step quantization error.
-
-The same mechanism **does** help on **short-sequence** models: LTX2's aggressive
-VAE compression gives it short attention sequences, so FP4 linear reaches −24%
-there (#1594). The rule: **on the GB10, the lever that matters is attention
-(sparse or FP4), not the linear layers** — unless the model has short sequences.
-
-### FP4 on the GB10 (opt-in)
-
-Block-scaled FP4 works on `sm_121` under CUDA 13:
-
-- **FP4 attention** (`FASTVIDEO_ATTENTION_BACKEND=ATTN_QAT_INFER`, #1598) is
-  numerically correct on the GB10 and ~6% faster end-to-end generation, but it only preserves
-  quality on a **quantization-aware-distilled checkpoint** (e.g.
-  `FastVideo/FastWan-QAD-1.3B`) — stock weights aren't trained to tolerate it.
-- **FP4 linear** helps only where sequences are short (LTX2, above).
-
-The [`qad_fp4_ab.py`](#reproduce-these-numbers) harness reproduces the FP4
-attention A/B on the QAD checkpoint.
-
-## Running safely (don't lock the box)
-
-The GB10 is easy to make **unreachable** — a heavy build or an untiled high-res
-decode starves the ~20 ARM cores and unified memory, `sshd` can't get cycles, and
-you're locked out at *"Connection timed out during banner exchange"* until the box
-is power-cycled. To avoid it:
-
-- **Inference:** keep **VAE tiling on** (the default), use sane resolution/frames,
-  and run under `nice -n 19`:
-
-  ```bash
-  nice -n 19 nohup python your_script.py > run.log 2>&1 &
-  ```
-
-- **Builds** (flash-attn, kernel): `nice -n 19`, `MAX_JOBS=2`, `nohup`. Never a
-  bare foreground high-parallelism build.
-- Leave `*_cpu_offload` at the example defaults — "CPU" offload is the *same*
-  unified RAM on the GB10, so the win is tiling + sane resolution, not offloading.
-
-## Gotchas specific to the GB10
-
-A few things that surprise people on this box (beyond the memory notes above):
-
-- **Don't force `TORCH_SDPA` on a VSA checkpoint** (FastWan, LTX2.3-distilled).
-  The SDPA path builds a model without the gate weights the checkpoint carries and
-  fails to load. Run the model natively — VSA auto-routes to its Triton kernel on
-  `sm_121`.
-- **Few-step timings are noisy run-to-run** (~±30%) — one-time startup dominates a
-  3-step run. Compare in-process / as medians, never two separate single runs (the
-  benchmark script does this).
-- **`nvidia-smi` shows `[N/A]` for memory** — see [Memory](#memory-one-unified-128-gb-pool).
-- **Cosmos-2.5** uses a Qwen2.5-VL text encoder; make sure you're on a FastVideo
-  build recent enough to include its `transformers`-compatibility handling before
-  running it.
-
-## Reproduce these numbers
-
-Two scripts under `examples/inference/optimizations/` reproduce the claims on
-your own GB10:
-
-```bash
-# Headline: few-step generation timing (median) + the bf16-vs-fp32 decode A/B.
-# FASTVIDEO_STAGE_LOGGING=1 also prints the denoise / decode / text split.
-FASTVIDEO_STAGE_LOGGING=1 nice -n 19 \
-    python examples/inference/optimizations/spark_benchmark.py
-
-# FP4 attention quality/speed A/B on the QAD checkpoint (one arm per run).
-QAD_LINEAR=0 FASTVIDEO_ATTENTION_BACKEND=ATTN_QAT_INFER nice -n 19 \
-    python examples/inference/optimizations/qad_fp4_ab.py
-```
-
-See also the [Optimizations](../../inference/optimizations.md) reference for the
-full list of attention backends and quantization options.

docs/getting_started/quick_start.md

@@ -23,21 +23,61 @@ Also optionally install flash-attn:
 uv pip install flash-attn --no-build-isolation -v
 ```
 
-## Choose a maintained recipe
+## Basic Usage
 
-The cookbook selects complete, checked-in recipes instead of mixing model,
-parallelism, offload, and attention settings independently.
+### Text-to-Video Generation
 
-[Open the inference cookbook](../cookbook/index.md){ .md-button .md-button--primary }
+```python
+from fastvideo import VideoGenerator
 
-!!! tip "Need more control?"
-    Start from a maintained recipe, then use the
-    [configuration](../inference/configuration.md) and
-    [optimization](../inference/optimizations.md) guides for supported changes.
+def main():
+    # Create a video generator with a pre-trained model
+    generator = VideoGenerator.from_pretrained(
+        "Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
+        num_gpus=1,  # Adjust based on your hardware
+    )
+
+    # Define a prompt for your video
+    prompt = "A curious raccoon peers through a vibrant field of yellow sunflowers, its eyes wide with interest."
+
+    # Generate the video
+    video = generator.generate_video(
+        prompt,
+        output_path="my_videos/",  # Controls where videos are saved
+        save_video=True
+    )
+
+if __name__ == '__main__':
+    main()
+```
+
+### Image-to-Video Generation
+
+```python
+from fastvideo import VideoGenerator, SamplingParam
+
+def main():
+    # Create the generator
+    model_name = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
+    generator = VideoGenerator.from_pretrained(model_name, num_gpus=1)
+
+    # Set up parameters with an initial image
+    sampling_param = SamplingParam.from_pretrained(model_name)
+    sampling_param.image_path = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/astronaut.jpg"
+    sampling_param.num_frames = 107
+
+    # Generate video based on the image
+    prompt = "A photograph coming to life with gentle movement"
+    generator.generate_video(prompt, sampling_param=sampling_param,
+                             output_path="my_videos/",
+                             save_video=True)
+
+if __name__ == '__main__':
+    main()
+```
 
 ## Next Steps
 
-- [Inference Cookbook](../cookbook/index.md) - Choose a maintained recipe
 - [Installation Guide](installation.md) - Detailed installation instructions
 - [Configuration](../inference/configuration.md) - Learn about configuration options
 - [Examples](../inference/examples/examples_inference_index.md) - Explore more

docs/index.md

@@ -5,7 +5,7 @@
 </div>
 
 <div style="text-align: center;">
-<strong>FastVideo is a unified post-training and real-time inference framework for accelerated video generation.</strong>
+<strong>FastVideo is a unified inference and post-training framework for accelerated video generation.</strong>
 </div>
 
 <div style="text-align: center;">
@@ -25,23 +25,14 @@ FastVideo is an inference and post-training framework for diffusion models. It f
 
 FastVideo has the following features:
 
-- End-to-end post-training support for bidirectional and autoregressive models
-  - Full finetuning and LoRA [finetuning](training/finetune.md) for state-of-the-art open video DiTs
-  - [Data preprocessing pipeline](training/data_preprocess.md) for video, image, and text data
-  - [Distribution Matching Distillation (DMD2)](distillation/dmd.md) stepwise distillation
-  - Sparse attention with [Video Sparse Attention](attention/vsa/index.md)
-  - [Sparse distillation](https://hao-ai-lab.github.io/blogs/fastvideo_post_training/) to achieve >50x denoising speedup
-  - [Attn-QAT training](training/attn_qat.md) for quantization-aware post-training
-  - Causal distillation through Self-Forcing
-  - Scalable training with FSDP2, sequence parallelism, and selective activation checkpointing
-  - See the [training overview](training/overview.md) for the full training workflow
 - State-of-the-art performance optimizations for inference
-  - Sequence parallelism for distributed inference
-  - Multiple state-of-the-art [attention backends](attention/index.md)
-  - User-friendly [CLI](inference/cli.md) and Python API
-  - See the [support matrix](inference/support_matrix.md) for supported models and [optimizations](inference/optimizations.md) for the full list
-- Realtime video generation and editing
-  - [Dreamverse](https://github.com/hao-ai-lab/FastVideo/tree/main/apps/dreamverse): stream and "vibe direct" video in realtime ([live demo](https://dreamverse.fastvideo.org/))
+  - [Sliding Tile Attention](attention/sta/index.md)
+  - [Sage Attention](https://arxiv.org/abs/2410.02367)
+- E2E post-training support
+  - Data preprocessing pipeline for video data
+  - [Sparse distillation](https://hao-ai-lab.github.io/blogs/fastvideo_post_training/) for Wan2.1 and Wan2.2 using [Video Sparse Attention](https://arxiv.org/pdf/2505.13389) and [Distribution Matching Distillation](https://tianweiy.github.io/dmd2/)
+  - Support full finetuning and LoRA finetuning for state-of-the-art open video DiTs.
+  - Scalable training with FSDP2, sequence parallelism, and selective activation checkpointing, with near linear scaling to 64 GPUs.
 
 ## Documentation