mirror of
https://github.com/Bing-su/adetailer.git
synced 2026-01-26 11:19:53 +00:00
feat(ui): use InputAccodian
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
|
||||
20
aaaaaa/ui.py
20
aaaaaa/ui.py
@@ -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,
|
||||
|
||||
@@ -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",
|
||||
|
||||
@@ -1 +1 @@
|
||||
__version__ = "24.5.2-dev.0"
|
||||
__version__ = "24.6.0-dev.0"
|
||||
|
||||
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user