diff --git a/src/frontend/src/utils/reactflowUtils.ts b/src/frontend/src/utils/reactflowUtils.ts
index 4cd19bf7e..49aae5b7a 100644
--- a/src/frontend/src/utils/reactflowUtils.ts
+++ b/src/frontend/src/utils/reactflowUtils.ts
@@ -1234,6 +1234,19 @@ export function templatesGenerator(data: APIObjectType) {
   }, {});
 }
 
+export function extractFieldsFromComponenents(data:APIObjectType ) {
+  const fields = new Set();
+  Object.keys(data).forEach((key) => {
+    Object.keys(data[key]).forEach((kind) => {
+      Object.keys(data[key][kind].template).forEach((field) => {
+        if(data[key][kind].template[field].display_name && data[key][kind].template[field].show)
+        fields.add(data[key][kind].template[field].display_name);
+      });
+    });
+  });
+  return Array.from(fields);
+}
+
 export function downloadFlow(
   flow: FlowType,
   flowName: string,