Added support for watermark, and regionalPrompt

2026-03-12 14:21:38 -04:00
parent c17d46d70b
commit aa5f547b99
13 changed files with 1219 additions and 616 deletions
@@ -3871,6 +3871,137 @@ function imageInferenceAdvancedFeaturesToggleHandler(advancedFeaturesNode) {
    if (useHiresFixWidget && hiresFixWidget) toggleWidgetState(useHiresFixWidget, hiresFixWidget, "hiresFix");
 }

+function watermarkAdvancedFeatureToggleHandler(watermarkNode) {
+    if (!watermarkNode?.widgets) return;
+
+    function toggleWidgetState(useWidget, paramWidget, paramName) {
+        if (!useWidget || !paramWidget) return;
+
+        function toggleEnabled() {
+            const enabled = useWidget.value === true;
+
+            if (paramWidget.inputEl) {
+                paramWidget.inputEl.disabled = !enabled;
+                paramWidget.inputEl.style.opacity = enabled ? "1" : "0.5";
+                paramWidget.inputEl.style.cursor = enabled ? "text" : "not-allowed";
+                paramWidget.inputEl.readOnly = !enabled;
+            }
+
+            paramWidget.disabled = !enabled;
+
+            if (!paramWidget.inputEl && paramName) {
+                const nodeElement = watermarkNode.htmlElements?.widgetsContainer || watermarkNode.htmlElements;
+                if (nodeElement) {
+                    const input = nodeElement.querySelector(`input[name="${paramName}"], textarea[name="${paramName}"], select[name="${paramName}"]`);
+                    if (input) {
+                        input.disabled = !enabled;
+                        input.style.opacity = enabled ? "1" : "0.5";
+                        input.style.cursor = enabled ? "text" : "not-allowed";
+                        input.readOnly = !enabled;
+                    }
+                }
+            }
+
+            watermarkNode.setDirtyCanvas(true);
+        }
+
+        appendWidgetCB(useWidget, () => {
+            setTimeout(toggleEnabled, 50);
+        });
+
+        setTimeout(toggleEnabled, 100);
+    }
+
+    const pairs = [
+        ["useText", "text"],
+        ["useImage", "image"],
+        ["useDisplayPosition", "displayPosition"],
+        ["useTiled", "tiled"],
+        ["useOpacity", "opacity"],
+        ["useFontColor", "fontColor"],
+        ["useBgColor", "bgColor"],
+    ];
+
+    for (const [useName, paramName] of pairs) {
+        const useWidget = watermarkNode.widgets.find(w => w.name === useName);
+        const paramWidget = watermarkNode.widgets.find(w => w.name === paramName);
+        if (useWidget && paramWidget) {
+            toggleWidgetState(useWidget, paramWidget, paramName);
+        }
+    }
+}
+
+function regionalPromptingRegionsToggleHandler(regionsNode) {
+    if (!regionsNode?.widgets) return;
+
+    function toggleWidgetState(useWidget, paramWidgets, paramNames) {
+        if (!useWidget || !paramWidgets || paramWidgets.length === 0) return;
+
+        function toggleEnabled() {
+            const enabled = useWidget.value === true;
+
+            for (let i = 0; i < paramWidgets.length; i++) {
+                const paramWidget = paramWidgets[i];
+                const paramName = paramNames[i];
+
+                if (!paramWidget) continue;
+
+                if (paramWidget.inputEl) {
+                    paramWidget.inputEl.disabled = !enabled;
+                    paramWidget.inputEl.style.opacity = enabled ? "1" : "0.5";
+                    paramWidget.inputEl.style.cursor = enabled ? "text" : "not-allowed";
+                    paramWidget.inputEl.readOnly = !enabled;
+                }
+
+                paramWidget.disabled = !enabled;
+
+                if (!paramWidget.inputEl && paramName) {
+                    const nodeElement = regionsNode.htmlElements?.widgetsContainer || regionsNode.htmlElements;
+                    if (nodeElement) {
+                        const input = nodeElement.querySelector(`input[name="${paramName}"], textarea[name="${paramName}"], select[name="${paramName}"]`);
+                        if (input) {
+                            input.disabled = !enabled;
+                            input.style.opacity = enabled ? "1" : "0.5";
+                            input.style.cursor = enabled ? "text" : "not-allowed";
+                            input.readOnly = !enabled;
+                        }
+                    }
+                }
+            }
+
+            regionsNode.setDirtyCanvas(true);
+        }
+
+        appendWidgetCB(useWidget, () => {
+            setTimeout(toggleEnabled, 50);
+        });
+
+        setTimeout(toggleEnabled, 100);
+    }
+
+    for (let idx = 1; idx <= 4; idx++) {
+        const useWidget = regionsNode.widgets.find(w => w.name === `useRegion${idx}`);
+        const promptWidget = regionsNode.widgets.find(w => w.name === `region${idx}Prompt`);
+        const x0Widget = regionsNode.widgets.find(w => w.name === `region${idx}_x0`);
+        const y0Widget = regionsNode.widgets.find(w => w.name === `region${idx}_y0`);
+        const x1Widget = regionsNode.widgets.find(w => w.name === `region${idx}_x1`);
+        const y1Widget = regionsNode.widgets.find(w => w.name === `region${idx}_y1`);
+
+        const paramWidgets = [promptWidget, x0Widget, y0Widget, x1Widget, y1Widget];
+        const paramNames = [
+            `region${idx}Prompt`,
+            `region${idx}_x0`,
+            `region${idx}_y0`,
+            `region${idx}_x1`,
+            `region${idx}_y1`,
+        ];
+
+        if (useWidget) {
+            toggleWidgetState(useWidget, paramWidgets, paramNames);
+        }
+    }
+}
+
 function audioInferenceInputsToggleHandler(audioInputsNode) {
    // Find widgets
    const useVideoWidget = audioInputsNode.widgets.find(w => w.name === "useVideo");
@@ -3990,6 +4121,8 @@ export {
    useParameterToggleHandler,
    imageInferenceToggleHandler,
    imageInferenceAdvancedFeaturesToggleHandler,
+    watermarkAdvancedFeatureToggleHandler,
+  regionalPromptingRegionsToggleHandler,
    upscalerToggleHandler,
    videoUpscalerToggleHandler,
    audioInferenceToggleHandler,