Files
revanced-bots/server/events/addTrainData.js
2022-11-16 13:59:55 +03:00

23 lines
566 B
JavaScript

import { readFileSync, writeFileSync } from 'node:fs';
import { join } from 'node:path';
export default function addTrainData(eventData) {
const file = readFileSync(
join(global.__dirname, global.config.fasttext.trainFile),
'utf-8'
);
const data = file.split('\n');
const { label, text } = eventData;
const labelIndex = data.findIndex((data) => data.startsWith(label));
data.splice(labelIndex === -1 ? 0 : labelIndex, 0, `${label} ${text}`);
writeFileSync(
join(global.__dirname, global.config.fasttext.trainFile),
data.join('\n')
);
return;
}