|
|
import copy |
|
|
import gradio as gr |
|
|
from modules import script_callbacks, progress, shared, errors |
|
|
from replacer.options import (EXT_NAME, EXT_NAME_LOWER, needHideSegmentAnythingAccordions, |
|
|
getDedicatedPagePath, on_ui_settings, needHideAnimateDiffAccordions, hideVideoInMainUI, |
|
|
) |
|
|
from replacer.ui.tools_ui import IS_WEBUI_1_5 |
|
|
from replacer.ui import replacer_main_ui |
|
|
from replacer.tools import getReplacerFooter |
|
|
from replacer.ui.tools_ui import watchOutputPanel, watchSetCustomScriptSourceForComponents |
|
|
from replacer.extensions import replacer_extensions |
|
|
|
|
|
|
|
|
|
|
|
def on_ui_tabs(): |
|
|
result = [] |
|
|
replacer_main_ui.reinitMainUIAfterUICreated() |
|
|
tab = replacer_main_ui.replacerMainUI.getReplacerTabUI() |
|
|
result.append((tab, EXT_NAME, EXT_NAME)) |
|
|
if not hideVideoInMainUI(): |
|
|
video_tab = replacer_main_ui.replacerMainUI.getReplacerVideoTabUI() |
|
|
video_title = f"{EXT_NAME} - video" |
|
|
result.append((video_tab, video_title, video_title)) |
|
|
return result |
|
|
|
|
|
script_callbacks.on_ui_tabs(on_ui_tabs) |
|
|
|
|
|
|
|
|
def mountDedicatedPage(demo, app): |
|
|
try: |
|
|
path = getDedicatedPagePath() |
|
|
app.add_api_route(f"{path}/internal/progress", |
|
|
progress.progressapi, methods=["POST"], |
|
|
response_model=progress.ProgressResponse) |
|
|
replacer_extensions.image_comparison.preloadImageComparisonTab() |
|
|
|
|
|
with gr.Blocks(title=EXT_NAME, analytics_enabled=False) as replacerUi: |
|
|
gr.Textbox(elem_id="txt2img_prompt", visible=False) |
|
|
gr.Textbox(value=shared.opts.dumpjson(), elem_id="settings_json", visible=False) |
|
|
|
|
|
with gr.Tabs(elem_id='tabs'): |
|
|
with gr.Tab(label=f"{EXT_NAME} dedicated", elem_id=f"tab_{EXT_NAME_LOWER}_dedicated"): |
|
|
tab = replacer_main_ui.replacerMainUI_dedicated.getReplacerTabUI() |
|
|
tab.render() |
|
|
with gr.Tab(label="Video", elem_id=f"tab_video"): |
|
|
tab_video = replacer_main_ui.replacerMainUI_dedicated.getReplacerVideoTabUI() |
|
|
tab_video.render() |
|
|
replacer_extensions.image_comparison.mountImageComparisonTab() |
|
|
|
|
|
footer = getReplacerFooter() |
|
|
gr.HTML(footer, elem_id="footer") |
|
|
|
|
|
loadsave = copy.copy(demo.ui_loadsave) |
|
|
loadsave.finalized_ui = False |
|
|
video_title = f"{EXT_NAME} - video" |
|
|
loadsave.add_block(tab, EXT_NAME) |
|
|
loadsave.add_block(tab_video, video_title) |
|
|
loadsave.dump_defaults() |
|
|
replacerUi.ui_loadsave = loadsave |
|
|
gr.mount_gradio_app(app, replacerUi, path=path) |
|
|
except Exception as e: |
|
|
errors.report(f'[{EXT_NAME}] error while creating dedicated page: {e}', exc_info=True) |
|
|
|
|
|
script_callbacks.on_app_started(mountDedicatedPage) |
|
|
|
|
|
|
|
|
def hideSegmentAnythingAccordions(component, **kwargs): |
|
|
if type(component) is gr.Accordion and\ |
|
|
getattr(component, 'label', "") == "Segment Anything": |
|
|
|
|
|
component.visible = False |
|
|
print(f"[{EXT_NAME}] Segment Anything accordion has been hidden") |
|
|
|
|
|
if needHideSegmentAnythingAccordions(): |
|
|
script_callbacks.on_after_component(hideSegmentAnythingAccordions) |
|
|
|
|
|
|
|
|
def hideAnimateDiffAccordions(component, **kwargs): |
|
|
if type(component) is gr.Accordion and\ |
|
|
getattr(component, 'label', "") == "AnimateDiff": |
|
|
|
|
|
component.visible = False |
|
|
print(f"[{EXT_NAME}] AnimateDiff accordion has been hidden") |
|
|
|
|
|
if needHideAnimateDiffAccordions(): |
|
|
script_callbacks.on_after_component(hideAnimateDiffAccordions) |
|
|
|
|
|
|
|
|
script_callbacks.on_before_ui(replacer_main_ui.initMainUI) |
|
|
script_callbacks.on_after_component(replacer_extensions.controlnet.watchControlNetUI) |
|
|
script_callbacks.on_after_component(replacer_extensions.soft_inpainting.watchSoftInpaintUI) |
|
|
script_callbacks.on_after_component(watchOutputPanel) |
|
|
script_callbacks.on_after_component(watchSetCustomScriptSourceForComponents) |
|
|
script_callbacks.on_ui_settings(on_ui_settings) |
|
|
script_callbacks.on_after_component(replacer_extensions.image_comparison.addButtonIntoComparisonTab) |
|
|
script_callbacks.on_after_component(replacer_extensions.image_comparison.watchImageComparison) |
|
|
|
