dataextraction.py

from pdf2image import convert_from_path
import os
import pytesseract
from PIL import Image
import pandas as pd


class text_extract:
    pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'
    fields = []

    def retrieve_text(self, field_name, image_path):
        try:
            img = Image.open(image_path)
            tessdata_dir_config = r'--tessdata-dir "./ocr-layers" --psm 6'
            img_text = pytesseract.image_to_string(
                img,  lang='eng_layer', config=tessdata_dir_config)
            print(field_name + ':' + img_text)

            self.fields.append([field_name, img_text])
        except Exception as e:
            # print(e)
            print(field_name + ' not found')

    def save_fields(self, file_path):
        df = pd.DataFrame(self.fields, columns=["Field", "Value"])
        df.to_csv(file_path+"\\fields.csv")