+ UV-jGC^RIHt^RIt^RIt^RIt^RIHt^RIHt^RI H t ^RI H t ]P!R4tRR R RR R R//tRR R RR R^RR/R^R^/tRRRRRR RR./R]RRR R^/R/]CR^/C/RRR.RR/tRRRRRR RR./R]RRR /RRR R^/R/]CR^/C/R.R]ORR/tRRRRRR R^/R RR R^/R!RR R^/R"RR R^/R/]CR^/C/R.R^ORR/tRRRRRR R^/R RR R^/R"RR R^/R#RR R^/R/]CR^/C/R.R_ORR/tRRRR$RR R^/R%R&]]./R'RRRR(RR R^/R)RR R R&]].///RR(R).RR//RR'.RR/t]!R*R*R+7!R,R-44t]!R*R*R+7!R.R/44t!R0R1]4tR2R3ltR4R5ltR6R*R7R*/R8R9lltR6R*R:RR;RRlltR?R/R@RAlltRBtRCRDlt RERFlt!RGRHlt"RIRJlt#RKRLlt$RMRNlt%RORPlt&RQRRlt'RSRTlt(RURVlt)RWRXlt*R`RYRZllt+R[R\lt,R#)a) annotationsN)Mapping) dataclass)gcd)Anyz^#[0-9A-F]{6}$typearrayitemsstringpatternintegerminimummaximumminItemsmaxItemsobject propertiesenumobjbboxdesc minLength color_paletterequiredadditionalPropertiesFtext aestheticslightingphotomedium art_stylehigh_level_descriptionstyle_descriptionanyOfcompositional_deconstruction backgroundelementsT)frozenslotscx]tRt^]t$R]R&R]R&R]R&R]R&RtR]R&R tR ]R &R tR]R &R tR ]R&Rt R #)NormalizedPromptstrrboolis_json_captionis_structured_caption was_wrappedztuple[str, ...]warningsN str | Noneprompt_expansion_modelFprompt_expansion_usedprompt_expansion_error) __name__ __module__ __qualname____firstlineno____annotations__r2r4r5r6__static_attributes__r7s/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_vlm/models/ideogram4/prompting.pyr,r,]sD I "Ho")-J-"'4')-J-r>r,c6]tRt^it$R]R&R]R&R]R&RtR#)PromptExpansionResultr-rraw_textmodelr7N)r8r9r:r;r<r=r7r>r?rArAis IM Jr>rAc]tRt^ptRtR#)PromptExpansionCaptionErrorr7N)r8r9r:r;r=r7r>r?rErEpsr>rEc V^8dQhRRRR/#)captionMapping[str, Any]returnr-r7)formats"r? __annotate__rLtsJJ-J#Jr>c4\P!VRRR7#),F) separators ensure_ascii)rN:)jsondumps)rHs&r?format_captionrTts ::g*5 IIr>c V^8dQhRRRR/#rGpromptr-rJr.r7)rKs"r?rLrLxs**#*$*r>cD\V4pVfR#\V4'*#)NF)_parse_json_caption_caption_warnings)rWrHs& r?r0r0xs"!&)G ) ))r>auto_json_captionwarnc(V^8dQhRRRRRRRR/#)rGrWr-r[r.r\rJr,r7)rKs"r?rLrLs2         r>cnVP4p\V4'd^\V4p\\ V44pV'd"VFp\ P !V^R7K \VRV'*RVR7#V'g\VRRRR7#\\\V44RRRR7#)rG stacklevelTF)rr/r0r1r2)rr/r0r1) strip_looks_like_json_object_loads_json_captiontuplerZr2r\r,rT_minimal_caption)rWr[r\strippedrHissuesissues&$$ r?normalize_promptris ||~Hx((%h/(12  e2  &,*    !"'    ,X6 7"  r>r4widthheightc4V^8dQhRRRRRRRRR RR RR R /#) rGrWr-r[r.r4r3rj int | Nonerkr\rJr,r7)rKs"r?rLrLsP< < < < ' <  <  <  < < r>c VP4p\V4'gVf\VVVR7#\VV\ W44R7p\VP RVR7p\ VP VPVPRVPVPRR7# \dp T'g \R4T hT'd\P!RT 2^R7\TRTR7p \ T P T PT PT PT P\T4R\T 4R 7uRp ? #Rp ? ii;i) N)r[r\)rC aspect_ratioFT)rr/r0r1r2r4r5zPrompt expansion failedzVPrompt expansion failed; falling back to the minimal Ideogram 4 JSON caption wrapper. r_)rr/r0r1r2r4r5r6)rarbri!generate_prompt_expansion_caption_aspect_ratio_from_sizerr,r/r0r2rCrE ValueErrorr\r1r-) rWr[r4rjrkr\rf expansionpreparedexcfallbacks &$$$$$ r?prepare_promptrwsP||~Hx((,B,J /  + 5 (0? $ NN#   $44"*"@"@&&#,??"&   '  67S @  MM4475:  $ "   $44"*"@"@ ,,&&#&'=#>"'#&s8   s%A2B(( E3EA?EEEroc(V^8dQhRRRRRRRR/#)rGrWr-rCror3rJrAr7)rKs"r?rLrLs277 7 7 7  7r>c N^RIp^RIHp^RIHp^RIHp^RIHp^RI H pRp Rp V!V4wrRRR\/RR R\W4/.p V!V V PV 4p \V R 4'd V PMT p V!V \ 4pV!V V V V.R R R 7pVP"P%4p\'\)\+V44V\-V4R7? ? VP/4VP14# ? ? TP/4TP14i;i)rN)generate)apply_chat_template)"build_json_schema_logits_processor)loadrolesystemcontentuser tokenizerFT)logits_processorsverboseskip_special_tokens)rrBrC)gcmlx.corecoregenerate.dispatchrz prompt_utilsr{ structuredr|utilsr}PROMPT_EXPANSION_SYSTEM_PROMPT_prompt_expansion_user_promptconfighasattrrIDEOGRAM4_CAPTION_SCHEMArrarArT_load_prompt_expansion_captionr-collect clear_cache)rWrCrormxrzr{r|r} model_obj processormessagesformatted_promptrlogits_processorresultrBs&$$ r?rprps2 -3@II&#E{ Xy*H I8N  /       $+9k#B#BI    >  $    /0 $  ;;$$&$ >x HIe* y   y  s B5D$D$aYou prepare structured JSON captions for Ideogram 4 image generation. Return only JSON matching the provided schema. Preserve the user's intent, requested wording, and constraints while making the visual description more specific and useful to the image model. Always include a concrete high_level_description and compositional_deconstruction. Write descriptions as observations of the desired image, never as commands or as a copy of the user's request. The background must describe the actual scene, not a generic placeholder. Use one obj element for each explicitly named visual subject. Use one text element for every quoted string or other visible wording the user requests. Copy each text field verbatim, including capitalization, punctuation, line breaks, and non-ASCII characters. Do not hide requested lettering inside an obj description. Bounding boxes are optional. Include them only when useful for layout, using integer normalized [0, 1000] coordinates as [y_min, x_min, y_max, x_max] with y_min < y_max and x_min < x_max. If style_description is included, use exactly one of photo or art_style. Use only uppercase #RRGGBB values in color palettes. c$V^8dQhRRRRRR/#)rGrWr-ror3rJr7)rKs"r?rLrL3s&UU#UZUCUr>c2V'dRV R2MRpRV RV 2#)z Target aspect ratio: zH. Use it only to plan the composition; do not add an aspect_ratio field.z4Convert this prompt into an Ideogram 4 JSON caption: r7)rWroaspects&& r?rr3s:  ",09 9  B&F8 TTr>c$V^8dQhRRRRRR/#)rGrjrmrkrJr3r7)rKs"r?rLrL=s!??:?z?j?r>cV'd V'gR#\\V4\V44p\V4V, R\V4V, 2#)NrQ)rint)rjrkdivisors&& r?rqrq=sC #e*c&k*G%jG# $Ac&kW&<%= >>r>c V^8dQhRRRR/#rVr7)rKs"r?rLrLDs""C"D"r>c$VPR4#){) startswithrWs&r?rbrbDs   S !!r>c V^8dQhRRRR/#)rGrWr-rJzdict[str, Any] | Noner7)rKs"r?rLrLHs(=r>cVP4p\V4'gR#\V4# \dR#i;iN)rarbrcrr)rWrfs& r?rYrYHs=||~H "8 , ,"8,, s 0 ??c V^8dQhRRRR/#rGrWr-rJdict[str, Any]r7)rKs"r?rLrLRsr>c\P!V4p\ T\ 4'g \R4hT# \Pdp\R4ThRp?ii;i)zInvalid Ideogram 4 JSON captionNz)Ideogram 4 JSON caption must be an object)rRloadsJSONDecodeErrorrr isinstancedict)rWvaluerus& r?rcrcRs^E 6" eT " "DEE L   E:;DEs;A! AA!c V^8dQhRRRR/#)rGrr-rJrr7)rKs"r?rLrL\sr>c\P!V4p\ T\ 4'g \R4h\ T4pT'd\RT^, 24hT# \Pdp\R4ThRp?ii;i)z0Prompt expansion model did not return valid JSONNz3Prompt expansion model did not return a JSON objectz4Prompt expansion model returned an invalid caption: )rRrrrErrrZ)rrrurgs& r?rr\s 4  eT " ") A  u %F )B6!9+ N   L   ) >  sA##B 8 BB c V^8dQhRRRR/#rr7)rKs"r?rLrLos  S ^ r>c  RVRRRRRRRV/.//#)r#r&r'zHThe setting, environment, and surrounding context implied by the prompt.r(rrrr7rs&r?rereos4 &&  &%89 )   r>c V^8dQhRRRR/#)rGrHrIrJ list[str]r7)rKs"r?rLrL|s8808Y8r>c.pVPR4p\V\4'gVPR4M\ VPR44'gVPR4VPR4p\V\ 4'gVPR4M.\ V4FwrEVP\WT44K! VPR4p\V\4'dRV9pRV9pWx8XdVPR 4Rp V F:p \ VPV 44'dK%VPR V R 24K< V'd2\ VPR44'gVPR 4V'd2\ VPR44'gVPR 4MVeVPR4VP\V44V#)r&zOIdeogram 4 JSON caption should include a 'compositional_deconstruction' object.r'zgIdeogram 4 JSON caption should include 'compositional_deconstruction.background' as a non-empty string.r(zYIdeogram 4 JSON caption should include 'compositional_deconstruction.elements' as a list.r$r r"zaIdeogram 4 JSON caption 'style_description' should include exactly one of 'photo' or 'art_style'.z?M mW - -  5 $M$5$5l$CDD MMS !$$Z0(D)) MME  !*( 3  /=>!4 KK+ ,E%!!u$ #u,  % MM9 8C' #77 u46 1%))G2DEE MM&  !5eii 6L!M!M MM&    N  MM)'23 Mr>c$V^8dQhRRRRRR/#)rGrrrrrJrr7)rKs"r?rLrLs!Ssyr>cRV R2p\V\4'gRV R2.#.pVPR4pVR 9dVPRV R24\ VPR44'gVPRV R24VR8Xd;\VPR4\ 4'gVPRV R 24R V9d&VP \VR ,V R 244V#) z&compositional_deconstruction.elements[]Ideogram 4 JSON caption 'z' should be an object.rrz!.type' should be 'obj' or 'text'.rz$.desc' should be a non-empty string.z.text' should be a string.rz.bbox>rr)rrrrrr-r_bbox_warnings)rrpathrg element_types&& r?rrs 3C5 :D eW % %+D61GHIIF99V$L?* 'v-N O   & 1 2 2 'v-Q R vj61BC&H&H 1$7QRS  nU6]tfENCD Mr>c$V^8dQhRRRRRR/#rGrrrr-rJrr7)rKs"r?rLrLs!#SYr>c\V\4'd\V4^8wdRV R2.#\;QJdRV4F 'gK RM RM !RV44'dRV R2.#Vwr#rEW$8gW58dRV R2.#.#)rz*' should contain four integer coordinates.c3"TFUp\V\4;'g9\V\4'*;'g^Tu;8*;'dR8*Mu'*xKW R#5i)rrN)rr.r).0items& r? !_bbox_warnings..sL D 4TTjs&;";TT1CTCTPTCT?TTsAAA ATFz4' coordinates should be integers between 0 and 1000.z1' should satisfy y_min < y_max and x_min < x_max.)rrlenany)rry_minx_miny_maxx_maxs&& r?rrs eT " "c%jAo'v.    s  sss   (v." "  "'E% ~'v.    Ir>c V^8dQhRRRR/#)rGrrrJr.r7)rKs"r?rLrLs::::r>cd\V\4;'d\VP44#r)rr-r.ra)rs&r?rrs" eS ! 9 9d5;;=&99r>c$V^8dQhRRRRRR/#rr7)rKs"r?rLrLs!  3 c I r>c .p\V\4'd`VP4FIwr4V RV 2pVR8XdVP\ WE44K/VP\ WE44KK V#\V\ 4'd5\V4F%wrdVP\ WA RV R244K' V#).r[r)rrr r_validate_color_paletterrr)rrrgrnested nested_pathrs&& r?rrsF%!! ;;=KC!F!C5/Ko% 5fJK 5fJK ) M E4 $U+KC MM1&F!C5:JK L, Mr>c$V^8dQhRRRRRR/#rr7)rKs"r?rLrLs!  3 c i r>c\V\4'gRV R2.#.p\V4FLwr4\V\4'd\P V4eK4VP RV RV R24KN V#)rz!' should be a list of hex colors.rz,]' should be an uppercase #RRGGBB hex color.)rrrr- _HEX_COLOR_RE fullmatchr)rrrgrcolors&& r?rrs eT " "+D61RSTT F& %%%)@)@)G)O MM63%KM ' Mr>)rrr)rrr r!)rrr!r")$)- __future__rrRrer2collections.abcr dataclassesrmathrtypingrcompiler_COLOR_PALETTE_SCHEMA _BBOX_SCHEMA_OBJECT_ELEMENT_SCHEMA_TEXT_ELEMENT_SCHEMA_PHOTO_STYLE_SCHEMA_ART_STYLE_SCHEMArr,rArrrErTr0rirwrprrrqrbrYrcrrerZrrrrrr7r>r?rs" #! ,-  G fh +< =  G fiAy$ ?   H6E73 ;2A1A:qA   E  H6F84 ";2A1A:qA (E  Hvxa8VX{A6&(K368[!4B1B:rB =E  Hvxa8VX{A668[!4fh Q7B1B:rB AE  H 68[!"Dg(;=N'OP& H vxa@Gg(>@T'UV z2 "E )  /0E'. $d#..$. $d#$  * J* #  F< #< *. <  <  < < ~7 $ 7t"0U?"& 8v,.:  r>