__init__.py

import os
import filecmp
import shutil
from .face_swap import MultisubjectFaceSwapNode
from .regional_prompting import RegionalPromptingNode, RegionalAttentionProcessorNode, GetRegionalMaskNode

import __main__


# Update to javascripts files
javascript_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "javascript")
extentions_folder = os.path.join(os.path.dirname(os.path.realpath(__main__.__file__)), "web" + os.sep + "extensions" + os.sep + "SALT" + os.sep)
if not os.path.exists(extentions_folder):
    os.mkdir(extentions_folder)

result = filecmp.dircmp(javascript_folder, extentions_folder)
if result.left_only or result.diff_files:
    file_list = list(result.left_only)
    file_list.extend(x for x in result.diff_files if x not in file_list)

    for file in file_list:
        src_file = os.path.join(javascript_folder, file)
        dst_file = os.path.join(extentions_folder, file)
        if os.path.exists(dst_file):
            os.remove(dst_file)
        shutil.copy(src_file, dst_file)


NODE_DISPLAY_NAME_MAPPINGS = {
    "MultisubjectFaceSwapNode": "Multisubject Face Swap",
    "RegionalPromptingNode": "Regional Prompting",
    "RegionalAttentionProcessorNode": "Regional Attention Processor",
    "GetRegionalMaskNode": "Get Regional Mask Node",
}

NODE_CLASS_MAPPINGS = {
    "MultisubjectFaceSwapNode": MultisubjectFaceSwapNode,
    "RegionalPromptingNode": RegionalPromptingNode,
    "RegionalAttentionProcessorNode": RegionalAttentionProcessorNode,
    "GetRegionalMaskNode": GetRegionalMaskNode,
}