1 files changed, 31 insertions, 3 deletions
diff --git a/src/background/run-ocr.ts b/src/background/run-ocr.ts
index f6d1339..1bf605b 100644
--- a/src/background/run-ocr.ts
+++ b/src/background/run-ocr.ts
@@ -1,13 +1,41 @@
 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 () => {
-			const { data: { text, confidence } } = await Tesseract.recognize(message.dataURL, "eng+osd", {
-				logger: m => console.log(m),
+			console.log(message);
+			const worker = await Tesseract.createWorker("eng", 1, {
+				workerBlobURL: false,
+				logger: console.log,
 			});
-			sendResponse({ text, confidence });
+
+			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;
 	}