@customize-agent/knowledge 4.0.14 → 4.0.16

package/dist/extraction/ocr-providers.js

@@ -0,0 +1,134 @@
+/**
+ * OCR 提供者 — tesseract.js 跨平台 WASM
+ *
+ * 使用 tesseract.js v7（WASM），bundled traineddata，
+ * 真正跨平台（Windows/macOS/Linux），无需系统依赖。
+ */
+import * as fs from 'node:fs';
+import * as os from 'node:os';
+import * as path from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { resolveAndImport, resolvePackage } from './module-resolver.js';
+// ─── 路径工具 ───────────────────────────────────────────────────
+const knowledgeDir = path.dirname(fileURLToPath(import.meta.url));
+function tessdataDir() {
+    if (process.env.TESSDATA_PREFIX)
+        return process.env.TESSDATA_PREFIX;
+    const pkg = path.resolve(knowledgeDir, '..', '..', 'models', 'tessdata');
+    if (fs.existsSync(pkg))
+        return pkg;
+    return pkg;
+}
+// ─── Tesseract.js Provider（主 OCR 引擎，跨平台） ──────────────
+export class TesseractJsProvider {
+    id = 'tesseract.js';
+    _available = null;
+    worker = null;
+    workerPromise = null;
+    warnings = [];
+    get available() {
+        if (this._available !== null)
+            return this._available;
+        // 检查 traineddata 和 tesseract.js 是否可用
+        const td = tessdataDir();
+        const hasChiSim = fs.existsSync(path.join(td, 'chi_sim.traineddata'));
+        try {
+            resolvePackage('tesseract.js');
+            this._available = hasChiSim;
+        }
+        catch {
+            this._available = false;
+        }
+        return this._available;
+    }
+    async recognize(input) {
+        let pngPath;
+        let tmpDir = null;
+        // 如果传了 filePath，直接使用；否则 raw pixels → PNG
+        if (input.filePath && fs.existsSync(input.filePath)) {
+            pngPath = input.filePath;
+        }
+        else {
+            const sharpMod = await resolveAndImport('sharp');
+            const sharpFn = sharpMod.default ?? sharpMod;
+            tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'ocr-'));
+            pngPath = path.join(tmpDir, 'input.png');
+            const channels = input.channels ?? 3;
+            await sharpFn(Buffer.from(input.data), {
+                raw: { width: input.width, height: input.height, channels },
+            })
+                .removeAlpha().normalize().linear(3.0, -150)
+                .withMetadata({ density: 288 })
+                .png().toFile(pngPath);
+        }
+        try {
+            const worker = await this.getWorker();
+            const result = await worker.recognize(pngPath);
+            const text = (result.data.text ?? '').trim();
+            const lines = (result.data.lines ?? []);
+            const regions = lines
+                .filter((l) => l.text?.trim())
+                .map((l) => ({
+                text: l.text.trim(),
+                confidence: l.confidence ?? 0,
+                box: {
+                    x: l.bbox?.x0 ?? 0,
+                    y: l.bbox?.y0 ?? 0,
+                    width: (l.bbox?.x1 ?? 0) - (l.bbox?.x0 ?? 0),
+                    height: (l.bbox?.y1 ?? 0) - (l.bbox?.y0 ?? 0),
+                },
+            }));
+            return { text, confidence: result.data.confidence ?? 0, regions, warnings: this.getWarnings() };
+        }
+        finally {
+            if (tmpDir)
+                fs.rmSync(tmpDir, { recursive: true, force: true });
+        }
+    }
+    getWarnings() {
+        return [...new Set(this.warnings)].slice(-20);
+    }
+    async getWorker() {
+        if (this.worker)
+            return this.worker;
+        if (!this.workerPromise)
+            this.workerPromise = this.createReusableWorker();
+        this.worker = await this.workerPromise;
+        return this.worker;
+    }
+    async createReusableWorker() {
+        const tessMod = await resolveAndImport('tesseract.js');
+        const { createWorker, OEM, setLogging } = tessMod;
+        if (typeof setLogging === 'function')
+            setLogging(false);
+        const worker = await createWorker('chi_sim', OEM?.LSTM_ONLY ?? 1, {
+            langPath: tessdataDir(),
+            gzip: false,
+            logger: () => undefined,
+            errorHandler: (error) => {
+                const message = error instanceof Error ? error.message : String(error);
+                if (message.trim())
+                    this.warnings.push(message.trim());
+            },
+        });
+        if (typeof worker.setParameters === 'function') {
+            await worker.setParameters({ preserve_interword_spaces: '0', user_defined_dpi: '300' });
+        }
+        return worker;
+    }
+    async dispose() {
+        if (this.worker)
+            await this.worker.terminate();
+        this.worker = null;
+        this.workerPromise = null;
+    }
+}
+// ─── 工厂 ───────────────────────────────────────────────────────
+export async function createOcrProvider() {
+    const tess = new TesseractJsProvider();
+    if (tess.available)
+        return tess;
+    const td = tessdataDir();
+    throw new Error(`OCR 不可用。请将 chi_sim.traineddata 和 eng.traineddata 放置到 ${td}，` +
+        '并确保 tesseract.js 已安装。');
+}