From bb7b565ea4324ddf4f2b7071ca1606393e4a9fd1 Mon Sep 17 00:00:00 2001
From: Kasisnu <5044120+kasisnu@users.noreply.github.com>
Date: Thu, 11 Jul 2024 11:30:54 +0530
Subject: [PATCH] comfy: normals inputs jsonpathing fix

---
 workflows/comfy/ComfyLauncher/ComfyRunnerForSidecar.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/workflows/comfy/ComfyLauncher/ComfyRunnerForSidecar.py b/workflows/comfy/ComfyLauncher/ComfyRunnerForSidecar.py
index 5f7d7f3..d393cbc 100644
--- a/workflows/comfy/ComfyLauncher/ComfyRunnerForSidecar.py
+++ b/workflows/comfy/ComfyLauncher/ComfyRunnerForSidecar.py
@@ -457,10 +457,11 @@ def get_jsonpath_mods(style_mods: Dict[str, Any],
         modifications["$.509.inputs.text"] = travel_in
         
     # TODO The 3 Mappings for depth outlines and normals apparently uses an array.
-    if depth_video_filename != None and outline_video_filename != None and normals_video_filename !=None:
-        modifications["$.3731.widgets_values"][0] = depth_video_filename 
-        modifications["$.4120.widgets_values"][0] = normals_video_filename 
-        modifications["$.4122.widgets_values"][0] = outline_video_filename
+    if pipeline_type.isin([PipelineType.IPA, PipelineType.BASE]):
+        if depth_video_filename != None and outline_video_filename != None and normals_video_filename !=None:
+            modifications["$.3731.inputs.text"] = depth_video_filename
+            modifications["$.4120.inputs.text"] = normals_video_filename
+            modifications["$.4122.inputs.text"] = outline_video_filename
         
     return modifications