"""연결 갤러리: 각 샷의 최종 이미지 ← 실제 첨부된 가이드/배경/캐릭터(input_image_ids) 를
나란히 보여줘 '가이드가 샷/배경에 연결되어 생성됨' 을 육안 확인. 재생성 결과 SOT=DB."""
import os, shutil, json, psycopg2
ROOT = "/Users/manta/Documents/Projects/TheRoad-I1"
EP = "286f3ba4-5738-49b7-a7e4-ce0295fd4352"
OUT = os.path.join(ROOT, "backend/scratchpad/conn_gallery")
IMG = os.path.join(OUT, "img")
os.makedirs(IMG, exist_ok=True)
conn = psycopg2.connect(host="localhost", user="theroad", password="theroad_dev_2026", dbname="theroad")
cur = conn.cursor()
# 최신 scene_still asset per still (재생성분) + still 씬/샷
cur.execute("""
SELECT DISTINCT ON (ia.still_id) ss.scene_index, ss.shot_index, ia.id, ia.file_path,
ia.input_image_ids, LEFT(ss.shot_description,60)
FROM image_asset ia JOIN scene_still ss ON ss.id=ia.still_id
WHERE ia.episode_id=%s AND ia.pipeline_role='scene_still' AND ia.still_id IS NOT NULL
ORDER BY ia.still_id, ia.created_at::timestamptz DESC
""", (EP,))
stills = cur.fetchall()
ROLE_KO = {
"outdoor_camera_sketch": "야외 마네킹 스케치", "registered_pose_guide": "등록 pose 가이드",
"indoor_pose_guide": "실내 pose 가이드", "background_render": "배경",
"reference_composite": "캐릭터 합성", "reference_face": "얼굴",
"character_state_variant": "상태 변형", "reference_outlook": "아웃룩",
"outdoor_aerial_base": "항공뷰", "outdoor_shot_blocking": "블로킹",
}
GUIDE_ROLES = {"outdoor_camera_sketch", "registered_pose_guide", "indoor_pose_guide"}
def resolve(asset_id):
cur.execute("SELECT file_path, pipeline_role FROM image_asset WHERE id=%s", (asset_id,))
r = cur.fetchone()
return r if r else (None, None)
def cp(fp, tag):
if not fp:
return None
src = os.path.join(ROOT, fp)
if not os.path.exists(src):
return None
dst_name = f"{tag}_{os.path.basename(fp)}"
shutil.copy(src, os.path.join(IMG, dst_name))
return "img/" + dst_name
cards = []
n_guide = n_total = 0
for sc, sh, aid, fp, iii, desc in stills:
n_total += 1
final_rel = cp(fp, f"S{sc}sh{sh}_final")
refs = []
has_guide = False
try:
ids = json.loads(iii) if iii and iii not in ("null", "") else []
except Exception:
ids = []
for rid in ids:
rfp, rrole = resolve(rid)
if rrole in GUIDE_ROLES:
has_guide = True
rel = cp(rfp, f"S{sc}sh{sh}_{rrole}")
refs.append((rel, ROLE_KO.get(rrole, rrole or "?")))
if has_guide:
n_guide += 1
cards.append((sc, sh, desc, final_rel, refs, has_guide))
cards.sort(key=lambda c: (0 if c[5] else 1, c[0], c[1])) # 가이드 샷 먼저
def ref_html(refs):
if not refs:
return '첨부 참조 없음(환경/인서트)'
out = []
for rel, label in refs:
img = f'
' if rel else '파일없음
'
out.append(f'{img}{label}')
return "".join(out)
rows = []
for sc, sh, desc, final_rel, refs, has_guide in cards:
badge = '가이드 연결' if has_guide else ''
fimg = f'
' if final_rel else '파일없음
'
rows.append(f"""
S{sc} · shot {sh} {badge}{desc or ''}
{fimg}최종 샷
←
{ref_html(refs)}
""")
html = f"""
연결 갤러리 — 최종 샷 ← 첨부 가이드/배경/캐릭터
연결 갤러리 — 최종 샷 ← 실제 첨부된 가이드/배경/캐릭터
에피소드 재생성 후 input_image_ids(=실제 생성 입력) 기준. 가이드 연결 샷 {n_guide} / 총 {n_total}. 가이드 샷을 위로 정렬.
{''.join(rows)}
"""
with open(os.path.join(OUT, "index.html"), "w") as f:
f.write(html)
print(f"gallery: {OUT}/index.html (가이드연결 {n_guide}/{n_total} 샷, img {len(os.listdir(IMG))}개)")
cur.close(); conn.close()