1580158
/
Stable-Diffusion-Webui-Civitai-Helper
mirror of https://github.com/butaixianran/Stable-Diffusion-Webui-Civitai-Helper.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update button color

pull/39/head
butaixianran 2023-03-10 17:05:07 +08:00
parent 0a08fb5b6a
commit eacdbcd897
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/civitai_helper.py
View File

@ -84,14 +84,15 @@ def on_ui_tabs():
model_name_drop = gr.Dropdown(choices=no_info_model_names, label="Model", value="ckp", multiselect=False) model_name_drop = gr.Dropdown(choices=no_info_model_names, label="Model", value="ckp", multiselect=False)
model_url_or_id = gr.Textbox(label="Civitai URL or Model ID", lines=1, value="") model_url_or_id = gr.Textbox(label="Civitai URL or Model ID", lines=1, value="")
get_civitai_model_info_by_id_btn = gr.Button(value="Get 1 Model Info from Civitai") get_civitai_model_info_by_id_btn = gr.Button(value="Get 1 Model Info from Civitai", variant="primary")
get_model_by_id_log_md = gr.Markdown("") get_model_by_id_log_md = gr.Markdown("")
with gr.Box(): with gr.Box():
with gr.Column(): with gr.Column():
gr.Markdown("### Check models' new version") gr.Markdown("### Check models' new version")
with gr.Row():
model_types_ckbg = gr.CheckboxGroup(choices=model_types, label="Model Types", value=["lora"]) model_types_ckbg = gr.CheckboxGroup(choices=model_types, label="Model Types", value=["lora"])
check_models_new_version_btn = gr.Button(value="Check New Version from Civitai") check_models_new_version_btn = gr.Button(value="Check New Version from Civitai", variant="primary")
check_models_new_version_log_md = gr.Markdown("It takes time, just wait. Check console log for detail") check_models_new_version_log_md = gr.Markdown("It takes time, just wait. Check console log for detail")
@ -101,7 +102,7 @@ def on_ui_tabs():
with gr.Row(): with gr.Row():
open_url_with_js_ckb = gr.Checkbox(label="Open Url At Client Side", value=open_url_with_js, elem_id="ch_open_url_with_js_ckb") open_url_with_js_ckb = gr.Checkbox(label="Open Url At Client Side", value=open_url_with_js, elem_id="ch_open_url_with_js_ckb")
save_setting_btn = gr.Button(value="Save Setting", variant="primary", elem_id="ch_save_setting_btn") save_setting_btn = gr.Button(value="Save Setting", elem_id="ch_save_setting_btn")
general_log_md = gr.Markdown(value="", elem_id="ch_general_log_md") general_log_md = gr.Markdown(value="", elem_id="ch_general_log_md")
# hidden component for js, not in any tab # hidden component for js, not in any tab

3
scripts/lib/civitai.py
View File

@ -503,6 +503,7 @@ def check_models_new_version_by_model_types(model_types:list, delay:float=0.5) -
if ext in model.exts: if ext in model.exts:
# find a model # find a model
r = check_model_new_version_by_path(item) r = check_model_new_version_by_path(item)
# delay before next request, to prevent to be treat as DDoS # delay before next request, to prevent to be treat as DDoS
util.printD(f"delay:{delay} second") util.printD(f"delay:{delay} second")
time.sleep(delay) time.sleep(delay)
@ -514,4 +515,6 @@ def check_models_new_version_by_model_types(model_types:list, delay:float=0.5) -
new_versions.append(r) new_versions.append(r)
return new_versions return new_versions