From 7dd23bb2f3fbddc9c8e7a9c9d83e37977d1bfa8e Mon Sep 17 00:00:00 2001
From: Daniel Mauricio Flores <dnielfs@gmail.com>
Date: Mon, 26 Aug 2024 22:44:28 +0200
Subject: [PATCH] fix: use existing worflow inputFields when parsing it

---
 .../services/settings/workflows/parseWorkflow.ts | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/packages/app/src/services/settings/workflows/parseWorkflow.ts b/packages/app/src/services/settings/workflows/parseWorkflow.ts
index d5f1ca62..c6eb1ffd 100644
--- a/packages/app/src/services/settings/workflows/parseWorkflow.ts
+++ b/packages/app/src/services/settings/workflows/parseWorkflow.ts
@@ -49,10 +49,18 @@ export function parseWorkflow(
       throw new Error(`the workflow data seems invalid`)
     }
     if (maybeWorkflow.engine == ClapWorkflowEngine.COMFYUI_WORKFLOW) {
-      const { inputFields, inputValues } =
-        convertComfyUiWorkflowApiToClapWorkflow(maybeWorkflow.data)
-      maybeWorkflow.inputFields = inputFields
-      maybeWorkflow.inputValues = inputValues
+      // The `data` comfyui workflow doesn't have info about custom
+      // inputFields/inputValues, it gets only the default ones based on the nodes
+      const {
+        inputFields: defaultInputFields,
+        inputValues: defaultInputValues,
+      } = convertComfyUiWorkflowApiToClapWorkflow(maybeWorkflow.data)
+      // Use the already existing inputFields/inputValues, otherwise use the default
+      // ones based on the raw comfyui workflow data
+      maybeWorkflow.inputFields =
+        maybeWorkflow.inputFields || defaultInputFields
+      maybeWorkflow.inputValues =
+        maybeWorkflow.inputValues || defaultInputValues
     }
     return maybeWorkflow
   } catch (err) {