T'j_UdZddlmZddlmZmZddlmZmZm Z m Z e dhZ de d<dZd e d <d Zd e d <dZd e d<e hdZde d<GddeZd*dZ d+dZ d,dZdZdZdZdZe ddhZde d< d-dZed d.d Zd*d!Z d/d"Zd0d#Z d$eed% d1d&Z! d2d'Z"d$d( d3d)Z#y$)4uW21B (2026-06-08) — dwelling_zone_map pure core. The cross-space consistency problem: the edge-judge (``bg_space_partition``) cannot group bgs of the SAME dwelling that are framed from different camera angles (on an observed multi-room dwelling it scored only one bg pair as ``same_space`` while the cross-angle pairs fell to ``adjacent_related`` confidence below the strong gate) — so the same dwelling renders as many independent plates and looks different in every shot. This module owns the DETERMINISTIC core of the alternative grouping path: group bgs into dwelling candidates, validate the VLM zone-mapping output, assemble the per-dwelling **zone map** contract (zones + bg/shot zone assignments) by an exact-string join, compare it diagnostically against the edge-judge grouping, and provide a synthetic fixture. It performs NO LLM / image / VLM / DB I/O — the clean black-and-white floor-plan image generation and the gpt-5.5 vision call live in ``dwelling_zone_map_provider`` and the step, which inject their output through ``validate_vlm_zone_output`` / ``assemble_zone_map``. Boundaries (scenario-leakage guard): - Code joins on exact bg_id / fp_id / shot_id strings and on closed structural vocabularies (``surface_role``). It NEVER lexically inspects labels or prose to infer a zone — zone identity comes only from the VLM assignment + the structured space analysis. - The CLEAN floor-plan reference (the one a downstream renderer may use as an i2i anchor) is kept strictly separate from the ANNOTATED reference (shot-number overlay) which is ``must_not_be_used_for_render`` — the contract carries both refs and the flag, never a single ambiguous ref. ) annotations)Counter defaultdict)AnyDictListOptional interior_room frozensetINTERIOR_SURFACE_ROLESdint GRID_SIZEg?floatCONFIDENCE_FLOOR_ZONE_BBOX_PAD>okfallback ambiguousASSIGNMENT_STATESceZdZdZy)DwellingZoneMapErrorz2Fail-closed signal for the dwelling-zone-map core.N)__name__ __module__ __qualname____doc__\/Users/manta/Documents/Projects/TheRoad-I1/backend/app/modules/pipeline/dwelling_zone_map.pyrr8s`` id, NOT the label: two rooms can share a label (e.g. two bedrooms) and a label key would merge them. The raw label is carried verbatim for the FP prompt / VLM but never used as a key or parsed. Empty / malformed input or zero structural areas yields no rooms, so the caller falls back to the bg-derived analysis. Nz#floor_plan_prompt.numbered_elements)rIrJrKrLglobal_fixturesrMcategorybase_layer_decisionrUnumberrKErSrLrh) r#rXr6r5r7_NE_AREA_CATEGORY_NE_AREA_DECISIONr'r:_NE_OPENING_CATEGORY_NE_OPENING_DECISIONrf)numbered_elementsrFr\catdecrUnums r %build_analysis_from_numbered_elementsrus0<tbb)NC ' . !T" eeJee)*",QUU7^S"Agt # #/@(@%%/C3< L  C5 "'qY!% ! ( (S4H-HJ&&u- , ,%&--e4/2 Jr) grid_sizec rg}t|xsg}t|tsddt|jdgidSi}t}t |D]\}}d|d} t|t s|j| d2|jd} t| tr| s|j| d j| |vr|j| d | d | |vr|j| d | d |j| |jd } t| tr| js|j| d| d|jd} dx} }t| t s|j| d| dn| jd| jd}} t| r t|s|j| d| ddx} }n8d| cxkr|krnn d|cxkr|ksn|j| d| d| d|d|jd}t|rdt|cxkrdksn|j| d| d| t|jdtr|jdndt| tr| nd| | |dndt|r t|ndt|jd tr|jd ndd!|| <t||z }|D]} |jd"| |rd|idSd#g|dS)$uStrictly validate the per-bg VLM zone-mapping output. The VLM returns a list of per-bg records ``{bg_id, room, zone_group, grid{x,y}, confidence, evidence}`` (the raw grouping uses opaque ``zone_group`` strings; the deterministic Zxx assignment happens in ``assemble_zone_map``). This checks: list shape; every record a dict with a known bg_id; exact coverage (no missing / unexpected bg); a non-empty ``zone_group`` string; ``grid`` an ``{x,y}`` pair inside ``[0, grid_size]``; ``confidence`` a number in ``[0, 1]``. Returns ``{"ok", "blockers": [str], "normalized": {bg_id: record}}`` — ``normalized`` is ``{}`` whenever ``ok`` is False. Fzvlm output is not a list ())rblockers normalizedzvlm[]z not a dictr@z.bg_id missing or non-stringz.bg_id=z not an expected bgz duplicated zone_groupz #z zone_group missing or emptygridNz grid not an {x,y} objectxyz grid x/y not numericrz grid out-of-bound (, confidencegg?z confidence not in [0,1]room)r~revidence)r@rr|r}rrzmissing zone assignment for bg T)r9r#rXtyper enumerater6r:r5r7r;strip_is_numrr4)rawexpected_bg_idsrvryexpectedrzseenidxrecpfxr@groupr}gxgyconfmissings r validate_vlm_zone_outputrs($H?(b)H c4 5d3i6H6H5IKL  "$JDcNSSEm#t$ OOse;/ 0  %%U OOse#?@ A   OOse75)3FG H  D= OOse75);? @   %%%U[[] OOse2eW,HI JwwvR$% OOse2eW,GH IXXc]DHHSMB2;gbk3%r%0E FGR2**qB/C)/Ce2eW$8AbTCEww|$t}SE$K%>3%> OOse2eW,DE F'1#''&/3'GCGGFOR#-eS#9%r*,."2&d)0%+D/9#''*:Ms/S +Y[  5M#^X_%G9%ABDDBj AArcTt|ttfxrt|t Sr")r#rrr$r%s r rrAs" a#u & Bz!T/B+BBrc|Dcgc]L}t||jdtr(||jdr||jdN}}|syt|j dddScc}w)Nrr)r#r5r7r most_common) member_idsvlm_normalizedbrKs r _most_common_roomrEs  A nQ'++F3S 9 1  ! !& ) qf%    5> % %a ( +A .. sAA;cv|d|cxkxr|d|dzkncxr|d|cxkxr|d|dzkScS)Nr~wrhr)bboxr~rs r _point_in_bboxrSsP S Q/$s)d3i// 4 I 3d3i$s)3 3 3rN)space_analysisrvconfidence_floorc  -./i}i} t|D];} || jdxsd} | |vrdt|dzd|| <|| | | <=tt} t|D]} | | | j | i} t |tr|jdxsgD]m}t |tst |jdts4|jdxsgDcgc]}t |ts|c}| |d<oi/|jD]Y\} .| .}|Dcgc],}t ||jd tr||d .}}|Dcgc]}|d  }}|Dcgc]}|d  }}|rqtd t|tz }td t|tz }t|t|tz}t|t|tz}n d d ||f\}}}}t||xs| }|Dcgc]'}t||jd r||d )}}|||||z ||z d| j|g|r t|nd|d/.<\g}i}t|D]} || }| | .t |jd tr|jd nd-|jd }-d} nMt| xst||k}!t!-./fd/D}"|!r|j | |!s|"rdnd} .t|j| xsg-t|r|nd| d|| <tt}#|xsijD]=\} }$| |vr |$xsgD]*}%t |%ts|%s|#|%j | ,?i}&t|#D]O}'tt#|#|'}(|(s|(d } || d} t|(dkDrd} | || d|| d| d|&|'<Qt/})|)dk\}*|*rdn|)dk(rdnd}+|d||ddd d!|i/||&|)|)dk|dd"|*|+d# },dg|,d$Scc}wcc}wcc}wcc}wcc}w)%uDeterministically join the validated VLM zone output into the per-dwelling zone-map contract. Steps (exact-string / numeric joins only — no lexical inference): 1. Assign ``Zxx`` ids in first-seen order over bgs sorted by bg_id, so the first bg's zone is always ``Z01`` (stable, intuitive). 2. Derive each zone's ``grid_bbox`` from its members' grid points (padded), ``label`` from the members' most-common ``room``, ``structure_cues`` from the space analysis by an EXACT room-name match (empty when no exact match), ``confidence`` = member min. 3. ``bg_zone_assignments``: zone, member grid_focus, bound shots, ``assignment_state`` (``ok`` / ``ambiguous`` — low confidence or a grid point landing inside another zone's bbox / ``fallback`` — no grid). Low-confidence bgs are also listed in diagnostics. 4. ``shot_zone_assignments``: invert ``bg_shot_map``; a shot bound to two bgs of this dwelling is ``ambiguous``. The CLEAN render ref and the ANNOTATED (number-overlay, render-forbidden) ref are carried as separate fields so a downstream i2i anchor can never grab the numbered image. Returns ``{"ok", "blockers", "zone_map"}``. r|rZr02drKrTrVr}r~rrr)r~rrrN)rU grid_bboxstructure_cuesr member_bg_idsrc3^K|]$}|k7xrt|ddd&yw)rr~rN)r).0z2r}r[rNs r z$assemble_zone_map..sH Bc Q"59[#949d3iPQs*-rrrOr1 grid_focusrassignment_staterrOr)r@rOrrr/ single_zoneno_zonesFTvlm_mapping_onlyrefmust_not_be_used_for_render asset_rolesize) zone_countrlow_confidenceedge_judge_comparison r+ synthetic clean_fp_refannotated_fp_refr}rNbg_zone_assignmentsshot_zone_assignments diagnosticsr2r3)rryzone_map)r4r5r=rrXr:r#r6r7r<maxminrrrranyr9)0r+r bg_shot_maprrrrvrzone_id_by_groupbg_zoner@rmembers_by_zone room_fixturesrfrrptspxsysx0y0x1y1rUconfsr bg_assignrrstatelowr shot_to_bgsshotsrE shot_assignshotbgsrr2r3rr}r[rNs0 @@@r assemble_zone_maprZs@B(* G'u%)),7=2 ( (()#.>*?!*CC)H&I U #)%0 ( -8,=O''..u5(+-M.$'##G,22A!T"z!%%-'E !j 1 7R 7, 7!Jq# 1 f %  " "cafc "! "cafc " QB.01BQB.01BYB. 89BYB. 89B9i7NBB!*n=F  ~a(,,\:; 1 l +  !brBwG+//r:(-#e*4'  c -/>!#N "I'U#en",SWWV_d"Cswwvww|$ <Edm#EuT{5E'EC I %%e,$'9K4E{u5;<")$-$T %  %#(4)4D(9K$*113 u  & "A!S!aA%%e,4 #%K{#ST*+, A% !34 s8a<E ' 2#E*<8 %  D$UJqJ*/z$#+/, #(!,$%>,%)  !!6)H,BH ==g, # " s$9Q5Q51Q:7 Q? R),R c 8|jdxsijDcic]\}}||jd}}}i}t|tr|jdxsi}t |}|r t t |t |zng}g}g} d} tt|D]t} t| dzt|D]W} || || } }|||| k(}|||| k(}|r|s|j|| g;|r|s| j|| gS| dz } Yv||r"tt |jndtt |j|| | t|t|dz zdzdScc}}w)uDiagnostic-only comparison of the VLM zone grouping against the edge-judge plate grouping (``bg_space_partition.node_assignments``). The edge-judge cannot unify same-dwelling bgs framed from different angles; this surfaces exactly the pairs the zone map merged (``merged_by_zone_map``) and any it split that the edge-judge had unified (``split_by_zone_map``). It NEVER changes an assignment — the zone map stays authoritative. Returns empty lists when no plan exists. rrOnode_assignmentsrrr/)edge_judge_availableedge_judge_group_countzone_map_group_countmerged_by_zone_mapsplit_by_zone_mapagreement_pair_countcompared_pair_count) r5r<r#r6r$r4r9ranger=r:values)rpartition_planbgr^rnode availablecommonmergedsplit agreementijrz_samee_sames r compare_to_edge_judgers ll#89?RFFHHEB AEE) H D.$'!!"45;T I1:VCL3t9, -F FEI 3v; q1uc&k*A!9fQiqAQZ71:-F!WQ'Ff q!f% aV$Q + !*=F#c$++-&8"9A #C(8$9 :$" )"6{c&kAo>!C 1sF)rc|xsi}t|Dcic]'}|dt|j|xsgdddd)}}|dddddddtii|iddgdd d d d d Scc}w)u Non-authoritative placeholder zone map (default-off / no-real-provider path). Asserts NOTHING — every bg is ``fallback`` with no zone — so a synthetic run can never be mistaken for a real VLM observation, while the contract shape stays uniform for downstream readers. NrrTrrrrzNsynthetic_fixture: non-authoritative placeholder; no real VLM zone observation)rrrrnoteFsynthetic_fixturer)r4r5r)r+r,rrrs r compute_synthetic_zone_mapr3s#K. !B {r28b9 *   ! +/, #(!# %)'  !41  s,A )r&rreturnr$)r>Dict[str, Any]rzDict[str, Dict[str, Any]])rYrrr)rqrrr)rrr List[str]rvrrr)rrrrrz Optional[str])rzDict[str, int]r~rrrrr$)r+r7rrrzDict[str, List[str]]rrrrrOptional[Dict[str, Any]]rvrrrrr)rrrrrr)r+r7r,rrzOptional[Dict[str, List[str]]]rr)$r __future__r collectionsrrtypingrrrr r r __annotations__rrrr Exceptionrr'rGr_rmrnrorprfrurrrrrrrrrr rs:#,,, %..?$@ @ 3 %()HI9I=9=:-&--`...j* %&/ ;<'>)><<<F RB RBRB RB  RBjC / /+9 / /04.e> e>#e>& e>  e>  e>-e>e>e>e>P..-. .j37 / / /0 /  /r