#!/usr/bin/env python3
import json, re, time, importlib.util
from pathlib import Path
PROJECT=Path('/data/video-pipeline/Youtube-Video-Maker/Project/29062026-001-Le-Trung-Hung')
RUN=PROJECT/'scripts_run_step_06_images.py'; LOG=PROJECT/'logs/step06_missing_retry_run.log'; MANIFEST=PROJECT/'logs/step06_missing_retry_manifest.json'
def load_runner():
    spec=importlib.util.spec_from_file_location('step06', RUN); mod=importlib.util.module_from_spec(spec); spec.loader.exec_module(mod); return mod
def log(msg):
    line=time.strftime('%Y-%m-%d %H:%M:%S ')+msg; print(line, flush=True); LOG.parent.mkdir(parents=True, exist_ok=True)
    with LOG.open('a',encoding='utf-8') as f: f.write(line+'\n')
def reset_wait(err):
    m=re.search(r'reset after\s+(?:(\d+)m\s*)?(\d+)s', err or '', re.I)
    if not m: return 75
    return max(45,int(m.group(1) or 0)*60+int(m.group(2) or 0)+35)
def main():
    LOG.write_text('',encoding='utf-8'); mod=load_runner(); refs=mod.load_refs(); scenes=json.loads(mod.SCENES.read_text(encoding='utf-8'))['scenes']; results=[]
    for scene in scenes:
        out=mod.OUT_SCENES/f"{scene['scene_id']}.png"
        if out.exists() and out.stat().st_size>1000: continue
        for attempt in range(1,6):
            log(f"RETRY {scene['scene_id']} attempt={attempt}"); r=mod.run_one(scene,refs); results.append(r)
            if r['status'] in ('ok','skipped_exists'): log(f"OK {scene['scene_id']} bytes={r.get('bytes')}"); break
            wait=reset_wait(r.get('error','')); log(f"FAILED {scene['scene_id']} wait={wait}s error={(r.get('error','') or '')[:500].replace(chr(10),' | ')}"); time.sleep(wait)
        if not (out.exists() and out.stat().st_size>1000):
            MANIFEST.write_text(json.dumps({'status':'blocked','failed_scene':scene['scene_id'],'results':results},ensure_ascii=False,indent=2),encoding='utf-8'); raise SystemExit('blocked '+scene['scene_id'])
    thumbs=[]
    for kind,name in [('16x9','thumb_16x9_final.png'),('9x16','thumb_9x16_final.png')]:
        out=mod.OUT_THUMBS/name
        for attempt in range(1,6):
            log(f"THUMB {kind} attempt={attempt}"); r=mod.gen_thumb(kind,out,refs); thumbs.append(r)
            if r['status'] in ('ok','skipped_exists'): log(f"THUMB_OK {kind} bytes={r.get('bytes')}"); break
            wait=reset_wait(r.get('error','')); log(f"THUMB_FAILED {kind} wait={wait}s error={(r.get('error','') or '')[:500].replace(chr(10),' | ')}"); time.sleep(wait)
        if not (out.exists() and out.stat().st_size>1000):
            MANIFEST.write_text(json.dumps({'status':'blocked','results':results,'thumbs':thumbs},ensure_ascii=False,indent=2),encoding='utf-8'); raise SystemExit('thumb blocked '+kind)
    scene_results=[]
    for scene in scenes:
        out=mod.OUT_SCENES/f"{scene['scene_id']}.png"; scene_results.append({'scene_id':scene['scene_id'],'status':'skipped_exists','output':str(out),'bytes':out.stat().st_size})
    thumb_results=[]
    for kind,name in [('16x9','thumb_16x9_final.png'),('9x16','thumb_9x16_final.png')]:
        out=mod.OUT_THUMBS/name; thumb_results.append({'kind':kind,'status':'skipped_exists','output':str(out),'bytes':out.stat().st_size})
    mod.MANIFEST.write_text(json.dumps({'project_id':PROJECT.name,'step':6,'status':'completed','workers':'polite-missing-retry','total_scenes':len(scenes),'scene_results':scene_results,'thumbnail_results':thumb_results},ensure_ascii=False,indent=2),encoding='utf-8')
    check=json.loads((PROJECT/'checklist_19_steps.json').read_text(encoding='utf-8')); check['steps'][4]['status']='completed'; check['steps'][5]['status']='completed'; check['current_step']=6
    (PROJECT/'checklist_19_steps.json').write_text(json.dumps(check,ensure_ascii=False,indent=2),encoding='utf-8'); (PROJECT/'checklist_19_steps.md').write_text('# Checklist 19 Steps\n\n'+'\n'.join(f"- [{'x' if s['status']=='completed' else ' '}] Step {s['step']:02d}: {s['status']}" for s in check['steps'])+'\n',encoding='utf-8')
    MANIFEST.write_text(json.dumps({'status':'completed','scene_retry_results':results,'thumbnail_results':thumbs},ensure_ascii=False,indent=2),encoding='utf-8'); log('DONE completed')
if __name__=='__main__': main()