gui.py

import tkinter
import tkinter.ttk
import cv2
import PIL.Image
import PIL.ImageTk
import time
from tkinter import messagebox
from tkinter import filedialog
import os
import threading

import GoogleAPI as G
import ALPR as A
import checkout_system as C
import imageManagement as I


class MyApp:
    # consts
    IMAGE_DIR = "images/"
    NOFILEIMAGE_DIR = "noimagefile.jpg"
    CAMERA = 0
    OBS = 1

    def __init__(self, window=tkinter.Tk(), window_title="ALPR", video_source=CAMERA):

        # create a top-window
        self.window = window
        self.window.title(window_title)
        self.window.geometry("1300x680")
        self.video_source = video_source


        # Constants
        self.BASE_X = 900
        self.BASE_Y = 500

        # Variables
        self.timeStamp = ""
        self.fileName = tkinter.StringVar(value=os.listdir(self.IMAGE_DIR)[0])
        self.filedir = self.IMAGE_DIR + self.fileName.get()
        self.result = tkinter.StringVar(value="Result")
        self.last_index = 0  # listbox
        self.current_image = self.fileName.get()
        self.ENABLE_tesseract = 0
        self.ENABLE_charRecognition = 0

        # init. widgets
        self.initwidgets()
        self.makemenu()

        # After it is called once, the update method will be automatically called every delay milliseconds
        self.delay = 15
        self.update()

        # start the app
        self.window.mainloop()

    def initwidgets(self):

        # frame
        self.panel = tkinter.Frame(self.window)

        # camera
        self.vid = MyVideoCapture(self.video_source)

        # Tk Image
        self.preview = PIL.ImageTk.PhotoImage(file=self.NOFILEIMAGE_DIR)

        # canvas
        self.canvas = tkinter.Canvas(self.window, width=self.vid.width, height=self.vid.height)
        self.preview_canvas = tkinter.Canvas(self.window, width=self.vid.width, height=self.vid.height)
        self.preview_canvas.create_image(0, 0, image=self.preview, anchor=tkinter.NW)

        # message box
        self.messagebox = messagebox.Message()

        # label
        self.lb = tkinter.Label(self.panel, text="Result", textvariable=self.result, font=("arial", 15))

        # button
        self.btn_snapshot = tkinter.Button(self.panel, width=10, height=1, text="Snapshot", command=self.snapshot,
                                           font=("arial", 15), bg='green', fg='white')
        self.btn_send = tkinter.Button(self.panel, width=10, height=1, text="Google API",
                                       command=self.result_from_google, font=("arial", 15), bg='blue', fg='white')
        self.btn_fee = tkinter.Button(self.panel, width=10, height=1, text="Checkout", command=self.fee,
                                      font=("arial", 15), bg='green', fg='white')
        self.btn_quit = tkinter.Button(self.panel, width=10, height=1, text="QUIT", command=self.window.destroy,
                                       font=("arial", 15), bg='red', fg='white')
        self.btn_tesseract = tkinter.Button(self.panel, width=10, height=1, text='tesseract(off)',
                                            command=self.tesseract_enable, font=("arial", 15), bg='orange')
        self.btn_charRecognition = tkinter.Button(self.panel, width=20, height=1, text='charRecognition(off)',
                                                  command=self.charRecognition_enable, font=("arial", 15), bg='orange')
        self.btn_images = tkinter.Button(self.panel, width=10, height=1, text="Images", command=self.image_window,
                                         font=("arial", 15), bg='green', fg='white')

        # combobox
        self.cb = tkinter.ttk.Combobox(self.window, width=58, textvariable=self.fileName)

        # pre-process
        self.makeimageslist()

        # layout
        self.canvas.grid(column=0, row=0, sticky='w')
        self.preview_canvas.grid(column=1, row=0, sticky='e')
        self.cb.grid(column=0, row=1, sticky='ne')
        self.panel.grid(column=1, row=1, sticky='nw')

        # panel
        self.lb.grid(column=0, row=0)
        self.btn_tesseract.grid(column=0, row=1)
        self.btn_snapshot.grid(column=0, row=2)
        self.btn_send.grid(column=0, row=3)

        self.btn_charRecognition.grid(column=1, row=1)
        self.btn_images.grid(column=1, row=2, sticky='w')
        self.btn_fee.grid(column=1, row=3, sticky='w')
        self.btn_quit.grid(column=1, row=4, sticky='w')

    def makemenu(self):
        self.main_menu = tkinter.Menu(self.window)
        self.window.config(menu=self.main_menu)
        self.file_menu = tkinter.Menu(tearoff=False)
        self.main_menu.add("cascade", label="File", menu=self.file_menu)
        self.file_menu.add("command", label='Open Image', command=self.opentheimage)
        self.file_menu.add("command", label='Quit', command=self.window.destroy)
        self.main_menu.add("command", label="About", command=self.about)

    def opentheimage(self):
        askfilename = filedialog.askopenfilename(
            filetypes=(("all files", "*.*"), ("png files", "*.png"), ("jpg files", "*.jpg")))
        self.filedir = askfilename
        img = self.resize(askfilename)
        self.preview = PIL.ImageTk.PhotoImage(img)
        self.preview_canvas.create_image(0, 0, image=self.preview, anchor=tkinter.NW)

    def resize(self, filename):
        img = PIL.Image.open(filename)
        w, h = img.size

        if w > 640:
            r = 640 / w
            h *= r
            img = img.resize((640, int(h)), PIL.Image.ANTIALIAS)
        return img

    def about(self):  # under construction
        messagebox.showinfo(title='about',
                            detail='main window by Josh\ncheckout window by Falice\nimage window by c444569\nReference : \nPython OpenCV - show a video in a Tkinter window by Paul')

    def makeimageslist(self):
        imlist = os.listdir('./images')
        self.cb['values'] = imlist

    def selection_event(self):
        if self.last_index != self.cb.get():
            filename = self.IMAGE_DIR + self.cb.get()
            img = self.resize(filename)
            self.preview = PIL.ImageTk.PhotoImage(img)
            #  self.preview = self.preview.zoom(24)  # zoom in
            #  self.preview = self.preview.subsample(int(self.preview.height() / 250))  # zoom out

            self.preview_canvas.create_image(0, 0, image=self.preview, anchor=tkinter.NW)
            self.filedir = self.IMAGE_DIR + self.fileName.get()
            self.last_index = self.cb.get()

    def snapshot(self):
        self.timeStamp = self.getTimeStamp()
        self.fileName.set("snapshot.png")
        self.filedir = self.IMAGE_DIR + self.fileName.get()

        # Get a frame from the video source
        success, frame = self.vid.get_frame()

        if success:
            cv2.imwrite("images/" + self.fileName.get(), cv2.cvtColor(frame, cv2.COLOR_RGB2BGR))
        self.makeimageslist()

    def fee(self):  # under construction
        app2 = C.Checkout()

    def image_window(self):
        app3 = I.imageMangement()

    def getTimeStamp(self):
        return time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())

    def result_from_google(self):
        result = G.send(self.filedir)
        messagebox.showinfo(title="result", detail=result)
        with open("carinfo.txt", "a") as f:
            f.write(result + "," + self.getTimeStamp() + "\n")
            f.flush()
        cv2.imwrite(self.IMAGE_DIR + result + ".png", cv2.imread(filename=self.IMAGE_DIR + self.fileName.get()))

    def tesseract_enable(self):
        self.ENABLE_tesseract = ~self.ENABLE_tesseract
        self.ENABLE_charRecognition = 0
        if self.ENABLE_tesseract:
            self.btn_tesseract['text'] = "tesseract(on)"
            self.btn_charRecognition['text'] = 'charRecognition(off)'
        else:
            self.btn_tesseract['text'] = 'tesseract(off)'

    def charRecognition_enable(self):
        self.ENABLE_tesseract = 0
        self.ENABLE_charRecognition = ~self.ENABLE_charRecognition
        if self.ENABLE_charRecognition:
            self.btn_charRecognition['text'] = "charRecognition(on)"
            self.btn_tesseract['text'] = "'tesseract(off)'"
        else:
            self.btn_charRecognition['text'] = 'charRecognition(off)'

    def update(self):
        # Get a frame from the video source
        success, frame = self.vid.get_frame()

        if self.ENABLE_tesseract:
            time1 = time.time()
            text, frame = A.alpr(image=frame)
            time2 = time.time()
            print("time : ", time2 - time1)
            self.result.set(text)

        if self.ENABLE_charRecognition:
            time1 = time.time()
            text = A.characterRecognition(frame)
            time2 = time.time()
            print("time : ", time2 - time1)
            self.result.set(text)
            A.displayROI(frame)

        if success:
            self.photo = PIL.ImageTk.PhotoImage(image=PIL.Image.fromarray(frame))
            self.canvas.create_image(0, 0, image=self.photo, anchor=tkinter.NW)

        if os.path.isfile("f"):
            f = open("f", 'r')
            self.fileName.set(f.read())
            f.close()
            os.remove("f")


        self.selection_event()
        self.window.after(self.delay, self.update)


class MyVideoCapture:
    def __init__(self, video_source=0):
        # Open the video source
        self.vid = cv2.VideoCapture(video_source)
        if not self.vid.isOpened():
            raise ValueError("Unable to open video source", video_source)

        # Get video source width and height
        self.width = self.vid.get(cv2.CAP_PROP_FRAME_WIDTH)
        self.height = self.vid.get(cv2.CAP_PROP_FRAME_HEIGHT)

    def get_frame(self):
        if self.vid.isOpened():
            ret, frame = self.vid.read()
            if ret:
                # Return a boolean success flag and the current frame converted to BGR
                return (ret, cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
            else:
                return (ret, None)
        else:
            return (0, None)

    # Release the video source when the object is destroyed
    def __del__(self):
        if self.vid.isOpened():
            self.vid.release()


app = MyApp()