from .video_extract_key_frames.nodes import ExtractKeyFrames, ExtractKeyFramesAndSplit

# A dictionary that contains all nodes you want to export with their names
# NOTE: names should be globally unique
NODE_CLASS_MAPPINGS = {
    "ExtractKeyFrames": ExtractKeyFrames,
    "ExtractKeyFramesAndSplit": ExtractKeyFramesAndSplit,
}

# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
    "ExtractKeyFrames": "Extract Key Frames",
    "ExtractKeyFramesAndSplit": "Extract Key Frames & Split Video",
}

WEB_DIRECTORY = "./web"

__all__ = [
    "NODE_CLASS_MAPPINGS",
    "NODE_DISPLAY_NAME_MAPPINGS",
    "WEB_DIRECTORY",
]

__author__ = """guyouyue"""
__email__ = "guyuoyue@gmail.com"
__version__ = "1.0.0"