From eacdbcd897fe8b61b51b118d9d1d917e3055b699 Mon Sep 17 00:00:00 2001 From: butaixianran Date: Fri, 10 Mar 2023 17:05:07 +0800 Subject: [PATCH] update button color --- scripts/civitai_helper.py | 9 +++++---- scripts/lib/civitai.py | 3 +++ 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/scripts/civitai_helper.py b/scripts/civitai_helper.py index f958ba5..d6921a2 100644 --- a/scripts/civitai_helper.py +++ b/scripts/civitai_helper.py @@ -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_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("") with gr.Box(): with gr.Column(): gr.Markdown("### Check models' new version") - 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") + with gr.Row(): + 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", variant="primary") 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(): 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") # hidden component for js, not in any tab diff --git a/scripts/lib/civitai.py b/scripts/lib/civitai.py index 850e724..cb830d3 100644 --- a/scripts/lib/civitai.py +++ b/scripts/lib/civitai.py @@ -503,6 +503,7 @@ def check_models_new_version_by_model_types(model_types:list, delay:float=0.5) - if ext in model.exts: # find a model r = check_model_new_version_by_path(item) + # delay before next request, to prevent to be treat as DDoS util.printD(f"delay:{delay} second") 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) + + return new_versions