Added support for watermark, and regionalPrompt

clientlibs/main.js

@@ -1,6 +1,6 @@
 import { app } from "../../scripts/app.js";
 import { api } from "../../scripts/api.js";
-import { promptEnhanceHandler, syncDimensionsNodeHandler, searchNodeHandler, APIKeyHandler, captionNodeHandler, mediaUUIDHandler, save3DFilepathHandler, videoTranscriptionHandler, videoOutputsHandler, handleCustomErrors, videoInferenceDimensionsHandler, videoModelSearchFilterHandler, audioModelSearchFilterHandler, vectorizeModelSearchFilterHandler, vectorizeToggleHandler, useParameterToggleHandler, imageInferenceToggleHandler, imageInferenceAdvancedFeaturesToggleHandler, upscalerToggleHandler, videoUpscalerToggleHandler, audioInferenceToggleHandler, audioInferenceSpeechToggleHandler, audioSettingsToggleHandler, videoSettingsToggleHandler, acceleratorOptionsToggleHandler, bytedanceProviderSettingsToggleHandler, xaiProviderSettingsToggleHandler, viduProviderSettingsToggleHandler, sourcefulProviderSettingsToggleHandler, sourcefulProviderSettingsFontsToggleHandler, threeDInferenceToggleHandler, threeDInferenceSettingsToggleHandler, threeDInferenceSettingsLatToggleHandler, ultralyticsProviderSettingsToggleHandler, openaiProviderSettingsToggleHandler, lightricksProviderSettingsToggleHandler, klingProviderSettingsToggleHandler, lumaProviderSettingsToggleHandler, briaProviderSettingsToggleHandler, pixverseProviderSettingsToggleHandler, alibabaProviderSettingsToggleHandler, mireloProviderSettingsToggleHandler, googleProviderSettingsToggleHandler, syncProviderSettingsToggleHandler, syncSegmentToggleHandler, settingsToggleHandler, audioInputToggleHandler, speechInputToggleHandler, briaProviderMaskToggleHandler, wanAnimateAdvancedFeatureSettingsToggleHandler, videoAdvancedFeatureInputsToggleHandler, audioInferenceInputsToggleHandler } from "./utils.js";
+import { promptEnhanceHandler, syncDimensionsNodeHandler, searchNodeHandler, APIKeyHandler, captionNodeHandler, mediaUUIDHandler, save3DFilepathHandler, videoTranscriptionHandler, videoOutputsHandler, handleCustomErrors, videoInferenceDimensionsHandler, videoModelSearchFilterHandler, audioModelSearchFilterHandler, vectorizeModelSearchFilterHandler, vectorizeToggleHandler, useParameterToggleHandler, imageInferenceToggleHandler, imageInferenceAdvancedFeaturesToggleHandler, watermarkAdvancedFeatureToggleHandler, regionalPromptingRegionsToggleHandler, upscalerToggleHandler, videoUpscalerToggleHandler, audioInferenceToggleHandler, audioInferenceSpeechToggleHandler, audioSettingsToggleHandler, videoSettingsToggleHandler, acceleratorOptionsToggleHandler, bytedanceProviderSettingsToggleHandler, xaiProviderSettingsToggleHandler, viduProviderSettingsToggleHandler, sourcefulProviderSettingsToggleHandler, sourcefulProviderSettingsFontsToggleHandler, threeDInferenceToggleHandler, threeDInferenceSettingsToggleHandler, threeDInferenceSettingsLatToggleHandler, ultralyticsProviderSettingsToggleHandler, openaiProviderSettingsToggleHandler, lightricksProviderSettingsToggleHandler, klingProviderSettingsToggleHandler, lumaProviderSettingsToggleHandler, briaProviderSettingsToggleHandler, pixverseProviderSettingsToggleHandler, alibabaProviderSettingsToggleHandler, mireloProviderSettingsToggleHandler, googleProviderSettingsToggleHandler, syncProviderSettingsToggleHandler, syncSegmentToggleHandler, settingsToggleHandler, audioInputToggleHandler, speechInputToggleHandler, briaProviderMaskToggleHandler, wanAnimateAdvancedFeatureSettingsToggleHandler, videoAdvancedFeatureInputsToggleHandler, audioInferenceInputsToggleHandler } from "./utils.js";
 import { RUNWARE_NODE_TYPES, RUNWARE_NODE_PROPS, SEARCH_TERMS } from "./types.js";
 
 const nodeInitList = [];
@@ -80,6 +80,10 @@ app.registerExtension({
             audioInferenceToggleHandler(node);
         } else if(nodeClass === RUNWARE_NODE_TYPES.IMAGEINFERENCEADVANCEDFEATURES) {
             imageInferenceAdvancedFeaturesToggleHandler(node);
+        } else if(nodeClass === RUNWARE_NODE_TYPES.WATERMARKADVANCEDFEATURE) {
+            watermarkAdvancedFeatureToggleHandler(node);
+        } else if(nodeClass === RUNWARE_NODE_TYPES.REGIONALPROMPTINGADVFEATUREREGIONS) {
+            regionalPromptingRegionsToggleHandler(node);
         } else if(nodeClass === RUNWARE_NODE_TYPES.AUDIOINFERENCESPEECH) {
             audioInferenceSpeechToggleHandler(node);
         } else if(nodeClass === RUNWARE_NODE_TYPES.AUDIOSETTINGS) {

clientlibs/utils.js

@@ -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,