aboutsummaryrefslogtreecommitdiffstats
path: root/src/background/run-ocr.ts
blob: 1bf605b9feb16e4e53c206309c61cb3f5c7316b4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import Tesseract from "tesseract.js";
//
// chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
// 	if (message.type === "OCR") {
// 		(async () => {
// 			const { data: { text, confidence } } = await Tesseract.recognize(message.dataURL, "eng+osd", {
// 				logger: m => console.log(m),
// 			});
// 			sendResponse({ text, confidence });
// 		})();
// 		return true;
// 	}
//
// 	return false;
// });

chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
	if (message.type === "OCR") {
		(async () => {
			console.log(message);
			const worker = await Tesseract.createWorker("eng", 1, {
				workerBlobURL: false,
				logger: console.log,
			});

			await worker.setParameters({
				tessedit_char_whitelist: "ABCDEFGHJKLMNPQRSTUVWXYZ23456789",
			});

			const {
				data: { text, confidence }
			} = await worker.recognize(message.dataURL);

			await worker.terminate();

			sendResponse({ type: "OCR_RESULT", text, confidence });
		})();

		return true;
	}

	return false;
});