feat(ui): use InputAccodian

This commit is contained in:
Dowon
2024-06-02 13:00:19 +09:00
parent a79a472c6b
commit a479f60f40
5 changed files with 23 additions and 18 deletions

View File

@@ -1,9 +1,16 @@
from __future__ import annotations
msg = "[-] ADetailer: WebUI versions below 1.6.0 are not supported."
try:
from modules.processing import create_binary_mask # noqa: F401
except ImportError as e:
msg = "[-] ADetailer: WebUI versions below 1.6.0 are not supported."
raise RuntimeError(msg) from e
try:
from modules.ui_components import InputAccordion # noqa: F401
except ImportError as e:
raise RuntimeError(msg) from e

View File

@@ -7,7 +7,8 @@ from typing import Any
import gradio as gr
from adetailer import AFTER_DETAILER, __version__
from aaaaaa.conditional import InputAccordion
from adetailer import ADETAILER, __version__
from adetailer.args import ALL_ARGS, MASK_MERGE_INVERT
from controlnet_ext import controlnet_exists, controlnet_type, get_cn_models
@@ -123,17 +124,14 @@ def adui(
infotext_fields = []
eid = partial(elem_id, n=0, is_img2img=is_img2img)
with gr.Accordion(AFTER_DETAILER, open=False, elem_id=eid("ad_main_accordion")):
with InputAccordion(
value=False,
elem_id=eid("ad_main_accordion"),
label=ADETAILER,
visible=True,
) as ad_enable:
with gr.Row():
with gr.Column(scale=6):
ad_enable = gr.Checkbox(
label="Enable ADetailer",
value=False,
visible=True,
elem_id=eid("ad_enable"),
)
with gr.Column(scale=6):
with gr.Column(scale=8):
ad_skip_img2img = gr.Checkbox(
label="Skip img2img",
value=False,

View File

@@ -4,12 +4,12 @@ from .common import PredictOutput, get_models
from .mediapipe import mediapipe_predict
from .ultralytics import ultralytics_predict
AFTER_DETAILER = "ADetailer"
ADETAILER = "ADetailer"
__all__ = [
"__version__",
"ADetailerArgs",
"AFTER_DETAILER",
"ADETAILER",
"ALL_ARGS",
"PredictOutput",
"get_models",

View File

@@ -1 +1 @@
__version__ = "24.5.2-dev.0"
__version__ = "24.6.0-dev.0"

View File

@@ -33,7 +33,7 @@ from aaaaaa.p_method import (
from aaaaaa.traceback import rich_traceback
from aaaaaa.ui import WebuiInfo, adui, ordinal, suffix
from adetailer import (
AFTER_DETAILER,
ADETAILER,
__version__,
get_models,
mediapipe_predict,
@@ -110,7 +110,7 @@ class AfterDetailerScript(scripts.Script):
return f"{self.__class__.__name__}(version={__version__})"
def title(self):
return AFTER_DETAILER
return ADETAILER
def show(self, is_img2img):
return scripts.AlwaysVisible
@@ -846,7 +846,7 @@ def on_after_component(component, **_kwargs):
def on_ui_settings():
section = ("ADetailer", AFTER_DETAILER)
section = ("ADetailer", ADETAILER)
shared.opts.add_option(
"ad_max_models",
shared.OptionInfo(