jM ddlZddlZddlZddlZddlZddlmZej jdej jej je ddl m Z m Z ddlmZddlmZ ddlmZmZej jej jej je Zej j/edZej j/edZej j/ed Zej6ed ej6ed ej8ej:d ej<ej j/ed dej>gej@e!Z"dZ#dZ$dZ%d dZ&dZ'dZ(dZ)dZ*dZ+d!dZ,dZ-d!dZ.dZ/dZ0e!dk(re0yy#e$rdZdZY~wxYw)"N)datetime) ForgeClient VisionClient) PromptBuilder) VisionCritic)PromptDesignerLiterouterDesignerConfigOutputLogsTexist_okz'%(asctime)s [%(levelname)s] %(message)sz engine.logutf-8encoding)levelformathandlersctjjtd}t |dd5}t j |cdddS#1swYyxYw)Nzplumloom_engine.yamlrrr)ospathjoin CONFIG_DIRopenyaml safe_load) config_pathfs 8C:\Users\Usuario\Downloads\NEW\Engine\plumloom_engine.py load_configr!'s='',,z+ABK k3 1Q~~a  2 1 1s AAc J|dd}t|d|d}|dd}t|d|d|d}|jd i}|jd d }t tt j d ||dfS|d k(r/|dd }tjjtd d}g} tjj|r}t|dd5} | D]`} | j} | jds%d| vs*| jdddj} | sP| j!| b ddd| s!tj"jddg} t | |jddg|jdd|jdd} nZ|dd}tj"jdd}t|d||jdd |jdd!} ||| fS#1swYxYw)"N endpoints forge_neourltimeout)r%r&visionmodel)r%r(r&prompt_designerprimary literouterz:prompt_designer no disponible: menu IA (7/8/9) desactivadoCONTEXTOzapis literouter.txtrrrkey=LITEROUTER_API_KEYmodelszdeepseek-v3.2:free temperaturegffffff?)api_keysr2r&r4 omnirouteOMNIROUTE_API_KEYz#sk-9e3027503ec29237-25cded-bd502dd0zcohere/command-r-plus-08-2024) omniroute_url omniroute_keyr(r4)rrgetrr loggerwarningrrrBASE_DIRexistsrstrip startswithsplitappendenviron)config forge_cfgforge vision_cfgr' designer_cfgr*lit_cfg api_keys_filer5rliner-designeromni_cfgapi_keys r init_clientsrO-s{#K0I Ie,i 6J KE $X.J  u !9%F ::/4Ly,7G"4"<STfd"",%l3 Xz;PQ  77>>- (mS7;qD::)?@KK 2.$((<  +&{3**..!46[\!"5/!""7,KL$((<   &( ""7<;s''HH%H:HH"cFtdtdtdy)Nz8 =======================================================z/ PLUMLOOM ENGINE v2 - PANEL DE GENERACION7=======================================================printr print_headerrVbs / ;< (OrUctdtdtdtdtdtdtdtdtd | td td td y) Nz' [1] Generar lote sugerente (X/Twitter)u%[2] Generar lote explícito (Patreon)z[3] Generar imagen individualz[4] Ver personajes disponiblesz[5] Ver wildcards disponiblesu![6] Verificar conexión Forge Neou%[7] Verificar conexión Diseñador IAz[8] Generar prompts con IAz*[9] Generar lote con IA (prompts + imagen)z< [7][8][9] desactivados: modulo de diseno IA no instaladoz [0] Salirz7-------------------------------------------------------rR)rLs r print_menurXhsq 45 12 )* *+ )* -. 12 &' 67 LM + (OrUc |j}tdt|dt|dD]A\}}|j |}|j drdnd}td|dd |d |d Cy) Nz --- Personajes disponibles () ---r/outfitu✓× 3d. z [outfit: ])get_character_namesrSlen enumerate get_characterr:)buildernamesinamechar has_outfits r show_charactersrkxs|  ' ' )E *3u:,e <=UA&4$$T*"hhx0Ud  1R&4& :,a89'rUc dD]e}|j|}td|jdt|dt |dD]\}}td|dd|dd gy) N)posescenedetailsr[z ---  (z opciones) ---r/r]2dr_F)get_wildcard_optionsrSupperrbrc)rewc_typeoptionsrgopts r show_wildcardsrxsp9..w7 w}}'r#g,~FG+FAs BqfBs3Bxj) *,:rUctd|jr=td |j}|jdd}td|ytdy#t$rYywxYw)Nu' Verificando conexión con Forge Neo...u&✓ Forge Neo CONECTADO y respondiendosd_model_checkpoint desconocidoz Modelo activo: u?✗ Forge Neo NO responde. Verifica que esté abierto con --api)rS check_status get_optionsr: Exception)rFoptsr(s r check_forgerss 45  67 $$&DHH2MBE %eW- . OP   s0A%% A10A1c| tdyt|trdnd}td|d|j\}}|rtd|dtd|ytd |d |y) N; ✗ Diseno IA no disponible (DeepSeek retirado del stack).Literouter (DeepSeek)Omniroute (Command R+)u Verificando conexión con ...u✓ z CONECTADO y respondiendoz Respuesta: u✗ z NO responde: )rS isinstancer check_connection)rL designer_typeokmsgs r check_command_rrs LM/9(DV/W+]uM (s ;<'')GB  ]O#<=> cU#$ ]O>#78rUc | tdyt|trdnd}td|d|j}tdt |t dj }|s tdy|j|}|std |d ytd |d dd dt dj j}|dvrd}|dk(rdnd}t dj xsd}t dj xsd} td |j||d ||| \} } tddtdtdt| tdtd| tdt | dt dj j} | dk7r| S y#t$r} td | Yd} ~ yd} ~ wwxYw)!Nr Literouter Command R+u$ --- Generación de Prompts con IA (rZ Personajes disponibles: z Nombre del personaje: Debes especificar un personaje Personaje '' no encontradoz Tags base: basePrz6Nivel de contenido (sugerente/explicito) [sugerente]: ) sugerente explicitor suggestiveexplicit%Idea de outfit (vacío = IA decide): (Idea de escenario (vacío = IA decide): z$ Generando prompts con Command R+...)character_namecharacter_tags spice_level outfit_hint scene_hint rQzPROMPT GENERADO POR IA:zTokens usados: z Longitud: z caracteresu4 ¿Usar este prompt para generar imagen? (s/n) [s]: nzError generando prompt: ) rSrr rarbinputr?rdlower design_promptr~)rLrerrf personajerispicerrr prompt_texttokensusares r run_ai_prompt_generationrs LM$.x9K$LLR^M 1- FG  ' ' )E &s5zl 3401779I  ./   +D  I;o67 M$v,s+,C 01 J K Q Q S Y Y [E .."';"6,JK?@FFHPDKABHHJRdJ 12&44$<##! 5  V 8*o '(   k   x() 3{+,K89LMSSU[[] 3;     (,-s4B+G"" H+G>>Hc  | tdy|d}t|trdnd}td|d|d|j}tdt |t d j } | r#|j| } | sDtd | d ytj|} |j| } td | t d j } t| } t dj xsd} t dj xsd}td| d|d |j| | d|| | |\}}tdt |d|dtddtdtdt|dD]\}}td|d|dd dtddt d!j j}|d"k(r td#yt!||d$d%|d$d&'}|d(k(rd)}d*}nd+}d,}t"j$j't(|}t#j*|d-.d/}d/}t-j.j1d0}t|D]\}}|| k\rn|dz }td1|dzd2t |d3|d2|  |j2d4| dd5|d4|}|j4}td6|dd7d|j7|||d8|d9|d:|d;|d<|d=>\}} |s td?|d/}!| j9d@tj:ddA}"tdB|j=|!\}#}$}%|#rX|dz }|dC|dDdE|"dF}&t"j$j'||&}'|j?|!|'tdG|d2| dH|&nKt"j$j't(dI|dJ|dF}(|j?|!|(tdK|$tddtdOtdP|d2| tdQ|tddy#t$rd} YwxYw#t$r}td|Yd}~yd}~wwxYw#t$r7}t@jCdL|dzdM|tdN|Yd}~Wd}~wwxYw)RNr generationrrz --- Lote con IA (z): z ---ru, Nombre del personaje (vacío = aleatorio): rrzPersonaje aleatorio: uCantidad de imágenes [5]: rrz Generando z prompts con rr)rrrcountrrzPrompts generados: rpz tokens)zError generando prompts: rrQzPROMPTS GENERADOS:r/z [z] du3 ¿Generar imágenes con estos prompts? (s/n) [s]: r Canceladocritic max_retriesauto_approve_on_error vision_clientrrr campaigns_Xsuggestive, cleavage patreon_NSFW2undressing, bare shoulders, highly suggestive poseTr r %Y%m%d_%H%M%Sz [Imagen / ] Progreso: z, z, age up, mature female, Prompt: rwidthheightsteps cfg_scale sampler_name clip_skippromptnegative_promptrrrrrru Forge no devolvió imagenseedɚ; Enviando a crítico visual..._ai_03d_seed.png ✓ APROBADA [] → rejected_ai_rej_ ✗ RECHAZADA: zError en imagen : ✗ Error: z LOTE IA COMPLETADO Aprobadas: Carpeta: )"rSrr rarbrr?rdrandomchoiceint ValueError design_batchr~rcrrrrr OUTPUT_DIRmakedirsrnowstrftime master_promptrtxt2imgr:randint full_audit save_imager;error))rDrFr'rerLrgen_cfgrrfrri cantidad_strcantidadrrpromptsrrrgp confirmarr output_subdir extra_tags output_path aprobadasintentos timestamp prompt_aipositivenegativeimagesinfo image_bytesrapprovedverdictrofilenamefilepath rejected_paths) r run_ai_batchrso LM\"G$.x9K$LLR^M  c+d CD  ' ' )E &s5zl 34EFLLNI$$Y/ K {/: ; MM%( $$Y/ %i[1267==?L|$?@FFHPDKABHHJRdJ L - c BC "//$<##! 0  #CL>F88DE  Bxj/  XJ'1%1 A3b4C %&& Bxj/LMSSU[[]IC k 8$]3$X./FGF l"% + & I '',,z=9KKK d+IH ''8I!'* 9  A  1Q3%qWl9+QxjQR( '!//04<.@YZcYddfgqfrsH..H Jx}oS1 2 == (g&x(g&!+.$^4!+.) LFD34 )K88FFNN1i$@AD 3 4)/):):;)G &HgwQ '[YsO5dK77<< X>  h7( 1XJfXJOP " Z {RZ[cZddhEi j   m<)'34W+b Bxj/ " M)AhZ 01 K } %& XJb/U "  )!-.b ' LL+AaC51#6 7 M!% & & 'sJ- R 8RA3SC0S RR R?'R::R? T ,S==TcFtd|dtdj}|sd}tdj}|sd}tdj}|sd}tdj} t|}||||fS#t$rd}YwxYw)Nu --- Configuración del lote (rZu Personaje (vacío = aleatorio): uOutfit (vacío = aleatorio): Escenario (vacío = aleatorio): uCantidad de imágenes: r)rSrr?rr)rrr[rnrrs r get_batch_paramsrgs +K= >?89??AI  2 3 9 9 ;F  4 5 ; ; =E 2399;L|$ feX -- s B B B c  |d}t|\}}}} t||dd|dd} |dk(rd} d} nd } d } tjj t | } tj | d d }d }tjjd}tddtd| d|dtd|xsdtd|xsdtd|xsdtd| dtdd|| kr|dz }td|d|d|  |r|j|||| \}}nGtj|j}|j||\}}td|td |dd!d"|j!|||d#|d$|d%|d&|d'|d()\}}|s td*|d }|j#d+tj$dd,}td-| j'|\}}}|rX|dz }|d.|d/d0|d1}tjj | |}|j)||td2|d| d3|nKtjj t d4|d5|d1}|j)||td6||| krtddtd:td;|d| td<|td=||z d>zd?d@td| tddy#t*$r4}t,j/d7|d8|td9|Yd}~jd}~wwxYw)ANrrrrrrrrrrTr rrrrQz Iniciando lote: u imágenes ()z Personaje: ALEATORIOz Outfit: z Escenario: rrr/z [Intento #r)rr[rnr)rrz Personaje aleatorio: rrrrrrrrrru" ✗ Forge no devolvió imágenesrrr_rrrrrr_rej_rzError en intento rrz LOTE COMPLETADOrz Intentos totales: u Tasa de éxito: rz.1f%)rrrrrrrrrrrS build_promptrrra build_randomrr:rrrr~r;r)rDrFr'rerrrr[rnrrrrrrrr prompt_pos prompt_neg random_charrrrrrrrorrrrs r run_batchr s\"G)9+)F&Ivuh 8$]3$X./FGF l"% + & I '',,z=9KKK d+IH ''8I Bxj/ xj [M CD M)2{3 45 Jv,- ./ M%.;/ 01 K a () XJb/ h A  8*L 1XJGH5 )0)=)=#,!) *>*& J%mmG,G,G,IJ )0)=)=#. +*>*& J/ }=> Jz#2/s3 4 ==! *g&x(g&!+.$^4!+.) LFD:; )K88FFNN1i$@AD 3 4)/):):;)G &HgwQ '[)CdV4H77<< X>  h7( 1XJfXJOP " Z {RWX`WaaeEf g   m<)'34i h v Bxj/  M)AhZ 01  +, y1#5c:! <= K } %& XJb/  LL,XJb< = M!% &  s 3B/M#C0M N)M==Nc Jtdtdj}|s tdytdjxsd}tdjxsd}tdjxsd}tdjxsd}|d} |j||||| \} } td | dd d td |j | | | d| d| d| d| d| d\} } | r| j dd}t jjd}d|d|d}tjjtd|}|j| d|td|ytdy)Nu --- Generación Individual ---z Personaje: ru!Outfit (vacío = del personaje): ruPose (vacío = aleatoria): uDetalles (vacío = aleatorio): r)rr[rnrmroz Prompt: rrz Generando...rrrrrrrrrrsingle_rrru✓ Guardada: u ✗ No se generó ninguna imagen)rSrr?rrr:rrrrrrrr)rDrFr'rerr[rnrmrorr r rrrrrrs r run_singlers ,-m$**,I  ./ 6 7 = = ? G4F 4 5 ; ; = EE . / 5 5 7 ?4D56<<>F$G\"G$11   2J  Jz$3'( ,- .=="gx g+&^,+&! LFDxx"LLN++O< YKuTF$777<< M8D H- xj)* 01rUc t}t|\}}}t t }|d}d}n@t |trdnd}t |tr|jn |j}ttdt|jtdt|jdtd |d |d  t|td j!}|d k(rt#||||dn|dk(rt#||||dn|dk(rt%||||n|dk(r t'|nw|dk(r t)|nf|dk(r t+|nU|dk(r t-|nD|dk(r t/||n2|dk(rt1|||||dn|dk(r tdytd#t$r}td|Yd}~yd}~wwxYw)NuError cargando configuración: zninguno (DeepSeek retirado)-rrz Personajes: z Wildcards: z tipos cargadosu Diseñador: rpruSelecciona una opción: 1r2r34567890z Hasta luego!zOpcion no valida)r!r~rSrOrr=rr _get_current_modelr(rVrbra wildcardsrXrr?r rrkrxrrrr) rDrrFr'rLrerdesigner_modelopcions r mainr!s +62E68H%G5 3=hHZ3[/ay :DXOa:b446hphvhvN LW88:;< => KG--./ ?@ Lr.)9 ;< 81288: S= feVWl C s] feVWj A s] vufg 6 s] G $ s] 7 # s]   s] H % s] $Xw 7 s] < P s] " #  $ %5 '  /s34s G G! GG!__main__)N)r)1rsysrloggingrrrinsertdirnameabspath__file__ api_clientrrprompt_builderr vision_criticrr)rr ImportErrorr=rrrLOGS_DIRr basicConfigINFO FileHandler StreamHandler getLogger__name__r;r!rOrVrXrkrxrrrrrr rr!rTrUr r4s  277??277??8#<=>0(&B 77??277??277??8+DE F WW\\(H - WW\\(H - 77<<& )  J&  Ht$ ,, 4BGGLL<@7S   8 $! 2#j  :+ Q 96rEP.0bJ-2`0&f zFoNs>G88 HH