BatchCreativeInterpolationNodeDynamicSettings added

Author: komojini

__init__.py

@@ -30,6 +30,7 @@
     "ImageCropByRatio": ImageCropByRatio,
     "ImageCropByRatioAndResize": ImageCropByRatioAndResize,
     "ImagesCropByRatioAndResizeBatch": ImagesCropByRatioAndResizeBatch,
+    "BatchCreativeInterpolationNodeDynamicSettings": BatchCreativeInterpolationNodeDynamicSettings,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
@@ -55,6 +56,7 @@
     "ImageCropByRatio": "ImageCropByRatio",
     "ImageCropByRatioAndResize": "ImageCropByRatioAndResize",
     "ImagesCropByRatioAndResizeBatch": "ImagesCropByRatioAndResizeBatch",
+    "BatchCreativeInterpolationNodeDynamicSettings": "BatchCreativeInterpolationNodeDynamicSettings",
 }
 
 

js/widgethider.js

@@ -109,6 +109,9 @@ const nodeWidgetHandlers = {
     "UltimateVideoLoader (simple)": {
         "source": handleUltimateVideoLoaderSource,
     },
+    "BatchCreativeInterpolationNodeDynamicSettings": {
+        "image_count": handleBatchCreativeInterpolationNodeDynamicSettingsVisibility,
+    }
 };
 
 // In the main function where widgetLogic is called
@@ -145,13 +148,42 @@ function handleUltimateVideoLoaderVisibility(node, source) {
     }
 }
 
+function handleBatchCreativeInterpolationNodeDynamicSettingsVisibility(node, widget) {
+    handleVisibility(node, widget.value, "BatchCreativeInterpolationNodeDynamicSettings")
+}
+
+const MAX_COUNT_VALUE = 50
 
 function handleUltimateVideoLoaderSource(node, widget) {
     handleInputModeWidgetsVisibility(node, widget.value);
     handleUltimateVideoLoaderVisibility(node, widget.value);
 }
 
+function handleVisibility(node, countValue, nodeType) {
+    const baseNamesMap = {
+        "BatchCreativeInterpolationNodeDynamicSettings": [
+            "frame_distribution",
+            "key_frame_influence",
+            "min_strength_value",
+            "max_strength_value",
+        ],
+    }
+    const baseNames = baseNamesMap[nodeType]
 
+    
+    for (let i=1; i <= MAX_COUNT_VALUE; i++) {
+        const widgets = baseNames.map((n) => findWidgetByName(node, `${n}_${i}`))
+
+        if (i <= countValue) {
+            widgets?.forEach((w) => {
+                
+                toggleWidget(node, w, true)}
+            )
+        } else {
+            widgets?.forEach((w) => toggleWidget(node, w, false))
+        }
+    }
+}
 
 app.registerExtension({
     name: "komojini.widgethider",

komojini_server.py

@@ -214,6 +214,7 @@ async def get_path(request):
 def is_prompt_node_type_of(node_value, node_type: str) -> bool:
     return node_type in node_value.get("class_type", "") or node_type in node_value.get("_meta", {}).get("tile", "")
 
+
 def is_workflow_node_type_of(node_value, node_type: str) -> bool:
     return node_type in node_value.get("type", "")
 
@@ -247,22 +248,27 @@ def search_setter_getter_connected_nodes(json_data):
             inputs = v["inputs"]
 
             if is_prompt_node_type_of(v, "Get"):
-                key = inputs["key"]
+                key = inputs.get("key")
+                if not key:
+                    continue
 
-                if key and class_type.endswith("CachedGetter") and CACHED_MAP.get(key, None) is not None:
+                if class_type.endswith("CachedGetter") and CACHED_MAP.get(key, None) is not None:
                     continue
 
                 if key in key_to_getter_node_ids:
                     key_to_getter_node_ids[key].append(node_id)
                 else:
                     key_to_getter_node_ids[key] = [node_id]
             elif is_prompt_node_type_of(v, "Set"):
-                key = inputs["key"]
+                key = inputs.get("key")
+                if not key:
+                    continue
                 key_to_setter_node_id[key] = node_id
+
     return key_to_getter_node_ids, key_to_setter_node_id
 
 
-def search_setter_getter_from_workflow(json_data):
+def search_setter_getter_from_workflow_test(json_data):
     key_to_getter_node_ids = {}
     key_to_setter_node_id = {}
 
@@ -345,6 +351,22 @@ def search_setter_getter_from_workflow(json_data):
     print_info(f"{not_included_nodes_count} Nodes not included in prompt but is activated")
     return key_to_getter_node_ids, key_to_setter_node_id
 
+def search_setter_getter_from_workflow(json_data):
+    key_to_getter_node_ids = {}
+    key_to_setter_node_id = {}
+
+    workflow = json_data["extra_data"]["extra_pnginfo"]["workflow"]
+    nodes = workflow["nodes"]
+    prompt = json_data["prompt"]
+
+    not_included_nodes_count = 0
+    for node in nodes:
+        if node["mode"] == 0 and node["id"] not in prompt:
+            not_included_nodes_count += 1
+
+    print_info(f"{not_included_nodes_count} Nodes not included in prompt but is activated")
+    return key_to_getter_node_ids, key_to_setter_node_id
+
 
 def connect_to_from_nodes(json_data):
     prompt = json_data["prompt"]
@@ -384,7 +406,7 @@ def workflow_update(json_data):
 
 def on_prompt_handler(json_data):
     try:
-        test_prompt(json_data)
+        # test_prompt(json_data)
         search_setter_getter_from_workflow(json_data)
         connect_to_from_nodes(json_data)
 

nodes/__init__.py

@@ -22,5 +22,6 @@
     "DragNUWAImageCanvas",
     "ImageCropByRatio",
     "ImageCropByRatioAndResize",
-    "ImagesCropByRatioAndResizeBatch"
+    "ImagesCropByRatioAndResizeBatch",
+    "BatchCreativeInterpolationNodeDynamicSettings",
 ]

nodes/komojini_nodes.py

@@ -263,6 +263,54 @@ def run(self, image, tracking_points, **kwargs):
         return (image, tracking_points, )
 
 
+MAX_IMAGE_COUNT = 50
+
+class BatchCreativeInterpolationNodeDynamicSettings:
+    @classmethod
+    def INPUT_TYPES(s):
+        inputs = {
+            "required": {
+                "image_count": ("INT", {"default": 1, "min": 1, "max": MAX_IMAGE_COUNT, "step": 1}),
+            },
+        }
+
+        for i in range(1, MAX_IMAGE_COUNT):
+            if i == 1:
+                inputs["required"][f"frame_distribution_{i}"] = ("INT", {"default": 4, "min": 4, "max": 64, "step": 1})
+            else:
+                inputs["required"][f"frame_distribution_{i}"] = ("INT", {"default": 16, "min": 4, "max": 64, "step": 1})
+            
+            inputs["required"][f"key_frame_influence_{i}"] = ("FLOAT", {"default": 1.0, "min": 0.0, "max": 10.0, "step": 0.1})
+            inputs["required"][f"min_strength_value_{i}"] = ("FLOAT", {"default": 0.0, "min": 0.0, "max": 1.0, "step": 0.1})
+            inputs["required"][f"max_strength_value_{i}"] = ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.1})
+
+        return inputs
+    
+    RETURN_TYPES = ("STRING", "STRING", "STRING",)
+    RETURN_NAMES = ("dynamic_frame_distribution_values", "dynamic_key_frame_influence_values", "dynamic_strength_values",)
+
+    FUNCTION = "run"
+
+    def run(self, image_count, **kwargs):
+        dynamic_frame_distribution_values = ""
+        dynamic_key_frame_influence_values = ""
+        dynamic_strength_values = ""
+
+        previous_frame_distribution = 0
+
+        for i in range(1, image_count+1):
+            previous_frame_distribution += kwargs.get(f"frame_distribution_{i}", 0)
+
+            distribution_value = str(previous_frame_distribution) + ","
+            influence_value = str(kwargs.get(f"key_frame_influence_{i}")) + ","
+            strength_value = "({min},{max}),".format(min=kwargs.get(f"min_strength_value_{i}"), max=kwargs.get(f"max_strength_value_{i}"))
+            
+            dynamic_frame_distribution_values += distribution_value
+            dynamic_key_frame_influence_values += influence_value
+            dynamic_strength_values += strength_value
+
+        return (dynamic_frame_distribution_values[:-1], dynamic_key_frame_influence_values[:-1], dynamic_strength_values[:-1],)
+    
 __all__ = [
     "To",
     "From",
@@ -271,4 +319,5 @@ def run(self, image, tracking_points, **kwargs):
     "FlowBuilder",
     "FlowBuilderSetter",
     "DragNUWAImageCanvas",
+    "BatchCreativeInterpolationNodeDynamicSettings",
 ]