109 lines
5.7 KiB
JavaScript
109 lines
5.7 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
import { api } from "../../scripts/api.js";
|
|
import { promptEnhanceHandler, syncDimensionsNodeHandler, searchNodeHandler, APIKeyHandler, captionNodeHandler, mediaUUIDHandler, videoTranscriptionHandler, handleCustomErrors, videoInferenceDimensionsHandler, videoModelSearchFilterHandler, audioModelSearchFilterHandler, useParameterToggleHandler, imageInferenceToggleHandler, upscalerToggleHandler, videoUpscalerToggleHandler, audioInferenceToggleHandler, acceleratorOptionsToggleHandler, bytedanceProviderSettingsToggleHandler, openaiProviderSettingsToggleHandler, lightricksProviderSettingsToggleHandler, klingProviderSettingsToggleHandler, lumaProviderSettingsToggleHandler, briaProviderSettingsToggleHandler, pixverseProviderSettingsToggleHandler } from "./utils.js";
|
|
import { RUNWARE_NODE_TYPES, RUNWARE_NODE_PROPS, SEARCH_TERMS } from "./types.js";
|
|
|
|
const nodeInitList = [];
|
|
app.registerExtension({
|
|
name: "runware.ai",
|
|
async setup() {
|
|
api.addEventListener('runwareError', handleCustomErrors);
|
|
api.addEventListener('runwareImageCaption', captionNodeHandler);
|
|
api.addEventListener('runwareMediaUUID', mediaUUIDHandler);
|
|
api.addEventListener('runwareVideoTranscription', videoTranscriptionHandler);
|
|
},
|
|
|
|
async nodeCreated(node) {
|
|
const nodeClass = node.comfyClass;
|
|
let crNodeProps = false;
|
|
if(typeof nodeClass === "string" && Object.values(RUNWARE_NODE_TYPES).includes(nodeClass)) {
|
|
crNodeProps = RUNWARE_NODE_PROPS[nodeClass];
|
|
} else {
|
|
return;
|
|
}
|
|
|
|
node.bgcolor = crNodeProps.bgColor;
|
|
if(nodeClass === RUNWARE_NODE_TYPES.APIMANAGER) {
|
|
APIKeyHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.IMAGECAPTION) {
|
|
const captionInput = node.widgets[1].inputEl;
|
|
captionInput.style.outline = "none";
|
|
captionInput.readOnly = true;
|
|
return;
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.MEDIAUPLOAD) {
|
|
// Style the mediaUUID widget to show full UUID
|
|
const mediaUUIDWidget = node.widgets.find(widget => widget.name === "mediaUUID");
|
|
if(mediaUUIDWidget && mediaUUIDWidget.inputEl) {
|
|
mediaUUIDWidget.inputEl.style.width = "100%";
|
|
mediaUUIDWidget.inputEl.style.minWidth = "300px";
|
|
mediaUUIDWidget.inputEl.style.fontSize = "11px";
|
|
mediaUUIDWidget.inputEl.style.fontFamily = "monospace";
|
|
}
|
|
}
|
|
|
|
const nodeWidgets = node.widgets;
|
|
|
|
// Handle node-specific handlers first - each node has its own dedicated handler
|
|
// These must be called BEFORE the colorModeOnly check
|
|
if(nodeClass === RUNWARE_NODE_TYPES.IMAGEINFERENCE) {
|
|
imageInferenceToggleHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.VIDEOINFERENCE) {
|
|
videoInferenceDimensionsHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.UPSCALER) {
|
|
upscalerToggleHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.VIDEOUPSCALER) {
|
|
videoUpscalerToggleHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.AUDIOINFERENCE) {
|
|
audioInferenceToggleHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.ACCELERATOROPTIONS) {
|
|
acceleratorOptionsToggleHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.BYTEDANCEPROVIDERSETTINGS) {
|
|
bytedanceProviderSettingsToggleHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.OPENAIPROVIDERSETTINGS) {
|
|
openaiProviderSettingsToggleHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.LIGHTRICKSPROVIDERSETTINGS) {
|
|
lightricksProviderSettingsToggleHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.KLINGPROVIDERSETTINGS) {
|
|
klingProviderSettingsToggleHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.LUMAPROVIDERSETTINGS) {
|
|
lumaProviderSettingsToggleHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.BRIAPROVIDERSETTINGS) {
|
|
briaProviderSettingsToggleHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.PIXVERSEPROVIDERSETTINGS) {
|
|
pixverseProviderSettingsToggleHandler(node);
|
|
}
|
|
|
|
if(crNodeProps.colorModeOnly === true) return;
|
|
if(nodeWidgets.length <= 0) return;
|
|
|
|
if(nodeClass === RUNWARE_NODE_TYPES.VIDEOMODELSEARCH) {
|
|
videoModelSearchFilterHandler(node);
|
|
} else if(nodeClass === RUNWARE_NODE_TYPES.AUDIOMODELSEARCH) {
|
|
audioModelSearchFilterHandler(node);
|
|
}
|
|
|
|
for(const nodeWidget of nodeWidgets) {
|
|
const widgetName = nodeWidget.name;
|
|
const widgetType = nodeWidget.type;
|
|
|
|
if(crNodeProps.promptEnhancer === true && widgetType === "customtext") {
|
|
if (widgetName === "positivePrompt" || widgetName === "negativePrompt") {
|
|
nodeWidget.inputEl.addEventListener('keydown', promptEnhanceHandler);
|
|
}
|
|
} else if(crNodeProps.liveDimensions === true && widgetType === "combo" && widgetName === "dimensions") {
|
|
syncDimensionsNodeHandler(node, nodeWidget);
|
|
}
|
|
|
|
if(crNodeProps.liveSearch === true) {
|
|
if(widgetType === "text" && SEARCH_TERMS.includes(widgetName)) {
|
|
node.callback = function(){};
|
|
searchNodeHandler(node, nodeWidget);
|
|
nodeInitList.push(node);
|
|
}
|
|
}
|
|
}
|
|
},
|
|
loadedGraphNode(node) {
|
|
if(nodeInitList.includes(node)) node.callback();
|
|
}
|
|
}) |