a Aþdíã@sjddlZddlZddlmZmZmZddlmZddl m Z Gdd„dejƒZej ejejdœdd „ZdS) éN)ÚscriptsÚuiÚerrors)Úcmd_opts)Ú ToolButtonc@s<eZdZdZdZdd„Zdd„Zdd„Zd d „Zdd„Z d S)Ú ScriptSeedÚseedFcCsd|_d|_d|_dS©N)rÚ reuse_seedÚ reuse_subseed©Úself©rúG/var/www/html/stable-diffusion-webui/modules/processing_scripts/seed.pyÚ__init__szScriptSeed.__init__cCsdS)NÚSeedrrrrrÚtitleszScriptSeed.titlecCstjSr )rÚ AlwaysVisible)r Ú is_img2imgrrrÚshowszScriptSeed.showc sÆtjˆ d¡dtjr6tjddˆ d¡ddˆ_ntjddˆ d¡dd d ˆ_tt j ˆ d¡dd }tt jˆ d¡dd ‰tjdˆ d¡dd}Wdƒn1s¨0Ytj dˆ d¡d}tjˆ d¡dltjddˆ d¡d d‰tt j ˆ d¡d}tt jˆ d¡d‰tjddd ddˆ d ¡d!}Wdƒn1sP0Ytjˆ d"¡dHtjd d#d$d%d ˆ d&¡d'}tjd d#d$d(d ˆ d)¡d'}Wdƒn1s¼0YWdƒn1sÜ0Y|jdd*ˆ d¡d+dggd,|jdd*ˆ d¡d+dggd,|jd-d.„d|g|gd/ˆjdf|d0d.„fˆdf|d1f|d2f|d3fgˆ_ˆj‡‡fd4d.„d5ˆj›dˆj‡‡fd6d.„d5ˆj›dˆj|ˆ|||fS)7NZseed_row)Úelem_idrÚréd)ÚlabelÚvaluerÚ min_widthéÿÿÿÿr)rrrrÚ precisionÚrandom_seedzRandom seed)rrr z Reuse seedÚExtraZsubseed_showF)rrrÚseed_extras)ÚvisiblerZsubseed_rowúVariation seedÚsubseed)rrrrÚrandom_subseedrzVariation strengthgég{®Gáz„?Úsubseed_strength)rrÚminimumÚmaximumÚsteprZseed_resize_from_rowiézResize seed from widthÚseed_resize_from_w)r'r(r)rrrzResize seed from heightÚseed_resize_from_hzfunction(){setRandomSeed('z')}©ÚfnÚ_jsÚ show_progressÚinputsÚoutputscSstj|dS)N)r!)ÚgrÚupdate©ÚxrrrÚ3ózScriptSeed.ui..)r0r1r2cSsd|vpd|vS)Nr"úSeed resize from-1r)Údrrrr77r8zVariation seed strengthr9zSeed resize from-2cstˆjˆ|jdƒS)NF)Úconnect_reuse_seedrÚ componentr5)r r rrr7>r8Úgeneration_info_cstˆˆ|jdƒS)NT)r;r<r5)rr#rrr7?r8)r3ÚRowrrÚuse_textbox_seedÚTextboxrÚNumberrrÚ random_symbolÚreuse_symbolÚCheckboxÚGroupÚSliderÚclickÚchangeÚinfotext_fieldsÚon_after_componentÚtabname) r rrÚ seed_checkboxr r$r&r+r,r)r rr r#rrs:4<\"" ú z ScriptSeed.uicCsB||_|r|dkr||_||_|r>|dkr>|dkr>||_||_dS)Nr)rr#r&r+r,)r ÚprrLr#r&r+r,rrrÚsetupCszScriptSeed.setupN) Ú__name__Ú __module__Ú__qualname__ÚsectionÚcreate_grouprrrrrNrrrrr s*r)rr Úgeneration_infocs2tdœ‡fdd„}|j|dd||g||gddS)a Connects a 'reuse (sub)seed' button's click event so that it copies last used (sub)seed value from generation info the to the seed field. If copying subseed and subseed strength was 0, i.e. no variation seed was used, it copies the normal seed value instead.)Úgen_info_stringcsÜd}z t |¡}|| dd¡8}ˆrl| dd¡dkrl| ddg¡}|d|kr\t|ƒkrdnn|nd}n6| ddg¡}|d|kr”t|ƒkrœnn|nd}Wn*tjjyÎ|rÊt d|›¡Yn0|t ¡gS)NrÚindex_of_first_imagerr&Úall_subseedsÚ all_seedsz$Error parsing JSON generation info: ) ÚjsonÚloadsÚgetÚlenÚdecoderÚJSONDecodeErrorrÚreportr3r4)rUÚindexÚresZgen_inforWrX©Ú is_subseedrrÚ copy_seedUs *,z%connect_reuse_seed..copy_seedz'(x, y) => [x, selected_gallery_index()]Fr-N)ÚstrrG)rr rTrcrdrrbrr;Psûr;)rYÚgradior3ÚmodulesrrrZmodules.sharedrZmodules.ui_componentsrÚScriptBuiltinUIrrAÚButtonr@r;rrrrÚsF