tkinter_gui.py

import sys
import cv2
import os
import time

from PIL import Image as Img
from PIL import ImageTk
from tkinter import filedialog
from ocr import perform_ocr
import gtts as gTTS
from tkinter import *

import tkinter.messagebox
# from tkinter import Tk, Label, Button, BOTTOM, LEFT, X
import webbrowser

def openHelp():
	webbrowser.open('https://github.com/NJACKWinterOfCode/Printed-Text-recognition-and-conversion/blob/master/README.md')


class Window:

	global imageWindow
	global path
	def __init__(self, master):
	
		master.title("Optical Character Recognition")
		master.configure(background='black')
		master.geometry("800x800+500+300")
		# image window
		self.imageWindow = None
		# input image path 
		self.path = ''

		# Load image button
		self.load_button = Button(master, text="Load image", command=self.select_image, bg="orange", relief=RAISED)
		self.load_button.pack(side=TOP, anchor=E)

		# Extract Text button
		self.extract_button = Button(master, text="Extract text", command=self.extract_text, bg="orange", relief=RAISED)
		self.extract_button.pack(side=TOP, anchor=E, expand="yes")

		# Play Text button
		self.playText_button = Button(master, text="Play text", bg="orange", relief=RAISED)
		self.playText_button.pack(side=TOP, anchor=E)

		# Quit Button
		self.quit_button = Button(master, text="Quit", command=master.quit, bg="orange", relief=RAISED)
		self.quit_button.pack(side=TOP, anchor=E, expand="yes")

	# function to extract text from input image
	def extract_text(self):
		perform_ocr(self.path)

		# pop-up to show extraction is completed and text file is saved
		tkinter.messagebox.showinfo('Success', 'Text file saved')

	# function to select and load image 
	def select_image(self):

		# get image path
		self.path = filedialog.askopenfilename()
		
		if len(self.path) > 0:
			image = cv2.imread(self.path) #read image
			image = cv2.resize(image, (500, 600)) #resize image
			cv2.imwrite("original_image.jpg", image) #save original image
			# swap channels
			image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

			# convert image to PIL format
			image = Img.fromarray(image)

			# convert image to ImageTk format
			image = ImageTk.PhotoImage(image)

		# if the image window is None, initialize it

	def __init__(self, master):
		# frame = Frame(master, width=600, height=600)
		# frame.pack()
		
		# imageWindow = Label()
		# imageWindow.pack(side=TOP, padx=10, pady=10, anchor=S)
		master.title("Optical Character Recognition")
		master.configure(background='black')
		master.geometry("800x800+500+300")
		self.imageWindow = None

		self.load_button = Button(master, text="Load image", command=self.select_image, bg="orange", relief=RAISED)
		self.load_button.pack(side=TOP, anchor=E)

		self.extract_button = Button(master, text="Extract text", bg="orange", relief=RAISED)
		self.extract_button.pack(side=TOP, anchor=E, expand="yes")

		self.playText_button = Button(master, text="Play text", bg="orange", relief=RAISED)
		self.playText_button.pack(side=TOP, anchor=E)

		self.quit_button = Button(master, text="Quit", command=master.quit, bg="orange", relief=RAISED)
		self.quit_button.pack(side=TOP, anchor=E, expand="yes")

	def select_image(self):

		path = filedialog.askopenfilename()
		# imageWindow
		if len(path) > 0:
			image = cv2.imread(path)
			image = cv2.resize(image, (500, 600))

			image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

			image = Img.fromarray(image)

			image = ImageTk.PhotoImage(image)


		if self.imageWindow is None:
			self.imageWindow = Label(image=image)
			self.imageWindow.image = image
			self.imageWindow.pack(side=LEFT)


		else:
			self.imageWindow.configure(image=image)
			self.imageWindow.image = image


root = Tk()

menu = Menu(root)
root.config(menu=menu)
menu.add_command(label="Help", command=openHelp)

b = Window(root)
root.mainloop()