import { extractBytes, initWasm, registerOcrBackend, TesseractWasmBackend } from "@kreuzberg/wasm";

async function extractWithProgressTracking() {
  await initWasm();

  const backend = new TesseractWasmBackend();

  backend.setProgressCallback((progress: number) => {
    const progressBar = document.getElementById("progress");
    if (progressBar) {
      progressBar.style.width = `${progress}%`;
      progressBar.textContent = `${progress}%`;
    }
  });

  await backend.initialize();
  registerOcrBackend(backend);

  const bytes = new Uint8Array(await fetch("document.png").then((r) => r.arrayBuffer()));

  const result = await extractBytes(bytes, "image/png", {
    ocr: {
      backend: "tesseract-wasm",
      language: "eng",
    },
  });

  console.log("OCR complete");
  console.log(result.content);
}

extractWithProgressTracking().catch(console.error);