Hj 0 dZddlZddlZddlZddlmZej jdeee jddl Z ddl Ze jdz ZdZddddd idd idd id d d dd idd idd idd dgidgdddddddd idd idd idgdddddddgddZdZdedefdZdZdefdZdZdefdZd Zdefd!Zd"ed#efd$Zd%Zed&k(reyy)'u%s2 — 실내 체인: I0 계획 → I1 탑레이어 → I2 fp변형 → 임시샷 → A/B/C 경로 → 최종. 사용: .venv/bin/python s2_indoor.py [--only plan|toplayer|fpvar|temp|paths] [--alpha-shot 첫샷 임시배경 α 변형] [--mannequin-preview] N)PathindooruYou are the background production planner for the INDOOR shots of one real-world place. You are given: the place's aerial-sketch element plan (the spatial root contract), the indoor floor plan identifiers available, and the FULL content of every indoor shot. Plan the background generation like a set photographer: 1. For each shot, state what background it needs (which part of the indoor space, from roughly what direction). 2. Merge those needs into AT MOST TWO "top-layer" master backgrounds — wide, clean, empty-room views that together cover all the shots. Prefer ONE if the space is small enough to read in a single wide view. 3. Give the generation order. If there are two, the second must chain on the first (same dwelling, same materials). 4. For each top-layer, judge whether attaching the floor plan image as a reference would help (geometry anchoring) or hurt (style leakage), and say why. Ground every statement in the given data only; do not invent rooms or furniture. Use generic physical terms, no proper names.objectarraytypestringzFwhat the master background shows, phrased as an empty-room camera view)r descriptionbooleanintegernull)keyloc_idpurposeview_description attach_fpattach_fp_reasonorderchain_prev_keyF)r propertiesrequiredadditionalProperties)ritems)shot_key top_layer_keyneeded_background) top_layersshot_assignmentsrrc|djDcic]\}}|ds ||}}}djd|djD}dtj|dzdzt j |d zd z|zd zd jdt |jDzdz}tjdt|td}tjd|tdt|ddt|dd|dD]+}td|dd|dd|dd|d d!|d" -|Scc}}w)#Nshots is_indoor c3^K|]%\}}d|ddjd|D'yw)z- loc z: floor plan(s) z, c3&K|] }|d yw)fp_idN).0xs s2_indoor.py z$i0_plan...Es/HCq' CsN)join)r&locfpss r(r)zi0_plan..Ds:42HC %dii/HC/H&H%IJ2s+- fp_by_loczPLACE MEMBERS: membersz' AERIAL ROOT CONTRACT (element plan): F ensure_asciiz AVAILABLE FLOOR PLANS: z ALL INDOOR SHOTS: z c3NK|]\}}tj||yw)N)F shot_block)r&kvs r(r)zi0_plan..Ms"R5QTQall1a(5Qs#%z% Plan the top-layer backgrounds now.forest_indoor_plangpt)model indoor_planzI0: rz top layers; rz shot assignmentsz [rz] r z loc=rz attach_fp=rz chain=r) rr*r2 members_blockjsondumpssortedllm I0_SYSTEM I0_SCHEMA save_planprintlen) recon root_planr4r5 indoor_shotsfp_linesuserplantls r(i0_planrKBs%*7^%9%9%;N%;TQq~AqD%;LNyy4k*00244H Q__U9-=>> 6 7 **YU 3 4 ) ),4 4 &  & ++RVL?<  BwK=2e9+U2h<.Ak?+726F3G2HJ K! K)Os EEchainedreturnctfd|dDi}ddddd|jdd |jd d d d dddt|jddg }|r|gdz }dj|S)Nc3:K|]}|ddk(s|yw)rNr%)r&mrJs r(r)z#top_layer_prompt..[s#K+aq{bl/J+sr.zECreate ONE photorealistic EMPTY interior master background plate of auIreal dwelling interior — a wide, clean view with NO people, no text, noz8markers. Real-world human scale (doorways adult height).z"THE PLACE (from production data): labelz: summaryzTHIS PLATE'S PURPOSE: rz THE VIEW: rz5Site orientation notes from the aerial root contract: sketch_notes)rQuCThe LAST attached image shows ANOTHER part of THE SAME dwelling —zBmatch its wall colours, floor material, trim, door style, lightingzAmood and renovation age EXACTLY so both read as the same home. Doz)NOT copy its camera angle or composition.r!)nextgetstrr*)rDrJrErLr+liness ` r(top_layer_promptrYZs K5+KR PCOSB ,SWWW-=,>bAS@TU I0 R*+,- ? IMM." -. E    99U u The FIRST attached image is the TOP-DOWN floor plan of this dwelling — use it ONLY to get the room geometry, openings and furniture positions right. NEVER copy its drawing style, numbered circles or any marks into the photorealistic output.force_abc d}t|dd}t|D]_\}}|dj|dxsg}|r|ddnd} t|jdxr|du} t |||| } g} t|jd xr| } | r| t z}| j | n| }| r| j |tdz |d d z }tjd |d || xsdd ||}|s|dk(s| s| s;tjd |d d| t z| gd tdz |d dz /tjd |d d| dd tdz |d dz by)Nrc |dSNrr%ts r(zi1_top_layers..}sqzrZr r-rrpng_pathrrr .pngtop_ 1536x1024refssizeout_path_fpONz _fpON.png_fpOFFz _fpOFF.png) r= enumeraterVboolrY FP_CLAUSEappendOUTDIRr2img_gpt)rDrIrEr[prev_pngorderedidxrJfp_listfp_pngrLpromptrhuse_fp prompt_fullouts r( i1_top_layersr|{sHT,'-ABGW%R $((H6<"+2J'rvv./0IXT5I!%Y@bff[)*5v  9,K KK  K  KK !|#E 4&88 DE $k "S 2 qV DE 516I3E &xk#)L#8bi[ ;R#RT DE 62F*#)L#8bi[ ;S#SU5&rZu7The attached image is a TOP-DOWN floor plan of one dwelling: numbered circles mark fixed rooms and furniture. Keep the plan EXACTLY as it is — same walls, rooms, furniture and numbered circles, do not redraw or restyle anything. ADD exactly these annotations for ONE camera setup: - ONE green camera icon with a green triangular view-CONE showing where the camera stands and what it looks at, following this staging description: {cam} {figures}Add NOTHING else — no new rooms, no new furniture, no text other than the existing numbers and the new capital letters.c|jdxsi}|jdxsijdxsg}td|D}|dkrt|jdxsg}|dkrydjd t |D}d |d |d d jd|jdxsgDzd zS)Nstagingframe_spatial_contract constraintsc3jK|]+}t|ts|jddk(rd-yw) target_kind characterN) isinstancedictrV)r&cs r(r)z!figures_clause..s0 5t!z!T2m$ 3 ts33rcharacter_anglesu7- This shot has NO people — add no lettered circles.  c3@K|]}dtd|zdyw)(A)N)chr)r&is r(r)z!figures_clause..s"<8a3rAv;-q)8sz- z$ red circle(s) with capital letters z; inside the view-cone where the staging places the people: r!c3PK|]}dtj|dz yw)z * Fr/N)r;r<)r&cas r(r)z!figures_clause..s)F#CBBU!CC#Cs$&)rVsumrCr*range)sstconsnletterss r(figures_clausers y  RB VV, - 3 8 8 G M2D  5t 5 5AAv )*0b 1AvIhh<58<D" EC  # #^A5F # GF 2+ #C 99t6]F'!*Z2H1I% 55rZcb|dD]}|d|k(s |dcSt|ddddS) Nrrrrc |dSr^r%r_s r(razassigned_top..sAgJrZrbrr )r=)rIras r( assigned_toprsH $ % Z=B _% %& $|$*> ? B5 IIrZalphamannequin_previewc |d|}t|z }t||}tdz |dz }|dz } tj||} tj|d||| |dz } |rdj dd d d d d tj |tj|g} tj|d| | |g|dz } tj|d|| | |dz tj|d||| |dz | tj|| }tj|d|||dz |}tj|d||| |dz tj|d| |dz }tj|d|||dz |}tj|d ||| |d!z |rAtj|d"|||d#z |d$%}tj|d&||| |d'z yy)(Nrrrdr_tempztemp.pngr!z?Create the photorealistic EMPTY interior background seen by thezAcamera marked on the attached top-down diagram (green icon+cone),zCat eye level, real-world scale. Use the other attached image as thezBmaterial/lighting ground truth of the same dwelling. NO people, noztext, no markers.rQ_tempbgz temp_bg.png)rhrj _temp_alphaztemp_alpha.png_final_Az final_A.png) control_png_bgBzbg_B.png)comp_final_Bz final_B.png_contourz contour.png_bgCzbg_C.png) contour_png_final_Cz final_C.png_bgC_mz bg_C_m.pngT)r mannequin _final_C_mz final_C_m.png)rqrCpassports_for_shot temp_shotr*shot_content_lines staging_linesr2rr final_shotvlm_compositionbg_regencontour)rDrIrrrrdtop_keyenvfpvar passportstemp tb_prompttbrbg_bcontbg_cbg_cms r(run_shotrs$ grA A4$G < WIT"2 2C OE$$UA.I ;;"U|QUA N CD II M O Q P  $$Q');   YY"W~ys| !M 13 rd+&2ua:J6JKLLB4x!S)Q5F"$  R &D ::Dk1c1z> EDLLB4x!T9a-6GH 99t8_dA ,= >D ::Dk1c1z>t LDLLB4x!T9a-6GH bT=!S!l2B'+t= t:&5)Q=PQrZc tj}|jddgd|jdd|j}t j }t j d}t j d d }|jd vr t||}nt j d }|jd vrt||||j |jdvrDt|D]6\}}t|||jdvs!t||||dk(|dk(8t jdd|j|dy)Nz--onlyall)rrItoplayerrrpaths)defaultchoicesz --no-force-ab store_true)actionrEsubsetr)rrIr9)rr)r[)rrrr)rrrr)rrstage s2_indoor)kindrdoner)argparseArgumentParser add_argument parse_argsr2 load_recon load_planonlyrKr| no_force_abrmrrrunlog)apargsrDrErrIrrs r(mainrs,  "BOOHeQSOOOLO9 ==?D LLNE K(I [[ "8 ,F yyO#ui({{=) yy''eT94;K;K7KL yy55v&EAr UB yy44ba,-F5' HHg TYY !rZ__main__)__doc__rr;syspathlibrpathinsertrW__file__parent forest_libr2 forest_chainrOUTrqr?r@rKrnrYror|rrrrrr__name__r%rZr(rs  3tH~,,-.   0 & &(H%x())11W!X!),!' 2i(%&'9: S<+%*2+,&-#X."((!3&,h%79K$) 8+,!012!7  >0DS4" UDUF 5 N N 5JcJ'RT'Rd'RT!4 zFrZ