521 lines
31 KiB
Python
521 lines
31 KiB
Python
import gradio as gr
|
||
import launch
|
||
import modules
|
||
import modules.generation_parameters_copypaste as parameters_copypaste
|
||
import os
|
||
import sys
|
||
import typing as t
|
||
from modules import shared
|
||
from modules.call_queue import wrap_queued_call
|
||
from modules.sd_models import CheckpointInfo
|
||
from modules.ui_components import ToolButton
|
||
|
||
from scripts.msai_logging.msai_logger import Logger
|
||
from scripts.runtime.msai_prelude import MiaoshouPrelude
|
||
from scripts.runtime.msai_runtime import MiaoshouRuntime
|
||
from . import widget
|
||
|
||
|
||
class MiaoShouAssistant(object):
|
||
# default css definition
|
||
default_css = '#my_model_cover{width: 100px;} #my_model_trigger_words{width: 200px;}'
|
||
|
||
def __init__(self) -> None:
|
||
self.logger = Logger()
|
||
self.prelude = MiaoshouPrelude()
|
||
self.runtime = MiaoshouRuntime()
|
||
self.refresh_symbol = '\U0001f504'
|
||
self.coffee_symbol = '\U0001f9cb' # 🧋
|
||
self.folder_symbol = '\U0001f4c2' # 📂
|
||
|
||
def on_event_ui_tabs_opened(self) -> t.List[t.Optional[t.Tuple[t.Any, str, str]]]:
|
||
with gr.Blocks(analytics_enabled=False, css=MiaoShouAssistant.default_css) as miaoshou_assistant:
|
||
self.create_subtab_boot_assistant()
|
||
self.create_subtab_model_management()
|
||
self.create_subtab_model_download()
|
||
self.create_subtab_update()
|
||
|
||
return [(miaoshou_assistant.queue(), "Miaoshou Assistant", "miaoshou_assistant")]
|
||
|
||
def create_subtab_boot_assistant(self) -> None:
|
||
with gr.TabItem('Boot Assistant', elem_id="boot_assistant_tab") as boot_assistant:
|
||
with gr.Row():
|
||
with gr.Column(elem_id="col_model_list"):
|
||
gpu, theme, port, chk_args, txt_args, webui_ver = self.runtime.get_default_args()
|
||
gr.Markdown(value="Argument settings")
|
||
with gr.Row():
|
||
self.drp_gpu = gr.Dropdown(label="VRAM Size", elem_id="drp_args_vram",
|
||
choices=list(self.prelude.gpu_setting.keys()),
|
||
value=gpu, interactive=True)
|
||
self.drp_theme = gr.Dropdown(label="UI Theme", choices=list(self.prelude.theme_setting.keys()),
|
||
value=theme,
|
||
elem_id="drp_args_theme", interactive=True)
|
||
self.txt_listen_port = gr.Text(label='Listen Port', value=port, elem_id="txt_args_listen_port",
|
||
interactive=True)
|
||
|
||
with gr.Row():
|
||
|
||
self.chk_group_args = gr.CheckboxGroup(choices=list(self.prelude.checkboxes.keys()), value=chk_args,
|
||
show_label=False)
|
||
self.additional_args = gr.Text(label='COMMANDLINE_ARGS (Divide by space)', value=txt_args,
|
||
elem_id="txt_args_more", interactive=True)
|
||
|
||
with gr.Row():
|
||
with gr.Column():
|
||
txt_save_status = gr.Markdown(visible=False, interactive=False, show_label=False)
|
||
drp_choose_version = gr.Dropdown(label="WebUI Version",
|
||
choices=['Official Release', 'Python Integrated'],
|
||
value=webui_ver, elem_id="drp_args_version",
|
||
interactive=True)
|
||
gr.HTML(
|
||
'<div><p>*Save your settings to webui-user.bat file. Use Python Integrated only if your'
|
||
' WebUI is extracted from a zip file and does not need python installation</p></div>')
|
||
save_settings = gr.Button(value="Save Boot Settings", elem_id="btn_arg_save_setting")
|
||
|
||
with gr.Row():
|
||
# with gr.Column():
|
||
# settings_submit = gr.Button(value="Apply settings", variant='primary', elem_id="ms_settings_submit")
|
||
# with gr.Column():
|
||
btn_apply = gr.Button(value='Apply Settings', variant='primary',
|
||
elem_id="ms_settings_restart_gradio")
|
||
restart_gradio = gr.Button(value='Restart WebUI', variant='primary',
|
||
elem_id="ms_settings_restart_gradio")
|
||
|
||
'''def mod_args(drp_gpu, drp_theme, txt_listen_port, chk_group_args, additional_args):
|
||
global commandline_args
|
||
|
||
get_final_args(drp_gpu, drp_theme, txt_listen_port, hk_group_args, additional_args)
|
||
|
||
print(commandline_args)
|
||
print(sys.argv)
|
||
#if '--xformers' not in sys.argv:
|
||
#sys.argv.append('--xformers')
|
||
|
||
settings_submit.click(mod_args, inputs=[drp_gpu, drp_theme, txt_listen_port, chk_group_args, additional_args], outputs=[])'''
|
||
|
||
save_settings.click(self.runtime.change_boot_setting,
|
||
inputs=[drp_choose_version, self.drp_gpu, self.drp_theme, self.txt_listen_port, self.chk_group_args,
|
||
self.additional_args], outputs=[txt_save_status])
|
||
btn_apply.click(
|
||
self.save_cmdline_args,
|
||
inputs=[self.drp_gpu, self.drp_theme, self.txt_listen_port, self.chk_group_args,
|
||
self.additional_args],
|
||
outputs=[txt_save_status],
|
||
)
|
||
|
||
def request_restart():
|
||
shared.state.interrupt()
|
||
shared.state.need_restart = True
|
||
launch.prepare_environment()
|
||
#launch.start()
|
||
|
||
restart_gradio.click(
|
||
request_restart,
|
||
_js='restart_reload',
|
||
inputs=[],
|
||
outputs=[],
|
||
)
|
||
|
||
with gr.Column():
|
||
with gr.Row():
|
||
machine_settings = self.prelude.get_sys_info()
|
||
txt_sys_info = gr.TextArea(value=machine_settings, lines=20, max_lines=20,
|
||
label="System Info",
|
||
show_label=False, interactive=False)
|
||
with gr.Row():
|
||
sys_info_refbtn = gr.Button(value="Refresh")
|
||
|
||
with gr.Row():
|
||
md_vram_release = gr.Markdown(visible=False, interactive=False, value='Memory Released', show_label=False)
|
||
with gr.Row():
|
||
chk_auto_release = gr.Checkbox(value=self.prelude.boot_settings['auto_vram'], label='Enable Auto Memory Release')
|
||
reload_button = gr.Button('Forc VRAM Release')
|
||
|
||
|
||
self.drp_gpu.change(self.runtime.update_xformers, inputs=[self.drp_gpu, self.chk_group_args], outputs=[self.chk_group_args])
|
||
sys_info_refbtn.click(self.prelude.get_sys_info, None, txt_sys_info)
|
||
chk_auto_release.change(self.runtime.change_auto_vram, inputs=[chk_auto_release])
|
||
reload_button.click(self.runtime.force_mem_release, outputs=[md_vram_release])
|
||
|
||
def create_subtab_model_management(self) -> None:
|
||
with gr.TabItem('Model Management', elem_id="model_management_tab") as tab_model_manager:
|
||
with gr.Row():
|
||
with gr.Column():
|
||
with gr.Row():
|
||
gr.Markdown(value="If you want your model directory to be other than [your_webui_dir]\\models, select a new directory down below. "
|
||
"Your default models directory needs to be removed first before you apply. Make sure do your backups!")
|
||
md_result = gr.Markdown(visible=False, value="")
|
||
with gr.Row():
|
||
model_folder_path = gr.Textbox("", label="Model path", placeholder="Copy & paste your model destination folder here", interactive=True)
|
||
#open_folder_button = ToolButton(value=self.folder_symbol, elem_id="hidden_element" if shared.cmd_opts.hide_ui_dir_config else "open_folder_metadata_editor")
|
||
refresh_models_button = ToolButton(value=self.refresh_symbol, elem_id="hidden_element")
|
||
with gr.Row():
|
||
btn_connect_modeldir = gr.Button(value="Apply Virtual Model Folder")
|
||
|
||
with widget.Row(equal_height=True):
|
||
my_search_text = gr.Textbox(
|
||
label="Model name",
|
||
show_label=False,
|
||
max_lines=1,
|
||
placeholder="Enter model name",
|
||
)
|
||
btn_my_search = gr.Button("Search")
|
||
|
||
with gr.Row():
|
||
my_model_source_dropdown = widget.Dropdown(
|
||
choices=["civitai.com", "liandange.com"],
|
||
value=self.runtime.my_model_source,
|
||
label="Select Model Source",
|
||
type="value",
|
||
show_label=True,
|
||
elem_id="my_model_source",
|
||
show_progress="full")
|
||
|
||
mtypes = list(self.prelude.model_type.keys())
|
||
my_model_type = widget.Radio(choices=mtypes,
|
||
show_label=False, value='Checkpoint', elem_id="my_model_type",
|
||
interactive=True, elem_classes="full")
|
||
|
||
with gr.Row():
|
||
my_models = self.runtime.get_local_models('', my_model_type.value)
|
||
self.runtime.ds_my_models = gr.Dataset(
|
||
components=[gr.Image(visible=False, label='Cover', elem_id='my_model_cover'),
|
||
gr.Textbox(visible=False, label='ModelId'),
|
||
gr.Textbox(visible=False, label='Name/Version'),
|
||
gr.Textbox(visible=False, label='File Name')],
|
||
elem_id='my_model_lib',
|
||
label="My Models",
|
||
headers=None,
|
||
samples=my_models,
|
||
samples_per_page=50)
|
||
with gr.Column():
|
||
self.runtime.ds_my_model_covers = gr.Dataset(components=[gr.HTML(visible=False)],
|
||
elem_id='my_model_covers',
|
||
label="Model Cover",
|
||
headers=None,
|
||
samples=[],
|
||
samples_per_page=10)
|
||
with gr.Row(variant='panel'):
|
||
c_image = gr.Image(elem_id="pnginfo_image", label="Source", source="upload", interactive=True,
|
||
type="pil", visible=True)
|
||
|
||
with gr.Row(variant='panel'):
|
||
with gr.Column():
|
||
|
||
btn_load_model = gr.HTML(
|
||
value=f'<div class="lg secondary gradio-button svelte-cmf5ev" style="text-align: center;"' \
|
||
f'onclick="return selectCheckpoint()">Load Model</div>',
|
||
visible=True)
|
||
with gr.Column():
|
||
btn_delete_model = gr.Button(visible=True, value='Delete Model')
|
||
with gr.Column():
|
||
with gr.Row():
|
||
btn_set_all_covers = gr.Button(visible=True, value='Download Cover for Listed Models')
|
||
with gr.Row():
|
||
btn_set_cover = gr.Button(visible=False, value='Set as Cover')
|
||
|
||
with gr.Row(variant='panel'):
|
||
generation_info = gr.Textbox(label='prompt', interactive=False, visible=True, elem_id="imginfo_generation_info")
|
||
with gr.Row(variant='panel'):
|
||
display_text = 'Select a model and type some text here, ChatGPT will generate prompt for you. Supports different text in different languages.'
|
||
display_value = ''
|
||
|
||
if self.prelude.boot_settings['openai_api'] == '':
|
||
display_text = 'Set your OpenAI api key in Setting & Update first: https://platform.openai.com/account/api-keys'
|
||
display_value = display_text
|
||
|
||
self.txt_main_prompt = gr.Textbox(label='Let ChatGPT write your prompt', placeholder=display_text, value=display_value, interactive=True, visible=True, elem_id="txt_main_prompt")
|
||
with gr.Row(variant='panel'):
|
||
with gr.Row():
|
||
btn_generate_prompt = gr.Button(value="Use GPT to Generate Prompt")
|
||
with gr.Row():
|
||
buttons = parameters_copypaste.create_buttons(["txt2img", "img2img", "inpaint", "extras"])
|
||
|
||
for tabname, button in buttons.items():
|
||
parameters_copypaste.register_paste_params_button(parameters_copypaste.ParamBinding(
|
||
paste_button=button, tabname=tabname, source_text_component=generation_info,
|
||
source_image_component=c_image,
|
||
))
|
||
|
||
with gr.Row(variant='panel'):
|
||
html_my_model = gr.HTML(visible=False)
|
||
|
||
btn_delete_model.click(self.runtime.delete_model, inputs=[self.runtime.ds_my_models, my_search_text, my_model_type], outputs=[self.runtime.ds_my_models])
|
||
btn_set_all_covers.click(self.runtime.set_all_covers, inputs=[my_search_text, my_model_type], outputs=[self.runtime.ds_my_models])
|
||
btn_set_cover.click(self.runtime.set_cover, inputs=[self.runtime.ds_my_models, c_image, my_search_text, my_model_type], outputs=[self.runtime.ds_my_models])
|
||
#open_folder_button.click(self.runtime.open_folder, inputs=[model_folder_path], outputs=[model_folder_path])
|
||
btn_connect_modeldir.click(self.runtime.change_model_folder, inputs=[model_folder_path], outputs=[md_result])
|
||
refresh_models_button.click(self.runtime.refresh_local_models, inputs=[my_search_text, my_model_type], outputs=[self.runtime.ds_my_models])
|
||
my_model_source_dropdown.change(self.switch_my_model_source,
|
||
inputs=[my_model_source_dropdown, my_model_type],
|
||
outputs=[self.runtime.ds_my_models])
|
||
|
||
btn_my_search.click(self.runtime.search_my_model, inputs=[my_search_text, my_model_type], outputs=[self.runtime.ds_my_models])
|
||
my_model_type.change(self.runtime.update_my_model_type, inputs=[my_search_text, my_model_type], outputs=[self.runtime.ds_my_models])
|
||
btn_generate_prompt.click(self.runtime.get_gpt_prompt, inputs=[self.runtime.ds_my_models, my_model_type, self.txt_main_prompt], outputs=[generation_info])
|
||
|
||
self.runtime.ds_my_models.click(self.runtime.get_my_model_covers,
|
||
inputs=[self.runtime.ds_my_models, my_model_type],
|
||
outputs=[self.runtime.ds_my_model_covers, html_my_model, btn_load_model])
|
||
|
||
self.runtime.ds_my_model_covers.click(self.runtime.update_cover_info,
|
||
inputs=[self.runtime.ds_my_models, self.runtime.ds_my_model_covers],
|
||
outputs=[btn_set_cover, generation_info, c_image])
|
||
|
||
|
||
|
||
def tab_model_manager_select():
|
||
self.runtime.active_model_set = 'my_model_set'
|
||
|
||
tab_model_manager.select(tab_model_manager_select, inputs=[], outputs=[])
|
||
|
||
|
||
def create_subtab_model_download(self) -> None:
|
||
with gr.TabItem('Model Download', elem_id="model_download_tab") as tab_downloads:
|
||
with gr.Row():
|
||
with gr.Column(elem_id="col_model_list"):
|
||
with widget.Row(equal_height=True):
|
||
model_source_dropdown = widget.Dropdown(choices=["civitai.com", "liandange.com", "official_models", 'hugging_face', "controlnet"],
|
||
value=self.runtime.model_source,
|
||
label="Select Model Source",
|
||
type="value",
|
||
show_label=True,
|
||
elem_id="model_source",
|
||
show_progress="full")
|
||
|
||
#btn_fetch = gr.Button("Fetch")
|
||
|
||
with widget.Row(equal_height=True):
|
||
search_text = gr.Textbox(
|
||
label="Model name",
|
||
show_label=False,
|
||
max_lines=1,
|
||
placeholder="Search keywords in model name, description or file name",
|
||
)
|
||
btn_search = gr.Button("Search")
|
||
|
||
with widget.Row(equal_height=True):
|
||
rad_model_tags = widget.Radio(choices=['All'] + self.prelude.model_tags,
|
||
show_label=False, value='All', elem_id="rad_model_tags",
|
||
interactive=True, elem_classes="full")
|
||
|
||
with widget.Row(equal_height=True):
|
||
nsfw_checker = gr.Checkbox(label='NSFW', value=False, elem_id="chk_nsfw", interactive=True)
|
||
with gr.Accordion(label="Base Model", open=False):
|
||
rad_all_none = gr.Radio(label='', choices=['All', 'None'], default='All', value='All', interactive=True)
|
||
ckg_base_model = gr.CheckboxGroup(label='', choices=self.prelude.base_model_group,
|
||
default=self.prelude.base_model_group,
|
||
value=self.prelude.base_model_group,
|
||
elem_id="ckg_base_model", interactive=True)
|
||
btn_bm_apply = gr.Button(value="Apply fiter")
|
||
with gr.Accordion(label="Model Type", open=False):
|
||
model_type = gr.Radio(choices=["All"] + list(self.prelude.model_type.keys()),
|
||
show_label=False, value='All', elem_id="rad_model_type",
|
||
interactive=True, elem_classes="full")
|
||
with gr.Accordion(label="Sort", open=False):
|
||
rad_sort = gr.Radio(choices=['Default', 'downloadCount', 'favoriteCount', 'commentCount', 'rating'],
|
||
show_label=False, value='Default', elem_id="rad_sort",
|
||
interactive=True, elem_classes="full")
|
||
|
||
images = self.runtime.get_images_html(base_model=self.prelude.base_model_group)
|
||
self.runtime.ds_models = widget.Dataset(
|
||
components=[gr.HTML(visible=False)],
|
||
headers=None,
|
||
type="values",
|
||
label="Models",
|
||
samples=images,
|
||
samples_per_page=45,
|
||
elem_id="model_dataset",
|
||
elem_classes="gallery",
|
||
container=True)
|
||
|
||
|
||
|
||
with gr.Column(elem_id="col_model_info"):
|
||
with gr.Row():
|
||
self.runtime.ds_cover_gallery = widget.Dataset(
|
||
components=[gr.HTML(visible=False)],
|
||
headers=None,
|
||
type="values",
|
||
label="Cover",
|
||
samples=[],
|
||
samples_per_page=10,
|
||
elem_id="ds_cover_gallery",
|
||
elem_classes="gallery",
|
||
container=True)
|
||
|
||
with gr.Row():
|
||
with gr.Column():
|
||
download_summary = gr.HTML('<div><span>No downloading tasks ongoing</span></div>')
|
||
downloading_status = gr.Button(value=f"{self.refresh_symbol} Refresh Downloading Status",
|
||
elem_id="ms_dwn_status")
|
||
with gr.Row():
|
||
model_version = gr.Dropdown(choices=['Select Version'], label="Version", show_label=False,
|
||
value='Select Version', elem_id='dwn_vname',
|
||
interactive=True)
|
||
model_dropdown = gr.Dropdown(choices=['Select Model'], label="Models", show_label=False,
|
||
value='Select Model', elem_id='dwn_mname',
|
||
interactive=True)
|
||
model_des_folder = gr.Dropdown(choices=[], lable='Destination Folder', show_label=False,
|
||
value='', elem_id='dwn_des_folder', interactive=True)
|
||
|
||
is_civitai_model_source_active = self.runtime.model_source == "civitai.com"
|
||
with gr.Row(variant="panel"):
|
||
dwn_button = gr.Button(value='Download',
|
||
visible=is_civitai_model_source_active, elem_id='ms_dwn_button')
|
||
open_url_in_browser_newtab_button = gr.HTML(
|
||
value='<div class="lg secondary gradio-button svelte-cmf5ev" style="text-align: center;">'
|
||
'<a style="text-align: center;" href="http://www.liandange.com/models" '
|
||
'target="_blank">Download</a></div>',
|
||
visible=not is_civitai_model_source_active)
|
||
with gr.Row():
|
||
model_info = gr.HTML(visible=True)
|
||
|
||
rad_all_none.change(self.runtime.set_basemodel, inputs=[rad_all_none], outputs=[ckg_base_model])
|
||
rad_model_tags.change(self.runtime.search_model, inputs=[search_text, nsfw_checker, ckg_base_model, model_type, rad_model_tags], outputs=self.runtime.ds_models)
|
||
nsfw_checker.change(self.runtime.set_nsfw, inputs=[search_text, nsfw_checker, ckg_base_model, model_type, rad_model_tags],
|
||
outputs=self.runtime.ds_models)
|
||
btn_bm_apply.click(self.runtime.search_model, inputs=[search_text, nsfw_checker, ckg_base_model, model_type, rad_model_tags], outputs=self.runtime.ds_models)
|
||
model_type.change(self.runtime.search_model, inputs=[search_text, nsfw_checker, ckg_base_model, model_type, rad_model_tags], outputs=self.runtime.ds_models)
|
||
|
||
#btn_fetch.click(self.runtime.refresh_all_models, inputs=[], outputs=self.runtime.ds_models)
|
||
rad_sort.change(self.runtime.sort_dataset, inputs=[search_text, nsfw_checker, ckg_base_model, model_type, rad_model_tags, rad_sort], outputs=self.runtime.ds_models)
|
||
btn_search.click(self.runtime.search_model, inputs=[search_text, nsfw_checker, ckg_base_model, model_type, rad_model_tags], outputs=self.runtime.ds_models)
|
||
|
||
self.runtime.ds_models.click(self.runtime.get_model_info,
|
||
inputs=[self.runtime.ds_models],
|
||
outputs=[
|
||
self.runtime.ds_cover_gallery,
|
||
model_version,
|
||
model_dropdown,
|
||
model_info,
|
||
open_url_in_browser_newtab_button,
|
||
model_des_folder
|
||
])
|
||
|
||
model_version.change(self.runtime.select_version, inputs=[self.runtime.ds_models, model_version], outputs=[model_dropdown, self.runtime.ds_cover_gallery, open_url_in_browser_newtab_button])
|
||
dwn_button.click(self.runtime.download_model, inputs=[model_dropdown, model_des_folder], outputs=[download_summary])
|
||
downloading_status.click(self.runtime.get_downloading_status, inputs=[], outputs=[download_summary])
|
||
|
||
model_source_dropdown.change(self.switch_model_source,
|
||
inputs=[model_source_dropdown, search_text, nsfw_checker, ckg_base_model, model_type, rad_model_tags, rad_sort],
|
||
outputs=[self.runtime.ds_models, dwn_button, open_url_in_browser_newtab_button])
|
||
|
||
def tab_downloads_select():
|
||
self.runtime.active_model_set = 'model_set'
|
||
|
||
tab_downloads.select(tab_downloads_select, inputs=[], outputs=[])
|
||
|
||
def create_subtab_update(self) -> None:
|
||
with gr.TabItem('Setting & Update', elem_id="about_update") as tab_update:
|
||
with gr.Row():
|
||
md_api_res = gr.Markdown(visible=False)
|
||
with gr.Row():
|
||
if self.prelude.boot_settings['openai_api'] == '':
|
||
display_text = 'Enter you OpenAI API Key here, you can get it from https://platform.openai.com/account/api-keys'
|
||
else:
|
||
display_text = self.prelude.boot_settings['openai_api']
|
||
txt_gptapi = gr.Textbox(label='OpenAI API Key', value=display_text)
|
||
with gr.Row():
|
||
btn_update_gptapi = gr.Button(value="Update API Key")
|
||
with gr.Row():
|
||
txt_update_result = gr.Markdown(visible=False)
|
||
with gr.Row():
|
||
btn_check_update = gr.Button(value="Check for Update")
|
||
with gr.Row():
|
||
chk_dont_update_ms = gr.Checkbox(visible=False, label="Do not update model source", value=False)
|
||
btn_update = gr.Button(visible=False, value="Update Miaoshouai Assistant")
|
||
with gr.Row():
|
||
gr.Markdown(value="About")
|
||
with gr.Row():
|
||
gr.HTML(
|
||
f"""
|
||
<div><p>
|
||
This extension is created to improve some of the use experience for automatic1111 webui.</br>
|
||
It is free of charge, use it as you wish, please DO NOT sell this extension.</br>
|
||
Follow us on github, discord and give us suggestions, report bugs. support us with love or coffee~</br></br>
|
||
Cheers~</p>
|
||
<p style="text-align: left;">
|
||
<a target="_blank" href="https://github.com/miaoshouai/miaoshouai-assistant"><img src="https://img.shields.io/github/followers/miaoshouai?style=social" style="display: inline;" alt="MiaoshouAI GitHub"/></a>
|
||
<a href="https://discord.gg/S22Jgn3rtz"><img src="https://img.shields.io/discord/1086407792451129374?label=Discord" style="display: inline;" alt="Discord server"></a>
|
||
<a target="_blank" href="https://jq.qq.com/?_wv=1027&k=p5ZhOHAh">【QQ群：256734228】</a>
|
||
</p>
|
||
|
||
"""
|
||
)
|
||
|
||
btn_check_update.click(self.runtime.check_update, inputs=[], outputs=[txt_update_result, chk_dont_update_ms, btn_update])
|
||
btn_update_gptapi.click(self.runtime.update_gptapi, inputs=[txt_gptapi], outputs=[md_api_res, self.txt_main_prompt])
|
||
btn_update.click(self.runtime.update_program, inputs=[chk_dont_update_ms], outputs=[txt_update_result])
|
||
|
||
def save_cmdline_args(self, drp_gpu, drp_theme, txt_listen_port, chk_group_args, additional_args):
|
||
#print(drp_gpu, drp_theme, txt_listen_port, chk_group_args, additional_args)
|
||
self.runtime.get_final_args(drp_gpu, drp_theme, txt_listen_port, chk_group_args, additional_args)
|
||
#print('request_restart: cmd_arg = ', self.runtime.cmdline_args)
|
||
#print('request_restart: sys.argv = ', sys.argv)
|
||
|
||
# reset args
|
||
sys.argv = [sys.argv[0]]
|
||
os.environ['COMMANDLINE_ARGS'] = ""
|
||
#print('remove', sys.argv)
|
||
|
||
for arg in list(dict.fromkeys(self.runtime.cmdline_args)):
|
||
sys.argv.append(arg)
|
||
|
||
print('saved args', sys.argv)
|
||
#launch.start()
|
||
return gr.Markdown.update(value="Settings Saved", visible=True)
|
||
|
||
def switch_model_source(self, new_model_source: str, search, chk_nsfw, base_model, model_type, model_tag, sort_by):
|
||
def sort_key(item, key):
|
||
if key in item['stats']:
|
||
return item['stats'][key]
|
||
|
||
self.runtime.model_source = new_model_source
|
||
show_download_button = self.runtime.model_source != "liandange.com"
|
||
|
||
if sort_by != 'Default':
|
||
self.runtime.sorted_model_set = sorted(self.runtime.model_set, key=lambda item: sort_key(item, sort_by), reverse=True)
|
||
else:
|
||
self.runtime.sorted_model_set = self.runtime.model_set
|
||
|
||
print(len(self.runtime.sorted_model_set))
|
||
if self.runtime.ds_models:
|
||
self.runtime.ds_models.samples = self.runtime.sorted_model_set
|
||
self.runtime.ds_models.update(samples=self.runtime.sorted_model_set)
|
||
else:
|
||
self.logger.error(f"ds models is null")
|
||
|
||
images = self.runtime.get_images_html(search, chk_nsfw, base_model, model_type, model_tag)
|
||
#images = self.runtime.sort_dataset(search, chk_nsfw, base_model, model_type, model_tag)
|
||
|
||
self.runtime.ds_models.samples = images
|
||
self.runtime.ds_models.update(samples=self.runtime.sorted_model_set)
|
||
|
||
if self.runtime.model_source not in ['official_models', 'hugging_face', 'controlnet']:
|
||
self.runtime.update_boot_setting('model_source', self.runtime.model_source)
|
||
|
||
return (
|
||
gr.Dataset.update(samples=images),
|
||
gr.Button.update(visible=show_download_button),
|
||
gr.HTML.update(visible=not show_download_button)
|
||
)
|
||
|
||
def switch_my_model_source(self, new_model_source: str, model_type):
|
||
self.runtime.my_model_source = new_model_source
|
||
my_models = self.runtime.get_local_models('', model_type)
|
||
self.runtime.ds_my_models.samples = my_models
|
||
|
||
if self.runtime.my_model_source not in ['official_models', 'hugging_face', 'controlnet']:
|
||
self.runtime.update_boot_setting('my_model_source', self.runtime.my_model_source)
|
||
|
||
return gr.Dataset.update(samples=my_models)
|
||
|
||
def release_mem(self) -> None:
|
||
self.runtime.mem_release()
|
||
|
||
def introception(self) -> None:
|
||
self.runtime.introception()
|
||
|
||
|