Update endpoints/antibot.js

endpoints/antibot.js

@@ -1,6 +1,25 @@
 const Tesseract = require("tesseract.js");
 const sharp = require("sharp");
-const levenshtein = require("levenshtein-edit-distance");
+
+function levenshtein(s, t) {
+    if (s === t) {
+        return 0;
+    }
+    var n = s.length, m = t.length;
+    if (n === 0) return m;
+    if (m === 0) return n;
+    var d = [];
+    for (var i = 0; i <= n; i++) d[i] = [];
+    for (var i = 0; i <= n; i++) d[i][0] = i;
+    for (var j = 0; j <= m; j++) d[0][j] = j;
+    for (var i = 1; i <= n; i++) {
+        for (var j = 1; j <= m; j++) {
+            var cost = (s.charAt(i - 1) === t.charAt(j - 1)) ? 0 : 1;
+            d[i][j] = Math.min(d[i - 1][j] + 1, d[i][j - 1] + 1, d[i - 1][j - 1] + cost);
+        }
+    }
+    return d[n][m];
+}
 
 const DICTIONARIES = {
     wordnumber: { ONE: '1', TWO: '2', THREE: '3', FOUR: '4', FIVE: '5', SIX: '6', SEVEN: '7', EIGHT: '8', NINE: '9', TEN: '10' },
@@ -13,15 +32,11 @@ const DICTIONARIES = {
     xxx: { '--X': '--+', '-X-': '-+-', 'X--': '+--', 'XX-': '++-', '-XX': '-++', 'X-X': '+-+', '---': '---', 'XXX': '+++', 'X-X-': '+-+-', '-X-X': '-+-+' },
     xox: { '--X': 'OO+', '-X-': 'O+O', 'X--': '+OO', 'XX-': '++O', '-XX': 'O++', 'X-X': '+O+', '---': 'OOO', 'XXX': '+++', 'X-X-': '+O+O', '-X-X': 'O+O+' },
     oxo: { OOX: '--+', OXO: '-+-', XOO: '+--', XXO: '++-', OXX: '-++', XOX: '+-+', OOO: '---', XXX: '+++', XOXO: '+-+-', OXOX: '-+-+' },
-    zoo: { ZOO: '200', OZO: '020', OOZ: '002', SOO: '500', OSO: '050', OOS: '005', LOL: '101', SOS: '505', ZOZ: '202', LLL: '111' },
+    zoo: { ZOO: '200', OZO: '020', OOZ: '002', SOO: '500', '0SO': '050', OOS: '005', LOL: '101', SOS: '505', ZOZ: '202', LLL: '111' },
     ooz: { '200': 'ZOO', '020': 'OZO', '002': 'OOZ', '500': 'SOO', '050': 'OSO', '005': 'OOS', '101': 'LOL', '505': 'SOS', '202': 'ZOZ', '111': 'LLL' },
     animals: { CAT: 'C@T', DOG: 'D0G', LION: '1!0N', TIGER: 'T!G3R', MONKEY: 'M0NK3Y', ELEPHANT: '31EPH@NT', COW: 'C0W', FOX: 'F0X', MOUSE: 'M0US3', ANT: '@NT' }
 };
 
-const LEET_GENERAL = {
-    '0': 'O', '1': 'I', '3': 'E', '4': 'A', '5': 'S', '7': 'T', '8': 'B', '@': 'A', '$': 'S', '!': 'I'
-};
-
 function applyVariations(word, type) {
     let w = word.toUpperCase();
     
@@ -75,12 +90,12 @@ function findBestMatch(scannedText) {
 
         for (const [key, val] of Object.entries(dict)) {
             if (val === variedText) return { type, value: key, original: variedText };
-            
+
             const distKey = levenshtein(variedText, key);
             const distVal = levenshtein(variedText, val);
-            
-            if (distKey <= 1) return { type, value: val, original: key };
-            if (distVal <= 1) return { type, value: key, original: val };
+
+            if (distKey <= 1) return { type, value: val, original: key }; 
+            if (distVal <= 1) return { type, value: key, original: val }; 
         }
     }
     
@@ -113,7 +128,7 @@ async function preprocessImage(imageBuffer, type = 'main') {
 async function ocr(imageBuffer, psmMode = '6') {
     try {
         const result = await Tesseract.recognize(imageBuffer, "eng", {
-            tessedit_char_whitelist: "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+-*=@!$",
+            tessedit_char_whitelist: "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+-*=@!$,.:;",
             tessedit_pageseg_mode: psmMode
         });
         return result.data.text.trim();
@@ -129,9 +144,9 @@ module.exports = async (data) => {
 
         const mainBuffer = buf(data.main);
         const processedMain = await preprocessImage(mainBuffer, 'main');
-        const rawMainText = await ocr(processedMain, '7');
-
-        const mainWords = rawMainText.split(/\s+/).filter(w => w.length > 0);
+        const rawMainText = await ocr(processedMain, '7'); 
+        
+        const mainWords = rawMainText.replace(/[,.]/g, ' ').split(/\s+/).filter(w => w.length > 0);
         console.log("📝 Raw Main Text: ", rawMainText);
         console.log("📝 Parsed Words: ", mainWords);
 
@@ -142,20 +157,20 @@ module.exports = async (data) => {
                 targetConcepts.push(match);
             }
         }
-
+        
         if (targetConcepts.length === 0) {
             targetConcepts = mainWords.map(w => ({ type: 'raw', value: normalize(w) }));
         }
 
-        console.log("🎯 Target Concepts: ", targetConcepts);
+        console.log("🎯 Target Concepts: ", targetConcepts.map(t => t.value));
 
         const botResults = [];
         for (const bot of data.bots) {
             const botBuffer = buf(bot.img);
             const processedBot = await preprocessImage(botBuffer, 'bot');
-            const rawBotText = await ocr(processedBot, '6');
+            const rawBotText = await ocr(processedBot, '6'); 
             const cleanBotText = normalize(rawBotText);
-
+            
             const botAnalysis = findBestMatch(cleanBotText);
             
             botResults.push({
@@ -178,7 +193,7 @@ module.exports = async (data) => {
 
                 if (target.value === bot.analysis.value || target.value === bot.clean) {
                     bestBotId = bot.id;
-                    minDistance = 0;
+                    minDistance = -1;
                     break;
                 }