nodes/JeyzerMC-comfyui-flakes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: Hide Model Family dropdown when flake_data input is connected (#204)

Browse Source
This commit is contained in:
Mehdi Chaid
2026-05-20 10:09:25 -04:00
parent 64ebdc6a81
commit 36620bdebf
2 changed files with 128 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/widgets/flake-combo.js
+64
View File
@@ -287,6 +287,64 @@ export function setupFlakeComboWidget(node) {
return familyWidget?.value || "SDXL/Base";
}
// Hide model_family widget when flake_data input is connected, and infer
// the family value from the upstream node.
let _familyWidgetVisible = true;
function _getFamilyWidgetEl() {
if (!familyWidget) return null;
if (familyWidget.element) return familyWidget.element;
if (familyWidget.inputEl) return familyWidget.inputEl;
const nodeEl = node.htmlEl || node.element;
if (!nodeEl) return null;
const allEls = nodeEl.querySelectorAll("label, span, div");
for (const el of allEls) {
const text = el.textContent || "";
if (text.includes("model_family") || text.includes("Model Family") || text.includes("model family")) {
const container = el.closest(".comfy-widget, .litegraph-widget, [class*='widget']");
return container || el;
}
}
return null;
}
function _updateFamilyVisibility() {
if (!familyWidget) return;
const flakeDataInput = node.inputs.find(i => i.name === "flake_data");
const isConnected = flakeDataInput && flakeDataInput.link != null;
if (isConnected) {
const link = node.graph.links[flakeDataInput.link];
if (link) {
const upstreamNode = node.graph.getNodeById(link.origin_id);
if (upstreamNode) {
const upstreamFamily = upstreamNode.widgets?.find(w => w.name === "model_family");
if (upstreamFamily && upstreamFamily.value !== familyWidget.value) {
familyWidget.value = upstreamFamily.value;
if (familyWidget.callback) familyWidget.callback(upstreamFamily.value);
}
}
}
}
if (isConnected === _familyWidgetVisible) return;
_familyWidgetVisible = isConnected;
const el = _getFamilyWidgetEl();
if (el) {
el.style.display = isConnected ? "none" : "";
const parent = el.closest(".comfy-widget, .litegraph-widget, [class*='widget']");
if (parent && parent !== el) parent.style.display = isConnected ? "none" : "";
}
if (isConnected) {
familyWidget.computeSize = () => [0, -4];
familyWidget.type = "hidden";
familyWidget.hidden = true;
} else {
familyWidget.computeSize = undefined;
familyWidget.type = "combo";
familyWidget.hidden = false;
}
node.setSize(node.computeSize());
node.setDirtyCanvas(true, true);
}
const _familyPoll = setInterval(_updateFamilyVisibility, 200);
if (!node.properties) node.properties = {};
if (!node.properties._combo_flakes) node.properties._combo_flakes = [];
if (node.properties._combo_active_index == null) node.properties._combo_active_index = 0;
@@ -577,4 +635,10 @@ export function setupFlakeComboWidget(node) {
};
updateActiveFlake();
render();
const origOnRemoved = node.onRemoved;
node.onRemoved = function () {
clearInterval(_familyPoll);
return origOnRemoved?.apply(this, arguments);
};
}
web/widgets/flake-stack.js
+64
View File
@@ -311,6 +311,64 @@ export function setupFlakeWidget(node) {
return familyWidget?.value || "SDXL/Base";
}
// Hide model_family widget when flake_data input is connected, and infer
// the family value from the upstream node.
let _familyWidgetVisible = true;
function _getFamilyWidgetEl() {
if (!familyWidget) return null;
if (familyWidget.element) return familyWidget.element;
if (familyWidget.inputEl) return familyWidget.inputEl;
const nodeEl = node.htmlEl || node.element;
if (!nodeEl) return null;
const allEls = nodeEl.querySelectorAll("label, span, div");
for (const el of allEls) {
const text = el.textContent || "";
if (text.includes("model_family") || text.includes("Model Family") || text.includes("model family")) {
const container = el.closest(".comfy-widget, .litegraph-widget, [class*='widget']");
return container || el;
}
}
return null;
}
function _updateFamilyVisibility() {
if (!familyWidget) return;
const flakeDataInput = node.inputs.find(i => i.name === "flake_data");
const isConnected = flakeDataInput && flakeDataInput.link != null;
if (isConnected) {
const link = node.graph.links[flakeDataInput.link];
if (link) {
const upstreamNode = node.graph.getNodeById(link.origin_id);
if (upstreamNode) {
const upstreamFamily = upstreamNode.widgets?.find(w => w.name === "model_family");
if (upstreamFamily && upstreamFamily.value !== familyWidget.value) {
familyWidget.value = upstreamFamily.value;
if (familyWidget.callback) familyWidget.callback(upstreamFamily.value);
}
}
}
}
if (isConnected === _familyWidgetVisible) return;
_familyWidgetVisible = isConnected;
const el = _getFamilyWidgetEl();
if (el) {
el.style.display = isConnected ? "none" : "";
const parent = el.closest(".comfy-widget, .litegraph-widget, [class*='widget']");
if (parent && parent !== el) parent.style.display = isConnected ? "none" : "";
}
if (isConnected) {
familyWidget.computeSize = () => [0, -4];
familyWidget.type = "hidden";
familyWidget.hidden = true;
} else {
familyWidget.computeSize = undefined;
familyWidget.type = "combo";
familyWidget.hidden = false;
}
node.setSize(node.computeSize());
node.setDirtyCanvas(true, true);
}
const _familyPoll = setInterval(_updateFamilyVisibility, 200);
function readEntries() {
try {
const arr = JSON.parse(flakesHidden.value || "[]");
@@ -594,4 +652,10 @@ export function setupFlakeWidget(node) {
};
writeEntries(readEntries());
render();
const origOnRemoved = node.onRemoved;
node.onRemoved = function () {
clearInterval(_familyPoll);
return origOnRemoved?.apply(this, arguments);
};
}