FlowState-Creator-Nodes/FS_Mappings.py at main · flowstatelabs/FlowState-Creator-Nodes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
# Project: FlowState Node Mappings
# Description: Node mappings for ComfyUI registry.
# Author: Johnathan Chivington
# Contact: flowstateeng@gmail.com | youtube.com/@flowstateeng


##
# FS IMPORTS
##
from .FS_Nodes import *


##
# SYSTEM STATUS
##
print(f'  3. 💾 Loading node name mappings...')


##
# MAPPINGS
##
NODE_CLASS_MAPPINGS = {
    'FlowState_SimpleLatent': FlowState_SimpleLatent,
    'FlowState_VideoCreator': FlowState_VideoCreator,
    'FlowState_VideoPreview': FlowState_VideoPreview,
    'FlowState_LatentSource': FlowState_LatentSource,
    'FlowState_FluxEngine': FlowState_FluxEngine,
    'FlowState_WANStudio': FlowState_WANStudio,
    # 'FlowState_WANStudio_Pro': FlowState_WANStudio_Pro,
    'FlowState_Chef': FlowState_Chef,
    'FlowState_Chef_Ingredients': FlowState_Chef_Ingredients,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    'FlowState_VideoPreview': '🌊📺 FlowState Video Preview',
    'FlowState_VideoCreator': '🌊🎥 FlowState Video Creator',
    'FlowState_SimpleLatent': '🌊👌 FlowState Simple Latent',
    'FlowState_LatentSource': '🌊🌱 FlowState Latent Source',
    'FlowState_FluxEngine': '🌊🚒 FlowState Flux Engine',
    'FlowState_WANStudio': '🌊🍿 FlowState WAN Studio',
    # 'FlowState_WANStudio_Pro': '🌊🎬 FlowState WAN Studio Pro',
    'FlowState_Chef': '🌊👩🏻‍🍳 FlowState Chef',
    'FlowState_Chef_Ingredients': '🌊🥗 FlowState Chef Ingredients',
    # 'FlowState_AssetForge': '🌊 FlowState Asset Forge',
}


##
# SYSTEM STATUS
##
# for fs_node in NODE_CLASS_MAPPINGS:
#     print(f'\t - 🟢 {fs_node}: {NODE_DISPLAY_NAME_MAPPINGS[fs_node]}')

print(f'\t   - ✅ Mappings Loaded.')