Preserve mask when sending output into image upload

2023-05-27 22:39:07 -05:00
parent cb9e8540a0
commit 3b9d4533f9
5 changed files with 59 additions and 28 deletions
--- a/src/lib/nodes/ComfyPickImageNode.ts
+++ b/src/lib/nodes/ComfyPickImageNode.ts
@@ -99,9 +99,9 @@ export default class ComfyPickImageNode extends ComfyGraphNode {
    override onExecute() {
        const data = this.getInputData(0)
        let index = this.getInputData(1);
-        if (this.properties.imageTagFilter && Array.isArray(data))
+        if (this.properties.imageTagFilter != "" && Array.isArray(data))
            index = data.findIndex(i => i.tags?.includes(this.properties.imageTagFilter))
-        else
+        else if (index == null)
            index = 0;
        this.setValue(data, index);