bj< dZddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z e e jjjZejj!dee e jjddlmZedz dz dz Zedz dz d z Zedz d z d z Zed z d z ZdZdZedz dz Zedz dz Zejj!dee e jjddlmZeeedz edz edz edz edz edz edz edz edz edz d ZdZgdZ dZ!d)dZ"efdZ#dZ$dZ%d Z&d!Z'd"Z(d#Z)d$Z*d*d%Z+ d+d&Z,d'Z-e.d(k(re-yy),a Test de BLOQUES: mette un bloque curado de wd14_blocks.py tras un personaje del catalogo y comprueba si Stable Diffusion respeta la pose. La diferencia con test_harvest_tags.py: aqui NO se recombinan pose/outfit/scene por separado. Se coge el bloque entero tal cual (una linea de bloques.txt) y se pega DETRAS de la linea del personaje. Asi se mide lo que el bloque pide, no lo que nosotros reconstruimos. # 1. Ver los prompts sin gastar GPU python Engine/test_blocks.py --dry-run # 2. Un bloque concreto, 2 variaciones python Engine/test_blocks.py --char Seed --block 0 --variations 2 # 3. Barrido: 6 bloques sobre varios personajes python Engine/test_blocks.py --n 6 # 4. A/B focalizado de ropa: mismo indice en ambos pools python Engine/test_blocks.py --block 5 --pool desnudo python Engine/test_blocks.py --block 5 --pool vestido Salida en Output/test_blocks/ + manifest_*.json con el prompt exacto de cada imagen, y una copia del .txt original para comparar lado a lado. N)datetime)Path) PromptBuilderArchivoswildcards_harvestz bloques.txtzbloques_vestidos.txtwd14tagsOutput test_blockszhttp://127.0.0.1:7860ct|z SN)OUT_ROOT)pools Engine/test_blocks.pyout_dirr0s d?wildcards_danbooru)strip_weights_ofzbloques_sfw.txtzbloques_spicy.txtzbloques_explicit.txtzbloques_sfw_vestidos.txtzbloques_spicy_vestidos.txtzbloques_explicit_vestidos.txt) desnudovestidosfwspicyexplicit sfw_vestido spicy_vestidoexplicit_vestidodan_sfw dan_spicy dan_explicit dan_todosz1boy, (faceless male:1.4), no face, blank face, featureless face, no facial features, eyes closed, black hair, short hair, muscular body, broad shoulders, large hands, masculine physique, nude male, larger male, taller dominant),Hoseki_ZenlessZoneZero_Seed_IllustriousXL_v1 ArlecchinoYelanremReZero_Rem_IlluXLznilou-illu-nvwls-v1Furinaz,ai-hoshino-s1-illustriousxl-lora-nochekaiser Raiden_shogunhu_tao_genshin_impactSkirkkugisaki_nobarasanaeFischlsakamata_chloefirefly_honkai_star_rail robin_hsrgold_ship_illust_scarxzyszM(full body:1.2), wide shot, whole body visible, both characters fully visiblec\tj|t}|jst d|dg}|j dj D]8}|j}|s|jdr(|j|:|st dtd|S)Nz[!] No existe z. Ejecuta antes wd14_blocks.pyutf-8encoding#[!] z no tiene bloques) POOLSgetBLOCKSexists SystemExit read_text splitlinesstrip startswithappend)rpathoutlines r load_blocksrDes 99T6 "D ;;=>$/MNOO C0;;=zz| , JJt > 4x'89:: Jrc ^ddlm}m}|\}}}t|}t |j dD]}|j djdD cgc]#} | js| j%} } || D cgc]0} |j| j| j2c} ||\} } tg}} | D])} | | vs| j| |j| +dj||k(s|cSycc} wcc} w) zLocaliza el .txt de Archivos/wd14/tags del que salio este bloque. Compara SIN pesos: bloques.txt lleva '(ass_focus:0.7)' pero el .txt original no, asi que hay que normalizar antes de comparar. r)load_character_tags clean_blockz*.txtr2r3,, N) wd14_blocksrFrG strip_weightssortedrglobr<splitr>r8lowersetaddr@join)blocktags_dirrFrGcharscopiesaliaswantftrawkeep_seenuniqs r find_originalr`ss =.0E65  D HNN7+ ,"#++w+"?"E"Ec"JX"JQaggiqwwy"JXL1uyyAGGI>LeU[\aUBdA}  A 99T?d "H- YLsD%5D%5D* c0tjdd|S)zH'(ass_focus:0.7), nude' -> 'ass_focus, nude'. Para comparar con el .txt.z\(([^:()]+):[\d.]+\)z\1)resub)rSs rrKrKs 66)5% 88rc td}|jsy|jdDchc]}|jj }}|j |vry|j dddj |vScc}w)Nz?D:\Data\Packages\Stable Diffusion WebUI Forge - Neo\models\Loraz *.safetensorsT-r)rr:rMstemrOrsplit)namedpstemss r lora_on_diskrms| OPA 88:%&WW_%= >%=QVV\\^%=E > zz|u ;;sA q ! ' ' )U 22 ?s!B cVddlmtfd|jdDS)zFTrue si el bloque trae alguna prenda (mismo criterio que wd14_blocks).r) DROP_OUTFITc3\K|]#}j|j%ywr )searchrO).0rZros r z%block_has_clothing..s%G6F{!!!''),6Fs),rH)rJroanyrN)rSros @rblock_has_clothingrus ' Gekk#6FG GGrzF1boy, (faceless male:1.4), muscular body, broad shoulders, taller malez1boy, muscular body)1girl1boyznude, completely_nude, nakedct|}d|jdDchc] }|jj"c}v}d|vxsd|v}|rKt j dd|}t j dd|jjd}d|g} |r| j d |d k(s|d k(r t|s| j t|d j| g} |r| j |rtnt| j |t} g} | D]} | jdDcgc]#}|js|j%}}g}|D]A}|j}|tvr|| vr | j||j |C| j d j|d jd| D| fScc}wcc}w)z-> (prompt_con_BREAK, [tramos])povrHrwheteroz\bsolo\bz,\s*,rvzage up, mature femalealwaysautorIz BREAK c3BK|]}|js|ywr )r>)rrxs rrsz"assemble_prompt..s:A as)rrNr>rOrbrcr@ru NUDE_TAGSrRMALE_POV MALE_FULLrP COUNT_TAGSrQ)master char_linerS force_nudeage_up block_cleanrZis_pov two_peoplegirlsegsr^rBstokskeptkeys rassemble_promptrs"5)K +2C2CC2HI2HQqwwy(2HI IFK'B8{+BJ R0#v.446<>s%G6G;G;c,t|||||\}}|S)zCompatibilidad: usa el ensamblador nuevo. cam_brake ya no se aplica (los tags de cuerpo completo se descartaron por decision del usuario).)rr)r)rrrS cam_brakerrposr]s r build_promptrs#VY(26CFC Jrc tjd}|jdtdd|jdd|jd td d |jd d d|jdtdd|jdtddd|jdtdd|jdt t dd|jdtd |jd!t d |jd"d d#|jd$d d%|jd&gd'd(d)|jd*d +|j}t|j}ttt}ddl } |jtd,d-.}|j!jd/d0}t#d1|t&Dcgc]$}|j)|st+|d4us#|&} }| s%t&Dcgc]}|j)|s|} }t#d5t-|d6|jd7t-| t&D]0}|j)|st+|d4us#t#d8|2|j.rvt1t-|D cgc]} | |j2k\s| } } |j4r| D cgc]} | |j4vs| } } t#d9t-| d:| n|j6md|j6cxkrt-|ks*nt9d;|j6dtOjPjSd?}g}g}tU| | D]3\}}t1|j<D]}|jW|||f5tY|dD]\} \}}}|j)|}|st#d@|dA.|dB}|jZ%t]j^dCdD|jZdE|}||}ta|jb|||jd |jf|jhF}|jj}|j}||jdrdGndHz }| dIdJ|dIdK|ddLdK|dM|dzdN }tm|}|jW||||j||r t|nd|dOt#dP| dQt-|dR|t#dS|ddTdUt#dV||jZrt#dW|jZ|rt#dX|jndY|jprt#dZ|dd[dU||d\d]d^d_d`dd|jrdadb } tujt}|jwtdc|dde}|jy|j!} ||z j{t}j~| dfd t!j| jdgdhjdid0}!t#djtujt|z dkdl|!|dn|doz jt!j|dpdqdrst#dtt-|du|t#dv|dwy#t$$r}t#d2|d3Yd}~d}~wwxYwcc}wcc}wcc} wcc} wcc} w#t$$rd0}!YwxYw#t$$r}t#dm|Yd}~zd}~wwxYw)xNz%Test de bloques WD14 sobre personajes) descriptionz--blockzindice de bloque (0-based))typedefaulthelpz--charz,personaje concreto (si no, se elige al azar))rz--nz cuantos bloques probar (al azar)z--all store_truezOrecorre TODOS los bloques del pool en orden 0..N-1, para auditarlos uno por uno)actionrz--skiprz2con --all, empieza por este indice (para reanudar)z--onlyr@zHcon --all, procesa solo estos indices (repetible). Ej: --only 3 --only 7)rrrrz --variationsrfzseeds por bloquez--poolrzzdesnudo = bloques.txt (sin ropa) | vestido = bloques_vestidos.txt (ropa del WD14 sin color). Ambos pools comparten indice.)choicesrrz--seed)rrz --lora-weightz--no-cam-brakez4quita el freno de camara (A/B: ver si deforma menos)z--age-upusañade 'age up, mature female'. FUERA por defecto: fuerza rasgos maduros que chocan con personajes de busto pequenoz --force-nude)r}r|neverr}zauto (defecto) = fuerza nude solo si el bloque no trae ropa (para LoRA con ropa horneada, p.ej. Yelan). always = siempre. never = nunca.z --dry-run)rz/sdapi/v1/options)timeoutsd_model_checkpoint?z[-] Forge OK | checkpoint: z[!] Forge NO responde (z). Arranca webui-user.bat --apiTz[-] z bloques (pool: z) | personajes usables: z AVISO sin LoRA en disco -> z[-] modo --all: z bloques -> z [!] bloque z fuera de rango (0-))parentsexist_okz %Y%m%d_%H%M%Sr6z$ no esta en personajes.txt, saltandobasez(]+):[^>]*>z\g<1>:>)rrr_sincamr{02d_br]_vz.png)file block_idxcharrrS original_txtpromptz [/z] z bloque : dz...z char : z lora w : z original: z <- comparar con estez prompt : i@ig@zDPM++ 2M Karras) rnegative_promptwidthheightsteps cfg_scale sampler_name batch_sizen_iterseed clip_skipz/sdapi/v1/txt2imgi,)jsonrimagesinfoz{}rz [OK] z.0fzs seed=z [FAIL] manifest_z.jsonF) ensure_asciiindentr2r3z [OK] z en z[OK] manifest_z..json (bloque + prompt exacto + .txt original))CargparseArgumentParser add_argumentintlistr7float parse_argsrDrrstrBASE_DIRrequestsr8FORGErprint ExceptionCHARS get_characterrmlenallrangeskiponlyrSr;max variationsrandomsampleminnrchoicermkdirrnowstrftimezipr@ enumerate lora_weightrbrcr master_prompt no_cam_brakerrrr`ridry_runrtimepostraise_for_status write_bytesbase64 b64decodeloads write_textdumps)"rkargsblocksBrrckptecusableiidxsr]rUOUTstampmanifestjobsbickvrrrSrnegtagrisrcpayloadt0respdatasds" rmainrs,STANN93;WNXNN8"PNQNN5sA4VNWNN7<67NN8#qLNNN8#h01NN>Q=ONPNN8T%[)STNN8#rN2NN?N=NN#LNPNN:lTUNN>+FPV;<NN;|N4 <<>D  #Fc(m$AL LLE7"34bL Avvx||137 +D623 QA!//!"4aD9PaF Q "9Uaooa&8!U9 DV -dii[8PQTU[Q\P] ^_  ??1 ,q/"= 3A37 8  xx!V-@-adii-@ 99#6t!qDII~AtD6  T <v>?  DJJ,V,{4::,6I#f+VW-XYZ[ [ |c$//155}}U3v;/TVVS[1IJ'+yyTYYK#d) #(,-1v -  $)) CIIdTI* LLN # #O 4EH DdE"Bt'A KKR $(#$D!,;BAr" D@A B L    '7"()9)9(:!E"r2tyyCH  A3aD {"TF+, uTc{m3/0 rd#$    N4#3#3"45 6  N388*,BC D << N3t9+S1 2 cD2C-Q!IIA   %B==E7*;!<7TW=XD  ! ! #99;D 4Z $ $V%5%5d8nQ6G%H I ZZ 67;;FCH Idiik"nS1= >k-r YugU ##// 8%:W0N GCM?$se ,- N5'!O PQE L 's*IJKKLR9A6 .|   % Ks# $ $ %s"A_2` ``'` >` `$`:```)A<`/&5`'`/ `(_==` `,)`/+`,,`// a8a  a__main__)r)r}F)TFr})/__doc__rrrrrbsysrrpathlibr__file__resolveparentrrAinsertrprompt_builderrr9BLOCKS_DRESSEDTAGS_DIRrrrHARVEST HARVEST_DANrJrr7 MALE_TAGSr CAMERA_BRAKErDr`rKrmrurrrrrrr__name__rrr!s2  > ! ! # * * 1 13tH~--/6678( J !4 4} DJ&)<