|
|
from modules.script_callbacks import on_ui_settings, on_before_ui |
|
|
from modules.shared import opts, OptionInfo |
|
|
|
|
|
|
|
|
def add_ui_settings(): |
|
|
from gradio import Slider |
|
|
|
|
|
args = {"section": ("ui_tabs_ex", "Tabs Extension"), "category_id": "ui"} |
|
|
|
|
|
opts.add_option( |
|
|
"tabs_ex_delay", |
|
|
OptionInfo( |
|
|
50, |
|
|
"Delay (ms) before moving the Extensions", |
|
|
Slider, |
|
|
{"minimum": 0, "maximum": 500, "step": 25}, |
|
|
**args, |
|
|
).needs_reload_ui(), |
|
|
) |
|
|
|
|
|
opts.add_option( |
|
|
"tabs_ex_act_color", |
|
|
OptionInfo( |
|
|
"greenyellow", |
|
|
"Color for active Extensions", |
|
|
**args, |
|
|
) |
|
|
.link("CSS", "https://www.w3schools.com/cssref/css_colors.php") |
|
|
.needs_reload_ui(), |
|
|
) |
|
|
|
|
|
opts.add_option( |
|
|
"tabs_ex_version", |
|
|
OptionInfo( |
|
|
False, |
|
|
"Hide the version number", |
|
|
**args, |
|
|
).needs_reload_ui(), |
|
|
) |
|
|
|
|
|
opts.add_option( |
|
|
"tabs_ex_forge", |
|
|
OptionInfo( |
|
|
False, |
|
|
'Hide the "Integrated" text', |
|
|
**args, |
|
|
) |
|
|
.info('for <a href="https://github.com/lllyasviel/stable-diffusion-webui-forge">Forge</a>') |
|
|
.needs_reload_ui(), |
|
|
) |
|
|
|
|
|
opts.add_option( |
|
|
"tabs_ex_sort", |
|
|
OptionInfo( |
|
|
False, |
|
|
"Sort Extensions based on Configs", |
|
|
**args, |
|
|
).needs_reload_ui(), |
|
|
) |
|
|
|
|
|
opts.add_option( |
|
|
"tabs_ex_toggle", |
|
|
OptionInfo( |
|
|
False, |
|
|
"Allow hiding the extension content when clicking on the same tab button again", |
|
|
**args, |
|
|
).needs_reload_ui(), |
|
|
) |
|
|
|
|
|
opts.add_option( |
|
|
"tabs_ex_scripts_toggle", |
|
|
OptionInfo( |
|
|
False, |
|
|
"Allow Ctrl + Click to toggle the Scripts dropdown as well", |
|
|
**args, |
|
|
).needs_reload_ui(), |
|
|
) |
|
|
|
|
|
opts.add_option( |
|
|
"tabs_ex_rmb", |
|
|
OptionInfo( |
|
|
False, |
|
|
"Use Right Click instead of Ctrl + Click for toggling Extensions", |
|
|
**args, |
|
|
).needs_reload_ui(), |
|
|
) |
|
|
|
|
|
opts.add_option( |
|
|
"tabs_ex_open", |
|
|
OptionInfo( |
|
|
True, |
|
|
"Automatically show the first extension tab on startup", |
|
|
**args, |
|
|
).needs_reload_ui(), |
|
|
) |
|
|
|
|
|
opts.add_option( |
|
|
"tabs_ex_container", |
|
|
OptionInfo( |
|
|
False, |
|
|
"Hide the Extension container", |
|
|
**args, |
|
|
) |
|
|
.info("In certain configurations, the original Extension container will show up as an empty space in the Webui. You can enable this to hide the container") |
|
|
.needs_reload_ui(), |
|
|
) |
|
|
|
|
|
|
|
|
def load_ui_settings(): |
|
|
import os |
|
|
|
|
|
color = getattr(opts, "tabs_ex_act_color", "greenyellow").strip().lower() |
|
|
|
|
|
css_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "style.css") |
|
|
if not os.path.isfile(css_path): |
|
|
|
|
|
with open(css_path, "w", encoding="utf-8") as f: |
|
|
f.write("body {\n --tabs-highlight-color: %s;\n}\n" % color) |
|
|
return |
|
|
|
|
|
with open(css_path, "r", encoding="utf-8", errors="ignore") as f: |
|
|
lines = f.readlines() |
|
|
|
|
|
key = "--tabs-highlight-color:" |
|
|
idx = next((i for i, line in enumerate(lines) if key in line), None) |
|
|
|
|
|
if idx is not None: |
|
|
prefix = lines[idx].split(":")[0] |
|
|
lines[idx] = f"{prefix}: {color};\n" |
|
|
else: |
|
|
|
|
|
insert_at = len(lines) |
|
|
for i, line in enumerate(lines): |
|
|
if line.strip().startswith("body"): |
|
|
insert_at = i + 1 |
|
|
break |
|
|
lines.insert(insert_at, f" --tabs-highlight-color: {color};\n") |
|
|
|
|
|
with open(css_path, "w", encoding="utf-8") as f: |
|
|
f.writelines(lines) |
|
|
|
|
|
|
