Files

43 lines
1.6 KiB
Python

from .final_frame_selector import (
FinalFrameSelector,
NODE_CLASS_MAPPINGS as FINAL_FRAME_SELECTOR_MAPPINGS,
NODE_DISPLAY_NAME_MAPPINGS as FINAL_FRAME_SELECTOR_DISPLAY_MAPPINGS,
NODE_CLASS_MAPPINGS as FIRST_FRAME_SELECTOR_MAPPINGS,
NODE_DISPLAY_NAME_MAPPINGS as FIRST_FRAME_SELECTOR_DISPLAY_MAPPINGS
)
from .video_merge import (
VideoMerge,
NODE_CLASS_MAPPINGS as VIDEO_MERGE_MAPPINGS,
NODE_DISPLAY_NAME_MAPPINGS as VIDEO_MERGE_DISPLAY_MAPPINGS
)
from .youtube_video_downloader import (
YouTubeVideoDownloader,
NODE_CLASS_MAPPINGS as YOUTUBE_VIDEO_DOWNLOADER_MAPPINGS,
NODE_DISPLAY_NAME_MAPPINGS as YOUTUBE_VIDEO_DOWNLOADER_DISPLAY_MAPPINGS
)
from .reverse_frame_sequence import (
ReverseFrameSequence,
NODE_CLASS_MAPPINGS as REVERSE_FRAME_SEQUENCE_MAPPINGS,
NODE_DISPLAY_NAME_MAPPINGS as REVERSE_FRAME_SEQUENCE_DISPLAY_MAPPINGS
)
from .prompt_journal import (
PromptJournal,
NODE_CLASS_MAPPINGS as PROMPT_JOURNAL_MAPPINGS,
NODE_DISPLAY_NAME_MAPPINGS as PROMPT_JOURNAL_DISPLAY_MAPPINGS
)
NODE_CLASS_MAPPINGS = {
**FINAL_FRAME_SELECTOR_MAPPINGS,
**FIRST_FRAME_SELECTOR_MAPPINGS,
**VIDEO_MERGE_MAPPINGS,
**YOUTUBE_VIDEO_DOWNLOADER_MAPPINGS,
**REVERSE_FRAME_SEQUENCE_MAPPINGS,
**PROMPT_JOURNAL_MAPPINGS
}
NODE_DISPLAY_NAME_MAPPINGS = {
**FINAL_FRAME_SELECTOR_DISPLAY_MAPPINGS,
**FIRST_FRAME_SELECTOR_DISPLAY_MAPPINGS,
**VIDEO_MERGE_DISPLAY_MAPPINGS,
**YOUTUBE_VIDEO_DOWNLOADER_DISPLAY_MAPPINGS,
**REVERSE_FRAME_SEQUENCE_DISPLAY_MAPPINGS,
**PROMPT_JOURNAL_DISPLAY_MAPPINGS
}