jdZddlZddlZddlZddlmZeejjjZ e dz dz Z ejdZ dZ dZd Zed k(reyy) a* Reune en un solo .txt los PROMPTS de todas las imagenes APROBADAS de un pool. Cada PNG de APROBADO/ tiene un prompt que se busca asi, por orden de prioridad: 1. regen_*.json -> prompt corregido a mano (la version buena) 2. manifest del lote --all -> prompt original con el que se genero Marca cuales siguen la especificacion nueva de hombre y cuales son los antiguos, para que se vea de un vistazo que prompts hay que regenerar. python Engine/dump_aprobados.py # explicit python Engine/dump_aprobados.py --pool spicy N)PathOutput test_blockszW1boy,\s*\(faceless male:1\.4\),\s*muscular body(,\s*broad shoulders)?(,\s*taller male)?)z full bodyz wide shotzwhole body visiblezboth characters fully visiblec0i}ttjdD]c} tj|j d}|D]6}|jdd}|s||vs|jds/|d||<8e|S#t $rYswxYw)z-> {nombre_png: prompt}. Busca en TODOS los manifests, no solo los del pool: algunas imagenes fueron generadas en corridas viejas cuyo manifest quedo en la raiz de Output/test_blocks/ y su pool no era subcarpeta.zmanifest_*.json utf-8-sigencodingfileprompt)sorted TEST_BLOCKSrglobjsonloads read_text Exceptionget)outmdataefs 7C:\Users\Usuario\Downloads\NEW\Engine\dump_aprobados.pyload_all_promptsrs C K%%&78 9 ::akk;k?@DAfb!AQc\aeeHo8A : J    s%B  BBc tjd}|jdd|j}t|j z dz }|j std|td|jd D}i}|jd D]R} tj|jd }|jd s;|jdd||d <Tt}g}|D]} |j| xs|j| ds|j!| ddf?t#t$j'} t)fdt*D} | rdnd} | r| dz } |j!| | f|d|j dz } d|j dt-|dd|j dddddd d!d"d#dg }|D]\} } |j!| d$|j!d|j!d%| d&|j!|j!d|j!d'|j!d| j/d(j1|d) t3d*|D}t3d+|D}t3d,|D}t5d-t-|d.| j7t8t5d/|t5d0|t5d1|t5|D]\} }} t5d2| d3d4| d&y#t$rYwxYw)5NzVolcado de prompts APROBADOS) descriptionz--poolexplicit)defaultAPROBADOz[!] No existe c34K|]}|jywN)name).0rs r zmain..8s5"4Q!&&"4sz*.pngz regen_*.jsonrrr r r z SIN PROMPTc3&K|]}|v ywr")r$tr s rr%zmain..Ms59aa6k9sNUEVOantiguoz +tags camaraprompts_aprobados_z.txtz# PROMPTS APROBADOS de z (z imagenes)z# en Output/test_blocks/z /APROBADO/z*# formato: :\n\n\n#zB# 'NUEVO' = Corresponde con la especificacion actual del hombrezK# (1boy, (faceless male:1.4), muscular body, [broad shoulders],z<# [taller male]) y sin tags de cuerpo completo.zI# 'antiguo' = Especificacion larga de hombre + tags de camara. La imagenz@# fue aprobada asi, pero si la quieres con la nuevaz2# especificacion hay que regenerarla.:[]zN------------------------------------------------------------------------------ zutf-8c3VK|]!\}}}|jdsd|vsd#yw)r)camaraN startswithr$_r(s rr%zmain..ks*V$wq!Q!,,w*?HTUDU$s )))c3LK|]\}}}|jdsdyw)r*r3Nr4r6s rr%zmain..ls"B$wq!Q!,,y*A$s$$c32K|]\}}}d|vs dyw)r2r3Nr'r6s rr%zmain..ms 5tGAq!x1}ats z[OK] z prompts -> z& NUEVO (spec actual, sin camara) : z' antiguo (spec larga + camara) : z& con tags de camara : z 48z [)argparseArgumentParser add_argument parse_argsrpoolis_dir SystemExitr globrrrrrrappendboolNEW_MALEsearchany FULL_BODYlen write_textjoinsumprint relative_toBASE_DIR)pargsapprpngsregenjdbatchrowspngnew_spechas_camtagrlinesnuevosviejoscamr7r s @rmainra/s+,JKANN8ZN0 <<>D  "Z /D ;;=>$011 5$))G"45 5D E YY~ &  1;; ;<=A 55= uuXr2E!F)  '  E D35599S"#5 KKb,/ 0 /05955!gy  ? "C S&#&' %dii[5 5C !$))Bs4ykD "499+Z87 LUFSJ<  E!VS uAY R qQZ  V R X R!NN499U#gN6 V$V VF B$B BF 5t 5 5C E#d)L)B(C DE 26( ;< 3F8 <= 23% 89 G Q SHBse1%&i   s'%M MM__main__)__doc__r;rrepathlibr__file__resolveparentrOrcompilerErHrra__name__r'rrls   > ! ! # * * 1 1!M1  2::0 1 . "E'P zFrk