🤏 Moderate Release - WAN Studio Redesign & 🌊📺 FlowState Video Preview Introduction

v1.0.2 Release

* The new WAN Studio is completely streamlined and simplified.

* Guess what? We now have a Video Preview node! Novel concept.

* It's very minor but I removed some of the startup CMD/terminal output.

See Releases for more info.

Author: flowstatelabs

FS_Constants.py

@@ -8,7 +8,7 @@
 ##
 # OUTSIDE IMPORTS
 ##
-import os, folder_paths, importlib
+import importlib
 import nodes
 
 
@@ -18,7 +18,7 @@
 
 # SYSTEM INFO
 SYSTEM_NAME = 'FlowState Creator Suite'
-SYSTEM_VERSION = '0.1.5'
+SYSTEM_VERSION = '1.0.1'
 
 # IMAGE PARAMETERS
 MAX_RESOLUTION=16384
@@ -32,14 +32,15 @@
 
 KJNODES_INSTALLED = "PathchSageAttentionKJ" in nodes.NODE_CLASS_MAPPINGS
 
+SAGE_AVAILABLE = False
 SageAttention = None
 
-
 try:
     importlib.import_module("sageattention")
     SAGE_ATTENTION_INSTALLED = True
 except:
     SAGE_ATTENTION_INSTALLED = False
 
 if SAGE_ATTENTION_INSTALLED and KJNODES_INSTALLED:
+    SAGE_AVAILABLE = True
     SageAttention = nodes.NODE_CLASS_MAPPINGS['PathchSageAttentionKJ']()

FS_Mappings.py

@@ -22,19 +22,23 @@
 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': '🌊🍿 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',
@@ -44,7 +48,7 @@
 ##
 # SYSTEM STATUS
 ##
-for fs_node in NODE_CLASS_MAPPINGS:
-    print(f'\t - 🟢 {fs_node}: {NODE_DISPLAY_NAME_MAPPINGS[fs_node]}')
+# for fs_node in NODE_CLASS_MAPPINGS:
+#     print(f'\t - 🟢 {fs_node}: {NODE_DISPLAY_NAME_MAPPINGS[fs_node]}')
 
 print(f'\t   - ✅ Mappings Loaded.')

FS_Nodes.py

@@ -19,7 +19,9 @@
 from .FlowState_LatentSource import *
 from .FlowState_FluxEngine import *
 from .FlowState_VideoCreator import *
+from .FlowState_VideoPreview import *
 from .FlowState_WANStudio import *
+from .FlowState_WANStudio_Pro import *
 from .FlowState_Chef import *
 from .FlowState_Chef_Ingredients import *
 

FS_Types.py

@@ -86,6 +86,12 @@ def __ne__(self, __value: object) -> bool:
     f' - Defines the height of the image.\n\n'
 )})
 
+# VIDEO
+TYPE_VIDEO_IN = ('VIDEO', {'tooltip': (
+    f' Video Input\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - The video input.\n\n'
+)})
+
 # LATENT
 TYPE_LATENT_IN = ('LATENT', {'tooltip': (
     f' Latent Image\n {"-" * TOOLTIP_UNDERLINE}\n'
@@ -291,6 +297,11 @@ def pad_label(label):
     f' - This field is not functional. It is just a label for the group of settings below.\n\n'
 )})
 
+TYPE_FLOWSTATE_LABEL_SAVING = ('STRING', {'default': pad_label('💾 File Save Settings'), 'tooltip': (
+    f' Label\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - This field is not functional. It is just a label for the group of settings below.\n\n'
+)})
+
 TYPE_FLOWSTATE_LABEL_SAMPLING = ('STRING', {'default': pad_label('🧪 Sampling Settings'), 'tooltip': (
     f' Label\n {"-" * TOOLTIP_UNDERLINE}\n'
     f' - This field is not functional. It is just a label for the group of settings below.\n\n'
@@ -316,6 +327,11 @@ def pad_label(label):
 # FLOWSTATE CREATOR GENERIC TYPES
 ##
 
+# BOOLEAN
+TYPE_SEED_SELECT = ('BOOLEAN', {'default': True, 'tooltip': (
+    f' Fixed Output\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - Choose to get the same output every time or get variations.\n\n'
+)})
 
 # SAGE ATTENTION
 enabled_sage_modes = [
@@ -344,44 +360,56 @@ def pad_label(label):
 
 
 # VIDEO
-TYPE_NUM_VIDEO_FRAMES = ('INT', {'default': 48, 'min': 1, 'max': nodes.MAX_RESOLUTION, 'step': 1, 'tooltip': (
+TYPE_NUM_VIDEO_FRAMES = ('INT', {'default': 25, 'min': 1, 'max': nodes.MAX_RESOLUTION, 'step': 1, 'tooltip': (
     f' Number of Video Frames\n {"-" * TOOLTIP_UNDERLINE}\n'
     f' - The number of frames you want in your final video.\n\n'
 )})
-
-
-##
-# FLOWSTATE VIDEO CREATOR
-##
 TYPE_FPS = ('INT', {'default': 12, 'min': 1, 'max': 120, 'tooltip': (
     f' Frames Per Second\n {"-" * TOOLTIP_UNDERLINE}\n'
     f' - The number of frames per second in the created video.\n\n'
 )})
-TYPE_FRAMES_IN = ('IMAGE', {'tooltip': (
-    f' Video Frames\n {"-" * TOOLTIP_UNDERLINE}\n'
-    f' - The frames used to create the video.\n\n'
-)})
 TYPE_AUDIO_IN = ('AUDIO', {'tooltip': (
     f' Video Audio\n {"-" * TOOLTIP_UNDERLINE}\n'
     f' - Optional audio to be added to the video.\n\n'
 )})
-TYPE_FILENAME_PREFIX = ('STRING', {'default': 'video/ComyUI', 'tooltip': (
-    f' Filename Prefix\n {"-" * TOOLTIP_UNDERLINE}\n'
-    f' - The prefix for the file to save.\n'
-    f' - This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes.\n\n'
-)})
 TYPE_VIDEO_FORMAT = (['mp4', 'auto'], {
     'tooltip': (
         f' Video Format\n {"-" * TOOLTIP_UNDERLINE}\n'
         f' - The format to save the video as.\n\n'
     )
 })
+TYPE_SAVE_VIDEO_TYPE = (['mp4', 'gif'], {
+    'tooltip': (
+        f' Video Save Type\n {"-" * TOOLTIP_UNDERLINE}\n'
+        f' - Save the video as an MP4 or a GIF.\n\n'
+    )
+})
 TYPE_VIDEO_CODEC = (['h264', 'auto'], {
     'tooltip': (
         f' Video Codec\n {"-" * TOOLTIP_UNDERLINE}\n'
         f' - The codec to use for the video.\n\n'
     )
 })
+TYPE_FRAMES_IN = ('IMAGE', {'tooltip': (
+    f' Video Frames\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - The frames used to create the video.\n\n'
+)})
+TYPE_BOOLEAN_SAVE_VIDEO = ('BOOLEAN', {'default': False, 'tooltip': (
+    f' Save Video\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - Choose whether to save the video.\n\n'
+)})
+TYPE_FILENAME_PREFIX = ('STRING', {'default': 'video/WANStudio', 'tooltip': (
+    f' Filename Prefix\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - The prefix for the file to save.\n'
+    f' - This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes.\n\n'
+)})
+
+
+##
+# FLOWSTATE VIDEO CREATOR
+##
+
+# NONE
 
 
 ##
@@ -510,7 +538,13 @@ def pad_label(label):
     f' CLIP Vision Output\n {"-" * TOOLTIP_UNDERLINE}\n'
     f' - Optionally, use a CLIP Vision model.\n\n'
 )})
-TYPE_WAN_STUDIO_OUT = ('IMAGE', 'LATENT')
+TYPE_WAN_STUDIO_OUT = ('VIDEO', )
+TYPE_WAN_STUDIO_FILENAME_PREFIX = ('STRING', {'default': 'video/FlowState_WANStudio', 'tooltip': (
+    f' Filename Prefix\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - The prefix for the file to save.\n'
+    f' - This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes.\n'
+    f' - This is ignored if you choose not to save the video.\n\n'
+)})
 
 
 ##

FlowState_Chef.py

@@ -8,7 +8,7 @@
 ##
 # SYSTEM STATUS
 ##
-print(f'\t - 🟢 👩🏻‍🍳 Loaded Chef.')
+print(f'\t - 🟢 👩🏻‍🍳 Loaded FlowState Chef.')
 
 
 ##

FlowState_Chef_Ingredients.py

@@ -8,7 +8,7 @@
 ##
 # SYSTEM STATUS
 ##
-print(f'\t - 🟢 🥗 Loaded Chef Ingredients.')
+print(f'\t - 🟢 🥗 Loaded FlowState Chef Ingredients.')
 
 
 ##

FlowState_FluxEngine.py

@@ -8,7 +8,7 @@
 ##
 # SYSTEM STATUS
 ##
-print(f'\t - 🟢 🚒 Loaded Flux Engine.')
+print(f'\t - 🟢 🚒 Loaded FlowState Flux Engine.')
 
 
 ##

FlowState_LatentSource.py

@@ -9,7 +9,7 @@
 ##
 # SYSTEM STATUS
 ##
-print(f'\t - 🟢 🌱 Loaded Latent Source.')
+print(f'\t - 🟢 🌱 Loaded FlowState Latent Source.')
 
 
 ##

FlowState_Node.py

@@ -10,6 +10,8 @@
 ##
 import os, sys, time
 
+from datetime import datetime
+
 
 ##
 # FS IMPORTS
@@ -63,7 +65,7 @@ def print_status(self, messages, init=False, end=False, error=False):
         if init:
             print(f'\n\n\n  --- STARTING {self.node_name} ---')
 
-        print(f'\n\n')
+        print(f'\n')
 
         if error: print('-' * 100)
 
@@ -90,3 +92,36 @@ def get_mins_and_secs(self, start_time):
         secs = int(duration - mins * 60)
         return round(duration, 4), mins, secs
 
+    def check_tensor_equality(self, t1, t2, tensor_type='image'):
+        self.print_status([(f'Checking Tensor ({tensor_type}) Equality...',)])
+
+        has_t1 = t1 != None
+        has_t2 = t2 != None
+
+        has_neither = not has_t1 and not has_t2
+        has_t2_but_not_t1 = has_t2 and not has_t1
+        has_t1_but_not_t2 = has_t1 and not has_t2
+        one_missing = has_t2_but_not_t1 or has_t1_but_not_t2
+
+        if one_missing:
+            return False
+        
+        if has_neither:
+            return True
+
+        same_shape = t1.shape == t2.shape
+
+        if not same_shape:
+            return False
+        
+        if same_shape:
+            equal_tensors = t1.equal(t2)
+
+            if not equal_tensors:
+                return False
+        
+        return True
+
+    def get_formatted_time(self):
+        return datetime.now().strftime('%Y-%m-%d %H-%M-%S.%f')[:-3]
+

FlowState_SaveTempVideo.py

@@ -0,0 +1,99 @@
+# Project: FlowState Video Preview
+# Description: Simple preview video don't know why this isn't a node.
+# Version: 0.0.1
+# Author: Johnathan Chivington
+# Contact: flowstateeng@gmail.com | youtube.com/@flowstateeng
+
+
+##
+# OUTSIDE IMPORTS
+##
+import os
+import folder_paths
+
+from datetime import datetime
+from comfy_api.input import VideoInput
+from comfy_api.util import VideoContainer
+from comfy_api.latest import io, ui
+from comfy.cli_args import args
+
+
+class FlowState_SaveTempVideo(io.ComfyNode):
+    file_prefix = f'FlowState_VideoPreview'
+    format = 'mp4'
+    codec = 'h264'
+
+    temp_dir = folder_paths.get_temp_directory()
+
+    @classmethod
+    def define_schema(cls):
+        return io.Schema(
+            node_id='FlowState_SaveTempVideo',
+            display_name='FlowState Save Temp Video',
+            category='FlowState Creator Suite/Video',
+            description='Saves the video previews to your ComfyUI temp directory.',
+            inputs=[
+                io.Video.Input('video'),
+                io.String.Input('filename_prefix'),
+                io.Combo.Input('format'),
+                io.Combo.Input('codec')
+            ],
+            outputs=[],
+            hidden=[io.Hidden.prompt, io.Hidden.extra_pnginfo],
+            is_output_node=True,
+        )
+
+    @classmethod
+    def save_metadata(cls):
+        null_metadata = None
+
+        if args.disable_metadata:
+            return null_metadata
+
+        metadata = {}
+
+        if cls.hidden.extra_pnginfo != None:
+            metadata.update(cls.hidden.extra_pnginfo)
+
+        if cls.hidden.prompt != None:
+            metadata['prompt'] = cls.hidden.prompt
+
+        have_metadata = len(metadata) > 0
+        
+        return metadata if have_metadata else null_metadata
+
+    @classmethod
+    def save_video(cls, video, tmp_dir_path, filename, metadata):
+        full_path = os.path.join(tmp_dir_path, filename)
+
+        video.save_to(
+            full_path, format=cls.format, codec=cls.codec, metadata=metadata
+        )
+
+    @classmethod
+    def get_path_parts(cls, width, height):
+        tmp_dir_path, filename, counter, subfolder, prefix = folder_paths.get_save_image_path(
+            cls.file_prefix, cls.temp_dir, width, height
+        )
+
+        return tmp_dir_path, filename, subfolder
+
+    @classmethod
+    def execute(cls, video: VideoInput) -> io.NodeOutput:
+        width, height = video.get_dimensions()
+
+        tmp_dir_path, filename, subfolder = cls.get_path_parts(width, height)
+
+        saved_metadata = cls.save_metadata()
+        video_extension = VideoContainer.get_extension(cls.format)
+
+        timestamp = datetime.now().strftime('%Y-%m-%d %H-%M-%S.%f')[:-3]
+        tmp_filename = f'{filename} - {timestamp}.{video_extension}'
+
+        cls.save_video(video, tmp_dir_path, tmp_filename, saved_metadata)
+
+        saved_result = ui.SavedResult(tmp_filename, subfolder, io.FolderType.temp)
+        video_preview = ui.PreviewVideo([saved_result])
+
+        return io.NodeOutput(ui=video_preview)
+