Geo Face.py

# Import modules -------------------------------------------

import os
from tkinter import *
from tkinter import filedialog
from tkinter.messagebox import showerror
from PIL import ImageTk, Image
import time

# Import custom modules ------------------------------------

import src.backend.calculateFace as searchFace

# Initialization -------------------------------------------

window = Tk()
window.configure(bg="white")
window.geometry("1040x720")
window.resizable(False, False)
window.title("GeoFace")
window.iconbitmap("src/frontend/img/geoface.ico")

# Variables ------------------------------------------------

# 1. Dataset
DATASET = StringVar() 
DATASET.set("")

DATASET_PLACEHOLDER = StringVar()
DATASET_PLACEHOLDER.set("No data chosen")

# 2. Image input
YOUR_IMAGE_PATH = StringVar() 
YOUR_IMAGE_PATH.set("")

YOUR_IMAGE_PLACEHOLDER = StringVar()
YOUR_IMAGE_PLACEHOLDER.set("No file chosen")

# 3. Result Percent
RESULT_STR = StringVar()
RESULT_STR.set("0% Similar")

# 4. Counter
total_counter = 0
COUNTER_STR = StringVar()
COUNTER_STR.set(f"{total_counter:.2f} seconds")

# 5. Result Image
RESULT_IMG_PATH = StringVar()
RESULT_IMG_PATH.set("")

# Functions ------------------------------------------------

def openFolder():
    DATASET_STR = filedialog.askdirectory(
        initialdir=".\\",
        title = "Choose data set",
    )
    if DATASET_STR:
        DATASET.set(DATASET_STR)
        DATASET_PLACEHOLDER.set(os.path.basename(DATASET_STR))

def openFile():
    YOUR_IMAGE_STR = filedialog.askopenfilename(
        initialdir=".\\",
        title = "Choose file",
        filetypes = (("jpg files", "*.jpg"),
        ("png files", "*.png"),
        ("all files", "*.*"))
    )
    if YOUR_IMAGE_STR:
        YOUR_IMAGE_PATH.set(YOUR_IMAGE_STR)
        YOUR_IMAGE_PLACEHOLDER.set(os.path.basename(YOUR_IMAGE_STR))

        yourImage_img = Image.open(YOUR_IMAGE_PATH.get()).resize((280, 280))
        yourImage_img = ImageTk.PhotoImage(yourImage_img)
        yourImage_label.configure(image = yourImage_img)
        yourImage_label.image = yourImage_img



def calculateFace():
    if (DATASET.get() != "") & (YOUR_IMAGE_PATH.get() != ""):
        
        # Start timer
        start_counter = time.time()
        
        # Loading (not working)
        counter_label.config(text="Loading...")
        
        # Call main backend
        print("Calculating face...")
        print("*do not minimize the program*")
        TEMP, SIMILARITY = searchFace.main(DATASET.get(), YOUR_IMAGE_PATH.get())
        RESULT_IMG_PATH.set(TEMP)

        # Configure similarity label
        RESULT_STR.set(f"{SIMILARITY:.2f}% Similar")

        # Configure output iamge label
        resultImage_img = Image.open(RESULT_IMG_PATH.get()).resize((280, 280))
        resultImage_img = ImageTk.PhotoImage(resultImage_img)
        resultImage_label.configure(image = resultImage_img)
        resultImage_label.image = resultImage_img
        
        # End timer
        end_counter = time.time()
        total_counter = end_counter - start_counter
        COUNTER_STR.set(f"{total_counter:.2f} seconds")
    else:
        showerror(title="Error", message="Data set and Image are invalid")

# Components -----------------------------------------------

# 1. Canvas
canvas = Canvas(
    window,
    bg = "#fff",
    height = 720,
    width = 1080,
    bd = 0,
    highlightthickness = 0,
    relief = "ridge"
)
canvas.place(x=0,y=0)

# 2. Background
background_img = PhotoImage(file = f"src/frontend/img/main-background.png")
background = canvas.create_image(
    540.0, 360.0,
    image = background_img
)

# 3. Buttons
button1_img = PhotoImage(file = f"src/frontend/img/button1.png")
button1_entry = Button(
    image = button1_img,
    borderwidth = 0,
    highlightthickness = 0,
    command = openFolder,
    relief = "flat"
)
button1_entry.place(
    x = 76, y = 270,
    width = 115,
    height = 40
)

button2_img = PhotoImage(file = f"src/frontend/img/button2.png")
button2_entry = Button(
    image = button2_img,
    borderwidth = 0,
    highlightthickness = 0,
    command = openFile,
    relief = "flat"
)
button2_entry.place(
    x = 76, y = 375,
    width = 100,
    height = 40
)

button3_img = PhotoImage(file = f"src/frontend/img/search.png")
button3_entry = Button(
    image = button3_img,
    borderwidth = 0,
    highlightthickness = 0,
    command = calculateFace,
    relief = "flat"
)
button3_entry.place(
    x = 76, y = 475,
    width = 210,
    height = 35
)

# 4. Labels
label1_label = Label(
    canvas,
    textvariable = DATASET_PLACEHOLDER,
    font = ("Poppins", 11),
    fg = "#8B8B8B",
    borderwidth = 0,
    background = "#fff",
    highlightthickness= 0,
    anchor = "w",
    padx = 10
)
label1_label.place(
    x = 190, y = 270,
    width = 190,
    height = 40,
)

label2_label = Label(
    canvas,
    textvariable = YOUR_IMAGE_PLACEHOLDER,
    font = ("Poppins", 11),
    fg = "#8B8B8B",
    borderwidth = 0,
    background = "#fff",
    highlightthickness= 0,
    anchor = "w",
    padx = 10
)
label2_label.place(
    x = 180, y = 375,
    width = 190,
    height = 40,
)

result_label = Label(
    canvas,
    textvariable= RESULT_STR,
    font = ("Poppins", 14, "bold"),
    fg = "#86FFA8",
    borderwidth = 0,
    background = "#fff",
    highlightthickness= 0,
    anchor = "w",
    padx = 10
)
result_label.place(
    x = 68, y = 555,
    width = 190,
    height = 40,
)

counter_label = Label(
    canvas,
    textvariable = COUNTER_STR,
    font = ("Poppins", 14),
    fg = "#000",
    borderwidth = 0,
    background = "#fff",
    highlightthickness = 0,
    anchor= "w",
    padx = 10
)
counter_label.place(
    x = 400, y = 555,
    width = 190,
    height = 40 
)

# 5. Image Label
yourImage_label = Label(
    canvas,
    width = 280, height = 280
)
yourImage_label.place(
    x = 408, y = 235,
    width = 280,
    height = 280
)

resultImage_label = Label(
    canvas,
    width = 280, height = 280
)
resultImage_label.place(
    x = 714, y = 235,
    width = 280,
    height = 280
)

window.mainloop()