Clean slate initiative.

New node suite for creative professionals.

Author: flowstateeng

.gitignore

@@ -1,7 +1,7 @@
 # Project: FlowState Assets
 # Description: Paths to assets needed by nodes.
 # Author: Johnathan Chivington
-# Contact: johnathan@flowstateengineering.com | youtube.com/@flowstateeng
+# Contact: flowstateeng@gmail.com | youtube.com/@flowstateeng
 
 
 ##
@@ -15,6 +15,7 @@
 # OUTSIDE IMPORTS
 ##
 import os
+import comfy
 import folder_paths
 
 
@@ -25,28 +26,19 @@
 FONT_PATH = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'fonts/ShareTechMono-Regular.ttf')
 
 
-UNET_LIST_PATH = None
-try:
-    UNET_LIST_PATH = folder_paths.get_filename_list('diffusion_models')
-    UNET_LIST_PATH = lambda: folder_paths.get_filename_list('diffusion_models')
-except ImportError:
-    UNET_LIST_PATH = None
+DIFFUSION_MODELS_LIST = lambda: folder_paths.get_filename_list('unet') + folder_paths.get_filename_list('diffusion_models')
+CHECKPOINTS_LIST = lambda: folder_paths.get_filename_list('checkpoints')
+CLIPS_LIST = lambda: folder_paths.get_filename_list('clip')
+VAES_LIST = lambda: get_vae_list()
 
-if UNET_LIST_PATH == None:
-    try:
-        UNET_LIST_PATH = folder_paths.get_filename_list('unet')
-        UNET_LIST_PATH = lambda: folder_paths.get_filename_list('unet')
-    except ImportError:
-        UNET_LIST_PATH = None
+CONTROL_NETS_LIST = lambda: folder_paths.get_filename_list('controlnet')
+LORAS_LIST = lambda: folder_paths.get_filename_list('loras')
 
-print(f'    - Setting UNET_LIST_PATH to: {UNET_LIST_PATH}')
+SAMPLERS_LIST = lambda: comfy.samplers.KSampler.SAMPLERS
+SCHEDULERS_LIST = lambda: comfy.samplers.KSampler.SCHEDULERS
 
+ALL_MODELS_LIST = lambda: DIFFUSION_MODELS_LIST() + CHECKPOINTS_LIST()
 
-NF4_LIST_PATH = lambda: folder_paths.get_filename_list('checkpoints')
-CKPT_LIST_PATH = lambda: folder_paths.get_filename_list('checkpoints')
-CLIP_LIST_PATH = lambda: folder_paths.get_filename_list('clip')
-VAE_LIST_PATH = lambda: get_vae_list()
-CONTROL_NET_LIST_PATH = lambda: folder_paths.get_filename_list('controlnet')
-LORA_LIST_PATH = lambda: folder_paths.get_filename_list('loras')
-ALL_MODEL_LIST_PATHS = lambda: UNET_LIST_PATH() + CKPT_LIST_PATH()
+INPUT_FILES = lambda: get_input_files()
+OUTPUT_FILES = lambda: get_output_files()
 

FS_Assets.py

@@ -1,7 +1,7 @@
 # Project: FlowState Constants
 # Description: Global constants for all nodes.
 # Author: Johnathan Chivington
-# Contact: johnathan@flowstateengineering.com | youtube.com/@flowstateeng
+# Contact: flowstateeng@gmail.com | youtube.com/@flowstateeng
 
 
 ##
@@ -13,15 +13,5 @@
 # CONSTANTS
 ##
 MAX_RESOLUTION=16384
-GLOBAL_MODELS_DIR = os.path.join(folder_paths.models_dir, 'llm_gguf')
-DEFAULT_INSTRUCTIONS = 'Generate a prompt from "{prompt}"'
-FS_LLM_PROMPT_TAG = '@FLOWSTATE_LLM_PROMPT'
 
-CLIP_PRESETS = {
-    'default': [1, 1, 1, 1],
-    'clarity_boost': [0.9, 0.9, 1.4, 8.4],
-    'even_flow': [1, 1, 1, 4],
-    'subtle_focus': [0.9, 0.9, 1, 4],
-    'sharp_detail': [1, 1, 1, 8.4]
-}
 

FS_Constants.py

@@ -1,7 +1,7 @@
 # Project: FlowState Node Mappings
 # Description: Node mappings for ComfyUI registry.
 # Author: Johnathan Chivington
-# Contact: johnathan@flowstateengineering.com | youtube.com/@flowstateeng
+# Contact: flowstateeng@gmail.com | youtube.com/@flowstateeng
 
 
 ##
@@ -13,38 +13,27 @@
 ##
 # SYSTEM STATUS
 ##
-print(f'  - Loading node name mappings...')
+print(f'  3. 💾 Loading node name mappings...')
 
 
 ##
 # MAPPINGS
 ##
 NODE_CLASS_MAPPINGS = {
-    'FlowStateUnifiedVideoSampler': FlowStateUnifiedVideoSampler,
-    'FlowStateUnifiedSampler': FlowStateUnifiedSampler,
-    'FlowStateUnifiedModelLoader': FlowStateUnifiedModelLoader,
-    'FlowStateUnifiedPrompt': FlowStateUnifiedPrompt,
-    'FlowStatePromptOutput': FlowStatePromptOutput,
-    'FlowStateLatentChooser': FlowStateLatentChooser,
-    'FlowStateUnifiedStyler': FlowStateUnifiedStyler
+    'FlowState_LatentSelector': FlowState_LatentSelector,
+    # 'FlowState_FluxSampler': FlowState_FluxSampler,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    'FlowStateUnifiedVideoSampler': 'FlowState Unified Video Sampler',
-    'FlowStateUnifiedSampler': 'FlowState Unified Sampler',
-    'FlowStateUnifiedModelLoader': 'FlowState Unified Model Loader',
-    'FlowStateUnifiedPrompt': 'FlowState Unified Prompt',
-    'FlowStatePromptOutput': 'FlowState Prompt Output',
-    'FlowStateLatentChooser': 'FlowState Latent Chooser',
-    'FlowStateUnifiedStyler': 'FlowState Unified Styler'
+    'FlowState_LatentSelector': '🌊 FlowState Latent Selector',
+    # 'FlowState_FluxSampler': '🌊 FlowState Flux Sampler',
 }
 
 
 ##
 # SYSTEM STATUS
 ##
-print(f'  - Mappings Loaded. Available nodes:')
-
 for fs_node in NODE_CLASS_MAPPINGS:
-    print(f'    - {fs_node}: {NODE_CLASS_MAPPINGS[fs_node]}')
+    print(f'\t - ✅ {fs_node}: {NODE_DISPLAY_NAME_MAPPINGS[fs_node]}')
 
+print(f'\t - ✅ Mappings Loaded.')

FS_Mappings.py

@@ -1,29 +1,24 @@
-# Project: FlowState Nodes
+# Project: FlowState Creator Nodes
 # Description: A collection of custom nodes to solve problems I couldn't find existing nodes for.
 # Author: Johnathan Chivington
-# Contact: johnathan@flowstateengineering.com | youtube.com/@flowstateeng
+# Contact: flowstateeng@gmail.com | youtube.com/@flowstateeng
 
 
 
 ##
 # SYSTEM STATUS
 ##
-print(f'  - Loading custom nodes...')
+print(f'  2. 💾 Loading custom nodes...')
 
 
 ##
 # NODES
 ##
-from .FlowStateUnifiedVideoSampler import *
-from .FlowStateUnifiedSampler import *
-from .FlowStateUnifiedPrompt import *
-from .FlowStateUnifiedModelLoader import *
-from .FlowStateLatentChooser import *
-from .FlowStateUnifiedStyler import *
+from .FlowState_LatentSelector import *
 
 
 ##
 # SYSTEM STATUS
 ##
-print(f'  - Nodes Loaded.')
+print(f'\t - ✅ All nodes Loaded.')