From fc76d349ebe7f56b250ed0a311cdb5bc21997444 Mon Sep 17 00:00:00 2001
From: atmaranto <35581227+atmaranto@users.noreply.github.com>
Date: Mon, 18 Mar 2024 16:23:26 -0500
Subject: [PATCH] Update comfyui_to_python_utils.py

---
 comfyui_to_python_utils.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/comfyui_to_python_utils.py b/comfyui_to_python_utils.py
index d51d879..6bc897c 100644
--- a/comfyui_to_python_utils.py
+++ b/comfyui_to_python_utils.py
@@ -161,10 +161,9 @@ def save_image_wrapper(cls):
                             if os.path.isdir(args.output):
                                 subfolder = args.output
                                 file = "output"
-                            elif os.pathsep in args.output:
-                                subfolder, file = os.path.split(args.output)
                             else:
-                                file = args.output
+                                subfolder, file = os.path.split(args.output)
+                            
                             if subfolder == "":
                                 subfolder = os.getcwd()
                             
@@ -179,6 +178,7 @@ def save_image_wrapper(cls):
                                     break
                         
                         img.save(os.path.join(subfolder, file), pnginfo=metadata, compress_level=self.compress_level)
+                        print("Saved image to", os.path.join(subfolder, file))
                         results.append({
                             "filename": file,
                             "subfolder": subfolder,