main.py

import cv2
import pytesseract
import numpy as np
from PIL import ImageGrab
import time

pytesseract.pytesseract.tesseract_cmd = 'C:\\Program Files\\Tesseract-OCR\\tesseract.exe'
img = cv2.imread('sample.png')
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
pytesseract


def txtRecog(choice):
    if (choice == 1):

        ##### Image to String   ######

        print(pytesseract.image_to_string(img))
        cv2.imshow('img', img)
        cv2.waitKey(0)

    elif (choice == 2):

        #### Detecting Characters  ######
        hImg, wImg, _ = img.shape
        boxes = pytesseract.image_to_boxes(img)
        for b in boxes.splitlines():
            print(b)
            b = b.split(' ')
            print(b)
            x, y, w, h = int(b[1]), int(b[2]), int(b[3]), int(b[4])
            cv2.rectangle(img, (x, hImg - y), (w, hImg - h), (50, 50, 255), 2)
            cv2.putText(img, b[0], (x, hImg - y + 25), cv2.FONT_HERSHEY_SIMPLEX, 1, (50, 50, 255), 2)
        cv2.imshow('img', img)
        cv2.waitKey(0)

    elif (choice == 3):

        ##### Detecting Words  ######
        # #[   0          1           2           3           4          5         6       7       8        9        10       11 ]
        # #['level', 'page_num', 'block_num', 'par_num', 'line_num', 'word_num', 'left', 'top', 'width', 'height', 'conf', 'text']
        boxes = pytesseract.image_to_data(img)
        for a, b in enumerate(boxes.splitlines()):
            print(b)
            if a != 0:
                b = b.split()
                if len(b) == 12:
                    x, y, w, h = int(b[6]), int(b[7]), int(b[8]), int(b[9])
                    cv2.putText(img, b[11], (x, y - 5), cv2.FONT_HERSHEY_SIMPLEX, 1, (50, 50, 255), 2)
                    cv2.rectangle(img, (x, y), (x + w, y + h), (50, 50, 255), 2)

        cv2.imshow('img', img)
        cv2.waitKey(0)

    elif (choice == 4):

        ##### Detecting ONLY Digits  ######
        ##############################################
        hImg, wImg, _ = img.shape
        conf = r'--oem 3 --psm 6 outputbase digits'
        boxes = pytesseract.image_to_boxes(img, config=conf)
        for b in boxes.splitlines():
            print(b)
            b = b.split(' ')
            print(b)
            x, y, w, h = int(b[1]), int(b[2]), int(b[3]), int(b[4])
            cv2.rectangle(img, (x, hImg - y), (w, hImg - h), (50, 50, 255), 2)
            cv2.putText(img, b[0], (x, hImg - y + 25), cv2.FONT_HERSHEY_SIMPLEX, 1, (50, 50, 255), 2)

        cv2.imshow('img', img)
        cv2.waitKey(0)

    elif(choice==5):
        print('Exiting....\n')
        exit()

    else:
        print('choose from the given options\n')
        txtRecog(choice)


requirement= int(input('\nchoose your requirement:\n\n1.Extract text from the image\n2.Detect individual characters in image\n3.'
                   'Detect words in image\n4.Detect only digits in image\n5.Exit\n'))

txtRecog(requirement)
exit()