From a75f22fe2131f15b24d605968872e450067c5516 Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Thu, 27 Apr 2023 19:29:50 -0300
Subject: [PATCH] update template working for import

---
 src/frontend/src/contexts/tabsContext.tsx | 29 ++++++++++++++++-------
 1 file changed, 21 insertions(+), 8 deletions(-)

diff --git a/src/frontend/src/contexts/tabsContext.tsx b/src/frontend/src/contexts/tabsContext.tsx
index 9903bda30..776f63077 100644
--- a/src/frontend/src/contexts/tabsContext.tsx
+++ b/src/frontend/src/contexts/tabsContext.tsx
@@ -66,13 +66,14 @@ export function TabsProvider({ children }: { children: ReactNode }) {
 			let cookieObject: LangFlowState = JSON.parse(cookie);
 			cookieObject.flows.forEach((flow) => {
 				flow.data.nodes.forEach((node) => {
-					node.data.node.template = updateObject(
-						node.data.node.template as TemplateVariableType,
-						templates[node.data.type][
-							"template"
-						] as unknown as TemplateVariableType
-					);
-					console.log(node)
+					if (Object.keys(templates[node.data.type]["template"]).length>0) {
+						node.data.node.template = updateObject(
+							node.data.node.template as TemplateVariableType,
+							templates[node.data.type][
+								"template"
+							] as unknown as TemplateVariableType
+						);
+					}
 				});
 			});
 			setTabIndex(cookieObject.tabIndex);
@@ -127,7 +128,19 @@ export function TabsProvider({ children }: { children: ReactNode }) {
 				// read the file as text
 				file.text().then((text) => {
 					// parse the text into a JSON object
-					addFlow(JSON.parse(text));
+					let flow: FlowType = JSON.parse(text);
+					flow.data.nodes.forEach((node) => {
+						if (Object.keys(templates[node.data.type]["template"]).length>0) {
+							node.data.node.template = updateObject(
+								node.data.node.template as TemplateVariableType,
+								templates[node.data.type][
+									"template"
+								] as unknown as TemplateVariableType
+							);
+						}
+					});
+
+					addFlow();
 				});
 			}
 		};