A bit of housekeeping

__init__.py

@@ -58,94 +58,7 @@ from . import pytorch_nodes
 from . import youtube_nodes
 from . import torchvision_nodes
 from . import hf_diffusers_nodes
-
+from . import audio_video_folder
 
 
 __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
-
-# import os
-# import importlib
-#
-# WEB_DIRECTORY = "./web"
-# __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
-# OLD_NODE_CLASS_MAPPINGS = {
-#     "Interpolation": "LatentInterpolation",
-#     "Image Pad To Match": "PadToMatch",
-#     "Image Stack": "StackImages",
-#     "Image B/C": "ImageBC",
-#     "Image RGBA Mod": "RGBA_MOD",
-#     "Image RGBA Lower Clip": "rgba_lower_clip",
-#     "Image RGBA Upper Clip": "rgba_upper_clip",
-#     "Image RGBA Merge": "rgba_merge",
-#     "Image RGBA Split": "rgba_split",
-#     "Select From Batch": "select_from_batch",
-#     "Quantize": "Quantize",
-#     "Select From RGB Similarity": "SelectFromRGBSimilarity",
-#     "Load Layered Image": "LoadImage",
-#     "Code Execution Widget": "ExecWidget",
-#     "Image Distance Mask": "ImageDistanceMask",
-#     "Image Text": "TextRender",
-#     "function": "FuncBase",
-#     "function render": "FuncRender",
-#     "function render image": "FuncRenderImage",
-#     "Tiny Txt 2 Img": "TinyTxtToImg",
-#     "Preview Image Test": "PreviewImageTest",
-#     "ETK_KSampler": "KSampler",
-# }
-# # reverse the dict
-# OLD_NODE_CLASS_MAPPINGS = {v: k for k, v in OLD_NODE_CLASS_MAPPINGS.items()}
-#
-# NODE_CLASS_MAPPINGS = {}
-# NODE_DISPLAY_NAME_MAPPINGS = {}
-# # Try importing optional dependencies
-#
-# import sys
-#
-# if os.getenv("UNIT_TEST", False) or 'pytest' in sys.modules:
-#     pass
-# else:  # if not a unit test
-#     try:
-#         import comfy.samplers
-#
-#         use_generic_nodes = False
-#     except ImportError:
-#         use_generic_nodes = True
-#         print("ETK> Failed to import comfy.samplers, assuming no comfyui skipping SD related nodes")
-#
-#     # Load classes from modules in the custom_nodes.EternalKernelLiteGraphNodes package
-#     folder = os.path.dirname(os.path.abspath(__file__))
-#
-#     for file in os.listdir(folder):
-#         if file == "nodes.py":
-#             raise ValueError("nodes.py is not an allowed name in this folder")
-#         if file.endswith(".py") and file[0] != "_":
-#             # import the NODE_CLASS_MAPPINGS from the file
-#             try:
-#                 module = importlib.import_module(f".{file[:-3]}", __name__)
-#             except ImportError:
-#                 try:
-#                     module = importlib.import_module(f"{file[:-3]}")  # needed for testing?
-#                 except ModuleNotFoundError:
-#                     module = importlib.import_module(f".{file[:-3]}", __name__)
-#
-#             if hasattr(module, "NODE_CLASS_MAPPINGS"):
-#                 mappings = module.NODE_CLASS_MAPPINGS
-#                 NODE_CLASS_MAPPINGS.update(mappings)
-#                 for k, v in mappings.items():
-#                     print(f"ETK: updated with {k}")
-#
-#             if hasattr(module, "NODE_DISPLAY_NAME_MAPPINGS"):
-#                 NODE_DISPLAY_NAME_MAPPINGS.update(module.NODE_DISPLAY_NAME_MAPPINGS)
-#
-#     # use OLD_NODE_CLASS_MAPPINGS to add the old names to the new nodes
-#     new_node_class_mappings = {}
-#     for k, v in OLD_NODE_CLASS_MAPPINGS.items():
-#         if k in NODE_CLASS_MAPPINGS:
-#             new_node_class_mappings[v] = NODE_CLASS_MAPPINGS[k]
-#             # mark it for removeal
-#             NODE_CLASS_MAPPINGS[k] = None
-#         else:
-#             print(f"ETK> Failed to find node {k} in new nodes, skipping...")
-#     # remove the nodes with None
-#     NODE_CLASS_MAPPINGS = {k: v for k, v in NODE_CLASS_MAPPINGS.items() if v is not None}
-#     NODE_CLASS_MAPPINGS.update(new_node_class_mappings)

audio_video_folder.py

@@ -1,5 +1,5 @@
-NODE_CLASS_MAPPINGS = {}
-NODE_DISPLAY_NAME_MAPPINGS = {}
+from . import NODE_CLASS_MAPPINGS
+from . import NODE_DISPLAY_NAME_MAPPINGS
 
 
 def ETK_AVF_base(cls):