From 69a31d376f8b049fc595e48437f09a0c827455f1 Mon Sep 17 00:00:00 2001
From: anapnoe <124302297+anapnoe@users.noreply.github.com>
Date: Fri, 20 Sep 2024 12:37:33 +0300
Subject: [PATCH] Update anapnoe_sd_uiux.py
---
.../scripts/anapnoe_sd_uiux.py | 169 +++++-------------
1 file changed, 48 insertions(+), 121 deletions(-)
diff --git a/extensions-builtin/anapnoe-sd-uiux/scripts/anapnoe_sd_uiux.py b/extensions-builtin/anapnoe-sd-uiux/scripts/anapnoe_sd_uiux.py
index ef869ccd..d685c7e0 100644
--- a/extensions-builtin/anapnoe-sd-uiux/scripts/anapnoe_sd_uiux.py
+++ b/extensions-builtin/anapnoe-sd-uiux/scripts/anapnoe_sd_uiux.py
@@ -1,133 +1,60 @@
-import os
-from pathlib import Path
import gradio as gr
import modules.scripts as scripts
from modules import script_callbacks, shared
-from modules.options import options_section, OptionInfo, OptionHTML, categories
+from modules.options import options_section, OptionInfo
categories.register_category("anapnoe", "Anapnoe")
+mapping = [
+ (info.infotext, k) for k, info in shared.opts.data_labels.items() if info.infotext
+]
-mapping = [(info.infotext, k) for k, info in shared.opts.data_labels.items() if info.infotext]
-#shared.options_templates.update(shared.options_section(('uiux_core', "Anapnoe UI-UX"), {
-shared.options_templates.update(options_section(('uiux_core', "Anapnoe UI-UX", "anapnoe"), {
- "uiux_enable_console_log": shared.OptionInfo(False, "Enable console log"),
- "uiux_max_resolution_output": shared.OptionInfo(2048, "Max resolution output for txt2img and img2img"),
- "uiux_show_input_range_ticks": shared.OptionInfo(True, "Show ticks for input range slider"),
- "uiux_no_slider_layout": shared.OptionInfo(False, "No input range sliders"),
- "uiux_disable_transitions": shared.OptionInfo(False, "Disable transitions"),
- "uiux_default_layout": shared.OptionInfo("Auto", "Layout", gr.Radio, {"choices": ["Auto","Desktop", "Mobile"]}),
- "uiux_mobile_scale": shared.OptionInfo(0.7, "Mobile scale", gr.Slider, {"minimum": 0.5, "maximum": 1, "step": 0.05}),
- "uiux_show_labels_aside": shared.OptionInfo(False, "Show labels for aside tabs"),
- "uiux_show_labels_main": shared.OptionInfo(False, "Show labels for main tabs"),
- "uiux_show_labels_tabs": shared.OptionInfo(False, "Show labels for page tabs"),
- "uiux_ignore_overrides": shared.OptionInfo([], "Ignore Overrides", gr.CheckboxGroup, lambda: {"choices": list(mapping)})
-}))
-
-
-basedir = scripts.basedir()
-html_folder = os.path.join(basedir, "html")
-
-layouts_folder = os.path.join(basedir, "layouts")
-javascript_folder = os.path.join(basedir, "javascript")
-
-def get_files(folder, file_filter=[], file_list=[], split=False):
- file_list = [file_name if not split else os.path.splitext(file_name)[0] for file_name in os.listdir(folder) if os.path.isfile(os.path.join(folder, file_name)) and file_name not in file_filter]
- return file_list
-
-html = """
-
-
- My First JavaScript
-
-
-
-
-"""
-
-scripts = """
-async () => {
- // set testFn() function on globalThis, so you html onlclick can access it
- globalThis.testFn = () => {
- document.getElementById('demo').innerHTML = "Hello"
- }
-}
-"""
-
-# with gr.Blocks() as demo:
- # input_mic = gr.HTML(html)
- # out_text = gr.Textbox()
- # # run script function on load,
- # demo.load(None,None,None,_js=scripts)
-# static_dir = Path('./static')
-# static_dir.mkdir(parents=True, exist_ok=True)
-
-# def predict(text_input):
- # file_name = f"{datetime.utcnow().strftime('%s')}.html"
- # file_path = html_folder / file_name
- # print(file_path)
- # with open(file_path, "w") as f:
- # f.write(f"""
- #
- #
- #
- # Hello {text_input} From Gradio Iframe
- # Filename: {file_name}
- #
- # """)
- # iframe = f""""""
- # link = f'{file_name}'
- # return link, iframe
-
-
+shared.options_templates.update(
+ options_section(
+ ("uiux_core", "Anapnoe UI-UX", "anapnoe"),
+ {
+ "uiux_enable_console_log": shared.OptionInfo(False, "Enable console log"),
+ "uiux_max_resolution_output": shared.OptionInfo(
+ 2048, "Max resolution output for txt2img and img2img"
+ ),
+ "uiux_show_input_range_ticks": shared.OptionInfo(
+ True, "Show ticks for input range slider"
+ ),
+ "uiux_no_slider_layout": shared.OptionInfo(False, "No input range sliders"),
+ "uiux_disable_transitions": shared.OptionInfo(False, "Disable transitions"),
+ "uiux_default_layout": shared.OptionInfo(
+ "Auto", "Layout", gr.Radio, {"choices": ["Auto", "Desktop", "Mobile"]}
+ ),
+ "uiux_mobile_scale": shared.OptionInfo(
+ 0.7,
+ "Mobile scale",
+ gr.Slider,
+ {"minimum": 0.5, "maximum": 1, "step": 0.05},
+ ),
+ "uiux_show_labels_aside": shared.OptionInfo(
+ False, "Show labels for aside tabs"
+ ),
+ "uiux_show_labels_main": shared.OptionInfo(
+ False, "Show labels for main tabs"
+ ),
+ "uiux_show_labels_tabs": shared.OptionInfo(
+ False, "Show labels for page tabs"
+ ),
+ "uiux_ignore_overrides": shared.OptionInfo(
+ [],
+ "Ignore Overrides",
+ gr.CheckboxGroup,
+ lambda: {"choices": list(mapping)},
+ ),
+ },
+ )
+)
def on_ui_tabs():
+ with gr.Blocks(analytics_enabled=False) as anapnoe_sd_uiux_core:
+ pass
- #print(mapping)
+ return ((anapnoe_sd_uiux_core, "UI-UX Core", "anapnoe_sd_uiux_core"),)
- with gr.Blocks(analytics_enabled=False) as anapnoe_sd_uiux_core:
- #override_settings = gr.CheckboxGroup(label="Ignore override settings", elem_id="ignore_overrides", choices=list(mapping))
- """ with gr.Row():
- with gr.Column():
- with gr.Row():
- layouts_dropdown = gr.Dropdown(label="Layout", elem_id="layout_drop_down", interactive=True, choices=get_files(layouts_folder,[".json"]), type="value")
- layout_save_as_filename = gr.Text(label="Save / Save as", elem_id="layout_save_as_name",)
- with gr.Row():
- layout_reset_button = gr.Button(elem_id="layout_reset_btn", value="Reset", variant="primary")
- layout_save_button = gr.Button(value="Save", elem_id="layout_save_btn", variant="primary")
- with gr.Row(elem_id="layout_hidden"):
- layout_json = gr.Textbox(label="Json", elem_id="layout_json", show_label=True, lines=7, interactive=False, visible=True)
+script_callbacks.on_ui_tabs(on_ui_tabs)
- def save_layout( layout_json, filename):
- with open(os.path.join(layouts_folder, f"{filename}.json"), 'w', encoding="utf-8") as file:
- file.write(layout_json)
- file.close()
- layouts_dropdown.choices=get_files(layouts_folder,[".json"])
- return gr.update(choices=layouts_dropdown.choices, value=f"{filename}.json")
-
- def open_layout(filename):
- with open(os.path.join(layouts_folder, f"{filename}"), 'r') as file:
- layout_json=file.read()
- no_ext=filename.rsplit('.', 1)[0]
- return [layout_json, no_ext]
-
-
- layout_save_button.click(
- fn=save_layout,
- inputs=[layout_json, layout_save_as_filename],
- outputs=layouts_dropdown
- )
-
- layouts_dropdown.change(
- fn=open_layout,
- inputs=[layouts_dropdown],
- outputs=[layout_json, layout_save_as_filename]
- ) """
-
-
-
- return (anapnoe_sd_uiux_core, 'UI-UX Core', 'anapnoe_sd_uiux_core'),
-
-
-
-script_callbacks.on_ui_tabs(on_ui_tabs)
\ No newline at end of file