#!/usr/bin/env python3
"""Final scene-still gallery grouped by scene + attached ref roles. 커밋금지."""
import json, shutil
import psycopg2
from pathlib import Path
PID="b0ad5c18-5140-4022-a2b0-805de0e5a385"
EID="0d30302a-51c8-4ac9-98e2-6cd9180512bf"
ROOT=Path("/Users/manta/Documents/Projects/TheRoad-I1")
OUT=ROOT/"scratchpad/e2e_final_gallery"; OUT.mkdir(parents=True, exist_ok=True)
IMGDIR=OUT/"img"; IMGDIR.mkdir(exist_ok=True)
c=psycopg2.connect(host='localhost',user='theroad',password='theroad_dev_2026',dbname='theroad'); cur=c.cursor()
cur.execute("""
SELECT ia.id, ia.file_path, ia.still_id, ss.scene_index, ss.shot_description, ia.pipeline_metadata_json
FROM image_asset ia LEFT JOIN scene_still ss ON ss.id = ia.still_id
WHERE ia.episode_id=%s AND ia.asset_type='scene'
ORDER BY (ss.scene_index)::int NULLS LAST, ia.created_at
""", (EID,))
rows=cur.fetchall()
from collections import defaultdict
by_scene=defaultdict(list)
role_flags_all=defaultdict(int)
for aid,fp,sid,scn,desc,meta in rows:
m=json.loads(meta or '{}')
refs=m.get('actual_attached_refs') or []
roles=[r.get('role') for r in refs]
for r in roles: role_flags_all[r]+=1
by_scene[scn].append((aid,fp,desc,roles))
def copy(fp, name):
src=ROOT/fp
if not src.exists(): src=ROOT/"projects"/fp
if src.exists(): shutil.copy(src, IMGDIR/name); return "img/"+name
return None
# highlight roles
HL={'composition_guide':'#fc6','dead_state_passport':'#f66','registered_pose_guide':'#6cf',
'character_state_variant':'#f9a','immobilized_pose_guide':'#6cf',
'scene_prev_frame':'#9f6','immobilized_prev_frame':'#f96'}
rows_html=[]
for scn in sorted(by_scene, key=lambda x:(int(x) if x is not None else 999)):
stills=by_scene[scn]
cells=""
for aid,fp,desc,roles in stills:
name=f"s{scn}_{aid[:8]}.png"; img=copy(fp,name)
rolestr=" ".join(f'{r}' for r in roles) or 'no-refs'
d=(desc or '')[:110]
if img: cells+=f'
{rolestr}
{d}
Scene {scn} ({len(stills)} stills)
{cells}
금월도 E2E 최종 스틸 ({len(rows)} stills, {len(by_scene)} scenes)
노랑=composition_guide(야외포즈/실내마네킹) · 빨강=dead passport · 파랑=registered pose guide
전체 attached role tally: {dict(role_flags_all)}
{''.join(rows_html)}"""
(OUT/"index.html").write_text(html)
print(f"final gallery -> {OUT}/index.html ({len(rows)} stills, {len(by_scene)} scenes)")
print("role tally:", dict(role_flags_all))
# print outdoor + corpse candidate stills
print("\n=== stills with composition_guide (outdoor pose / indoor mannequin) ===")
for scn in sorted(by_scene, key=lambda x:(int(x) if x is not None else 999)):
for aid,fp,desc,roles in by_scene[scn]:
if any('guide' in (r or '') or 'dead' in (r or '') or 'pose' in (r or '') for r in roles):
print(f" S{scn} {aid[:8]} roles={roles} :: {(desc or '')[:70]}")