From 4f427294944897045b444c77d12d0fcbd173b4a1 Mon Sep 17 00:00:00 2001
From: Rhys Yang <receyuki@gmail.com>
Date: Wed, 22 Nov 2023 03:22:40 +0800
Subject: [PATCH] Add `FILENAME` and `METADATA` output to the `Prompt Saver`
 node #16

---
 nodes.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/nodes.py b/nodes.py
index 3e6df9b..4f36c1b 100644
--- a/nodes.py
+++ b/nodes.py
@@ -95,7 +95,7 @@ class SDPromptReader:
         "CFG",
         "WIDTH",
         "HEIGHT",
-        "FILE_NAME",
+        "FILENAME",
         "SETTINGS",
     )
 
@@ -256,7 +256,8 @@ class SDPromptSaver:
             "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
         }
 
-    RETURN_TYPES = ()
+    RETURN_TYPES = ("STRING", "STRING")
+    RETURN_NAMES = ("FILENAME", "METADATA")
     FUNCTION = "save_images"
 
     OUTPUT_NODE = True
@@ -304,7 +305,9 @@ class SDPromptSaver:
             images[0].shape[0],
         )
 
-        results = list()
+        results = []
+        files = []
+        comments = []
         for image in images:
             # model_name_str, sampler_name_str, scheduler_str = None, None, None
 
@@ -392,8 +395,11 @@ class SDPromptSaver:
             results.append(
                 {"filename": file.name, "subfolder": str(subfolder), "type": self.type}
             )
+            files.append(str(file))
+            output_to_terminal("Saved file: " + str(file))
+            comments.append(comment)
 
-        return {"ui": {"images": results}}
+        return {"ui": {"images": results}, "result": (files, comments)}
 
     @staticmethod
     def calculate_model_hash(model_name):