aboutsummaryrefslogtreecommitdiff
path: root/gui.py
blob: 19d265cfd5f206b69aa8d5ea90956e07a3989f52 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
from tkinter import *
from tkinter import filedialog
from tkinter import ttk
from PIL import Image, ImageTk
import logging
from enum import Enum

class GuiMain(object):

    def __init__(self, args):
        self.__master = Tk()
        self.__master.title("file-tagger")
        self.__args = args
        self.__base = StringVar(self.__master, value=args["base"])
        self.__file_dir = StringVar(self.__master, value=args["file_dir"])
        self.__rename = StringVar(self.__master, value=str(args["rename"]))
        self.__tag_metadata = BooleanVar(self.__master, value=args["tag_metadata"])
        self.__predict_images = BooleanVar(self.__master, value=args["predict_images"])
        self.__predict_images_backend = StringVar(self.__master, value=str(args["predict_images_backend"]))
        self.__predict_images_top = StringVar(self.__master, value=str(args["predict_images_top"]))
        self.__predict_images_skip_detail = BooleanVar(self.__master, value=args["predict_images_skip_detail"])
        self.__predict_images_detail_factor = StringVar(self.__master, value=str(args["predict_images_detail_factor"]))
        self.__predict_videos = BooleanVar(self.__master, value=args["predict_videos"])
        self.__predict_videos_key_frames = StringVar(self.__master, value=str(args["predict_videos_key_frames"]))
        self.__gui_tag = BooleanVar(self.__master, value=args["gui_tag"])
        self.__gui_image_length = StringVar(self.__master, value=str(args["gui_image_length"]))
        self.__open_system = BooleanVar(self.__master, value=args["open_system"])
        self.__skip_prompt = BooleanVar(self.__master, value=args["skip_prompt"])
        self.__skip_tagged = BooleanVar(self.__master, value=args["skip_tagged"])
        self.__index = StringVar(self.__master, value=str(args["index"]))

        validate_number = (self.__master.register(self.__validate_number))

        Label(self.__master, text="Base directory with database:").grid(row=0, column=0)
        Entry(self.__master, textvariable=self.__base).grid(row=0, column=1, columnspan=2)
        Button(self.__master, text="Browse", command=lambda: self.__browse(self.__base)).grid(row=0, column=3)
        Label(self.__master, text="File directory for walking:").grid(row=1, column=0)
        Entry(self.__master, textvariable=self.__file_dir).grid(row=1, column=1, columnspan=2)
        Button(self.__master, text="Browse", command=lambda: self.__browse(self.__file_dir)).grid(row=1, column=3)
        Label(self.__master, text="File rename scheme:").grid(row=2, column=0)
        ttk.Combobox(self.__master, textvariable=self.__rename, values=("none", "sha1", "sha256", "cdate", "mdate"), state="readonly").grid(row=2, column=1)
        Checkbutton(self.__master, text="Use metadata as default tags", variable=self.__tag_metadata).grid(row=3, column=0, columnspan=4, sticky=W)
        Checkbutton(self.__master, text="Use prediction for image tagging", variable=self.__predict_images).grid(row=4, column=0, columnspan=4, sticky=W)
        Label(self.__master, text="Image prediction backend:").grid(row=5, column=0)
        ttk.Combobox(self.__master, textvariable=self.__predict_images_backend, values=("torch", "tensorflow", "keras"), state="readonly").grid(row=5, column=1)
        Label(self.__master, text="Number of top results:").grid(row=6, column=0)
        Entry(self.__master, textvariable=self.__predict_images_top, validate='all', validatecommand=(validate_number, '%P')).grid(row=6, column=1, columnspan=1)
        Checkbutton(self.__master, text="Skip detail scan in image prediction", variable=self.__predict_images_skip_detail).grid(row=7, column=0, columnspan=4, sticky=W)
        Label(self.__master, text="Width factor for detail scan:").grid(row=8, column=0)
        Entry(self.__master, textvariable=self.__predict_images_detail_factor, validate='all', validatecommand=(validate_number, '%P')).grid(row=8, column=1, columnspan=1)
        Checkbutton(self.__master, text="Use prediction for video tagging", variable=self.__predict_videos).grid(row=9, column=0, columnspan=4, sticky=W)
        Label(self.__master, text="Number of key frames:").grid(row=10, column=0)
        Entry(self.__master, textvariable=self.__predict_videos_key_frames, validate='all', validatecommand=(validate_number, '%P')).grid(row=10, column=1, columnspan=1)
        Checkbutton(self.__master, text="Show GUI for tagging", variable=self.__gui_tag).grid(row=11, column=0, columnspan=4, sticky=W)
        Label(self.__master, text="Image GUI preview size:").grid(row=12, column=0)
        Entry(self.__master, textvariable=self.__gui_image_length, validate='all', validatecommand=(validate_number, '%P')).grid(row=12, column=1, columnspan=1)
        Checkbutton(self.__master, text="Open all files with system default", variable=self.__open_system).grid(row=13, column=0, columnspan=4, sticky=W)
        Checkbutton(self.__master, text="Skip prompt for file tags", variable=self.__skip_prompt).grid(row=14, column=0, columnspan=4, sticky=W)
        Checkbutton(self.__master, text="Skip already tagged files", variable=self.__skip_tagged).grid(row=15, column=0, columnspan=4, sticky=W)
        Label(self.__master, text="Start at index:").grid(row=16, column=0)
        Entry(self.__master, textvariable=self.__index, validate='all', validatecommand=(validate_number, '%P')).grid(row=16, column=1, columnspan=1)
        Button(self.__master, text="Start", command=self.__master.destroy).grid(row=17, column=0, columnspan=4)

    def loop(self):
        self.__master.mainloop()

        self.__args["base"] = self.__base.get()
        self.__args["file_dir"] = self.__file_dir.get()
        self.__args["tag_metadata"] = self.__tag_metadata.get()
        self.__args["rename"] = self.__rename.get()
        self.__args["predict_images"] = self.__predict_images.get()
        self.__args["predict_images_backend"] = self.__predict_images_backend.get()
        self.__args["predict_images_top"] = int(self.__predict_images_top.get())
        self.__args["predict_images_skip_detail"] = self.__predict_images_skip_detail.get()
        self.__args["predict_images_detail_factor"] = int(self.__predict_images_detail_factor.get())
        self.__args["predict_videos"] = self.__predict_videos.get()
        self.__args["predict_videos_key_frames"] = int(self.__predict_videos_key_frames.get())
        self.__args["gui_tag"] = self.__gui_tag.get()
        self.__args["gui_image_length"] = int(self.__gui_image_length.get())
        self.__args["open_system"] = self.__open_system.get()
        self.__args["skip_prompt"] = self.__skip_prompt.get()
        self.__args["skip_tagged"] = self.__skip_tagged.get()
        self.__args["index"] = int(self.__index.get())
        return self.__args

    def __browse(self, filepath):
        filename = filedialog.askdirectory()
        filepath.set(filename)

    def __validate_number(self, P):
        if str.isdigit(P) or P == "":
            return True
        else:
            return False

class GuiTag(object):
    RETURN_NEXT = 0
    RETURN_ABORT = 1

    def __init__(self, index, file, tags):
        self.__ret = self.RETURN_NEXT
        self.__master = Tk()
        self.__master.title("Tag file {}".format(file))
        self.__tags = StringVar(self.__master, value=','.join(tags))
        Label(self.__master, text="Index: {}".format(index)).grid(row=0, column=0, columnspan=2)
        Label(self.__master, text="File: {}".format(file)).grid(row=1, column=0, columnspan=2)
        Entry(self.__master, textvariable=self.__tags).grid(row=2, column=0, columnspan=2, sticky="we")
        Button(self.__master, text="Next", command=self.__handle_next).grid(row=3, column=0)
        Button(self.__master, text="Abort", command=self.__handle_abort).grid(row=3, column=1)

    def loop(self):
        self.__master.mainloop()
        return (self.__ret, self.__tags.get().split(","))

    def __handle_next(self):
        self.__ret = self.RETURN_NEXT
        self.__master.destroy()

    def __handle_abort(self):
        self.__ret = self.RETURN_ABORT
        self.__master.destroy()

class GuiImage(object):
    RETURN_NEXT = 0,
    RETURN_ROTATE_90_COUNTERCLOCKWISE = 1,
    RETURN_ROTATE_90_CLOCKWISE = 2,
    RETURN_ABORT = 3

    def __init__(self, index, file, img, tags):
        self.__ret = self.RETURN_NEXT
        self.__master = Tk()
        self.__master.title("Tag file {}".format(file))
        self.__tags = StringVar(self.__master, value=','.join(tags))
        self.__image_pil = Image.fromarray(img)
        self.__image = ImageTk.PhotoImage(image=self.__image_pil)
        Label(self.__master, text="Index: {}".format(index)).grid(row=0, column=0, columnspan=4)
        Label(self.__master, text="File: {}".format(file)).grid(row=1, column=0, columnspan=4)
        self.__label = Label(self.__master, width=img.shape[1], height=img.shape[0], image=self.__image)
        self.__label.grid(row=2, column=0, columnspan=4)
        Entry(self.__master, textvariable=self.__tags).grid(row=3, column=0, columnspan=4, sticky="we")
        Button(self.__master, text="↺", command=self.__handle_rotate_90_counterclockwise).grid(row=4, column=0)
        Button(self.__master, text="↻", command=self.__handle_rotate_90_clockwise).grid(row=4, column=1)
        Button(self.__master, text="Next", command=self.__handle_next).grid(row=4, column=2)
        Button(self.__master, text="Abort", command=self.__handle_abort).grid(row=4, column=3)

    def loop(self):
        self.__master.mainloop()
        return (self.__ret, self.__tags.get().split(","))

    def __handle_rotate_90_counterclockwise(self):
        self.__image_pil = self.__image_pil.rotate(90)
        self.__image = ImageTk.PhotoImage(image=self.__image_pil)
        self.__label.config(image=self.__image)
    
    def __handle_rotate_90_clockwise(self):
        self.__image_pil = self.__image_pil.rotate(-90)
        self.__image = ImageTk.PhotoImage(image=self.__image_pil)
        self.__label.config(image=self.__image)

    def __handle_next(self):
        self.__ret = self.RETURN_NEXT
        self.__master.destroy()

    def __handle_abort(self):
        self.__ret = self.RETURN_ABORT
        self.__master.destroy()