From 2f0a6f688feb205bdcf7c715a436715bb799923d Mon Sep 17 00:00:00 2001
From: BennyKok <itechbenny@gmail.com>
Date: Wed, 27 Sep 2023 13:19:43 +0800
Subject: [PATCH] will auto check if there are any missing output slots, will
 auto create

---
 js/SideBar.js | 37 +++++++++++++++++++++++++------------
 1 file changed, 25 insertions(+), 12 deletions(-)

diff --git a/js/SideBar.js b/js/SideBar.js
index cae7c5c..2805364 100644
--- a/js/SideBar.js
+++ b/js/SideBar.js
@@ -24,21 +24,34 @@ const {
 } = van.tags;
 
 van.derive(() => {
-  if (showImageEditor.val && targetNode.val != undefined && targetNode.val.outputs) {
-    const outputNames = targetNode.val.outputs.map(x => x.name).slice(1)
-    const record = Object.keys(imagePromptsMulti.val)
+  if (
+    showImageEditor.val &&
+    targetNode.val != undefined &&
+    targetNode.val.outputs
+  ) {
+    const outputNames = targetNode.val.outputs.map((x) => x.name).slice(1);
+    const record = Object.keys(imagePromptsMulti.val);
 
-    const diff = outputNames.filter(x => !record.includes(x));
+    const diff = outputNames.filter((x) => !record.includes(x));
+    const missingDiff = record.filter((x) => !outputNames.includes(x));
 
-    if (diff.length === 0) return;
-
-    console.log("Cleaning up missing output slots", diff);
-    diff.forEach((x) => {
-      targetNode.val.removeOutput(targetNode.val.findOutputSlot(x))
-    })
-    targetNode.val.graph.change();
+    if (diff.length > 0) {
+      console.log("Cleaning up missing output slots", diff);
+      diff.forEach((x) => {
+        targetNode.val.removeOutput(targetNode.val.findOutputSlot(x));
+      });
+      targetNode.val.graph.change();
+    } else if (missingDiff.length > 0) {
+      missingDiff.forEach((x) => {
+        targetNode.val.addOutput(
+          x,
+          targetNode.val.type === "SAM" ? "IMAGE" : "SAM_PROMPT"
+        );
+      });
+      targetNode.val.graph.change();
+    }
   }
-}) 
+});
 
 export function SideBar() {
   const layer_to_delete = van.state("");