`gj?dZddlmZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z ddlmZmZmZdd lmZmZej*eZe eegdfZd Zd Zd Zehd Ze dGddZddZ d dddddd ddZ!ddZ"ddZ#dddddddd ddZ$y) uPet generation orchestration — the base-draft → hatch flow. Two steps, mirroring the UX across every surface: 1. :func:`generate_base_drafts` — a handful of prompt-only "what should this pet look like" variants. Cheap; the user picks one (or retries for a fresh set). 2. :func:`hatch_pet` — takes the chosen base and generates one grounded row strip per Hermes state, slices each into frames, composes the atlas, validates it, and writes the pet into the store. Splitting it this way bounds cost (4 cheap base calls per round; the ~6 row calls happen once, on the pet you actually keep) and gives each UI a natural preview/loading point. ) annotationsN)ThreadPoolExecutor as_completed) dataclass)Path)Callable)atlasimagegenprompts)GenerationErrorSpriteProvider>idlewaving running-rightT)frozencDeZdZUdZded<ded<ded<ded<d ed <y ) HatchResultz*Outcome of a successful :func:`hatch_pet`.strslug display_namer spritesheet list[str]statesdict validationN)__name__ __module__ __qualname____doc____annotations__Q/root/.hermes/venv/lib/python3.12/site-packages/agent/pet/generate/orchestrate.pyrr.s"4 I r%rcxddlm} |j|5}tj|j d}dddtj }|jd}|j|d|S#1swYDxYw#t$r#}tjd|||cYd}~Sd}~wwxYw) uKey out any solid backdrop the provider painted; save as an RGBA PNG. ``background=transparent`` is requested on every call, but image models honor it inconsistently — some still paint a flat (often near-white) backdrop. We run the same chroma-key pass the row extractor uses so every base draft the user picks between (and the reference the rows are grounded on) is a clean cutout. Best-effort: a decode failure leaves the original untouched. r)ImageRGBANz.pngPNG)formatz3base draft transparency hardening failed for %s: %s) PILr(openr remove_backgroundconvert_clear_transparent_rgb with_suffixsave Exceptionloggerdebug)pathr(openedkeyedoutexcs r&_harden_transparencyr;9s  ZZ  D++FNN6,BCE D,,U3v& 3u %  D D  JDRUV s4B %BAB B B B9B4.B94B9auto)nstylereference_imagesprovideron_draft is_cancelledc`|xsd|xstjt|xsdtj d|d fd }t dt |t}i} g} t|5} t|D cgc]} | j|| } } t| D]s}r.tj d| D]}|jn>|j\}}}||r| j|b|| |<|j |||udddt#| D cgc]} | |  }} |sst%t'| |Scc} w#t$r }tj!d |Yd}~d}~wwxYw#1swYtxYwcc} w) aGenerate *n* candidate base looks for *concept*; returns image paths. Each draft is hardened to a transparent cutout (see :func:`_harden_transparency`). Drafts are generated concurrently and *on_draft(index, path)* fires as each one finishes (not at the end) so callers can stream previews to the UI instead of leaving it blank until the whole batch is done. *is_cancelled*, when supplied, is polled cooperatively: a draft that hasn't started yet is skipped, and once it trips we stop staging/streaming further drafts and cancel any queued work (already-in-flight provider calls can't be hard-killed, but their results are dropped). Nrequire_referencescyNFr$r$r%r&z&generate_base_drafts..lr%z6pet generate: drafting %d base looks for %r (style=%s)cZr|ddfStj}tj|t tjz}tj  |} t j|d d}|stjd||ddfStjd|tj|z |t|d dfS#t$rE}tjd|tj|z ||dt|fcYd}~Sd}~wwxYw) N)r> variationpet_base)r=r?r@prefixz-pet generate: draft %d failed after %.1fs: %sz(pet generate: draft %d produced no imagez$the image provider returned no imagez%pet generate: draft %d ready in %.1fsr)time monotonicr BASE_VARIATIONSlenbuild_base_promptr generater3r4warningrinfor;) indext0rKpromptr9r: cancelledconceptrefsspriter>s r&_onez"generate_base_drafts.._oness ;$$ $ ^^ ++EC8O8O4P,PQ **7%9U )##Fa$QW`jkC NNEu M$ FF F ;UDNNDTWYDYZ*3q62D88 ) NNJESWSaSaScfhShjm n$C( ( )s(C D*%:D%D*%D*rL max_workersu5pet generate: cancelled — dropping remaining draftszon_draft callback failed: %s)rWintreturnz#tuple[int, Path | None, str | None])r resolve_providerboolr4rVmaxmin_MAX_PARALLEL_GENERATIONSrrangesubmitrcancelresultappendr3r5sortedr _drafts_failed_reason)r[r=r>r?r@rArBr^workersresultserrorspoolifuturesfutpendingrWr6errr:draftsrZr\r]s` ` @@@r&generate_base_draftsryRs0  #tD  Q22d4jQF/I  KKH!WV[\99"!S567G!GF  0FD16q:A4;;tQ':: ( FC{ ST&%GNN$%"zz| E4|MM#&!GEN#FUD) F F.#)/ 2Qgaj 2F 2 )+3F;<< M9;&!FLL!?EEF)FF.3sOFE.)A6F! E3*F F+.F3 F<FFFFF(c`|syddlm}t||jdddS)zDThe representative reason a draft round produced nothing, humanized.z*image generation produced no usable draftsr)CounterrL) collectionsr{_humanize_image_error most_common)rqr{s r&rnrns/ ;# !z(_humanize_image_error..s i18 i)moderation_blockedz safety systemzcontent policycontent_policyuThe image provider blocked this prompt — its safety filter rejects trademarked characters and real people. Try an original description.c3&K|]}|v ywrr$rs r&rz(_humanize_image_error..s H18 Hr)zapi key unauthorized401authuYThe image provider rejected the request — check your API key in Settings → Providers.z rate limit429uDThe image provider is rate-limiting — wait a moment and try again.rN)lowerany splitlinesstrip)errorrs @r&r}r}ss ++-C ih ii S  HG HHjseslU    a & & (# ..r%)r descriptionr[r> on_progressr@rBc  !t|jstd|xstjd!|xsd} |xsd|xs|xs i} t t j} tjd| d+ !fd } t jD cgc] } | dd k7s | }} td tt |t}d}t| 5}|D cgc]} |j| | }} t|D]i}r/tjd |D]}|j!n3|j#\}}|d z }| d |d|d| |se|| |<kdddr td| j%d}|rC|d z }| d d|d| t j&|| d <tjdntj)d| j%ds$| d dt j*dg| d<| ddtjdt | t j,t j.| }t j0|}|ds!tdj3|dxsdt5|d }t7t8|z }|rtd!d"j3|t |t:kr2td#t |d$t t jd%dd&lm}| d'tjd(|jA||xs|)}tC|jD|jF|jH|d |*Scc} wcc} w#1swYExYw),aITurn an approved base image into a full, installed Hermes pet. Generates a grounded row strip per state, extracts frames, composes + validates the atlas, and registers it. The idle row falls back to the base look so the pet always renders. Raises :class:`GenerationError` on failure. *is_cancelled*, when supplied, is polled cooperatively: rows that haven't started are skipped, queued rows are cancelled, and once every row is done we abort (raising :class:`GenerationError`) before composing/saving so a stopped hatch never writes a half-built pet. zbase image not found: TrDcyrr$)_s r&rHzhatch_pet..rIr%cyrGr$r$r%r&rHzhatch_pet..rIr%z*pet hatch %r: generating %d animation rowsc |\}}} r|dfStj}d}ttD]} r|dfcS|tdz k} t j t j||d gd|d}|rdnd} tj|d|| d } tjd |tj|z |dz|| fcStjd |tj|z ||dfS#t$r.} | }tjd ||dzt| Yd} ~ d} ~ wwxYw)NrL)r>pet_row_ landscape)r=r?r@rN aspect_ratio componentsr<rF)methodfitz0pet hatch %r: row %r ready in %.1fs (attempt %d)z-pet hatch %r: row %r attempt %d/%d failed: %sz,pet hatch %r: row %r gave up after %.1fs: %s)rOrPrh_ROW_GEN_ATTEMPTSr rTr build_row_promptr extract_strip_framesr4rVr3rU)specstate_rowcountrXlast_excattemptstrictstripsrframesr:baserZlabelrr]r>s r&_gen_rowzhatch_pet.._gen_rowsp!tU ;$;  ^^ %) ./ G{d{"0144F !**,,UE5N&*V#%eW-"- *0V33F1IuVY^_ F%!1B!6! f}$3 @  : %)B. d{ C%1.? sB D E #E  Erz running-leftrLr_u3pet hatch %r: cancelled — dropping remaining rowsrow:Nzhatch cancelledrz running-left:zpet hatch %r: no running-right to mirror; left walk left emptyrz idle-fallbackF)rcomposerz,pet hatch %r: composing atlas from %d statesokz; rqzatlas validation failed filled_statesz#missing required animation row(s): z, zonly /z' animation rows were usable; regenerate)storer2zpet hatch %r: saving pet)rrr)rrrrr)rztuple[str, int, int]rbztuple[str, list | None])%ris_filer r rcrRr ROW_SPECSr4rVrerfrgrrirrjrkget mirror_framesrU single_frame compose_atlasnormalize_cellsvalidate_atlasjoinsetrm_REQUIRED_STATES_MIN_FILLED_STATES agent.petrregister_local_petrrrr)" base_imagerrrr[r>rr@rBprogressframes_by_state total_rowsrrgenerated_specsrodonerrrtrurvrrrightsheetrrmissing_requiredrpetrrZrr]s" ` ` @@@@r& hatch_petrs.  D <<> 6tf=>>  K22dKF/H/I  +| +tE')OU__%J KKc/FZ[    5 5o F GE%%e,J d dii 8(<=ZAZ[[ ?34M.>? CDIIN^D_C`abb =..C &'qU__)=(>>e f    VT KK*D1  " " !)T # C  XX%%OO/*  GV L00s11 O?O6O;O A1OO OO)r6rrbr)r[rr=rar>rr?zlist[Path] | Noner@SpriteProvider | NonerAz"Callable[[int, Path], None] | NonerBCallable[[], bool] | Nonerbz list[Path])rqrrbr)rrrbr)rz str | Pathrrrrrrr[rr>rrzProgressFn | Noner@rrBrrbr)%r" __future__rloggingrOconcurrent.futuresrr dataclassesrpathlibrtypingragent.pet.generater r r agent.pet.generate.imagegenr r getLoggerrr4r ProgressFnrgrr frozensetrrr;ryrnr}rr$r%r&rs # ?!77G   8 $sCj$& ' @A $8*.&*37.2R R R  R ( R $ R1R,RRjG/0%)&*.2bb b b  b  b b#b$b,bbr%