From 62865a941e2fc9287385d4385d1dd680ef33eaea Mon Sep 17 00:00:00 2001
From: Emmanuel Morales <31712515+EmmanuelMr18@users.noreply.github.com>
Date: Sun, 7 Jul 2024 20:13:51 -0600
Subject: [PATCH] feat(save image): output the file paths in the node

Why and what?
When We upload images to S3, all is good, but will be nice to have the path of where our images were saved.

With this, you can connect the `show text` node and easily check the path.

So later you can use that path to know how to fetch your images.
---
 src/nodes/save_image_s3.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/nodes/save_image_s3.py b/src/nodes/save_image_s3.py
index 3614160..99c3cd4 100644
--- a/src/nodes/save_image_s3.py
+++ b/src/nodes/save_image_s3.py
@@ -28,15 +28,18 @@ class SaveImageS3:
             },
                 }
 
-    RETURN_TYPES = ()
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("s3_image_paths",)
     FUNCTION = "save_images"
     OUTPUT_NODE = True
+    OUTPUT_IS_LIST = (True,)
     CATEGORY = "ComfyS3"
 
     def save_images(self, images, filename_prefix="ComfyUI", prompt=None, extra_pnginfo=None):
         filename_prefix += self.prefix_append
         full_output_folder, filename, counter, subfolder, filename_prefix = S3_INSTANCE.get_save_path(filename_prefix, images[0].shape[1], images[0].shape[0])
         results = list()
+        s3_image_paths = list()
         
         for image in images:
             i = 255. * image.cpu().numpy()
@@ -62,8 +65,11 @@ class SaveImageS3:
 
                     # Upload the temporary file to S3
                     s3_path = os.path.join(full_output_folder, file)
-                    S3_INSTANCE.upload_file(temp_file_path, s3_path)
+                    file_path = S3_INSTANCE.upload_file(temp_file_path, s3_path)
 
+                    # Add the s3 path to the s3_image_paths list
+                    s3_image_paths.append(file_path)
+                    
                     # Add the result to the results list
                     results.append({
                         "filename": file,
@@ -77,4 +83,4 @@ class SaveImageS3:
                 if temp_file_path and os.path.exists(temp_file_path):
                     os.remove(temp_file_path)
 
-        return { "ui": { "images": results } }
+        return { "ui": { "images": results},  "result": (s3_image_paths,) }