l@j/ UdZddlmZddlZddlmZmZmZmZdZ dZ dEdZ dEdZ d Z d Zd Zd Ze eefZd ZdZdZdFdZ dGdZdHdZdd dIdZdZdZdZeeefZdZdZee ehZ eddhZ!dZ"dddddd d!id d"id"eegd#d d"id$gd$d%d&dddd d!id d"id d"id'gd'd%d&d"d(d)dd d"id&d*gd*d%Z#d+e$d,<dJd-Z%ed. dKd/Z&d0Z'd1Z(d2Z)d3Z*d4Z+d5Z,d6Z-d7Z.dddddd d!id d"id"e+e,e-gd#d8gd8d%d&dddd d!id d"id d"id d!id9gd9d%d&dddd d!id d"id d"id:gd:d%d&dddd d!id d"id;dgd>d%Z/d+e$d?<dZ0dEd@Z1 dLdAZ2 dMdBZ3de)dC dNdDZ4y)OuW21B-w5 STEP5-B — simplified (CV-readable) floor-plan prompt builder. Builds a GENERIC, deterministic simplified floor-plan ``t2i_prompt`` from a floor plan's structured ``numbered_elements`` (the ``floor_plan_prompt`` v6/v7 output). The simplified FP is rendered as an **I2I-reference sidecar** for ``background_render`` — it is NOT the projection-card / edge-judge substrate (the detailed FP stays that), so the W21B-w5 partition SOT is never disturbed (STEP5-B boundary lock, Codex + Claude consensus). Why a simpler FP: the production detailed FP draws every base element (furniture, decor, dense small numbers, colour zones) which the image model (I2I) struggles to read — rooms bleed into each other (a bathroom's fixtures float in the open living area). A thick-walled, enclosed, fixture-only partition diagram lets the renderer read independent rooms. The simplified FP carries SPATIAL ENCLOSURE (walls / rooms / openings / essential fixtures); the per-shot CONTENT is supplied by the zone-respecting background prompt (STEP6-C), so room-defining decorative furniture is intentionally dropped here. Absolute-rule compliance: * GENERIC — no scenario tokens / room names / proper nouns in this module. The only per-FP content is the runtime ``label`` / ``position_hint`` strings, carried through verbatim as opaque DATA (never matched on / pattern-parsed). * Element selection is by the closed ``base_layer_decision`` enum (the v6/v7 contract), NOT by any text/substring inspection of labels. ) annotationsN)AnyDictListOptional1u=Redraw this detailed floor plan as a SPARSE black-and-white architectural line-art PARTITION DIAGRAM for machine reading. PRESERVE EXACTLY, do not reinterpret: every wall position and room shape, every room location, all door and window openings, and every numbered marker with its EXACT number and position. Do NOT create, duplicate, move, merge, split, add, or renumber any wall, door, or numbered marker — exactly one circle per existing marker number, unchanged. RESTYLE aggressively: draw ALL walls as VERY THICK solid black lines so each room reads as a strongly enclosed cell; pure WHITE background; remove ALL color fills, tints, shading, gradients, and textures; remove EVERY text label, room name, and zone name. Redraw every furniture or fixture as a SINGLE minimal flat primitive icon only (a plain rectangle for a desk/bed/table, a minimal standard symbol for a sink/toilet) — NO interior detail, NO chairs, monitors, copier parts, pillows, cushions, drawers, or decoration. Make the numbered circles large and clearly readable. No 3D, perspective, shadow, or texture.ctS)u6Return the GENERIC deterministic restyle-edit instruction (v0 production). Pure and parameter-free — the per-fp content lives in the detailed FP reference image, never in this prompt (no scenario tokens, absolute-rule compliant). The step pairs this with ``images.edit(image=detailed_fp)``. )RESTYLE_EDIT_PROMPTb/Users/manta/Documents/Projects/TheRoad-I1/backend/app/modules/pipeline/floor_plan_light_prompt.pybuild_restyle_edit_promptrDs  r ctjtdztzj dj ddS)zCStable short hash of the restyle prompt (provenance + cache lever). utf-8N)hashlibsha256RESTYLE_PROMPT_VERSIONr encode hexdigestr r r restyle_prompt_hashrNs< >> $ &)< <DDWM ik#2 r base_structural_unit base_openingbase_persistent_fixturebase_persistent_furnitureaCreate a square flat top-down architectural FLOOR-PLAN PARTITION DIAGRAM. This is a clean CV-readable schematic for machine reading, NOT a decorated or artistic plan. WALLS AND ROOMS (most important): draw THICK solid black outer walls and THICK solid black interior partition walls. Each distinct room/area is a fully ENCLOSED cell bounded by walls on all sides, connected to its neighbours ONLY through clearly drawn door openings. Keep wall lines clean and straight. Do NOT merge separate rooms into one open area.aStyle: THICK black walls, thin single-weight black outlines for openings and fixtures, a pure WHITE background, NO colour fill, no pastel shading, no texture, no gradient, no 3D, no perspective, no isometric, no shadow. Do NOT write any text labels, room names, or legend inside the image; do NOT draw camera icons, field-of-view cones, or arrows except one small north arrow in a corner.zDo NOT draw decorative furniture or clutter (sofas, dining tables, chairs, TVs, lamps, wardrobes, framed pictures, bags, rugs, plants). Do NOT draw any story-state cues (blood, bodies, footprints, herbs, cups, coloured circles).cJt|trt|ts|SdSN) isinstanceintbool)ns r _intr#s1c":a+>1HDHr c |Dcic]}|g}}t|ts|S|D]}t|ts|jd}||vr*t |jd}|G||j |t|jdt r|jdndt|jdt r|jdndd|D]}||jd|Scc}w) zGroup numbered_elements by base_layer_decision (included layers only). Reads the closed enum + integer ``number`` + passthrough ``label`` / ``position_hint`` only. No text matching.base_layer_decisionnumberlabel position_hint)r&r'r)c |dSNr&r )es r z!_group_elements..sQx[r )key)rlistdictgetr#appendstrsort)numbered_elementsincluded_layerskgroupsellayernums r _group_elementsr<s?N.Noq"uoF.N ' . "d# ,-  ' 266(#$ ; u (2266'?C(HRVVG_bbff_5s;'AC    "q 01 M-/Os Dcg}|D]6}|drd|dnd}|dxsd}|jd|dd||8d j|S) Nr)u — r(r'z (unlabelled)z r&z: r)r2join)itemsoutitloclbls r _render_linesrDsk C/1//Bb)*+k+^ R8 ~RuSE23 99S>r F)include_furniturectt}|r|jtt |t |}t g}|jtxsg}|r|jdt|z|jtxsg}|r|jdt|z|jtxsg}|r|jdt|z|jtxsg}|r|jdt|z|jt|jd|jtdj|S)ucAssemble the GENERIC simplified-FP ``t2i_prompt`` from numbered_elements. ``include_furniture`` (default False) keeps the v0 boundary: spatial enclosure + essential fixtures only. Set True only for a canary comparison. Deterministic and pure — same input → same prompt (image generation is the separate, visually-validated step). ucRooms / areas — draw each as a separate fully enclosed cell, with a LARGE clear numbered circle: uOpenings — a door is a gap in the wall with a simple thin swing arc; a window is two short parallel lines in a wall. Draw each on the wall of the room it belongs to, with its numbered circle: uEssential fixtures — draw each as the simplest possible plan symbol, strictly INSIDE its own room (never floating in another room or the open area), with its numbered circle: umRoom-defining items — draw each as a single basic rectangle inside its own room, with its numbered circle: uNumbered circles must be LARGE, sparse, and clearly readable — one per listed element above, placed inside the room/on the element it marks.z )r/_DEFAULT_INCLUDED_LAYERSr2LAYER_PERSISTENT_FURNITUREr<tuple _PREAMBLEr1LAYER_STRUCTURAL_UNITrD LAYER_OPENINGLAYER_PERSISTENT_FIXTURE _EXCLUSIONS _STYLE_RULESr>) r5rEincludedr8partsroomsopeningsfixtures furnitures r build_simple_fp_promptrVsU,-H23 .h @F!{E JJ, - 3E   -/z+validate_light_fp_output..s)P aL$4$4Q$7;N$N=s''z-state-overlay elements selected (forbidden): z selected z > soft target uI (advisory — verbose but rendered; tighten the prompt if visually busy)rlrh)room_schematic_prompt too short / missingc3$K|]}|| ywrr )rr"s r rz+validate_light_fp_output..s#" ! sc3pK|].}t|trt|jd0yw)r&N)rr0r#r1)rms r rz+validate_light_fp_output..s/0P!a&QUU8_%0Ps46rk) rr0r/r#r1r?_MUST_KEEP_LAYERSsortedr2lenr3)rawr5r~diagsr9r"rselected selected_numsr: must_keepdropped_skeletonoverlay_selectedpromptrrs @r validate_light_fp_outputr{sd*E c4 b$&2 568 8 $&L#T*#B"d#)*=&(ff-B&CLO $'',-333A a  QUU8_3  !):1AMQM:#/#5#5#7V#7xq%5DU;U#7IVi-78 56F5G H JP PP DEUDVWX =.. M*+?;M:NOW X YWW, -F!&#.VBF 6{R @A  L&6"6 L3v;";LB!'"=1!" 038I0J0Pb0P"    ? ;Vs1HHH'H8H roomdoorwindowu4You convert a detailed apartment floor plan into a SPARSE, CV-readable top-down PARTITION DIAGRAM prompt for a TEXT-TO-IMAGE model (NO reference image is given). The detailed plan has far too many numbered markers for the model to render cleanly; a SHORT prompt with FEW markers renders far better. You are given the floor plan's numbered_elements (number, label, category, position_hint, base_layer_decision) and, for each, a camera_use_count: how many shot cameras actually FRAME that element. This is a frequency / importance signal — a HIGH count means cameras show it often (e.g. a TV, or a window/curtain repeatedly framed) and it MUST NOT be silently dropped. Classify EVERY numbered element into exactly ONE bucket: - geometry_skeleton: every room/area (base_structural_unit) and every opening (base_opening = door/window). These DEFINE the building shape. They are drawn, but WITHOUT numbered circles — the unnumbered structural shell. Set kind to room, door, or window. - essential_numbered_markers: the 8-10 fixtures/furniture that DEFINE a room's identity or are framed by cameras (high camera_use_count) — e.g. a toilet/sink for a bathroom, a bed for a bedroom, a counter/cooktop for a kitchen, a sofa/TV for a living room. Keep each element's ORIGINAL number; each appears EXACTLY ONCE; these get the LARGE numbered circles. Echo each element's camera_use_count. - high_frequency_candidates: any element with a HIGH camera_use_count that you nonetheless DROP. You MUST give a drop_reason — dropping something cameras frame is risky, so justify it. - metadata_only: decorative / minor / duplicate clutter (rugs, plants, framed pictures, small props) AND every story-state overlay (state_overlay_*). Not drawn. Aim for 8-10 essential_numbered_markers — NEVER the whole inventory; keeping every fixture is the mistake to avoid; but a high camera_use_count element should almost always be kept. Then write room_schematic_prompt — ONE SHORT image-generation prompt that: - draws THICK solid black outer + interior partition walls; each room a fully ENCLOSED cell connected to neighbours ONLY through clearly drawn door openings (a door = a gap in the wall with a thin swing arc; a window = two short parallel lines in the wall); explicitly says do NOT merge separate rooms into one open area; - MARKER PLACEMENT (critical): EVERY numbered circle sits INSIDE the building outline; a door/window marker sits ON its wall, touching from the inside, NEVER floating outside the building; a fixture marker sits INSIDE its own room, never floating in another room or the open area; EXACTLY ONE circle per number — if an element represents SEVERAL identical items (e.g. multiple chairs, desks, lockers, or cabinets), draw the symbols but place the numbered circle on only ONE representative item and do NOT repeat that number on the others; - draws each essential fixture as the simplest flat plan symbol strictly inside its own room; - uses LARGE clear numbered circles ONLY for the essential markers (their ORIGINAL numbers), big and sparse; - pure WHITE background, NO colour/shading/texture/gradient/3D/perspective/shadow; and the image must contain ZERO letters or words — NO room names, NO text labels, NO legend, NO captions anywhere (e.g. do not write 'kitchen', 'bedroom', 'entrance'); the ONLY characters allowed are the digits inside the numbered marker circles. Keep it SHORT — describe rooms by their position_hint relationships (which is north/east/etc., what connects to what). Use the given labels only to understand what each element is; do NOT write them into the image. Output strict JSON only.)r&r'kind)r&r'rdcamera_use_count)r&r' drop_reason)r&r'r&r')geometry_skeletonessential_numbered_markershigh_frequency_candidatesrYrlFREQ_SELECT_SCHEMAcddl}tdztzdz|jtdz}t j |jdjddS)zStable short hash of the NB2 freq-aware selection system prompt + schema. Provenance + cache-invalidation lever for the step config_hash. Pure.rNrT) sort_keysrr) rxFREQ_SELECT_PROMPT_VERSIONNB2_FREQ_SELECT_SYSTEMrzrrrrr)r{payloads r freq_select_prompt_hashrtsa #T),BBTI ++(D+ 9 :  >>'..1 2 < < >s CCr ci}t|trG|D]B}t|tst|j d}|1|j |dDt|tr|D]}t|ts|j d}t|ts6t }|D]=}t|}|||vr|j||j |ddz||<?|S)u Aggregate the per-element camera-framing frequency (importance signal). ``camera_use_count[n]`` = the number of ``camera_recommendations`` entries whose ``use_numbered_elements`` contains ``n``. This is a pure INTEGER signal (how many shot cameras frame the element) — never a label/text inspection (absolute-rule compliant). Every integer-numbered element is present (0 when no camera frames it). A number repeated inside one camera's ``use_numbered_elements`` counts once for that camera. r&ruse_numbered_elements)rr/r0r#r1 setdefaultsetadd) r5camera_recommendationscountsr9r"camused seen_in_camitems r compute_camera_use_countsrs F#T*#B"d#)*=%%a+ $ ($/)Cc4(7723DdD)%KJ9[ 0""JJq!,q0q * Mr cddl}t|tr|ni}g}t|tr|D]}t|tst |j d}|1|j ||j d|j d|j d|j dt|j |ddt|jd |id td S) u@Assemble the strict {system, user, schema} bundle for the NB2 selector. The user payload is the numbered_elements carried through verbatim, each annotated with its integer ``camera_use_count`` (the frequency signal). No scenario tokens live in this module — only the runtime data the model classifies.rNr&r'rqr)r%)r&r'rqr)r%rr5Frrrt) rxrr0r/r#r1r2r rrzr)r5camera_use_countsr{rr|r9r;s r !build_freq_aware_selection_bundlers",->"E 2FH#T*#Bb$'rvvh'(C{ OOFF:.!#!8')vv.C'D$' 3(:$;   $) 0(;% P$ r )rselect_hard_capcg}tts ddgggdgdSit|trI|D]D}t|tst|j d}|1|j d|<Ft|tr|ni}dfd }|d} t |d } t | } t |d t fd | D} | r|jd | | Dchc] }|vs| } }t | }t| t| k7rBt | Dchc]}| j|d kDs|c}}|jd|| Dchc]}j |tvr|}}t | |z }|r|jd|jDchc]\}}|tk(s|}}}t ||z | z }|r|jd|t fd| | zD}|r|jd|t||kDr"|jdt|d|dn7t|tkr%|jdt|dtd| |zt fd|jD}|r|jd|j d}t|tr|nd}t|dkr|jd| xr| xrt|dk\}|||t |||dScc}wcc}wcc}wcc}}w)uValidate the NB2 freq-aware selection against the STRUCTURAL guards. ``ok`` reflects STRUCTURAL INTEGRITY ONLY (Codex lock #1 — do not over-gate): every room (``base_structural_unit``) MUST be drawn (in the geometry skeleton or as an essential marker); no ``state_overlay_*`` element may be an essential numbered marker; the prompt must be present. The selection COUNT and the high-frequency protection are ADVISORY diagnostics — they never flip ``ok``, because a too-tight gate would fall back to the cluttered detailed FP (the very failure mode being fixed). Returns ``{ok, room_schematic_prompt, essential_numbers, skeleton_numbers, unjustified_high_freq, diagnostics}``. Fr(r)rrlessential_numbersskeleton_numbersunjustified_high_freqrr&r%cg}j|xsgD]A}t|tst|jd}|1|j |C|Sr+)r1rr0r#r2)r.r@rr"rs r _numsz,validate_freq_aware_selection.._numssU''#,$"$A!T"x)=JJqM %  r rrrc3,K|] }|vs| ywrr rs r rz0validate_freq_aware_selection..sQ-Q1L;Pq-s z+essential markers not in source (dropped): rz(duplicate essential markers (advisory): zBgeometry_skeleton has non-structural / unknown numbers (ignored): zroom(s) not drawn (forbidden): c3PK|]}j|tvr|ywrrrs r rz0validate_freq_aware_selection..s-70a   A "5 5 0s#&z-state-overlay selected as drawn (forbidden): z essential z > hard cap u( (advisory — busy, tighten the prompt)z < target min u! (advisory — may be too sparse)c3jK|]*\}}t|tr|tk\r |vr|vr|,ywr)rr NB2_HIGH_FREQ_MIN_CAMERAS)rr"cdrawn hf_justifieds r rz0validate_freq_aware_selection..,s<#$da a !'@"@ UNq 4 ns03zBhigh-frequency elements dropped without justification (advisory): rlrhr)r.r3returnz List[int])rr0r/r#r1rrr2rcountrr?rKNB2_SELECT_TARGET_MINr3)rr5rrrr9r"rr essential_raw skeleton_set essential_setunknown_essentialessential_valid_setessential_validdupsskeleton_validskeleton_invalidlrR rooms_missing overlay_drawnrrrrrrs` @@@r validate_freq_aware_selectionrs$E c4 b%'R)+=R"E 2F67Mu012L &Mu89:LQ-QQ 9:K9L M O&3ImqL7H1mI01O =S//-N-Q=3F3Fq3IA3Mq-NO ?vFG a   A "3 3 <l^;< P  " # (--/ N/41a18M3MQ/E N5>14GGHM 6}oFG7!L077M ;M? K M ?o- _-.l?:KL6 6 7 _  5 5 _-.n=R2!6  CJO Os*9 M M 4MM1 M MM)rr3)r"rrz Optional[int])r5rr6rIrzDict[str, List[Dict[str, Any]]])r?zList[Dict[str, Any]]rr3)r5rrEr!rr3)r5rrrn)rrr5rr~r rrn)r5rrrrzDict[int, int])r5rrrrrn) rrr5rrrrr rrn)5__doc__ __future__rrtypingrrrrrr rrrKrLrMrHrGrJrOrNr#r<rDrVRENDER_ROLE_SKELETONRENDER_ROLE_ESSENTIALRENDER_ROLE_METADATALIGHT_FP_RENDER_ROLESLIGHT_FP_SELECT_TARGETLIGHT_FP_SELECT_HARD_CAP frozensetrrryro__annotations__r}rrNB2_SELECT_TARGET_MAXNB2_SELECT_HARD_CAPrNB2_GEOMETRY_KIND_ROOMNB2_GEOMETRY_KIND_DOORNB2_GEOMETRY_KIND_WINDOWrrrrrrrr r r rs2#,,"(/ 48D  S I-2$@$999 9\-+&4mDE>! H ! (-%y1$h/,4,.C?E$F%x0 I   (-%y1$h/%x0 :   +3!D$+vx6HI=@G'#'T@7N N$'NNNT #7x ! (-%y1$h/%-.0F08238   (-%y1$h/%x0)/(;  N  '  (-%y1$h/$*H#5 ?  &  (-%y1$h/&w/  +3!Do8ry@&N@H! D!!47!!H  /2  N". { {{ {  {  {r