111 lines
4.4 KiB
Python
111 lines
4.4 KiB
Python
from .nodes.llm_api import LLMChat, LLMChatMultiModal
|
|
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.heygem import HeyGemF2F, HeyGemF2FFromFile
|
|
from .nodes.image import SaveImagePath
|
|
from .nodes.load_image_pro import LoadNetImg
|
|
from .nodes.log2db import LogToDB
|
|
from .nodes.nodes_bfl import (
|
|
FluxProUltraImageNode,
|
|
FluxKontextProImageNode,
|
|
FluxKontextMaxImageNode,
|
|
FluxProExpandNode,
|
|
FluxProFillNode,
|
|
FluxProCannyNode,
|
|
FluxProDepthNode
|
|
|
|
)
|
|
from .nodes.s3 import S3Download, S3Upload, S3UploadURL
|
|
from .nodes.string_empty_judgement import StringEmptyJudgement
|
|
from .nodes.text import *
|
|
from .nodes.traverse_folder import TraverseFolder
|
|
from .nodes.unload_all_models import UnloadAllModels
|
|
from .nodes.video import VideoCut, VideoCutByFramePoint, VideoChangeFPS
|
|
from .nodes.vod2local import VodToLocalNode
|
|
from .nodes.prompt_id_generator import TaskIdGenerate
|
|
from .nodes.random_line import RandomLineSelector
|
|
from .nodes.webhook_forward import PlugAndPlayWebhook, SaveImageWithOutput
|
|
|
|
# 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,
|
|
"SaveImagePath": SaveImagePath,
|
|
"LoadNetImg": LoadNetImg,
|
|
"FluxProUltraImageNode": FluxProUltraImageNode,
|
|
# "FluxProImageNode": FluxProImageNode,
|
|
"FluxKontextProImageNode": FluxKontextProImageNode,
|
|
"FluxKontextMaxImageNode": FluxKontextMaxImageNode,
|
|
"FluxProExpandNode": FluxProExpandNode,
|
|
"FluxProFillNode": FluxProFillNode,
|
|
"FluxProCannyNode": FluxProCannyNode,
|
|
"FluxProDepthNode": FluxProDepthNode,
|
|
"TaskIdGenerate": TaskIdGenerate,
|
|
"RandomLineSelector": RandomLineSelector,
|
|
"PlugAndPlayWebhook": PlugAndPlayWebhook,
|
|
"SaveImageWithOutput": SaveImageWithOutput,
|
|
"LLMChat": LLMChat,
|
|
"LLMChatMultiModal": LLMChatMultiModal
|
|
}
|
|
|
|
# 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, 传入文件路径)",
|
|
"SaveImagePath": "保存图片",
|
|
"LoadNetImg": "load_net_image",
|
|
"FluxProUltraImageNode": "Flux 1.1 [pro] Ultra Image",
|
|
# "FluxProImageNode": "Flux 1.1 [pro] Image",
|
|
"FluxKontextProImageNode": "Flux.1 Kontext [pro] Image",
|
|
"FluxKontextMaxImageNode": "Flux.1 Kontext [max] Image",
|
|
"FluxProExpandNode": "Flux.1 Expand Image",
|
|
"FluxProFillNode": "Flux.1 Fill Image",
|
|
"FluxProCannyNode": "Flux.1 Canny Control Image",
|
|
"FluxProDepthNode": "Flux.1 Depth Control Image",
|
|
"TaskIdGenerate": "TaskID生成器",
|
|
"RandomLineSelector": "随机选择一行内容",
|
|
"PlugAndPlayWebhook": "Webhook转发器",
|
|
"SaveImageWithOutput": "保存图片(带输出)",
|
|
"LLMChat": "LLM调用",
|
|
"LLMChatMultiModal": "多模态LLM调用"
|
|
}
|