121 lines
5.4 KiB
Python
121 lines
5.4 KiB
Python
from .nodes.image_modal_nodes import ModalEditCustom, ModalClothesMask, ModalMidJourneyGenerateImage, \
|
|
ModalMidJourneyDescribeImage
|
|
from .nodes.image_face_nodes import FaceDetect, FaceExtract
|
|
from .nodes.image_gesture_nodes import JMGestureCorrect, JMCustom
|
|
from .nodes.image_nodes import SaveImagePath, SaveImageWithOutput, LoadImgOptional, SaveImageAnywhere
|
|
from .nodes.llm_nodes import LLMChat, LLMChatMultiModalImageUpload, LLMChatMultiModalImageTensor, Jinja2RenderTemplate
|
|
from .nodes.object_storage_nodes import COSUpload, COSDownload, S3Download, S3Upload, S3UploadURL, S3UploadIMAGEURL
|
|
from .nodes.text_nodes import StringEmptyJudgement, LoadText, RandomLineSelector
|
|
from .nodes.util_nodes import LogToDB, TaskIdGenerate, TraverseFolder, UnloadAllModels, VodToLocalNode, \
|
|
PlugAndPlayWebhook
|
|
from .nodes.video_lipsync_nodes import HeyGemF2F, HeyGemF2FFromFile
|
|
from .nodes.video_nodes import VideoCut, VideoCutByFramePoint, VideoChangeFPS, VideoStartPointDurationCompute, \
|
|
VideoMerge
|
|
|
|
from .nodes.union_llm_node import LLMUionNode
|
|
from .nodes.img_agent import ImgSubmitNode
|
|
from .nodes.video_agent import VideoSubmitNode
|
|
from .nodes.save_node import ExtSaveNode
|
|
from .nodes.video_preview import VideoDownloaderNode
|
|
from .nodes.fetch_task_result import FetchTaskResult
|
|
from .nodes.file_upload import FileUploadNode
|
|
from .nodes.lip_sync_node import HedraLipNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FaceOccDetect": FaceDetect,
|
|
"FaceExtract": FaceExtract,
|
|
"COSUpload": COSUpload,
|
|
"COSDownload": COSDownload,
|
|
"S3Upload": S3Upload,
|
|
"S3UploadURL": S3UploadURL,
|
|
"S3UploadIMAGEURL": S3UploadIMAGEURL,
|
|
"S3Download": S3Download,
|
|
"VideoCutCustom": VideoCut,
|
|
"VideoCutByFramePoint": VideoCutByFramePoint,
|
|
"VideoChangeFPS": VideoChangeFPS,
|
|
"VodToLocal": VodToLocalNode,
|
|
"LogToDB": LogToDB,
|
|
"VideoPointCompute": VideoStartPointDurationCompute,
|
|
"StringEmptyJudgement": StringEmptyJudgement,
|
|
"unloadAllModels": UnloadAllModels,
|
|
"TraverseFolder": TraverseFolder,
|
|
"LoadTextCustom": LoadText,
|
|
"HeyGemF2F": HeyGemF2F,
|
|
"HeyGemF2FFromFile": HeyGemF2FFromFile,
|
|
"SaveImagePath": SaveImagePath,
|
|
"LoadImgCustom": LoadImgOptional,
|
|
"TaskIdGenerate": TaskIdGenerate,
|
|
"RandomLineSelector": RandomLineSelector,
|
|
"PlugAndPlayWebhook": PlugAndPlayWebhook,
|
|
"SaveImageWithOutput": SaveImageWithOutput,
|
|
"LLMChat": LLMChat,
|
|
"LLMChatMultiModalImageUpload": LLMChatMultiModalImageUpload,
|
|
"LLMChatMultiModalImageTensor": LLMChatMultiModalImageTensor,
|
|
"Jinja2RenderTemplate": Jinja2RenderTemplate,
|
|
"JMGestureCorrect": JMGestureCorrect,
|
|
"ModalClothesMask": ModalClothesMask,
|
|
"ModalEditCustom": ModalEditCustom,
|
|
"ModalMidJourneyGenerateImage": ModalMidJourneyGenerateImage,
|
|
"ModalMidJourneyDescribeImage": ModalMidJourneyDescribeImage,
|
|
"JMCustom": JMCustom,
|
|
"VideoMerge": VideoMerge,
|
|
"SaveImageAnywhere": SaveImageAnywhere,
|
|
"LLMUionNode": LLMUionNode,
|
|
"ImgSubmitNode": ImgSubmitNode,
|
|
"VideoSubmitNode": VideoSubmitNode,
|
|
"ExtSaveNode": ExtSaveNode,
|
|
"VideoDownloaderNode": VideoDownloaderNode,
|
|
"FetchTaskResult": FetchTaskResult,
|
|
"FileUploadNode": FileUploadNode,
|
|
"HedraLipNode": HedraLipNode
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FaceOccDetect": "面部遮挡检测",
|
|
"FaceExtract": "面部提取",
|
|
"COSUpload": "COS上传-返回key",
|
|
"COSDownload": "COS下载",
|
|
"S3Upload": "S3上传-返回key",
|
|
"S3UploadURL": "S3上传-返回URL",
|
|
"S3UploadIMAGEURL": "S3上传图片-返回URL",
|
|
"S3Download": "S3下载",
|
|
"VideoCutCustom": "视频剪裁",
|
|
"VideoCutByFramePoint": "视频剪裁(精确帧位)",
|
|
"VideoChangeFPS": "视频转换帧率",
|
|
"VodToLocal": "腾讯云VOD下载",
|
|
"LogToDB": "状态持久化DB",
|
|
"VideoPointCompute": "视频帧位计算",
|
|
"StringEmptyJudgement": "字符串是否为空",
|
|
"unloadAllModels": "卸载所有已加载模型",
|
|
"TraverseFolder": "遍历文件夹",
|
|
"LoadTextCustom": "读取文本文件(file_path优先)",
|
|
"HeyGemF2F": "HeyGem口型同步(API, 传入文件Tensor)",
|
|
"HeyGemF2FFromFile": "HeyGem口型同步(API, 传入文件路径)",
|
|
"SaveImagePath": "保存图片",
|
|
"LoadImgCustom": "加载图片(URL/本地, URL优先)",
|
|
"TaskIdGenerate": "TaskID生成器",
|
|
"RandomLineSelector": "随机选择一行内容",
|
|
"PlugAndPlayWebhook": "Webhook转发器",
|
|
"SaveImageWithOutput": "保存图片(带输出)",
|
|
"LLMChat": "LLM调用",
|
|
"LLMChatMultiModalImageUpload": "多模态LLM调用-图片Path",
|
|
"LLMChatMultiModalImageTensor": "多模态LLM调用-图片Tensor",
|
|
"Jinja2RenderTemplate": "Jinja2格式Prompt模板渲染",
|
|
"JMGestureCorrect": "人物侧身图片转为正面图-即梦",
|
|
"ModalClothesMask": "模特指定衣服替换为指定颜色",
|
|
"ModalEditCustom": "Prompt生/修图-Gemini",
|
|
"ModalMidJourneyGenerateImage": "Prompt生/修图-MJ",
|
|
"ModalMidJourneyDescribeImage": "反推生图提示词-MJ",
|
|
"JMCustom": "Prompt生视频",
|
|
"VideoMerge": "顺序合并视频",
|
|
"SaveImageAnywhere": "保存图片-任意路径",
|
|
"LLMUionNode": "LLM多模态节点",
|
|
"ImgSubmitNode": "提交图片生成",
|
|
"VideoSubmitNode": "提交视频生成",
|
|
"ExtSaveNode": "通用文件保存",
|
|
"VideoDownloaderNode": "视频下载",
|
|
"FetchTaskResult": "获取生成结果 (图片/视频链接)",
|
|
"FileUploadNode": "文件上传",
|
|
"HedraLipNode": HedraLipNode
|
|
}
|