wangbo/ComfyUI-Manager
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI-Manager.git synced 2026-01-02 10:10:49 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

add support for sharing outputs of the new SaveAnimatedWEBP node

Browse Source
This commit is contained in:
thecooltechguy 2023-11-23 23:20:20 -08:00
parent d00d3d374e
commit 2484ac97ae
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
js/comfyui-share.js
View File

@ -7,6 +7,7 @@ export const SUPPORTED_OUTPUT_NODE_TYPES = [
"SaveImage", "SaveImage",
"VHS_VideoCombine", "VHS_VideoCombine",
"ADE_AnimateDiffCombine", "ADE_AnimateDiffCombine",
"SaveAnimatedWEBP"
] ]
var docStyle = document.createElement('style'); var docStyle = document.createElement('style');
@ -33,6 +34,7 @@ docStyle.innerHTML = `
document.head.appendChild(docStyle); document.head.appendChild(docStyle);
export function getPotentialOutputsAndOutputNodes(nodes) { export function getPotentialOutputsAndOutputNodes(nodes) {
console.log({ nodes });
const potential_outputs = []; const potential_outputs = [];
const potential_output_nodes = []; const potential_output_nodes = [];
@ -88,7 +90,7 @@ export function getPotentialOutputsAndOutputNodes(nodes) {
const widgetValue = node.widgets[j].value; const widgetValue = node.widgets[j].value;
const parsedURLVals = widgetValue.params; const parsedURLVals = widgetValue.params;
if(!parsedURLVals.format.startsWith('image')) { if (!parsedURLVals.format.startsWith('image')) {
// video isn't supported format // video isn't supported format
continue; continue;
} }
@ -126,6 +128,17 @@ export function getPotentialOutputsAndOutputNodes(nodes) {
} }
} }
} }
else if (node.type === "SaveAnimatedWEBP") {
potential_output_nodes.push(node);
// check if node has an 'images' array property
if (node.hasOwnProperty("images") && Array.isArray(node.images)) {
// iterate over the images array and add each image to the potential_outputs array
for (let j = 0; j < node.images.length; j++) {
potential_outputs.push({ "type": "image", "image": node.images[j], "title": node.title });
}
}
}
} }
return { potential_outputs, potential_output_nodes }; return { potential_outputs, potential_output_nodes };
} }