Update server.js

server.js

@@ -1,37 +1,55 @@
-const express = require("express");
-const multer = require("multer");
-const tesseract = require("node-tesseract-ocr");
-const fs = require("fs");
-const path = require("path");
+const express = require('express');
+const Tesseract = require('tesseract.js');
+const fs = require('fs');
+const path = require('path');
+const { convert } = require('pdf-poppler');
 
 const app = express();
-
-// Pakai /tmp untuk folder upload (selalu bisa di-write di Linux container)
-const uploadDir = "/tmp/uploads";
-if (!fs.existsSync(uploadDir)) {
-  fs.mkdirSync(uploadDir, { recursive: true });
-}
-
-const storage = multer.diskStorage({
-  destination: (req, file, cb) => cb(null, uploadDir),
-  filename: (req, file, cb) => cb(null, Date.now() + path.extname(file.originalname))
+app.use(express.json({ limit: '50mb' }));
+
+// Buat folder uploads (dalam /tmp agar bisa ditulis di HuggingFace)
+const uploadDir = path.join('/tmp', 'uploads');
+if (!fs.existsSync(uploadDir)) fs.mkdirSync(uploadDir, { recursive: true });
+
+app.post('/ocr', async (req, res) => {
+    try {
+        const { file } = req.body;
+        if (!file) {
+            return res.status(400).json({ success: false, error: 'Base64 file string not provided' });
+        }
+
+        // Simpan base64 → PDF
+        const pdfPath = path.join(uploadDir, `file_${Date.now()}.pdf`);
+        fs.writeFileSync(pdfPath, Buffer.from(file, 'base64'));
+
+        // Convert PDF → PNG (halaman pertama saja)
+        const outputBase = pdfPath.replace('.pdf', '');
+        await convert(pdfPath, {
+            format: 'png',
+            out_dir: uploadDir,
+            out_prefix: path.basename(outputBase),
+            page: 1,
+            scale: 1024,
+        });
+
+        const imagePath = path.join(uploadDir, `${path.basename(outputBase)}-1.png`);
+
+        // OCR dengan Tesseract
+        const { data: { text } } = await Tesseract.recognize(imagePath, 'ind+eng');
+
+        // Bersihkan file sementara
+        fs.unlinkSync(pdfPath);
+        fs.unlinkSync(imagePath);
+
+        res.json({ success: true, text });
+    } catch (error) {
+        console.error('OCR Error:', error);
+        res.status(500).json({ success: false, error: error.message });
+    }
 });
 
-const upload = multer({ storage });
-
-app.post("/ocr", upload.single("file"), async (req, res) => {
-  try {
-    const config = { lang: "eng+ind" };
-    const text = await tesseract.recognize(req.file.path, config);
-
-    // Hapus file setelah OCR
-    fs.unlinkSync(req.file.path);
-
-    res.json({ success: true, text });
-  } catch (err) {
-    res.status(500).json({ success: false, error: err.message });
-  }
+// ⚠️ HuggingFace hanya menerima listen ke process.env.PORT
+const PORT = process.env.PORT || 7860;
+app.listen(PORT, '0.0.0.0', () => {
+    console.log(`🚀 OCR server running on http://0.0.0.0:${PORT}`);
 });
-
-const port = process.env.PORT || 7860;
-app.listen(port, () => console.log(`🚀 OCR server running on port ${port}`));