Added support for promptEnhance

2026-06-04 23:44:20 +05:30
parent 527d07e69b
commit 282daf4149
8 changed files with 143 additions and 13 deletions
@@ -5259,6 +5259,7 @@ export {
    imageInferenceSettingsColorPaletteToggleHandler,
    imageInferenceSettingsMoodboardsToggleHandler,
    imageInferenceSettingsStructuredPromptToggleHandler,
+    imageInferenceSettingsPromptEnhanceToggleHandler,
    audioInputToggleHandler,
    speechInputToggleHandler,
    briaProviderMaskToggleHandler,
@@ -5659,6 +5660,32 @@ function imageInferenceSettingsColorPaletteToggleHandler(paletteNode) {
    }
 }

+function imageInferenceSettingsPromptEnhanceToggleHandler(node) {
+    if (!node?.widgets) return;
+
+    const useTemperatureWidget = node.widgets.find((w) => w && w.name === "useTemperature");
+    const temperatureWidget = node.widgets.find((w) => w && w.name === "temperature");
+    const useTopPWidget = node.widgets.find((w) => w && w.name === "useTopP");
+    const topPWidget = node.widgets.find((w) => w && w.name === "topP");
+
+    function toggleWidgetState(useWidget, paramWidget) {
+        if (!useWidget || !paramWidget) return;
+        function toggleEnabled() {
+            toggleWidgetEnabled(paramWidget, useWidget.value === true, node);
+            node.setDirtyCanvas(true);
+        }
+        appendWidgetCB(useWidget, () => setTimeout(toggleEnabled, 50));
+        setTimeout(toggleEnabled, 100);
+    }
+
+    if (useTemperatureWidget && temperatureWidget) {
+        toggleWidgetState(useTemperatureWidget, temperatureWidget);
+    }
+    if (useTopPWidget && topPWidget) {
+        toggleWidgetState(useTopPWidget, topPWidget);
+    }
+}
+
 function imageInferenceSettingsMoodboardsToggleHandler(moodboardsNode) {
    if (!moodboardsNode?.widgets) return;