From f0c837252eec9d8dad0a063251dcef0eea3473d6 Mon Sep 17 00:00:00 2001
From: daxcay <daxtoncaylor@gmail.com>
Date: Fri, 5 Jul 2024 11:19:20 +0530
Subject: [PATCH] Bug Fix

---
 classes/DATASET_SaveImage.py     | 4 ++--
 classes/DataSet_TextFilesLoad.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/classes/DATASET_SaveImage.py b/classes/DATASET_SaveImage.py
index 1653bc4..80cf3ba 100644
--- a/classes/DATASET_SaveImage.py
+++ b/classes/DATASET_SaveImage.py
@@ -16,8 +16,8 @@ class DataSet_SaveImage:
         return {
             "required": {
                 "Images": ("IMAGE",),
+                "ImageFileNames": ("STRING", {"default": "Image"}),
                 "destination": ("STRING", {}),
-                "Filename": ("STRING", {"default": "Image"}),
             },
             "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
         }
@@ -27,7 +27,7 @@ class DataSet_SaveImage:
     OUTPUT_NODE = True
     CATEGORY = "🔶DATASET🔶"
 
-    def BatchSave(self, Images, destination, Filename, prompt=None, extra_pnginfo=None):
+    def BatchSave(self, Images, ImageFileNames, destination, prompt=None, extra_pnginfo=None):
 
         try:
 
diff --git a/classes/DataSet_TextFilesLoad.py b/classes/DataSet_TextFilesLoad.py
index 406e1cc..7c57fd3 100644
--- a/classes/DataSet_TextFilesLoad.py
+++ b/classes/DataSet_TextFilesLoad.py
@@ -26,7 +26,7 @@ class DataSet_TextFilesLoad:
         }
 
     INPUT_IS_LIST = True
-    RETURN_TYPES = ("STRING","STRING","STRING",)
+    RETURN_TYPES = ("STRING","STRING","STRING","STRING")
     RETURN_NAMES = ("TextFileNames","TextFileNamesWithoutExtension","TextFilePaths","TextFileContents",)
     OUTPUT_IS_LIST = (True, True, True, True,)
     FUNCTION = "LoadIT"