from .nodes.text import *
from .nodes.traverse_folder import TraverseFolder
from .nodes.unload_all_models import UnloadAllModels
from .nodes.string_empty_judgement import StringEmptyJudgement
from .nodes.compute_video_point import VideoStartPointDurationCompute
from .nodes.cos import COSUpload, COSDownload
from .nodes.face_detect import FaceDetect
from .nodes.face_extract import FaceExtract
from .nodes.log2db import LogToDB
from .nodes.videocut import VideoCut, VideoCutByFramePoint
from .nodes.vod2local import VodToLocalNode

# A dictionary that contains all nodes you want to export with their names
# NOTE: names should be globally unique
NODE_CLASS_MAPPINGS = {
    "FaceOccDetect": FaceDetect,
    "FaceExtract": FaceExtract,
    "COSUpload": COSUpload,
    "COSDownload": COSDownload,
    "VideoCutCustom": VideoCut,
    "VideoCutByFramePoint": VideoCutByFramePoint,
    "VodToLocal": VodToLocalNode,
    "LogToDB": LogToDB,
    "VideoPointCompute": VideoStartPointDurationCompute,
    "StringEmptyJudgement": StringEmptyJudgement,
    "unloadAllModels": UnloadAllModels,
    "TraverseFolder": TraverseFolder,
    "LoadTextCustom": LoadTextLocal,
    "LoadTextCustomOnline": LoadTextOnline
}

# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
    "FaceOccDetect": "面部遮挡检测",
    "FaceExtract": "面部提取",
    "COSUpload": "COS上传",
    "COSDownload": "COS下载",
    "VideoCutCustom": "视频剪裁",
    "VideoCutByFramePoint": "视频剪裁(精确帧位)",
    "VodToLocal": "腾讯云VOD下载",
    "LogToDB": "状态持久化DB",
    "VideoPointCompute": "视频帧位计算",
    "StringEmptyJudgement": "字符串是否为空",
    "unloadAllModels": "卸载所有已加载模型",
    "TraverseFolder": "遍历文件夹",
    "LoadTextCustom": "读取文本文件(本地)",
    "LoadTextCustomOnline": "读取文本文件(线上)"
}