diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx index bc5368bf1..714d61bb9 100644 --- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx +++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx @@ -107,6 +107,7 @@ export default function ParameterComponent({ let groupedObj = groupByFamily(myData, tooltipTitle, left, data.type, flow); if (groupedObj?.length === 0 && flow && flow.length > 0) { + groupedObj = groupByFamilyCustom( myData, tooltipTitle, @@ -114,16 +115,10 @@ export default function ParameterComponent({ data.type, flow ); - - if (groupedObj?.length === 0) { - groupedObj.push({ - family: "custom_components", - type: "", - component: "Please try adding a custom component.", - }); - } } + + if (groupedObj) { refNumberComponents.current = groupedObj[0]?.type?.length; @@ -180,6 +175,7 @@ export default function ParameterComponent({ useEffect(() => { renderTooltips(); }, [tooltipTitle, flow]); + return (
{ @@ -269,8 +271,10 @@ export function groupByFamilyCustom(data, baseClasses, left, type, flow) { } }); + const result = Object.values(groupedResult); + if (left === false) { let resultFiltered = []; flow.forEach((element) => { @@ -283,12 +287,23 @@ export function groupByFamilyCustom(data, baseClasses, left, type, flow) { resultFiltered.push({ family: "custom_components", type: n, - component: element["data"]["node"]["display_name"], + component: element["data"]["node"]["display_name"] }); }); } }); }); + + if(resultFiltered.length === 0) { + Object.keys(groupedResult).forEach((el) => { + resultFiltered.push({ + family: "custom_components", + type: groupedResult[el].type, + component: nodeNames["custom_components"] + }) + }) + } + return resultFiltered; } else { return result;