From 37480c2d217698f799f6388ff311b9f8c6c38804 Mon Sep 17 00:00:00 2001
From: Tim Pietrusky <timpietrusky@gmail.com>
Date: Tue, 28 May 2024 09:18:21 +0200
Subject: [PATCH] fix: images in subfolders are not working, fixes #12

---
 src/rp_handler.py | 2 +-
 test_input.json   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/rp_handler.py b/src/rp_handler.py
index 027b6831..b2014ab7 100644
--- a/src/rp_handler.py
+++ b/src/rp_handler.py
@@ -237,7 +237,7 @@ def process_output_images(outputs, job_id):
     for node_id, node_output in outputs.items():
         if "images" in node_output:
             for image in node_output["images"]:
-                output_images = image["filename"]
+                output_images = f"{image['subfolder']}/{image['filename']}"
 
     print(f"runpod-worker-comfy - image generation is done")
 
diff --git a/test_input.json b/test_input.json
index 1210cec3..7fdb0f20 100644
--- a/test_input.json
+++ b/test_input.json
@@ -53,7 +53,7 @@
       },
       "9": {
         "inputs": {
-          "filename_prefix": "ComfyUI",
+          "filename_prefix": "ComfyUI/test",
           "images": ["8", 0]
         },
         "class_type": "SaveImage"