toy_func_text.py

import tkinter as tk
import cv2
import os 
from PIL import Image, ImageTk
from pathlib import Path

# User input 
workDir = "testImg"
scaleParam = 0.2
default = "0"

# Directory stuff 
while Path(workDir).exists() == False:
    print("No such dir, try again: ")
    workDir = input("Enter dir: ")

os.chdir(workDir)

filename = workDir + ".csv"
outfile = Path.cwd().parent / filename

# Image files 
im_files = [f for f in os.listdir(".") if f[-3:] == 'jpg']
images = iter(im_files)
n_img = str(len(im_files))
count = 0

def next_img():
    try:
        global imgfile
        imgfile = next(images)  # get the next image from the iterator
        with open(outfile, "a") as f:
            f.write(imgfile + ";")
        image = cv2.imread(imgfile, flags=cv2.IMREAD_GRAYSCALE)
        image = cv2.resize(image, (0, 0), fx = scaleParam, fy = scaleParam) # Update to fit the screen
        img = Image.fromarray(image)
        imgtk = ImageTk.PhotoImage(image=img)
        label.imgtk = imgtk
        label.configure(image=imgtk)
        T.delete(1.0, tk.END)
        T.insert(1.0, imgfile)
        T.config(width = len(imgfile))
        global count
        count += 1
        counter_str = str(count) + "/" + n_img
        counter.delete(1.0,tk.END)
        counter.insert(1.0,counter_str)
        counter.config(width = len(counter_str))
    except StopIteration:
        top.destroy()

def func1entries(event, default=default):
    entry = w.get()
    if entry == "":
        entry = default 
    with open(outfile, "a") as f:
        f.write(entry + "\n")
    w.delete(0,"end")
    next_img()

def go_back(event):
    #This function should go back and erase the entry in the csv file, and return the iterator to the previous image, and then just do next image function. 
    with open(outfile, "r") as f:
        data = f.readlines()
        data.pop()
        data.pop()
    with open(outfile, "w") as f:
        f.writelines(data)
    i = im_files.index(imgfile) - 1 
    global images
    images = iter(im_files[i:])
    next_img()

# tkinter stuff 
top = tk.Tk()
top.bind('<Return>', func1entries)
top.bind("<Left>", go_back)

label = tk.Label(top)
label.pack()

# Text box
T = tk.Text(top, font=('calibre',15,'normal'), fg = "red", height=1) #UPDATE FONT SIZE IF NECESSARY
T.place(relx = 0.3, rely = 0.1) # PLACE THE FILENAME BOX

# Counter
counter = tk.Text(top, font=('calibre',15,'normal'), fg = "red" , height=1) #UPDATE FONT SIZE IF NECESSARY
counter.place(anchor=tk.NW) # PLACE THE FILENAME BOX

# Entry box 
w = tk.Entry(top, takefocus=True, font=('calibre',15,'normal')) #UPDATE FONT SIZE IF NECESSARY
w.place(relx = 0.2, rely = 0.9) # PLACE THE ENTRY BOX
w.focus()

next_img()

top.mainloop()