mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-15 05:57:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix extension widgets not working.

Browse Source
This commit is contained in:
comfyanonymous
2023-12-01 04:13:04 -05:00
parent 5d5c320054
commit ec7a00aa96
3 changed files with 17 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
web/scripts/app.js
View File

@@ -1,5 +1,5 @@
import { ComfyLogging } from "./logging.js";
import { ComfyWidgets, getWidgetType } from "./widgets.js";
import { ComfyWidgets } from "./widgets.js";
import { ComfyUI, $el } from "./ui.js";
import { api } from "./api.js";
import { defaultGraph } from "./defaultGraph.js";
@@ -1377,6 +1377,20 @@ export class ComfyApp {
await this.#invokeExtensionsAsync("registerCustomNodes");
}
getWidgetType(inputData, inputName) {
const type = inputData[0];
if (Array.isArray(type)) {
return "COMBO";
} else if (`${type}:${inputName}` in this.widgets) {
return `${type}:${inputName}`;
} else if (type in this.widgets) {
return type;
} else {
return null;
}
}
async registerNodeDef(nodeId, nodeData) {
const self = this;
const node = Object.assign(
@@ -1391,7 +1405,7 @@ export class ComfyApp {
const type = inputData[0];
let widgetCreated = true;
const widgetType = getWidgetType(inputData, inputName);
const widgetType = self.getWidgetType(inputData, inputName);
if(widgetType) {
if(widgetType === "COMBO") {
Object.assign(config, self.widgets.COMBO(this, inputName, inputData, app) || {});