Merge pull request #47 from KarryCharon/fix-#46

fix #46
This commit is contained in:
VALADI K JAGANATHAN
2023-12-08 17:54:35 +05:30
committed by GitHub
3 changed files with 6 additions and 6 deletions

View File

@@ -77,9 +77,9 @@ const ext = {
name: "efficiency.appearance",
nodeCreated(node) {
const title = node.getTitle();
if (NODE_COLORS.hasOwnProperty(title)) {
let colorKey = NODE_COLORS[title];
const nclass = node.comfyClass;
if (NODE_COLORS.hasOwnProperty(nclass)) {
let colorKey = NODE_COLORS[nclass];
if (colorKey === "random") {
// Check for a valid color key before popping

View File

@@ -18,7 +18,7 @@ const NODE_WIDGET_MAP = {
const SPECIFIC_WIDTH = 325; // Set to desired width
function setNodeWidthForMappedTitles(node) {
if (NODE_WIDGET_MAP[node.getTitle()]) {
if (NODE_WIDGET_MAP[node.comfyClass]) {
node.setSize([SPECIFIC_WIDTH, node.size[1]]);
}
}

View File

@@ -123,7 +123,7 @@ function handleInputModeWidgetsVisibility(node, inputModeValue) {
}
};
const inputModeVisibilityMap = nodeVisibilityMap[node.getTitle()];
const inputModeVisibilityMap = nodeVisibilityMap[node.comfyClass];
if (!inputModeVisibilityMap || !inputModeVisibilityMap[inputModeValue]) return;
@@ -319,7 +319,7 @@ const nodeWidgetHandlers = {
// In the main function where widgetLogic is called
function widgetLogic(node, widget) {
// Retrieve the handler for the current node title and widget name
const handler = nodeWidgetHandlers[node.getTitle()]?.[widget.name];
const handler = nodeWidgetHandlers[node.comfyClass]?.[widget.name];
if (handler) {
handler(node, widget);
}