a Ad n @sddlZddlZddlZddlZddlmZmZddlZddlZddl Z ddl Z ddl Z ddl m Z mZmZmZmZmZddlmZddlmZdgiadZdd Zd d Zd d ZddZddZd9ddZddZddZ ddZ!d:ddZ"ddZ#ddZ$d d!Z%d"d#Z&d$d%d&fd'd(d&fd'd)d&fd$d*d&fd'd+d&fd$d,d&fd$d-d&fd$d.d&fgZ'e(d/d0d1Z)d;d3d4Z*d5d6Z+d7d8Z,dS)<N)datetimetimezone) extensionssharedpaths config_stateserrorsrestart)config_states_dir)wrap_gradio_gpu_callrz" style="color: var(--primary-400)"cCstjjrJddS)Nz7extension access disabled because of command line flags)rcmd_optsdisable_extension_accessrr=/var/www/html/stable-diffusion-webui/modules/ui_extensions.py check_accesssrc Cstt|}t|tks*Jd|t|}t|tksNJd||rZtdt|}tjD]F}|j|vrxqhz | Wqht yt j d|jddYqh0qh|t j_|t j_t jt jtrtntdS)N/wrong disable_list data for apply_and_restart: z.wrong update_list data for apply_and_restart: zBackup (pre-update)zError getting updates for Texc_info)rjsonloadstypelistsave_config_statesetrnamefetch_and_reset_hard Exceptionrreportroptsdisabled_extensionsdisable_all_extensionssaveconfig_filenamer is_restartableZrestart_programZ stop_program) disable_listZ update_listZ disable_alldisabledupdateextrrrapply_and_restarts*       r(cCst}|sd}||d<td}tjt|d|d}t d|dt |dd d  }t j ||d d Wdn1s0Yt tttjd }d gttj}tjj||dd|dfS)NConfigrz%Y_%m_%d-%H_%M_%S_z.jsonz*Saving backup of webui/extension state to .wzutf-8)encoding)indentCurrent)valuechoicesz.Saved current webui/extension state to "z")r get_configrnowstrftimeospathjoinr printopenrdumplist_config_statesnextiterall_config_stateskeysrgrDropdownr&)rZcurrent_config_state timestampfilenamef new_valueZ new_choicesrrrr;s.rcCs|dkr dS|sdSttj|}td|d|dksD|dkr^|dtj_tjtj|d ksn|dkrxt |tj d S) Nr0z-Select a config to restore from.zCancelled.z'*** Restoring webui state from backup: z ***rbothfilepathwebui) rrr?r9rrrestore_config_state_filer!r"Zrestore_webui_configstaterequest_restart)Z confirmedZconfig_state_nameZ restore_type config_staterrrrestore_config_stateKs    rOc stt|ttks*Jd|fddtjD}t|tj_ |D]~}|j tj_ z | WnXt y}zdt|vrWYd}~n0d}~0tytjd|j ddYn0tjqNtdfS) Nrcs$g|]}|jdur|jvr|qSN)remoter.0r'r%rr iz!check_updates.. FETCH_HEADzError checking updates for TrrJ)rrrrrrlenrrL job_countrtextinfo check_updatesFileNotFoundErrorstrrrrnextjobextension_table)id_taskr$extsr'errTrr[cs        r[cCsZ|dur|dd}|drR|dr4|dd}|d|}d|d|dS|SdS) Nzhttps://github.com/.gitz/commit/ ) startswithendswith) commit_hashrQtexthrefrrrmake_commit_link|s     rncCsdtdtddtjDr$dndd}tjD]\}|dt|jpPdd t|jrdd n|jpldd }|jrd t|j d t|j d}n|j }d}t j j r|jrt jjdkr|jrt j jst jjdkrt}|j}|jr |jr t|j|j|j}|d|dt|j d|jr.dnddt|j d|d|jd|d|jrjt|jndd|jdurdndd|d7}q4|d7}|S)N rfrgzbuilt-inrhz>extraallz, z #
rVz"extension_table..checked="checked"rJa onchange="toggle_all_extensions(event)" /> Extension URL Branch Version Date Update
z"  z
)timertrread_info_from_repohtmlescaperQ is_builtin can_updaterstatusrr disable_extra_extensionsrr STYLE_PRIMARYversionrkrnrpbranch commit_dater fromtimestamp)coder'rQZ ext_statusstyleZ version_linkrrrr_sL  4$4   r_cCs\|dkrt}n tj|}|dd}tt|d}|dd}z|ddpZd }|dd }|dd pvd}|dd }|rtt|}nd}d t|dt|pd d} t ||} t |||} t } d } d }d }| d|krt } | d |kr t }| d |krt }dtd|d|d|d| d| d|d|d|d| d|d| d}ddt j D}|d D]`\}}|dpd }|d pd}|d}|d pd}|d }|rtt|}nd}d t|dt|pd d} t ||} t |||} d }d } d }d }||vr||}||j|kr`t }|j|krpt } |j|krt }|j|krt }|d|d|rdnd dt|d| d| d|d|d|d| d|d| d 7}q|d!7}Wn\tyV}zBtd"|d#|dtd|d|d|d$ }WYd}~n d}~00|S)%Nr0rr) created_atrHz rIrQrJrrkrrfrgrhroz -->

Config Backup: z

Filepath: z
Created at: a

WebUI State

URL Branch Commit Date

Extension State

cSsi|] }|j|qSr)rrRrrr rVz.update_config_states_table..rrpz# z
Extension URL Branch Commit Date
z[ERROR]: Config states , z&

This file is corrupted

)rr3r?getrxasctimegmtimerzr{rnZget_webui_configrritemsryrprQrrkrr9) state_namerN config_nameZ created_daterHZ webui_remoteZ webui_branchZwebui_commit_hashZwebui_commit_daterQZ commit_linkZ date_linkZ current_webuiZ style_remoteZ style_branchZ style_commitrZext_mapZext_nameZext_confZ ext_remoteZ ext_branchZ ext_enabledZext_commit_hashZext_commit_dateZ style_enabledZ current_extrbrrrupdate_config_states_tables      "    -$          rcCs|dur dS|dd}|S)NrJrd)replace)urlrrrnormalize_git_urlJs rc s@tt|tr|}t|tr*|}|s6Jd|dusF|dkrb|d^}}t|}|}tjt j |}tj |rJd|t|t fddt j Drt d|tjtjd|}z^t|d |s.z.Extension with this URL is already installed: tmpTz blob:none)filter)rrrzInstalled into z. Use Installed tab to restart.)$r isinstancer]stripsplitrr6r7r8rextensions_direxistsanyrr data_pathshutilrmtreegitRepo clone_fromrQfetch submodulesr&renameOSErrorerrnoZEXDEVmovelaunchrun_extension_installerlist_extensionsr_rzr{) dirnamer branch_namepartsZ last_part target_dirtmpdirrepo submoduleerrrrrrinstall_extension_from_urlRsP      .  ,  rcCs*td|\}}t|||\}}|||dfSNrJ)r&refresh_available_extensions_from_data)r hide_tags sort_column filter_textZ ext_tablemessagerr*rrrinstall_extension_from_indexsrcCslddl}|j|}|}Wdn1s20Yt|at||\}}||tj j |dddfS)Nr)r2rJ) urllib.requestrequesturlopenreadrravailable_extensionsrrA CheckboxGroupr&)rrrurllibresponserlrtagsrrrrefresh_available_extensionss & rcCst|||\}}|dfSrr)rrrrr*rrr%refresh_available_extensions_for_tagssrcCst|||\}}|dfSrr)rrrrr*rrrsearch_extensionssrTcCs |ddSNaddedzrrrrrrVrFcCs |ddSrrrrrrrrVcCs |ddSNrrrrrrrrrVcCs |ddSrrrrrrrrVcCsdS)NrrrrrrrrVcCs |ddS)N commit_timerJrrrrrrrVcCs |ddS)NrrJrrrrrrrVcCs |ddS)NstarsrrrrrrrrV)infoc CsFz(t||djtjddWStt fy@YdS0dS)Nz%Y-%m-%dT%H:%M:%SZ)tzinfoz%Y-%m-%drJ) rstrptimerrrutc astimezoner5 ValueError TypeError)rkeyrrrget_dates(rrJcsztd}ddtjD}tdit|d}dtd}td|krZttkrbnn|nd\}}t|||dD]} | d d } t| d d} | d d } t | d} t | d}| dd}| dd}| dg}|durqz|t |d}|r |dgn|}t fdd|Dr0|d7}qz|rz| rz| t|  vrz| t| vrz|d7}qzdt|d|rdndd|sdndd}dfdd |D}|d!t|d"t| d#|d$t|d%t| d&t| d't|d(| d)|d*7}fd+d |DD]}||<q:qz|d,7}|dkrn|d-|d.7}|tfS)/NrcSsi|]}t|j|jqSr)rrQr)rS extensionrrrrrVz:refresh_available_extensions_from_data..rrroz --> )rreverserZnonamerrunknownrrr descriptionrJ installedc3s|]}|vr|VqdSrPrr) tags_to_hiderrrqrVz9refresh_available_extensions_from_data..z5rcs&g|]}d|dd|dqS)z#rrrrrrUrVz:refresh_available_extensions_from_data..z/ csg|]}|vr|qSrrrrrrrUrVrwz

Extension hidden: z

)rrrrrx sort_orderingrXsortedintrrrrlowerrzr{r8r)rrrextlistZinstalled_extension_urlshiddenrZ sort_reverse sort_functionr'rrrZ update_time create_timerrZextension_tagsexistingZ install_codeZ tags_texttagr)rrrrsl  ,       02  rcCstjD] }|qdSrP)rry)rrrrpreload_extensions_git_metadata s rc" Csddl}ttjtdtjdd}tj ddbtj ddd tj d dtt rld nd }tj|d d}tjdd}tjdgdtjjdd}tjdddd}tjdddd}Wdn1s0Yd}tjjstjjstjjdkrFtjjrd} n"tjjdkr,d} ntjjr:d} d| d}t t|} Wdn1sp0Ytj dd tjd!d"d} Wdn1s0Y|jtg| gd#|jtd$|||ggd%|jtttgd&d'| |g| | gd%Wdn1s0Ytj d(d)d  t Xtjd*d d} tj d+d,} tj| d-dd.}tjd/dd0}tjd1dd0}Wdn1s0Yt >tj!gd2d3gd4d5}tjd6d7gd8d9d:}Wdn1s0Yt tjd;dd<}Wdn1s20Yt}t}| j|j"j#t$ttttgd&|||g|||||gd#|j|j"j#t%ttgd&||||g|| |gd#|j&|j"j#t'tgd&|||g||gd#|j&|j"j#t(tgd&|||g||gd#|j&|j"j#t(tgd&|||g||gd#Wdn1s`0Ytj d=d>d tjd?d@}tjdAdBdC}tjdDdEdC}tjdFd d}tjdGd}|j|j"j#dHdIttgd&|||g|| |gd#Wdn1s 0Yt dJ`tj dKdptj)dLdMdNdNgt*tj+,dO}|j"-|tjdPdIdQtjdRgdSdTdUd}tjdVd dWdX}Wdn1s0Ytj dYd,tj.ddZdd[}tjd\d}Wdn1s0Ytd}td!} |jt/|g| gd#|jt0|g||gd#tj1dd]}!|jt2d^|!||g|gd%|j&t/|g| gd#Wdn1s0YWdn1s0YWdn1s0Y|S)_Nr)targetF)analytics_enabledZtabs_extensions)elem_idrr)idZextensions_installed_topzApply and restart UIzApply and quitprimary)r1variantzCheck for updates)r1zDisable all extensions)nonersrtextensions_disable_all)labelr2r1rextensions_disabled_list)rvisible containerextensions_update_listrJrzK"--disable-all-extensions" was used, remove it to load all extensions againzR"Disable all extensions" was set, change it to "none" to load all extensions againzM"--disable-extra-extensions" was used, remove it to load all extensions againz)rzprogress-container) elem_classesz Loading...Zextensions_installed_html)fninputsoutputsZextensions_apply)r_jsrr) extra_outputsZextensions_checkZ Available availablez Load from:ZWEBUI_EXTENSIONS_INDEXzchttps://raw.githubusercontent.com/AUTOMATIC1111/stable-diffusion-webui-extensions/master/index.jsonzExtension index URL)r1rrextension_to_install)rrinstall_extension_button)ads localizationrzHide extensions with tags)scriptr r r)r1rr2 newest firstOrder)r z oldest firstza-zzz-azinternal orderz update timez create timerindex)r1rr2rZSearch)rrzInstall from URLZinstall_from_urlz"URL for extension's git repository)rzSpecific branch namez#Leave empty for default main branch)r placeholderzLocal directory namezLeave empty for autorZextension_install_resultcWstgt|SrP)rAr&r)argsrrrr~rVzcreate_ui..zBackup/RestoreZextensions_backup_top_rowz Saved ConfigsZextension_backup_saved_configsr0)rrr1r2cSsddgttjiS)Nr2r0)rrr?r@rrrrrrVZrefresh_config_stateszState to restore)rrIrGrZextension_backup_restore_typezRestore Selected ConfigZextension_backup_restore)r1rrZextensions_backup_top_row2z Config Name)r show_labelzSave Current Config)rZconfig_state_confirm_restore)3 modules.uirr< threadingThreadrstartrABlocksTabsTabItemRowr r#ButtonRadiorrr Textr rHTMLloadr_clickr(r r[r&r6environrruiwrap_gradio_callrrchangerrrBrr?r@create_refresh_buttonTextboxrrLabelrO)"modulesr"Z apply_labelapplycheckrrrrzmsgrZextensions_tableZ#refresh_available_extensions_buttonZextensions_index_urlZavailable_extensions_indexrr rrZsearch_extensions_textZinstall_resultZavailable_extensions_tableZ install_urlZinstall_branchZinstall_dirnameZinstall_buttonZconfig_states_listZconfig_restore_typeZconfig_restore_buttonZconfig_save_nameZconfig_save_buttonZconfig_states_infoZconfig_states_tabledummy_componentrrr create_uis ."    *.& . 6 .&  &  &"0,   fr-)N)N)rJ)-rr6rrxrrrgradiorArzrrr(rrrrrr modules.paths_internalr modules.call_queuer rrrr(rrOr[rnr_rrrrrrrrdictrrrr-rrrrsT   ! 9  :          K
Extension Description Action
Update: z Added: z Created: z*stars: z&

 z!