wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-22 12:20:16 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Fix for output images

Browse Source
This commit is contained in:
space-nuko 2023-06-08 21:29:08 -05:00
parent 0ba0a86aa7
commit e2895a98a5
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
web/scripts/app.js
View File

@ -132,7 +132,7 @@ export class ComfyApp {
} }
if(app.nodeOutputs[node.id + ""]) if(app.nodeOutputs[node.id + ""])
app.nodeOutputs[node.id + ""].images = node.images; app.nodeOutputs[node.id + ""] = [{ images: node.images }];
} }
if(ComfyApp.clipspace.imgs) { if(ComfyApp.clipspace.imgs) {
@ -393,12 +393,14 @@ export class ComfyApp {
let imgURLs = [] let imgURLs = []
let imagesChanged = false let imagesChanged = false
const output = app.nodeOutputs[this.id + ""]; const outputBatches = app.nodeOutputs[this.id + ""];
if (output && output.images) { if (outputBatches) {
if (this.images !== output.images) { if (this.images !== outputBatches) {
this.images = output.images; this.images = outputBatches;
const allImages = outputBatches.filter(batch => Array.isArray(batch.images))
.flatMap(batch => batch.images)
imagesChanged = true; imagesChanged = true;
imgURLs = imgURLs.concat(output.images.map(params => { imgURLs = imgURLs.concat(allImages.map(params => {
return "/view?" + new URLSearchParams(params).toString() + app.getPreviewFormatParam(); return "/view?" + new URLSearchParams(params).toString() + app.getPreviewFormatParam();
})) }))
} }
@ -426,7 +428,7 @@ export class ComfyApp {
}); });
}) })
).then((imgs) => { ).then((imgs) => {
if ((!output || this.images === output.images) && (!preview || this.preview === preview)) { if ((!outputBatches || this.images === outputBatches) && (!preview || this.preview === preview)) {
this.imgs = imgs.filter(Boolean); this.imgs = imgs.filter(Boolean);
this.setSizeForImage?.(); this.setSizeForImage?.();
app.graph.setDirtyCanvas(true); app.graph.setDirtyCanvas(true);
@ -530,6 +532,8 @@ export class ComfyApp {
} }
} else { } else {
// Draw individual // Draw individual
let w = this.imgs[imageIndex].naturalWidth;
let h = this.imgs[imageIndex].naturalHeight;
const scaleX = dw / w; const scaleX = dw / w;
const scaleY = dh / h; const scaleY = dh / h;
const scale = Math.min(scaleX, scaleY, 1); const scale = Math.min(scaleX, scaleY, 1);