from .nodes.heygem import HeyGemF2F, HeyGemF2FFromFile
from .nodes.s3 import S3Download, S3Upload, S3UploadURL
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.video import VideoCut, VideoCutByFramePoint, VideoChangeFPS
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,
    "S3Upload": S3Upload,
    "S3UploadURL": S3UploadURL,
    "S3Download": S3Download,
    "VideoCutCustom": VideoCut,
    "VideoCutByFramePoint": VideoCutByFramePoint,
    "VideoChangeFPS": VideoChangeFPS,
    "VodToLocal": VodToLocalNode,
    "LogToDB": LogToDB,
    "VideoPointCompute": VideoStartPointDurationCompute,
    "StringEmptyJudgement": StringEmptyJudgement,
    "unloadAllModels": UnloadAllModels,
    "TraverseFolder": TraverseFolder,
    "LoadTextCustom": LoadTextLocal,
    "LoadTextCustomOnline": LoadTextOnline,
    "HeyGemF2F": HeyGemF2F,
    "HeyGemF2FFromFile": HeyGemF2FFromFile,
}

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