GUI.py

import tkinter as tk
from tkinter.filedialog import askopenfilename
import os
import sys
import threading

import customtkinter as ctk
import PIL
from PIL import Image, ImageDraw

from ocr_passport import ocr_passport

ctk.set_default_color_theme("blue")  # Themes: "blue" (standard), "green", "dark-blue".
ctk.set_appearance_mode("Dark")  # Modes: "System" (standard), "Dark", "Light".


class Window(ctk.CTk):

    WIDTH = 1431
    HEIGHT = 805
    
    LARGE_FONT = ("Calibri", 42)
    MEDIUM_FONT = ("Calibri Light", 32)
    SMALL_FONT = ("Calibri Light", 28)

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        
        # Load the images used in the program.
        self.ARROW_DARK = Image.open(os.path.join(sys.path[0]+"\Icons", "Arrow_Dark.png"))
        self.ARROW_LIGHT = Image.open(os.path.join(sys.path[0]+"\Icons", "Arrow_Light.png"))
        self.arrow_image = ctk.CTkImage(self.ARROW_DARK, self.ARROW_LIGHT, (100, 100))
        
        self.CHECK_MRZ_DARK = Image.open(os.path.join(sys.path[0]+"\Icons", "Check_MRZ_Dark.png"))
        self.CHECK_MRZ_LIGHT = Image.open(os.path.join(sys.path[0]+"\Icons", "Check_MRZ_Light.png"))
        self.check_mrz = ctk.CTkImage(self.CHECK_MRZ_DARK, self.CHECK_MRZ_LIGHT, (110, 110))
        
        self.LOADING_DARK = Image.open(os.path.join(sys.path[0]+"\Icons", "Loading_Dark.png"))
        self.LOADING_LIGHT = Image.open(os.path.join(sys.path[0]+"\Icons", "Loading_Light.png"))
        self.loading_image = ctk.CTkImage(self.LOADING_DARK, self.LOADING_LIGHT, (115, 115))
        # --------------------------------------------------
        
        self.iconbitmap(os.path.join(sys.path[0]+"\Icons", "Check_MRZ_Light.ico"))
        
        self.WINDOW_CENTERING_X = int(self.winfo_screenwidth()/2 - self.WIDTH/2)
        self.WINDOW_CENTERING_Y = int(self.winfo_screenheight()/2 - self.HEIGHT/2)
        
        self.title("OCR Your Passport")
        self.geometry(
            f"{self.WIDTH}x{self.HEIGHT}+{self.WINDOW_CENTERING_X}+{self.WINDOW_CENTERING_Y}"
        )
        self.minsize(1400, 789)

        self.grid_rowconfigure((0, 1, 2, 3), weight=1)
        self.grid_columnconfigure((0, 1, 2), weight=1)
        
        self.create_widgets()
        
    def create_widgets(self):
        """
        Fill the window with widgets.
        """
        self.label_1 = ctk.CTkLabel(self, text="Optical Character\nRecognition of Your\nPassport", 
                                    font=self.LARGE_FONT, width=600, height=150, 
                                    fg_color=("gray83", "gray30"), corner_radius=10)
        self.label_1.grid(row=0, column=0, padx=(25, 10), pady=(40, 30), sticky="nswe")
        
        self.label_2 = ctk.CTkLabel(self, text="Select a Passport\nPhoto", 
                                    font=self.LARGE_FONT, width=600)
        self.label_2.grid(row=0, column=2, padx=(10, 25), pady=(40, 30), sticky="nswe")
        
        self.button = ctk.CTkButton(self, text="", width=170, height=80, corner_radius=10, 
                                    image=self.check_mrz, command=self.set_mrz)
        self.button.grid(row=1, column=2, padx=10, pady=(0, 30))
        
        self.loading = ctk.CTkLabel(self, text="Loading Results...", compound="left", 
                                    font=self.SMALL_FONT, image=self.loading_image, justify="left")
        
        self.frame = ctk.CTkFrame(self, height=330, width=600, fg_color="transparent", corner_radius=10)
        self.frame.grid(row=2, column=0, padx=10, pady=(0, 20))
        self.passport_label = ctk.CTkLabel(self.frame, text="", height=330, width=600)
        self.passport_label.pack()
        
        self.arrow_frame = ctk.CTkFrame(self, width=100, fg_color="transparent")
        self.arrow_frame.grid(row=2, column=1, padx=10)
        self.arrow_label = ctk.CTkLabel(self.arrow_frame, image=self.arrow_image, 
                                        text="", width=100)
        
        self.mrz_textbox = ctk.CTkTextbox(self, height=330, width=600, corner_radius=10, 
                                          font=("Consolas", 24, "bold"), wrap=tk.NONE, 
                                          padx=10, pady=10, border_spacing=10)

        self.frame_theme = ctk.CTkFrame(self, height=10, fg_color="transparent")
        self.frame_theme.grid(row=3, column=0, columnspan=3, sticky="we")

        self.label_theme = ctk.CTkLabel(self.frame_theme, text="Window Appearance:", font=("Calibri", 24))
        self.appearance_mode = ctk.CTkOptionMenu(self.frame_theme, font=("Calibri", 20), 
                                                 values=["Light", "Dark"], text_color=("#1f1f1f", "#ebebeb"), 
                                                 command=self.change_appearance_mode)
        self.appearance_mode.set("Dark")
        self.label_theme.pack(pady=(0, 20), padx=(20, 15), side=tk.LEFT, expand=1, anchor=tk.E)
        self.appearance_mode.pack(pady=(0, 20), padx=(15, 20), side=tk.LEFT, expand=1, anchor=tk.W)
                
    def set_mrz(self):
        self.passport_path = askopenfilename(parent=self, title="Select an Image File")
        
        if self.passport_path:
            self.arrow_label.pack_forget()
            self.passport_label.configure(image=None)
            self.mrz_textbox.grid_remove()
            
            self.loading.grid(row=1, column=0, padx=10, pady=(0, 30))
            
            self.mrz_textbox.delete(0.0, "end")
            
            threading.Thread(target=self.mrz_to_text).start()
        
    def mrz_to_text(self):
        try:
            passport_photo = ctk.CTkImage(self.add_corners(Image.open(self.passport_path), 10), 
                                          size=(600, 330))
            mrz_text = ocr_passport(self.passport_path)[1]
            self.mrz_textbox.insert(0.0, mrz_text)
            
            if mrz_text != "":
                self.loading.grid_remove()
                self.passport_label.configure(image=passport_photo)
                self.arrow_label.pack()
                self.mrz_textbox.grid(row=2, column=2, padx=(10, 25), pady=(0, 20))

            else:
                self.loading.grid_remove()
                self.mrz_textbox.insert(0.0, "Unable to Load Results\n\nMRZ Not Detected.")
                self.mrz_textbox.grid(row=2, column=2, padx=(10, 25), pady=(0, 20))
                
        except ValueError:
            self.loading.grid_remove()
            self.mrz_textbox.insert(0.0, "Unable to Load Results\n\nPlease select another photo.")
            self.mrz_textbox.grid(row=2, column=2, padx=(10, 25), pady=(0, 20))
            
        except PIL.UnidentifiedImageError:
                self.loading.grid_remove()
                self.mrz_textbox.insert(0.0, "Unable to Load Results\n\nPlease select another file.")
                self.mrz_textbox.grid(row=2, column=2, padx=(10, 25), pady=(0, 20))
                
        except:
            self.loading.grid_remove()
            self.mrz_textbox.insert(0.0, "Unable to Load Results\n\nPlease try again.")
            self.mrz_textbox.grid(row=2, column=2, padx=(10, 25), pady=(0, 20))

    def change_appearance_mode(self, new_appearance_mode):
        if new_appearance_mode == "Dark":
            ctk.set_appearance_mode("Dark")
        elif new_appearance_mode == "Light":
            ctk.set_appearance_mode("Light")
            
    def add_corners(self, im, rad, *args):
        circle = Image.new('L', (rad * 2, rad * 2), 0)
        draw = ImageDraw.Draw(circle)
        draw.ellipse((0, 0, rad * 2 - 1, rad * 2 - 1), fill=255)
        
        alpha = Image.new('L', im.size, 255)
        w, h = im.size
        
        alpha.paste(circle.crop((0, 0, rad, rad)), (0, 0))
        alpha.paste(circle.crop((0, rad, rad, rad * 2)), (0, h - rad))
        alpha.paste(circle.crop((rad, 0, rad * 2, rad)), (w - rad, 0))
        alpha.paste(circle.crop((rad, rad, rad * 2, rad * 2)), (w - rad, h - rad))
        
        im.putalpha(alpha)
        
        return im


if __name__ == "__main__":
    root = Window()
    root.mainloop()