Update ai_tag_extractor.py

script/ai_tag_extractor.py

@@ -18,7 +18,7 @@ def __init__(self):
         self.root = Tk()
         self.frame = None
         self.frame_index = 0
-        self.root.geometry('450x350')
+        self.root.geometry('500x350')
         self.root.title('Tag Extractor')
         self.root.protocol("WM_DELETE_WINDOW", self.quit)
         # maker info
@@ -50,7 +50,7 @@ def __init__(self, master):
         self.data_dict = {}
         self.novelai_dict = {}
         self.all_text = Text(self, width=30, height=10)
-        self.all_text.grid(column=2, row=0, sticky=N+W, columnspan=2, rowspan=5)
+        self.all_text.grid(column=2, row=0, sticky=N+W, columnspan=3, rowspan=5)
         self.all_label = Label(self, text="all info")
         self.all_label.grid(column=2, row=3)
 
@@ -70,7 +70,19 @@ def __init__(self, master):
         self.import_button.grid(column=0, row=6, sticky=N+W)
         self.save_button = Button(self, text='save', command=self.save_set)
         self.save_button.grid(column=1, row=6, sticky=N+W)
+        self.clear_button = Button(self, text='clear', command=self.text_clear)
+        self.clear_button.grid(column=2, row=6, sticky=N+W)
+
+        self.status_name = StringVar()
+        self.status = 'None'
+        self.now_status_str = 'Now status : '
+        self.status_display = Label(self, textvariable=self.status_name)
+        self.status_display.grid(column=0, row=7, sticky=N+W, rowspan=3)
+        self.set_label_text()
 
+    def set_label_text(self):
+        self.status_name.set(self.now_status_str + self.status)
+
     def add_text(self, text):
         if type(text) == dict:
             if 'Title' in text:# for novelai image
@@ -93,6 +105,8 @@ def text_clear(self):
         self.prompt_text.delete(1.0, 'end')
         self.negative_text.delete(1.0, 'end')
         self.parameter_text.delete(1.0, 'end')
+        self.status = 'had clear'
+        self.set_label_text()
 
     def get_info(self):# use info dict to get info
         if 'Title' in self.novelai_dict:# for novelai image
@@ -116,17 +130,23 @@ def dict_to_text(self, dict1):# change info dict to txt format
 
     def import_set(self):# import image to get info
         set_path = filedialog.askopenfilename(filetypes = (("png file","*.png"),("all files","*.*")))
+        self.status = 'loading'
+        self.set_label_text()
         if set_path:
             self.text_clear()# now will clear text box after import
             im = Image.open(set_path)
             im.load()
             self.data_dict['file_name'] = set_path.split('/')[-1].split('.')[0]
             self.add_text(im.info)
+            self.status = 'had imported set'
+            self.set_label_text()
 
     def save_set(self):# save to txt
         f = open(self.data_dict['file_name'] + ".txt", "w")
         f.write(str(self.data_dict['spell']))
         f.close()
+        self.status = 'save complete'
+        self.set_label_text()
 
 main = Gui_helper_main()
 main.run()