HjO% dZddlZddlZddlZddlmZej jdeee jddl Z e jdz ZdZddddd d d didgd d d diddd dgddddddd didd diddddgdddddd dd diddgdddZdZdedefdZdZddd diddd didd did didgdddZdefd Zd!Zed"k(reyy)#u&s1 — 뿌리(root): LLM 요소 선정(R1) → 통합 조감 스케치 생성(R2, A/B) → readback. 사용: .venv/bin/python s1_root.py [--only plan|image|readback] 산출: plans/root_plan.json, out/root/root_A.png(순수 T2I), root_B.png(배치도 I2I), plans/root_readback_{A,B}.json N)PathrootaYou are the master planner for a group of connected filming locations (one real-world place made of indoor and outdoor parts). You are given the place's member descriptions and the FULL content of every shot that happens there. Your job: design ONE comprehensive TOP-DOWN aerial site sketch that lets every later image of any of these shots stay spatially consistent. Select the elements that must appear on that single sketch. Hard rules: - The sketch shows the place from directly above: structures (roof outline), outdoor areas, approach paths, and every OPENING (door, gate, window, stair/level access). - NEVER expose interior floor plans or interior furniture. Indoor action may only be represented by its openings (the window/door through which it is visible or reached) and by elements visible from outside. - Every shot must be covered: for each shot, the elements a viewer would need to locate that shot's action on the sketch (its opening, area, or approach). - Keep it compact: at most 10 elements, each with a single capital letter ID (A, B, C, ...) and a SHORT English name (1-3 words) for the legend. - Placement phrases must be relative and physical (e.g. "on the north edge of the roof", "at the bottom of the exterior stair"), grounded in the given data only. - Do not invent features that contradict the member descriptions. Do not use any proper names in element names; use generic physical terms.objectarraystringzsingle capital letter)type descriptionr) structureopeningfixtureapproachsurround)renumz"which shots/member text justify it)idname_encategory placement groundingF)r propertiesrequiredadditionalProperties)ritems)shot_key element_idsrrz3overall massing/orientation guidance for the sketch)elementscoverage sketch_notes legend_orderc tj|d}djdt|dj D}d|zdz|zdz}tj dt |td }t|jd gd kDr+tjd ddd dt|d dtjd||jdgDchc]}|d }}tt|d|z }tdt|d dt|dt|dd|rd|ndz|Scc}w)Nmembersz c3NK|]\}}tj||yw)N)F shot_block).0kss s1_root.py zr1_plan..Ss%G'Etq! Q'Es#%shotszPLACE MEMBERS: z ALL SHOTS AT THIS PLACE: z7 Design the single aerial site sketch element set now.forest_root_plangptmodelr warns1z elements>: kindstagemsg root_planrrzR1: z elements, coverage /z shotsz MISSING=)r" members_blockjoinsortedrllm R1_SYSTEM R1_SCHEMAlengetrunlog save_plansetprint)reconr shot_blocksuserplanccoveredmissings r'r1_planrLQsvooeI./G++G'-eGn.B.B.D'EGGK W$ , -/: ; E F 55#Yiu MD 488J #$r) &4bTCZ(8$9#:;= >KK T"&*hhz2&>?&>q}&>G?Sw(723G DT*%&'';CL> c%.! "& *FMy -BSU WX K @s. E members_textreturnc djd|dD}djd|dD}djdddd d d d |d d |jdd d dddddd dd|d dd|d ddddgS)N c3DK|]}d|dd|dd|dyw)(rz) ru — rNr$es r'r(z sketch_prompt..hs8#!A AdG9Bq|nE!K.)9:!s rz, c3RK|]}|dd|dj!yw)rr1rN)upperrTs r'r(z sketch_prompt..ks3H6F1T7)2a l((*+,6Fs%'zHONE hand-drawn TOP-DOWN aerial SITE SKETCH of a single real-world place,zEin clean technical sketch style: dark ink outlines, light flat colourzAwashes, straight-down view. NOT a photograph, NOT a 3D render, nozperspective, no photo textures.r8uIThe place consists of these connected parts — draw them TOGETHER as onezCcoherent site (one main structure plus its immediate surroundings):z%Overall massing/orientation guidance:rzFDraw ONLY what is visible from above and from outside: the structure'szFroof outline and what stands on top of it, open ground areas, approachzHpaths and stairs, and each opening (door, gate, window, level access) aszHpart of the boundary it belongs to. Do NOT draw any interior room layoutu?or interior furniture — buildings show their roof plane only.zGMark each of the following elements at its position with a small circlez%containing its single capital letter:zEOn the RIGHT side of the image, add a clean legend panel listing eachzCcircled letter with its English name in small neat capital letters:zGThe circled letters and the legend panel are the ONLY text and the ONLYzFcircular marks allowed. No other labels, numbers, logos or watermarks.zHInclude a modest margin of the immediate surroundings so the place readszGin context; keep geometry simple, unambiguous and physically plausible.)r:r@)rHrMel_lines legend_liness r' sketch_promptrZgsyy#j!##H99H6::6FHHL 99ROK) SM / $ PPRRI Q/ OM QPRQ= c4tj|d}t||}tjd|dtdz |j dr*|dz}tjd||dgtdz ytj d d d d y) Nr root_A_t2iz root_A.png)refsout_pathbaseline_aerialuK The attached image is the existing clean top-down site plan of this place. KEEP its geometry, proportions and layout EXACTLY — same structures, same openings, same surroundings. Restyle it into the hand-drawn sketch style described above and ADD the circled capital letters and the right-side legend panel. Change nothing else.root_B_from_planz root_B.pngr/r0zno baseline aerial for R-Br2)r"r9rZimg_gptOUTDIRr@rA)rErHrMprompt edit_prompts r' r2_imagesrfs??5#34L 4 .FIIlF8MN yy"# M  $k?P9Q8R!L0 2 &48TUVr[boolean) letters_foundlegend_entriesinterior_layout_exposednotesvariantctd|dz }|jsydddtj|g}tjdd|t d }tj d |||d Dchc]}|d  }}t|jd xsg}td|dt|dt|d|jdd|jdddd ycc}w)Nroot_z.pngtextzThis is a top-down site sketch with circled capital letters and a legend panel. List every circled capital letter you can find, list the legend entries as written, and say whether any interior room layout is exposed.)rroforest_root_readbackz&You read technical sketches precisely.r+r,root_readback_rrrhz readback z : letters z vs plan z | interior_exposed=rjz | rkr8x) rcexistsr" png_data_urlr<READBACK_SCHEMArBrCr@rDr;)rHrlpngrGrbrUwantgots r'readbackrzs  U7)4( (C ::< " # s  D %(P_E 3BKK. *B/!*- .-AdG-D . bff_%+ ,C IgYj Yvd|nM ff%>?@BFF7SUDVW[X[D\C] _` /s7 C2ctj}|jddgd|j}t j }|j dvr t|}nt jd}|j dvr t|||j dvrd D]}t||t jd d |j d y) Nz--onlyall)r|rHimagerz)defaultchoices)r|rHr6)r|r})r|rz)ABr4s1_root)r3r4done) argparseArgumentParser add_argument parse_argsr" load_recononlyrL load_planrfrzrA)apargsrErHvs r'mainrs  "BOOHe@B ==?D LLNE yyO#u~{{;' yy$$% yy''A T1 HHg 499 EFr[__main__)__doc__rjsonsyspathlibrpathinsertstr__file__parent forest_libr"OUTrcr=r>rLrZrfrurzr__name__rSr[r'rs  3tH~,,-.  @ 0  #+P#Q(7(-  "*(]_!(FH3EFA!DI!K&  R,%c%c%PW* ")VX4FG#*fh5GH$*I#6(#  ! `C`*G$ zFr[