ocr.ts

// Menu: Optical Character Recognition
// Description: Extract text from images
// Author: Kent C. Dodds
// Twitter: @kentcdodds

import '@johnlindquist/kit'
import Tesseract from 'tesseract.js'

const clipboardImage = await clipboard.readImage()

if (clipboardImage.byteLength) {
  Tesseract.recognize(clipboardImage, 'eng', {
    logger: m => console.log(m),
  }).then(({data: {text}}) => {
    clipboard.writeText(text)
  })
} else {
  let selectedFiles = await getSelectedFile()
  let filePaths: Array<string>

  if (selectedFiles) {
    filePaths = selectedFiles.split('\n')
  } else {
    let droppedFiles = await drop({placeholder: 'Drop images to compress'})
    filePaths = droppedFiles.map(file => file.path)
  }
  for (const filePath of filePaths) {
    Tesseract.recognize(filePath, 'eng', {logger: m => console.log(m)}).then(
      ({data: {text}}) => {
        clipboard.writeText(text)
      },
    )
  }
}

notify({
  title: 'OCR finished',
  message: `Copied text to your clipboard`,
})