a Ad!@sJddlZddlZddlZddlmZddlmZddZGdddZ dS)N)errors) ToolButtoncCsddt|dgDS)NcSs"g|]}t|tr|dn|qS)r) isinstancetuple).0xr;/var/www/html/stable-diffusion-webui/modules/ui_loadsave.py z!radio_choices..choices)getattr)comprrr radio_choices src@sjeZdZdZddZddZdddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZdS) UiLoadsavez@allows saving and restoring default values for gradio componentsc Cs||_i|_i|_d|_d|_d|_d|_d|_ztj |jrJ| |_Wn6t y}zd|_t |dWYd}~n d}~00dS)NFTzloading settings)filename ui_settingscomponent_mapping error_loading finalized_uiui_defaults_viewui_defaults_applyui_defaults_reviewospathexistsread_from_file Exceptionrdisplay)selfrerrr __init__szUiLoadsave.__init__c sTjr Jdfdd }ttjtjtjtjtjtjt tj fvrVj rV|dttjkr|d|d|d|dttjkr|dfd d ttjkr|dttjkr|dttjkr|dttjkr(fd d }|d|t d dfdd}ttj krP|d|dS)z4adds component to the registry of tracked componentsNcs d|}t|dddur0d|jd|}t|ddr@dSj|d}|durht||j|<n~|rv||svnpttjr|dkrt|}n6ttjr|dkrz t |}Wnt yYdS0t ||||dur|||dkr|j vrj |<dS)N/custom_script_sourcez customscript/Zdo_not_save_to_configFvalue) r r#rgetrgrTextboxstrNumberfloat ValueErrorsetattrr)objfield condition init_fieldkeyZ saved_valuerrrrr apply_field*s,      z-UiLoadsave.add_component..apply_fieldvisibler$minimummaximumstepcs |tvSN)rvalrrr Rr z*UiLoadsave.add_component..cs6ttddr*tfdd|DS|vSdS)N multiselectFc3s|]}|vVqdSr8r)rr$r rr ar zCUiLoadsave.add_component..check_dropdown..)rr allr9r;r>r check_dropdown^s z0UiLoadsave.add_component..check_dropdownr0csfttddj}t|tkr6dd|D}||vSt|tkr^d|koXt|kSSdSdS)NcSs t|tjSr8)rr&TabItem)r rrr r<hr z@UiLoadsave.add_component..check_tab_id..cSsg|] }|jqSr)id)rtrrr r jr zBUiLoadsave.add_component..check_tab_id..rF)listfilterchildrentyper(intlen)Ztab_idZ tab_itemsZtab_idsr;rr check_tab_idgs  z.UiLoadsave.add_component..check_tab_idselected)NN)rrHr&SliderRadioCheckboxr'r)DropdownrButtonr4r Tabs)rrrr3rArKrr2r add_component%s. 0          zUiLoadsave.add_componentcCst|drRt|tjr8|jdur8||d|j||jD]}|||q>nR|jdurv||d|j|n.t|tj r|j dur||d|j |dS)zQadds all components inside a gradio block x to the registry of tracked componentsrGNz/Tabs@r") hasattrrr&rRelem_idrSrG add_blocklabelrQr$)rrrcrrr rWts   zUiLoadsave.add_blockcCs>t|jddd}t|WdS1s00YdS)Nrutf8encoding)openrjsonload)rfilerrr rszUiLoadsave.read_from_filecCsDt|jddd }tj||ddWdn1s60YdS)Nwr[r\)indent)r^rr_dump)rcurrent_ui_settingsrarrr write_to_fileszUiLoadsave.write_to_filecCs(|jrtj|jrdS||jdS)zqsaves default values to a file unless tjhe file is present and there was an error loading default values at startN)rrrrrrgrrrrr dump_defaultsszUiLoadsave.dump_defaultsccst|j|D]\\}}}||}t|}t|trd|rd|t|krJq||}t|trd|d}||krnq|dur~|dks|gkrq|||fVqdS)a given a dictionary with defaults from a file and current values from gradio elements, returns an iterator over tuples of values that are not the same between the file and the current; tuple contents are: path, old value, new value rNrT) zipritemsr%rrrIrJr)rrfvaluesr component new_value old_valuer rrr iter_changess   zUiLoadsave.iter_changesc Gsvdg}|||D]4\}}}|dur,d}|d|d|d|dqt|dkrb|d|dd |S) NzRz*Nonezz&zrT)rprappendrJjoin)rrltextrrornrrr ui_views    zUiLoadsave.ui_viewcGsZd}|}|||D]\}}}|d7}|||<q|dkrDdS||d|dS)Nrrqz No changes.zWrote z changes.)rrpcopyrg)rrlZ num_changedrfr_rnrrr ui_applys  zUiLoadsave.ui_applycCsptd|jdt4tjdddd|_tjddd d|_Wd n1sV0Ytd |_d S) zMcreates ui elements for editing defaults UI, without adding any logic to themzThis page allows you to change default values in UI elements on other tabs.
Make your changes, press 'View changes' to review the changed default values,
then press 'Apply' to write them to z>.
New defaults will apply after you restart the UI.
z View changesr secondary)r$rVvariantApplyrprimaryNrT)r&HTMLrRowrQrrrrhrrr create_uis 0zUiLoadsave.create_uicCsX|jr Jd|_|jj|jt|j|jgd|jj|j t|j|jgddS)z[adds logic to elements created with create_ui; all add_block class must be made before thisT)fninputsoutputsN) rrclickrurErrlrrrxrhrrr setup_uis "zUiLoadsave.setup_uiN)rT)__name__ __module__ __qualname____doc__r!rSrWrrgrirprurxrrrrrr rsO r) r_rgradior&modulesrmodules.ui_componentsrrrrrrr s   
PathOld valueNew value
z z
No changes