From 817fa92b583fcc247ce3ac91af3511b5aeb016b3 Mon Sep 17 00:00:00 2001
From: GramingFoxTeam <reis@turkeylist.gq>
Date: Tue, 15 Nov 2022 19:57:54 +0300
Subject: [PATCH] fix: addTrainData not being added to the events

---
 server/events/addTrainData.js | 4 ++--
 server/events/index.js        | 4 +++-
 server/index.js               | 7 ++++++-
 3 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/server/events/addTrainData.js b/server/events/addTrainData.js
index 8f2fbf5..d52f2e8 100644
--- a/server/events/addTrainData.js
+++ b/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;
 }
\ No newline at end of file
diff --git a/server/events/index.js b/server/events/index.js
index 3a00b73..32fc842 100644
--- a/server/events/index.js
+++ b/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
 }
\ No newline at end of file
diff --git a/server/index.js b/server/index.js
index c116777..f1d352b 100644
--- a/server/index.js
+++ b/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;