Render Animation Previews

1#!/usr/bin/env python3
2"""Render lightweight animated QA previews from extracted Codex pet frames."""
3
4from __future__ import annotations
5
6import argparse
7import json
8from pathlib import Path
9
10from PIL import Image
11
12ROW_DURATIONS = {
13    "idle": [280, 110, 110, 140, 140, 320],
14    "running-right": [120, 120, 120, 120, 120, 120, 120, 220],
15    "running-left": [120, 120, 120, 120, 120, 120, 120, 220],
16    "waving": [140, 140, 140, 280],
17    "jumping": [140, 140, 140, 140, 280],
18    "failed": [140, 140, 140, 140, 140, 140, 140, 240],
19    "waiting": [150, 150, 150, 150, 150, 260],
20    "running": [120, 120, 120, 120, 120, 220],
21    "review": [150, 150, 150, 150, 150, 280],
22}
23IMAGE_SUFFIXES = {".png", ".webp", ".jpg", ".jpeg"}
24
25
26def frame_files(state_dir: Path) -> list[Path]:
27    if not state_dir.is_dir():
28        return []
29    return sorted(path for path in state_dir.iterdir() if path.suffix.lower() in IMAGE_SUFFIXES)
30
31
32def load_frames(frames_root: Path, state: str, expected_count: int) -> list[Image.Image]:
33    files = frame_files(frames_root / state)
34    if len(files) != expected_count:
35        raise SystemExit(
36            f"{state} preview needs {expected_count} frames, found {len(files)} under {frames_root / state}"
37        )
38    frames = []
39    for path in files:
40        with Image.open(path) as opened:
41            frames.append(opened.convert("RGBA"))
42    return frames
43
44
45def save_preview(frames: list[Image.Image], durations: list[int], output: Path) -> None:
46    output.parent.mkdir(parents=True, exist_ok=True)
47    frames[0].save(
48        output,
49        save_all=True,
50        append_images=frames[1:],
51        duration=durations,
52        loop=0,
53        disposal=2,
54        optimize=False,
55    )
56
57
58def main() -> None:
59    parser = argparse.ArgumentParser(description=__doc__)
60    parser.add_argument("--frames-root", required=True)
61    parser.add_argument("--output-dir", required=True)
62    args = parser.parse_args()
63
64    frames_root = Path(args.frames_root).expanduser().resolve()
65    output_dir = Path(args.output_dir).expanduser().resolve()
66    previews = []
67    for state, durations in ROW_DURATIONS.items():
68        frames = load_frames(frames_root, state, len(durations))
69        output = output_dir / f"{state}.gif"
70        save_preview(frames, durations, output)
71        previews.append({"state": state, "path": str(output), "frames": len(frames)})
72
73    result = {"ok": True, "output_dir": str(output_dir), "previews": previews}
74    print(json.dumps(result, indent=2))
75
76
77if __name__ == "__main__":
78    main()