#!/usr/bin/env python3
from __future__ import annotations
import json, subprocess, shutil
from pathlib import Path
from weasyprint import HTML
ROOT = Path(__file__).resolve().parents[1]
DATA = ROOT/'assets/data/image-admin'
QA = ROOT/'qa_screenshots'
QA.mkdir(exist_ok=True)
manifest = json.load(open(DATA/'rc50_visual_review_manifest.json'))
counts = manifest['summary']['counts']
recommended = [e for e in manifest['exercises'] if e['reviewRecommended']][:6]
def html(width, height, name):
cards = ''.join(f'''
{e['label']}{e['exerciseKey']}
{e['runtimeGate']}{e['frameCount']} frames{e['mediapipeActiveFrames']} MP active
{v}{k}
' for k,v in {
'Exercises': counts['exercises'], 'Frames': counts['frames'], 'Review recommended': counts['reviewRecommendedExercises'], 'Runtime changed': 'No', 'MP active frame entries': counts['mediapipeActiveFrameEntries'], 'Missing poseKeys': counts['missingPoseKeys']
}.items())
return f'''NEXT Pose Atlas · RC50 · {name}
Visual Review + Correction Mode
Review-only layer on accepted RC49 base. Flags wrong foot/hand, direction, anchor, image candidate, needs-new-pose or hold without touching runtime.
{metrics}
Review recommended first
{cards}'''
for name,w,h in [('mobile',390,844),('ipad',820,1180),('desktop',1180,820)]:
html_path=QA/f'rc50_{name}_static_qa.html'
pdf_path=QA/f'rc50_{name}_static_qa.pdf'
png_base=QA/f'rc50_{name}_static_qa'
png_path=QA/f'rc50_{name}_static_qa.png'
html_path.write_text(html(w,h,name), encoding='utf-8')
HTML(filename=str(html_path), base_url=str(ROOT)).write_pdf(str(pdf_path))
subprocess.run(['pdftoppm','-png','-singlefile','-r','150',str(pdf_path),str(png_base)], check=True)
print(QA)