add scripts paths inside ui-config

Author: light-and-ray

README.md

@@ -88,4 +88,3 @@ When you press the "Apply hires fix" button, the extension regenerates the image
 
 ### small todo:
 - add hires fix args into metadata
-- add scripts paths inside ui-config

replacer/ui/make_advanced_options.py

@@ -6,7 +6,7 @@
     getAvoidancePromptExamplesNumber, getMaskColorStr
 )
 from replacer import replacer_scripts
-from replacer.ui.tools_ui import IS_WEBUI_1_9, AttrDict, IS_WEBUI_1_5
+from replacer.ui.tools_ui import IS_WEBUI_1_9, AttrDict, IS_WEBUI_1_5, setCustomScriptSourceForComponents
 
 
 
@@ -179,6 +179,7 @@ def makeAdvancedOptions(comp: AttrDict, isDedicatedPage: bool):
                         elem_id="replacer_mask_mode")
 
                 comp.soft_inpaint_inputs = []
+                setCustomScriptSourceForComponents("soft_inpainting")
                 if replacer_scripts.script_soft_inpaint:
                     try:
                         with gr.Row():
@@ -195,6 +196,7 @@ def makeAdvancedOptions(comp: AttrDict, isDedicatedPage: bool):
                     except Exception as e:
                         errors.report(f"Cannot add soft inpaint accordion {e}", exc_info=True)
                         replacer_scripts.script_soft_inpaint = None
+                setCustomScriptSourceForComponents(None)
 
 
             with gr.Tab('Avoidance'):
@@ -266,6 +268,7 @@ def makeAdvancedOptions(comp: AttrDict, isDedicatedPage: bool):
                     if IS_WEBUI_1_5:
                         comp.custom_mask_brush_color.visible = False
 
+            setCustomScriptSourceForComponents("inpaint_diff")
             with (gr.Tab('Inpaint Diff') if replacer_scripts.InpaintDifferenceGlobals
                     else gr.Group()) as comp.inpaint_diff_tab:
                 with gr.Row():
@@ -298,4 +301,5 @@ def makeAdvancedOptions(comp: AttrDict, isDedicatedPage: bool):
             if not replacer_scripts.InpaintDifferenceGlobals:
                 comp.inpaint_diff_tab.visible = False
                 comp.inpaint_diff_tab.render = False
+            setCustomScriptSourceForComponents(None)
 

replacer/ui/replacer_tab_ui.py

@@ -13,7 +13,8 @@
 from replacer.ui.make_advanced_options import makeAdvancedOptions
 from replacer.ui.make_hiresfix_options import makeHiresFixOptions
 from replacer.ui.tools_ui import ( update_mask_brush_color, get_current_image, unloadModels, AttrDict,
-    getSubmitJsFunction, sendBackToReplacer, IS_WEBUI_1_8, OuputPanelWatcher, ui_toprow
+    getSubmitJsFunction, sendBackToReplacer, IS_WEBUI_1_8, OuputPanelWatcher, ui_toprow,
+    setCustomScriptSourceForComponents,
 )
 
 
@@ -172,6 +173,7 @@ def init_tab(self, isDedicatedPage: bool):
                                 "extensions to increase consistency, if it fits to your scenario")
 
                     comp.cn_inputs = []
+                    setCustomScriptSourceForComponents("controlnet")
                     if replacer_scripts.script_controlnet:
                         try:
                             with gr.Row():
@@ -189,6 +191,7 @@ def init_tab(self, isDedicatedPage: bool):
                         except Exception as e:
                             errors.report(f"Cannot add controlnet accordion {e}", exc_info=True)
                             replacer_scripts.script_controlnet = None
+                    setCustomScriptSourceForComponents(None)
 
 
                 with gr.Column(scale=15):

replacer/ui/tools_ui.py

@@ -134,3 +134,16 @@ def prepareExpectedUIBehavior(gArgs: GenerationArgs):
 
     gArgs.detectionPrompt = gArgs.detectionPrompt.strip()
     gArgs.avoidancePrompt = gArgs.avoidancePrompt.strip()
+
+
+custom_script_source = None
+
+def setCustomScriptSourceForComponents(text: str):
+    global custom_script_source
+    custom_script_source = text
+
+
+def watchSetCustomScriptSourceForComponents(component, **kwargs):
+    global custom_script_source
+    if custom_script_source is not None:
+        component.custom_script_source = custom_script_source

scripts/replacer_main_ui.py

@@ -7,7 +7,7 @@
 from replacer.ui.tools_ui import IS_WEBUI_1_5
 from replacer.ui import replacer_tab_ui
 from replacer.tools import getReplacerFooter
-from replacer.ui.tools_ui import watchOuputPanel
+from replacer.ui.tools_ui import watchOuputPanel, watchSetCustomScriptSourceForComponents
 from replacer import replacer_scripts
 
 
@@ -67,6 +67,7 @@ def hideSegmentAnythingAccordions(component, **kwargs):
 script_callbacks.on_after_component(replacer_scripts.watchControlNetUI)
 script_callbacks.on_after_component(replacer_scripts.watchSoftInpaintUI)
 script_callbacks.on_after_component(watchOuputPanel)
+script_callbacks.on_after_component(watchSetCustomScriptSourceForComponents)
 script_callbacks.on_ui_settings(on_ui_settings)
 script_callbacks.on_after_component(replacer_scripts.addButtonIntoComparisonTab)
 script_callbacks.on_after_component(replacer_scripts.watchImageComparison)