diff --git a/server/events/addTrainData.js b/server/events/addTrainData.js new file mode 100644 index 0000000..8f2fbf5 --- /dev/null +++ b/server/events/addTrainData.js @@ -0,0 +1,16 @@ +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.trainData), '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.trainData), data.join('\n')); + + return; +} \ No newline at end of file diff --git a/server/events/ai.js b/server/events/ai.js index 75079b2..473a1e9 100644 --- a/server/events/ai.js +++ b/server/events/ai.js @@ -7,7 +7,7 @@ export default async function runAI(client, data) { id: data.id, predictions }; - console.log(predictions) + const bsonData = serialize(jsonData); client.write(bsonData); diff --git a/server/index.js b/server/index.js index 6f0e3d6..c116777 100644 --- a/server/index.js +++ b/server/index.js @@ -9,7 +9,6 @@ global.__dirname = dirname(__filename); const configJSON = readFileSync('./config.json', 'utf-8'); const config = JSON.parse(configJSON); global.config = config; -console.log(config); import { createServer } from 'node:net'; import { deserialize } from 'bson'; import FastText from 'fasttext.js';