j?:dZddlZddlZddlZddlZddlZddlmZddlmZejjddee jjjZejj!deedz ejj!deed z ddlZddlZdd lmZed z d z d z ZdZddddZdhZddefdZddefdZddefdZ ddefdZ!defdZ"ddZ#dedzfdZ$dddZ%dZ&e'dk(re&yy) aC batch_poses_chars: batch de personajes distintos x poses distintas x entornos distintos, con la estructura del piloto "Diferent poses". Estructura (identica al piloto de referencia): - 3 imagenes por personaje: 1 sugerente + 1 spicy + 1 explicito - camaras en el mismo orden: from above / from front / from side - un tema (entorno) distinto por personaje - salida consolidada por personaje: //img/*.png + manifest Regla dura: NINGUNA pose se repite en todo el lote. Las poses se reparten a mano sin reemplazo desde el banco y se fijan con `bank_ids`; no se deja al azar de `compose_theme_jobs` porque su contador de reuso es por personaje, no global. La semilla es aleatoria por imagen (`-1`): no se fija ninguna. python Engine/batch_poses_chars.py --chars 12 --dry-run python Engine/batch_poses_chars.py --chars 12 python Engine/batch_poses_chars.py --chars 12 --exclude-used --pick-seed 20261028 python Engine/batch_poses_chars.py --keys Makima_v2,Furina --themes oficina,playa Salida: Output/pilots/Diferent poses/BATCH_//img/*.png N)datetime)Pathutf-8backslashreplace)encodingerrorsEngineArchivos) PromptBuilderOutputpilotszDiferent poses) sugerentespicy explicito onsen_xilonenreturnc X|xsgDchc]#}|js|j%}}i}tD]b}|Dcgc]Q}|jd|k(r;|jd|vr(tj|jdd|s|Sc}||<d|Scc}wcc}w)zH{nivel: [entradas]} sin las blacklisteadas (avoid.jsonl) ni las vetadas.nivelid pose_line)stripLEVELSget pose_bank is_avoided)bankavoidskip_idsxskipoutlves 9C:\Users\Usuario\Downloads\SD\Engine\batch_poses_chars.py usable_posesr'1s (B =!1779AGGID = C"PdeeGn*uuT{$.$//k20FNdPB J >PsB"B"AB'c|r t|ntdz }t}|jdD]y} t j |j d}|jdxsgD]9}t|ts|jds&|j|d;{|jdD]y} t j |j d}t|tr|g}|D]9}t|ts|jds&|j|d;{|S#ttjf$rY'wxYw#ttjf$rYwxYw)zClaves ya usadas en cualquier lote/pack de `Output/`. Mira dos cosas, porque no todos los runs dejan el mismo artefacto: - `plan.json` (batches: `plan[].char`) - `pack_manifest.json` (packs: `char` en cada record) r plan.jsonrrplancharpack_manifest.json) rBASE_DIRsetrglobjsonloads read_textOSErrorJSONDecodeErrorr isinstancedictadd)rootusedpdocitemrecsrecs r&used_charactersr@=sH4:8h#6D 5D ZZ $ **Q[['[:;CWWV_**D$%$((6*:f&+ %ZZ, - ::akk7k; [(clave, linea)]z[!] personaje 'z' no esta en personajes.txtbaser) lora_stemsrz]+):rz"[-] personajes con LoRA en disco:  (z- descartados por LoRA ausente o sin , z ya usados) [!] pedias z personajes y solo hay z con LoRAN) get_character SystemExitappendcharacter_keysrCr/get_character_namesresearchgrouplowerprintlenshuffle)Bcountrngkeysexcluder#kcrC availablecandsdroppedskippedms r&pick_charactersr^[sz A"A ?1#5P!QRR JJ1V9~ &   ) IG EGG  " " $ OOA   < qLG  II(!F)/r : QWWQZ%5%5%7y%H qLG  a6^$% .s5zl;YC I[ "#KK s5z E7"9#e*YOP %=ctj}|r6|D]%}||vstd|dtjt |}n)|Dcgc] }|t vs |}}|j ||t|kDrtd|dt|dt|Dcgc]}||t|zc}Scc}wcc}w)zAUn tema distinto por personaje, sin temas ligados a un personaje.z [!] tema 'z' no existe en rEz temas y solo hay z disponibles) pb load_themesrG THEMES_FILElistTHEMES_BLOCKEDrQrPrOrange)rSrTwantedthemestpoolis r& pick_themesrls ^^ F A :aS?O!PQQF|!=6aQn%<6= D s4y E7"4SYK|LM).u 6ADSY  66 > 7s CC,C c tfdtD}t||kDrHtd|dtddtddtdd | |d |}|d |}g}t |D]:\}\}}|j ||||tD cic] } | | | c} d <|Scc} w) zReparte poses sin reemplazo: 1 por nivel y personaje. Devuelve [{char, theme, poses: {nivel: entrada}}]. El cuello de botella es `sugerente` (16 usables), asi que el tamano maximo del lote lo marca ese nivel. c3:K|]}t|ywN)rP).0r$poses_by_levels r& zdeal..s9&BN2&'&sz[!] el banco solo permite z( personajes sin repetir pose (sugerente=rz, spicy=rz , explicito=rz ); recorto a N)r, char_linethemeposes)minrrPrO enumeraterH) charsrhrqrTlimitr+rkkeyliner$s ` r&dealr|s 9&9 9E 5zE *5'2{ ;<=>>'2345~k:;>DAq K$  "t +D IIedQhq17 ;E6"2 B!d(a!t)q9:# JJxJ$#    s#CAC. C+*C+. C:9C:c`&tjd}|jdtdd|jddd |jd d d |jddd |jddd |jddd |jdtdd|jdtdd|jddd |jdtdd|jdd d |jdd d |j }t j |j}tj}tj}|jr7|jjd Dcgc]}|jc}nd}t|||}tD]4}t!d!|d"t#||d#|j%||6t't)t*} |j,rH|j,jd Dcgc]#}|js|j%c}nd} |j.rt1|j2nd} | r)| r't5t7| | z} | rt9d$| t;| |j<|| | } | s t9d%t?t#| ||j@rH|j@jd Dcgc]#}|js|j%c}nd}tC| |||}tE|tGjHjKd&}|jLrtO|jL}nS|jPr;t5tRjUd'd()}|st9d*tR|d+}n tRd,|z }t!d-t#|d.t#|d/zd0t!d1||D];&t!d2&d3d4d5&d6d7d8d9jW&fd:tDz=|jXrdt[j\}tDcic]}|t[j^|}}|D]"&tDcgc] }&d;|d<}}t[j`&d6gd/tctdd=id>d?|||||d@|A}t!dB&d3dC&d6dDtg|d@D]\}}t[jh| jj&dE|}t!dF|dG|jldH|jndI|jpdJ|jrdK|jtjwdL t!d2|jydMd9dNdz%yt[jzst9dOtZj|dPt[j\}tDcic]}|t[j^|}}|j~ |j~n dQt#|z}ggg}}}tg|d@D]g\}&&d3}tDcgc] }&d;|d<}}t!dR|dSt#|dG|dT&d6d9 dUjW|zt[j`&d6gd/tctdd=id>d?|||||d@|A}|Dcgc]}|jn} }| |k7rt!dV|dW| ||z }!|!dXz }"|jPr t|!|nd}#|#t!dYt#|#dZ|!nt[j|&dE| jj| j|"|j||[\}#}$|!jd\d\]|!d^z jtj|#d_d@`dab|!dcz jtj|$d_d@`dab|#D]M}%|jdd|i|%de|dS|%dei|j|"tO|%dejz O|j|&d6tDcic]}|&d;|d<&d;|dfdgc}|#D%cgc]}%|%dh|%di|%dj|%dk|%dl|%dedmc}%dnj|jd\d\]|doz jtj|d_d@`dab|dpz jtjtGjHjKdqt#||j|j|drd_d@`dabt||dsz t!dtt#|dSt#|d/zdu|t!dv|doz t!dw|dpz ycc}wcc}wcc}wcc}wcc}wcc}wcc}wcc}wcc}wcc}%w)xNzOBatch de personajes x poses x entornos (estructura del piloto 'Diferent poses')) descriptionz--chars z[cuantos personajes (1 por tema y 3 imagenes); maximo real = poses usables de sugerente (16))typedefaulthelpz--keyszBpersonajes concretos separados por coma (claves de personajes.txt))rrz--exclude-used store_truez@no repite personajes ya usados en cualquier lote/pack de Output/)actionrz --used-rootz8carpeta donde buscar los lotes previos (default Output/)z--themesz@temas concretos separados por coma (default: al azar, distintos)z--exclude-poseszCids de pose a vetar en este lote, separados por coma (p.ej. pb0009)z--seedzrsemilla FIJA para todas las imagenes (default: aleatoria por imagen, como -1); fijala solo para reproducir un lotez --pick-seedzjsemilla del reparto (personajes/temas/poses), para que el plan sea reproducible; no afecta a la generacionz--outzcarpeta de salidaz--budgetz7maximo de intentos de generacion (default 3 por imagen)z --dry-runz0imprime el plan y los prompts sin llamar a Forgez--resumezkno regenera los personajes que ya tienen manifest completo en la carpeta de salida (lo creado, creado esta),z [-] poses z: z usablesz([!] --exclude-used y ya estaban usados: z![!] no hay personajes disponiblesz %Y%m%d_%H%M%SzBATCH_*c|jSro)name)r;s r&zmain..(saffr_)rzz+[!] --resume sin --out y no hay BATCH_* en BATCH_z[-] z personajes x 3 niveles = z imagenesz [-] salida: z r,z<28z tema rtz<12 z | c3>K|]}|dd|dyw):rurN)rpr$r=s r&rrzmain..3s3-%+r!T4=#4T#:";<%+srurz per-imageautogr) max_variantsbank_idsz === rDz) ===rs[z] z pose:z cam:z dress:z outfit:outfit iz[!] Forge NO responde en z. Arranca webui-user.bat --api z [r~z | tema z, z[!] aviso: el plan pedia z y salieron imgz [skip] ya generado (z imagenes) -> )r,T)parentsexist_okr-F) ensure_asciiindentrr*z metrics.jsonrrr)rr{rrcameradressseed)rrrrrr)r,rtrujobszmanifest_all.jsonr)z%Y-%m-%d %H:%M:%S) generatedrxr pick_seedr+zcontact_all.jpgz [OK] z imagenes -> z manifest: z plan: )KargparseArgumentParser add_argumentint parse_argsrandomRandomrr load_bank load_avoid exclude_posessplitrr'rrOrPrQr strr.rU exclude_usedr@ used_rootrr/rGr^rxrlrhr|rrnowstrftimer#rresumeOUT_ROOTglobjoindry_runra load_camerasload_pool_linescompose_theme_jobsr7QUOTASrwbuild_run_prompt master_promptrrrr variant_keysrreplaceforge_availableFORGEbudgetrgeneratenegative_promptrmkdir write_textr1dumpsrHrr)'r;argsrTrrr!r"rqr$rRrUr:clashrxrhr+stamprun_dirdonecameraspoolsrrjjobpromptr all_recordsplan_outimagesnrzgotrimg_dirrmetricsr?r=s' @r&mainrsf / 0ANN93HINN8T\^NN#LZ\NN=$RTNN:tZ\NN$d)*NN8#tTUNN=sDKLNN7D/BNCNN:CQSNN;|JLNN:lTU <<>D -- 'C    D  "E!!!% 2 2 8 8 = > =1QWWY = >'+ !$t4N 2$b^B%7!8 9BC N2&' c(m$Ayy!%  4 B 41 QWWY 4 B" .2.?.??4>> *TD s4y4'( GwOP P Atzz3d ;E <== US KK.2[[->->s-CQ-Cqwwy!'')-CQ-13F v~s 3DD LLN # #O 4E xxtxx. hmmI.4DEJ8*UV Vr(veW-- DT 5c$i!m_I NO L "# T&\#&fT']3,?qA -%+--- .  ||//#6<=fR''++f=D5;OQTU!Bsyyk }E#**N"yyk#2B2B2F2Fx2P1QSTfv~~dE:45AAB -    4RXXJ?889 9ooG28 9&BR##B' '&E 9 KK3T[[T9JF$&B6KT1%46l1782tG}R &8 A3aD {"SE$w-D # $$ ']OQV k2 CueWcS* #''$Qqyy$' #: -cU,seD ES=U".2kk)Hc*t   ,S\N. S T!{{4k1B+,??A @ A MM'DV$5$:$:: ; ']!'( &"gr!24!8#'=#4[#ACC &( !()!( #7|I #H G !&k3v;@!()  K&b MM$M. "".. ;U1=/Q {&& !8!89L!M Y !%C!&a 1<$ : 9(<()sHg>hhh 3h h  h2hhh!!h&"h+__main__ro)NN)rN)r)(__doc__rr1rrKsysrpathlibrstdout reconfigure__file__resolveparentr.pathinsertrr pack_builderraprompt_builderr rrrrer7r'r/r@rdr^rlr|rrrr__name__rr_r&rs9.  0BC > ! ! # * * 1 13x(*+,3x*,-.( h  ),< < ,11 5!"  #<#t#L7D7"6Gt .%0i5X zFr_