diff --git a/server/events/addTrainData.js b/server/events/addTrainData.js index c253873..f530e7c 100644 --- a/server/events/addTrainData.js +++ b/server/events/addTrainData.js @@ -6,9 +6,12 @@ export default function addTrainData(eventData) { join(global.__dirname, global.config.fasttext.trainFile), 'utf-8' ); - const data = file.split('\n'); const { label, text } = eventData; + if (file.includes(text)) return; + + const data = file.split('\n'); + const labelIndex = data.findIndex((data) => data.startsWith(label)); data.splice(labelIndex === -1 ? 0 : labelIndex, 0, `${label} ${text}`);