Update endpoints/antibot.js

endpoints/antibot.js

@@ -1,184 +1,64 @@
-const fs = require("fs");
-const https = require("https");
-const path = require("path");
-const sharp = require("sharp");
+const fs = require('fs');
+const path = require('path');
+const { extractTextFromImage } = require('./imageProcessor');
 
-const buf = b => Buffer.from(b.replace(/^data:image\/\w+;base64,/, ""), "base64");
-
-// ==========================
-// AI GEMINI OCR (Original tanpa perubahan)
-// ==========================
-async function aiOcr(imageBuffer) {
-    return new Promise((resolve, reject) => {
-        const base64Image = imageBuffer.toString('base64');
-
-        const requestData = JSON.stringify({
-            "contents": [
-                {
-                    "parts": [
-                        {
-                            "inline_data": {
-                                "mime_type": "image/png",
-                                "data": base64Image
-                            }
-                        },
-                        {
-                            "text": "Berikan text yang ada di gambar ini saja, tidak ada informasi lain cukup yang ada di gambar saja"
-                        }
-                    ]
-                }
-            ]
-        });
-
-        const options = {
-            hostname: 'generativelanguage.googleapis.com',
-            path: '/v1beta/models/gemini-2.5-flash-lite:generateContent',
-            method: 'POST',
-            headers: {
-                'x-goog-api-key': 'AIzaSyBvQUzfkfgmdxMHmOVFEXdQq3DFi4siRVE',
-                'Content-Type': 'application/json',
-                'Content-Length': Buffer.byteLength(requestData)
-            }
-        };
-
-        const req = https.request(options, (res) => {
-            let data = '';
-
-            res.on('data', (chunk) => {
-                data += chunk;
-            });
-
-            res.on('end', () => {
-                try {
-                    const response = JSON.parse(data);
-                    if (response.candidates && response.candidates[0] && response.candidates[0].content) {
-                        const text = response.candidates[0].content.parts[0].text;
-                        resolve(text.trim());
-                    } else {
-                        resolve("");
-                    }
-                } catch (error) {
-                    reject(error);
-                }
-            });
-        });
-
-        req.on('error', (error) => {
-            reject(error);
-        });
-
-        req.write(requestData);
-        req.end();
-    });
-}
-
-// ==========================
-// OCR PREP (Preprocessing aja)
-// ==========================
-async function preprocessSoal(b) {
-  return sharp(b)
-    .resize({ width: 1000 })
-    .grayscale()
-    .normalize()
-    .sharpen({ sigma: 3 }) 
-    .median(3) 
-    .threshold(120) 
-    .toBuffer();
-}
-
-async function preprocessBot(b) {
-  return sharp(b)
-    .resize({ width: 800 })
-    .grayscale()
-    .normalize()
-    .sharpen({ sigma: 1.5 }) 
-    .median(1) 
-    .toBuffer();
-}
-
-// ==========================
-// SIMPLE MATCHING (Berdasarkan text asli Gemini)
-// ==========================
-function simpleMatch(soalText, botText) {
-  // Cek apakah ada substring yang sama
-  return soalText.includes(botText) || botText.includes(soalText);
-}
-
-// ==========================
-// MAIN FUNCTION - TEXT ASLI GEMINI
-// ==========================
-module.exports = async data => {
+async function antibot(data) {
   try {
-    // 1. PROSES SOAL & EKSTRAKSI
-    const soalImg = buf(data.main);
-    const soalProcessed = await preprocessSoal(soalImg);
-    const soalText = await aiOcr(soalProcessed);
-    
-    console.log("SOAL TEXT FROM GEMINI:", soalText);
-    
-    let soalRaw = [];
+    const { main, bots } = data;
 
-    // Pisahkan berdasarkan koma atau ambil 3 kata pertama
-    if (soalText.includes(',')) {
-        soalRaw = soalText.split(',').map(s => s.trim()).filter(s => s.length > 0);
-    } else {
-        soalRaw = soalText.split(' ').filter(s => s.length > 0);
-    }
+    // Process main image (soal)
+    const mainBuffer = Buffer.from(main, 'base64');
+    const mainText = await extractTextFromImage(mainBuffer);
     
-    // Ambil maksimal 3 soal
-    soalRaw = soalRaw.slice(0, 3);
+    const soalRaw = [mainText.response];
+    const soalLeet = [mainText.response]; // bisa ditambahkan konversi leetcode jika perlu
 
-    // 2. PROSES BOT
+    // Process bot images
     const botResults = [];
-    for (const b of data.bots) {
-      const d = buf(b.img);
-      const p = await preprocessBot(d);
-      const botText = await aiOcr(p); 
-
-      console.log(`BOT ${b.id} TEXT:`, botText);
-      
-      botResults.push({
-        id: b.id,
-        text: botText, // Text asli dari Gemini
-        original: botText // Simpan original juga
-      });
-    }
-
-    // 3. LOGIKA PENCERCOKAN SEDERHANA
-    const result = [];
-    const usedBots = new Set();
-
-    // Cari kecocokan untuk setiap soal
-    for (const soal of soalRaw) {
-      let foundBot = null;
-      
-      for (const bot of botResults) {
-        if (usedBots.has(bot.id)) continue;
+    
+    for (const bot of bots) {
+      try {
+        const botBuffer = Buffer.from(bot.img, 'base64');
+        const botText = await extractTextFromImage(botBuffer);
         
-        // Cek kecocokan sederhana
-        if (simpleMatch(soal.toLowerCase(), bot.text.toLowerCase())) {
-          foundBot = bot.id;
-          usedBots.add(bot.id);
-          break;
-        }
+        botResults.push({
+          id: bot.id,
+          text: botText.response,
+          value: botText.response // atau konversi ke leetcode jika perlu
+        });
+      } catch (error) {
+        console.error(`Error processing bot ${bot.id}:`, error);
+        botResults.push({
+          id: bot.id,
+          text: '',
+          value: '',
+          error: error.message
+        });
       }
-      
-      result.push(foundBot);
     }
 
-    // 4. KEMBALIKAN HASIL
     return {
-      soal: soalRaw,
-      botResults: botResults,
-      result: result
+      success: true,
+      data: {
+        soal: soalRaw,
+        soalLeet: soalLeet,
+        botResults: botResults,
+        result: botResults // atau hasil processing lainnya
+      }
     };
-    
+
   } catch (error) {
-    console.error("Main function error:", error);
+    console.error("Antibot function error:", error);
     return {
-      soal: [],
-      botResults: [],
-      result: []
+      success: false,
+      error: error.message,
+      data: {
+        soal: [],
+        botResults: [],
+        result: []
+      }
     };
   }
-};
+}
+
+module.exports = antibot;