fix: addTrainData not being added to the events

server/events/addTrainData.js

@@ -2,7 +2,7 @@ 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 file = readFileSync(join(global.__dirname, global.config.fasttext.trainFile), 'utf-8');
     const data = file.split('\n');
     const { label, text } = eventData;
 
@@ -10,7 +10,7 @@ export default function addTrainData(eventData) {
 
     data.splice(labelIndex === -1 ? 0 : labelIndex, 0, `${label} ${text}`);
 
-    writeFileSync(join(global.__dirname, global.config.fasttext.trainData), data.join('\n'));
+    writeFileSync(join(global.__dirname, global.config.fasttext.trainFile), data.join('\n'));
 
     return;
 }

server/events/index.js

@@ -1,9 +1,11 @@
 import runAI from './ai.js';
 import trainAI from './trainAI.js';
 import runOCR from './ocr.js';
+import addTrainData from './addTrainData.js';
 
 export {
     runAI,
     trainAI,
-    runOCR
+    runOCR,
+    addTrainData
 }

server/index.js

@@ -12,7 +12,7 @@ global.config = config;
 import { createServer } from 'node:net';
 import { deserialize } from 'bson';
 import FastText from 'fasttext.js';
-import { runAI, trainAI, runOCR } from './events/index.js';
+import { runAI, trainAI, runOCR, addTrainData } from './events/index.js';
 
 const ft = new FastText(global.config.fasttext);
 
@@ -31,6 +31,11 @@ const server = createServer(async (client) => {
                 break;
             };
 
+            case 3: {
+                addTrainData(eventData);
+                break;
+            };
+
             case 4: {
                 trainAI();
                 break;