Major update

Made changes to make nodes more cohesive
2024-07-05 09:04:31 +05:30
parent 37d33ca268
commit e8752c4f71
20 changed files with 441 additions and 543 deletions
@@ -0,0 +1,65 @@
+import os
+
+def read_text_file(file_path):
+    try:
+        with open(file_path, 'r') as file:
+            text = file.read()
+        return text
+    except FileNotFoundError:
+        print(f"Error: File '{file_path}' not found.")
+        return None
+    except Exception as e:
+        print(f"Error: {e}")
+        return None
+
+class DataSet_TextFilesLoadFromList:
+
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "TextFilePathsList": ("STRING", {"forceInput":True}),
+            },
+        }
+
+    INPUT_IS_LIST = True
+    RETURN_TYPES = ("STRING","STRING","STRING",)
+    RETURN_NAMES = ("TextFileNames","TextFileNamesWithoutExtension","TextFilePaths","TextFileContents",)
+    OUTPUT_IS_LIST = (True, True, True, True,)
+    FUNCTION = "LoadIT"
+    OUTPUT_NODE = True
+
+    CATEGORY = "🔶DATASET🔶"
+
+    def LoadIT(self, TextFilePathsList):
+        
+        try:
+
+            file_paths = [path for path in TextFilePathsList if path.endswith('.txt')]
+            file_names = [os.path.basename(path) for path in file_paths]
+            file_names_without_ext = [os.path.splitext(os.path.basename(path))[0] for path in file_paths]
+            file_contents = []
+
+            for file_path in file_paths:
+                content = read_text_file(file_path)
+                if(content):
+                    file_contents.append(content)
+                else:
+                    print(f'Error reading file: {file_path}')
+
+            return (file_names, file_names_without_ext, file_paths, file_contents,)        
+
+        except Exception as e:
+            return ([],[],[])
+
+
+N_CLASS_MAPPINGS = {
+    "DataSet_TextFilesLoadFromList": DataSet_TextFilesLoadFromList,
+}
+
+N_DISPLAY_NAME_MAPPINGS = {
+    "DataSet_TextFilesLoadFromList": "DataSet_TextFilesLoadFromList",
+}