zi UdZddlmZddlZddlZddlmZeejjjZ e dz Z e dz Z ddd Z d ed <d d hZddhZhdZddZddZddZddZedk(rej.eyy)uNIDS Phase 3 — Mobile Prototype Renderer. Open a device frame HTML, inject the chosen scenario into the .screen-content placeholder, and capture a viewport-sized PNG. Usage: python3 render_prototype.py \ --device iphone15pro \ --mode light \ --scenario dashboard \ --output /abs/path/output.png ) annotationsN)Pathframes scenarios)iiT)ii) iphone15pro pixel9prozdict[str, tuple[int, int]]DEVICE_VIEWPORTSrrlightdark> dashboard ai_analysis signup_step1 signup_step2 signup_step3cardnews_publishinsurance_comparec<tjd}|jddtt|jddtt |jddtt |jddd |jS) NzMobile prototype renderer) descriptionz--deviceT)requiredchoicesz--modez --scenarioz--outputzAbsolute output PNG path)rhelp)argparseArgumentParser add_argumentsorted VALID_DEVICES VALID_MODESVALID_SCENARIOS parse_args)parsers J/home/jay/workspace/skills/mobile-prototype-ko/scripts/render_prototype.pyrr+s  $ $1L MF  T6-;PQ 4 9LM  tVO=TU  T8RS    ct|dz }|jstd||jd}|j }|j d}|j d}d}|dk7r|dk7r|||t dz}|j d }|dk7r|j d |nd}|j d } d} |dk7r | dk7r||d z| } || zS) N.htmlzScenario not found: zutf-8)encodingzzz) SCENARIOS_DIRis_fileFileNotFoundError read_textlowerfindlen) scenariopathrawr. style_start style_end style_block body_start body_open_end body_close body_inners r!load_scenario_bodyr;4s hZu- -D <<>"6tf =>> ..'. *C IIKE**X&K :&IKbY"_+ C O(CD G$J1;r1ACHHS*-rMI&JJzR/*Z8  ##r"cX ddlm}t|d|dz }|jstd|t|\}}t|} t|} | jjd d 5} | jj!d } | j#||d d } | j%}|j'|j)d|j+d| |j-d|j/t1| dd| j3ddd| jst5d| y#t$r>}td|tj t j dYd}~d}~wwxYw#1swYrxYw)Nr)sync_playwrightzPlaywright not available: file_r$zFrame not found: T)parentsexist_ok)headless)widthheight)viewportdevice_scale_factorload) wait_untilz(html) => { const el = document.querySelector('.screen-content'); if (el) { el.innerHTML = html; } }i,F)r2 full_pageomit_backgroundzScreenshot was not written to )playwright.sync_apir= ImportErrorprintsysstderrexit FRAMES_DIRr+r,r r;rparentmkdirchromiumlaunch new_contextnew_pagegotoas_urievaluatewait_for_timeout screenshotstrclose RuntimeError)devicemoder1outputr=exc frame_pathrErF scenario_html output_pathpwbrowsercontextpages r!renderrmIs7 $u55J    "3J< @AA$V,ME6&x0Mv,KTD9  b++$$d$3%%$7 !& ! *##%& 9     c" S-PUV -0    ;K=IJJ !M  *3%0szzB  s$E;B4F  F3FF F)ct} t|j|j|j|j td|j y#t $r(}td|tjYd}~yd}~wwxYw)Nzrender failed: r>r)zOK: r) rrmrbrcr1rd ExceptionrOrPrQ)argsres r!mainrqvsn rs #  (^ # # % , , 3 3 ( " [( 0,  ,  $**KZ  z CHHTVr"