a Ad<@s*ddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z ddlmZdZeeZedZedZeeZiZgZGdd d Zd d Zd d ZddZddZd(ddZddZ ddZ!edddZ"ddZ#ddZ$dd Z%e&d!d"d#Z'gZ(d$d%Z)d&d'Z*dS))N) data_path)shared ui_tempdirscript_callbacks processing)Imagez/\s*([\w ]+):\s*("(?:\\.|[^\\"])+"|[^,]*)(?:,|$)z ^(\d+)x(\d+)$z\(([0-9a-f]+)\)$c@seZdZdddZdS) ParamBindingNcCs2||_||_||_||_||_||_|p*g|_dSN) paste_buttontabnamesource_text_componentsource_image_componentsource_tabnameoverride_settings_componentpaste_field_names)selfr r r r rrrrO/var/www/html/stable-diffusion-webui/modules/generation_parameters_copypaste.py__init__szParamBinding.__init__)NNNNN)__name__ __module__ __qualname__rrrrrrsrcCsttdSr ) paste_fieldsclearregistered_param_bindingsrrrrreset!srcCs6dt|vr(dt|vr(dt|vr(|Stj|ddS)N, :F) ensure_ascii)strjsondumpstextrrrquote&s$r%cCsPt|dks$|ddks$|ddkr(|Sz t|WStyJ|YS0dS)Nr")lenr!loads Exceptionr#rrrunquote-s $  r+cCs|dur dSt|tkrD|rDt|dtkrD|dddrD|d}t|tkr|ddr|d}ttj|}|s~Jd|ddd}t |St|tkrt |dkrdS|d}| dr|t dd}t |d }t t|}|S) Nris_fileFnamez8trying to open image file outside of allowed directories?zdata:image/png;base64,zutf-8)typelistdictgetrZcheck_tmp_filerdemorsplitropenr( startswithbase64 decodebytesencodeioBytesIO)ZfiledatafilenameZis_in_right_dirimagerrrimage_from_url_text7s&0     r?cCs>|||dt|<ddl}|dkr*||j_n|dkr:||j_dS)N)init_imgfieldsrrtxt2imgimg2img)rZ modules.uiuiZtxt2img_paste_fieldsZimg2img_paste_fields)r r@rArmodulesrrradd_paste_fieldsTs  rFcCs0i}|D]"}tjd||dd||<q|S)NzSend to Z_tab)elem_id)grButton)Z tabs_listbuttonstabrrrcreate_buttons_s rLc CsT|D]F\}}t|tjjr"|nd}t|tr4|nd}tt|||||dqdS)z[old function for backwards compatibility; do not use this, use register_paste_params_buttonN)r r r r r)items isinstancerH components Componentr register_paste_params_buttonr)rJZ send_imageZsend_generate_infor buttonr rrrr bind_buttonsfsrSbindingcCst|dSr )rappendrTrrrrQosrQcstD]}t|jd}t|jd}|jp8t|jd}tt|rPdd|Dngd}tt|rpdd|Dngd}|jr|rt|jtj r|rt nt }d}n|rt ndd }d}|j j |||jg|r|||gn|gd d |jdur|durt|j ||j||j|jdur|durgd tjjr:d gng|j|j j dd fddt|jdDfdd|Dd d|j j dd|jddd d qdS)Nr@rArcSsg|]\}}|dkr|qS)Size-1r.0fieldr-rrr zz0connect_paste_params_buttons..cSsg|]\}}|dkr|qS)Size-2rrXrrrr[{r\Zextract_image_from_gallerycSs|Sr rxrrrr\z.connect_paste_params_buttons..Ffn_jsinputsoutputs show_progress)PromptNegative promptStepszFace restorationSeedcWs|Sr rr^rrrr`r\csg|]\}}|vr|qSrrrXrrrr[r\csg|]\}}|vr|qSrrrXrkrrr[r\rbrdrerfZ switch_to_)rrr rnextiterr rNrHGallerysend_image_and_dimensionsr?r clickr connect_pasterropts send_seedr)rUZdestination_image_componentrArZdestination_width_componentZdestination_height_componentfuncZjsfuncrrkrconnect_paste_params_buttonsssH     " rvcCsVt|tjr|}nt|}tjjrr~rz Hires samplerzUse same samplerzHires checkpointzUse same checkpointz Hires promptzHires negative promptRNGGPUz Schedule type AutomaticzSchedule max sigmazSchedule min sigmaz Schedule rhoz VAE EncoderFullz VAE Decoder)stripsplitr(re_paramfindallrVr7rrsinfotext_styles prompt_stylesextract_styles_from_promptr+ re_imagesizematchgroupr*printr3r) r_rpromptnegative_promptZdone_with_promptlineslastlinelineZ found_styleskvmhypernetrrrparse_generation_parameterss             &            rc Csi}i}|D]"}|jddd\}}|||<q ddtjjD}|tD]0\}}||d} | durlqNtj|| ||<qN|S)a'creates processing's override_settings parameters from gradio's multiselect Example input: ['Clip skip: 2', 'Model hash: e6e99610c4', 'ENSD: 31337'] Example output: {'CLIP_stop_at_last_layers': 2, 'sd_model_checkpoint': 'e6e99610c4', 'eta_noise_seed_delta': 31337} rr/)maxsplitcSs g|]\}}|jr|j|fqSrinfotextrYrinforrrr[`r\z1create_override_settings_dict..N) rrrrs data_labelsrM infotext_to_setting_name_mappingr3 cast_value) Z text_pairsrparamspairrrmapping param_name setting_namevaluerrrcreate_override_settings_dictNs  rcsxfdd}|durt|j}|tkr|dkrd} n||} |tj| dWqxty,|tYqx0qx|S)Nz params.txtrutf8)encodingFalseF)r)rcmd_optshide_ui_dir_configospathjoinrexistsr6readrrinfotext_pasted_callbackcallabler3rVrHrzrNtype_of_gr_updater0rboolr*) rr=filerroutputkeyrZvaltypeval)rrr paste_funcms2  &       z!connect_paste..paste_funccSsi|]\}}|dqS)r/r)rY_rrrr r\z!connect_paste..csi}ddtjjD}|tD]h\}}|vr4q"||d}|durJq"|dkr\tjjr\q"tj||}ttj|d}||krq"|||<q"dd|D}t j j ||t |dS)NcSs g|]\}}|jr|j|fqSrrrrrrr[r\z9connect_paste..paste_settings..sd_model_checkpointcSsg|]\}}|d|qS)rr)rYrrrrrr[r\)rchoicesvisible) rrsrrMrr3disable_weights_auto_swaprgetattrrHDropdownrzr)rvalsrrrr current_valueZ vals_pairs)already_handled_fieldsrrpaste_settingss"  z%connect_paste..paste_settingscSsg|] }|dqS)rr)rYr_rrrr[r\z!connect_paste..FrlZrecalculate_prompts_ra)rq)rRrZ input_comprr rrr)rrrrrls$ $  rr)N)+r8r;r!rregradiorH modules.pathsrrErrrrPILrZ re_param_codecompilerrZre_hypernet_hashr0rzrrrrrr%r+r?rFrLrSrQrvrprr rrrrrrrrrs>          / m