Upload 2 files

Danbooru Prompt Selector/TEST/prompt_selector_auto_1208_testv6.exe

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e359edfa85abd142301b02cc3ebcc59d1040f2effeeb0443b02f6ccae56ccf88
-size 81784429
+oid sha256:2e4593d4f9fc5c8aaf0b3e291f874f76e1718d6c83cdfaca02dcc77d7e6e98e4
+size 84497981

Danbooru Prompt Selector/TEST/prompt_selector_auto_1208_testv6.py

@@ -148,13 +148,14 @@ def generate(width, height, positive, negative, button):
         with open(f"error_log.txt", "a") as file:
             file.write(error_message)
 
-    global access_token
+    global access_token, access_token_multi, error_count, multi_token_enable
     while (running_flag):
         time.sleep(1)
     try:
         running_flag = True
         tprint("Running : NAI request received")
-        zipped_bytes = generate_image(access_token, positive, "nai-diffusion-3", "generate", params)
+        if not (multi_token_enable): zipped_bytes = generate_image(access_token, positive, "nai-diffusion-3", "generate", params)
+        else: zipped_bytes = generate_image(access_token_multi, positive, "nai-diffusion-3", "generate", params)
         d = Path(f"output_NAI/{start_time}/txt2img")
         d.mkdir(parents=True, exist_ok=True)
         zipped = zipfile.ZipFile(io.BytesIO(zipped_bytes))
@@ -170,15 +171,23 @@ def generate(width, height, positive, negative, button):
             globals()['tk_image'] = None
         tk_image = ImageTk.PhotoImage(i_resized)
         image_label.config(image=tk_image)
+        image_label.original_image = i.copy()
         image_label.image = tk_image  # 참조 유지
         output_file_path = "output_image.jpg"
         i.save(output_file_path)
+        error_count = 0
     except Exception as e:
         #text_output.insert(tk.END, f"Error: {e}", fg='red')
-        log_error(e, "path_to_output_folder")
+        log_error(zipped_bytes.decode('utf-8')[2:-2], "path_to_output_folder")
         time.sleep(random.uniform(1.0, 2.5))
+        error_count += 1
+        if error_count > 5:
+            mac_var.set(0)
         button.config(state=tk.NORMAL)
 
+    global delay_offset
+    time.sleep(delay_offset)
+
     running_flag = False
     last_generation_seed = params['seed']
     entry_seed_value.set(last_generation_seed)
@@ -480,6 +489,24 @@ def random_function():
                 if " (" in keyword:
                     temp_first_keywords.append(keyword)
                     random_row_keywords.remove(keyword)
+        # 타율 개선 프롬프트
+        valid_keywords = []
+        for keyword in temp_first_keywords:
+            if "(" in keyword and "(artist)" not in keyword and "(style)" not in keyword:
+                start = keyword.find("(") + 1
+                end = keyword.find(")")
+                if start < end and keyword[start:end] in current_deep_search_keywords:
+                    valid_keywords.append(keyword)
+        for keyword in valid_keywords:
+            if keyword in temp_first_keywords:
+                temp_first_keywords.remove(keyword)
+        girl_index = next((i for i, k in enumerate(temp_first_keywords) if 'girl' in k), None)
+        if girl_index is not None:
+            for keyword in valid_keywords:
+                temp_first_keywords.insert(girl_index + 1, keyword)
+                girl_index += 1
+ # 결과 출력
+        print(temp_first_keywords)
 
         #last_selected_row_keywords =boy_keywords + girl_keywords[:0] + random_row_keywords
         #print("3 :", temp_first_keywords)
@@ -558,6 +585,11 @@ def turbo_stop(button):
     tprint("Stopping pipelines ... ")
 
 def exit_program():
+    global stop_event, auto_thread
+    if stop_event is not None:
+        stop_event.set()
+    if auto_thread is not None and auto_thread.is_alive():
+        auto_thread.join()  # 스레드가 종료될 때까지 기다림
     window.destroy()
 
 def save_settings():
@@ -843,11 +875,23 @@ def NAI_generation(width, height, button):
     global access_token_multi, NAI_width, NAI_height
     global button_stop, running_flag
 
+    def run_function():
+        time.sleep(1)
+        random_function() 
+        time.sleep(2)
+        window.event_generate(GENERATE_EVENT, when="tail")
+
     pretest = text_output.get("1.0", tk.END).split(', ')
     if (turbo_var.get() == 1):
         if ( 'sex' not in pretest or '1girl' not in pretest or '1boy' not in pretest):
-            tprint('현재 부스트 기능은 1girl, 1boy, sex 세가지 키워드가 있어야 동작합니다.')
-            return
+            tprint('부스트 기능은 1girl, 1boy, sex 세가지 키워드가 있어야 동작합니다.','color=red')
+            if not mac_var.get():
+                button.config(state='normal')
+                return
+            else:
+                thread = threading.Thread(target=run_function)
+                thread.start()
+                return
 
     if(rand_resolution_var.get() == 1):
         resolutions = ["1024 x 1024", "960 x 1088", "896 x 1152", "832 x 1216", "1088 x 960", "1152 x 896", "1216 x 832"]
@@ -866,14 +910,24 @@ def Auto_login_check():
     if os.path.exists('token.txt'):
         with open('token.txt', 'r', encoding='utf-8') as f:
             tokens = f.readlines()
-            if(tokens[0]): access_token = tokens[0].strip()
-    if(access_token): button_setting.config(state='disabled')
+            if(tokens[0]): 
+                access_token = tokens[0].strip()
+                tprint("1 Account automatically loaded : ",access_token[-5:])
+            try:
+                if(len(tokens)>=2): 
+                    access_token_multi = tokens[1].strip()
+                    tprint("2 Account automatically loaded : ",access_token[-5:]+", "+access_token_multi[-5:])
+            except:
+                access_token_multi = None
     if(access_token): button_generate.config(state='normal')
         
 def NAI_generation_normal(width, height, button):
     button.config(state=tk.DISABLED)
     positive = text_output.get("1.0", tk.END)
     negative = negative_text.get("1.0", "end-1c").strip()
+    global auto_count_left_flag, auto_count_left
+    if(auto_count_left_flag and auto_count_left > 0):
+        auto_count_left -= 1
     thread = threading.Thread(target=generate, args=(width, height, positive, negative, button))
     thread.start()
 
@@ -882,7 +936,6 @@ def NAI_generation_turbo(width, height, button):
     button.config(state=tk.DISABLED)
     positive = text_output.get("1.0", tk.END)
     negative = negative_text.get("1.0", "end-1c").strip()
-
     
     def SELECT(positive):
         lines = positive[:25]
@@ -1064,6 +1117,9 @@ def NAI_generation_turbo(width, height, button):
     sleep2 = sleep1 + random.uniform(13.5,19.5)
     sleep3 = sleep2 + random.uniform(13.5,19.5)
 
+    global auto_count_left_flag, auto_count_left, turbo_stop_bit
+    if(auto_count_left_flag and auto_count_left > 0):
+        auto_count_left -= 5
     thread_init = threading.Thread(target=generate_turbo, args=(seed,turbo_count, access_token, width, height, positive, negative, button, 0, 0))
     thread_init.start()
     thread0 = threading.Thread(target=generate_turbo, args=(seed,turbo_count, access_token, width, height, positive0, negative, button, 1, sleep0))
@@ -1149,7 +1205,7 @@ def generate_turbo(_seed, turbo_count, tb_access_token,width, height, positive,
 
         return new_image
     
-    def insert_queue(i, _number):
+    def insert_queue(i, _number, turbo_count):
         global image_queue, temp_clipboard_image
         while len(image_queue) < 4: image_queue.append(None)
         if _number <= 4:
@@ -1165,17 +1221,23 @@ def generate_turbo(_seed, turbo_count, tb_access_token,width, height, positive,
                 globals()['tk_image'] = None
             tk_image = ImageTk.PhotoImage(i_resized)
             image_label.config(image=tk_image)
+            image_label.original_image = i.copy()
             image_label.image = tk_image  # 참조 유지
         if all(img is not None for img in image_queue):
             time.sleep(5)
             merged_image = merge_images(image_queue)
             temp_clipboard_image = merged_image.copy()
             if merged_image is not None:
+                output_directory = Path(f"output_NAI/{start_time}/turbo/grid")
+                output_directory.mkdir(parents=True, exist_ok=True) 
+                file_path = output_directory / f"{turbo_count:05}.png"
+                merged_image.save(file_path)
                 i_resized = resize_and_fill(merged_image)
                 if 'tk_image' in globals():
                     globals()['tk_image'] = None
                 tk_image = ImageTk.PhotoImage(i_resized)
                 image_label.config(image=tk_image)
+                image_label.original_image = i_resized.copy()
                 image_label.image = tk_image  # 참조 유지
                 output_file_path = "output_image.jpg"
                 merged_image.save(output_file_path)
@@ -1223,7 +1285,8 @@ def generate_turbo(_seed, turbo_count, tb_access_token,width, height, positive,
     print('containing : ',pipe_number)
 
     running_flag = True
-    tprint("Running : request received, pipeline :",pipe_number)
+    tprint("Running : request received, pipeline : "+str(pipe_number))
+    global error_count
     try:
         zipped_bytes = generate_image(tb_access_token, positive, "nai-diffusion-3", "generate", params)
         if(pipe_number != 0):
@@ -1237,13 +1300,14 @@ def generate_turbo(_seed, turbo_count, tb_access_token,width, height, positive,
         (d / f"{turbo_count:05}_{pipe_number}.png" ).write_bytes(image_bytes)
         i = Image.open(io.BytesIO(image_bytes))
         i = ImageOps.exif_transpose(i).convert("RGB")
+        error_count = 0
         if(pipe_number != 0):
             if temp_clipboard_image is not None:
                 temp_clipboard_image.close()
             if(pipe_number == 1 and len(image_queue) < 4): image_queue.clear()
             temp_clipboard_image = i
-            insert_queue(i, pipe_number)
-            tprint("image inserted, pipe number : ",pipe_number)
+            insert_queue(i, pipe_number, turbo_count)
+            tprint("image inserted, pipe number : "+str(pipe_number))
         else:
             if temp_clipboard_image is not None:
                 temp_clipboard_image.close()
@@ -1253,12 +1317,16 @@ def generate_turbo(_seed, turbo_count, tb_access_token,width, height, positive,
                 globals()['tk_image'] = None
             tk_image = ImageTk.PhotoImage(i_resized)
             image_label.config(image=tk_image)
+            image_label.original_image = i.copy()
             image_label.image = tk_image  # 참조 유지
     except Exception as e:
         #text_output.insert(tk.END, f"Error: {e}", fg='red')
-        log_error(e, "path_to_output_folder")
+        log_error(zipped_bytes.decode('utf-8')[2:-2], "path_to_output_folder")
+        error_count += 1
+        if error_count > 5:
+            mac_var.set(0)
         time.sleep(random.uniform(3.0, 9.0))
-        insert_queue(temp_clipboard_image, pipe_number)
+        insert_queue(temp_clipboard_image, pipe_number, turbo_count)
 
     task_finished += 1 #터보 기능의 안전장치입니다. 
 
@@ -1269,6 +1337,9 @@ def generate_turbo(_seed, turbo_count, tb_access_token,width, height, positive,
     last_generation_seed = params['seed']
     entry_seed_value.set(last_generation_seed)
 
+    global delay_offset
+    time.sleep(delay_offset)
+
     stopped = False
     if (turbo_stop_bit):
         time.sleep(1.5)
@@ -1276,7 +1347,7 @@ def generate_turbo(_seed, turbo_count, tb_access_token,width, height, positive,
         button.config(state=tk.NORMAL)
         stopped = True
         if (pipe_number != 4):
-            tprint("Stopped : request stopped before assign pipeline :",pipe_number+1)
+            tprint("Stopped : request stopped before assign pipeline :"+str(pipe_number+1))
 
     if pipe_number == 4 and mac_var.get():
         wait_limit = 0
@@ -1596,7 +1667,6 @@ def apply_wildcard(artist_text, character_text, window):
 
     window.destroy()
 
-
 def load_wildcard(text_widget, window, _type):
     global artist_wildcard_saved, character_wildcard_saved
     window.iconify()
@@ -1643,13 +1713,236 @@ def get_random_keyword(req_keyword):
 
 def tprint(*args):
     ctext = ""
+    color = None
     for t in args:
-        ctext += str(t)
-    running_state.config(text=ctext)
+        if(type(t) is str and 'color=' in t):
+            color = t[6:]
+        else:
+            ctext += str(t)
+    if not color: running_state.config(text=ctext, fg='black')
+    else: running_state.config(text=ctext, fg='red')
+
+
+def show_automation_option():
+    global auto_thread, stop_event
+    top = tk.Toplevel(window)
+    top.title("Automation Stop Condition")
+
+    global var, timer_label, timer_entry, count_label, count_entry
+    var = tk.StringVar(value="unlimited")
+
+    def show_option():
+        if  var.get() == 'timer':
+            timer_label.grid(row=4, column=0)
+            timer_entry.grid(row=4, column=1)
+            count_label.grid_forget()
+            count_entry.grid_forget()
+        elif var.get() == 'count':
+            count_label.grid(row=4, column=0)
+            count_entry.grid(row=4, column=1)
+            timer_label.grid_forget()
+            timer_entry.grid_forget()
+        else:
+            timer_label.grid_forget()
+            timer_entry.grid_forget()
+            count_label.grid_forget()
+            count_entry.grid_forget()
+
+    label_time = tk.Label(top, text="이미지 생성당 지연시간 추가 (초)")
+    label_time.grid(row=0, column=0, columnspan=3)
+
+    delay_entry = tk.Entry(top)
+    delay_entry.grid(row=1, column=0, columnspan=3)
+
+    label = tk.Label(top, text="자동화 종료 조건")
+    label.grid(row=2, column=0, columnspan=3)
+
+    radio1 = tk.Radiobutton(top, text="무제한", variable=var, value="unlimited", command=show_option)
+    radio1.grid(row=3, column=0)
+
+    radio2 = tk.Radiobutton(top, text="타이머", variable=var, value="timer", command=show_option)
+    radio2.grid(row=3, column=1)
+
+    radio3 = tk.Radiobutton(top, text="생성 카운트", variable=var, value="count", command=show_option)
+    radio3.grid(row=3, column=2)
+
+    timer_label = tk.Label(top, text="자동화 동작 시간(분) :")
+    timer_entry = tk.Entry(top)
+
+    count_label = tk.Label(top, text="자동 생성 횟수:")
+    count_entry = tk.Entry(top)
+
+    # "A.설정" 창에서 버튼 클릭 이벤트 처리
+    def on_apply(top):
+        global auto_time_left_flag, auto_time_left, auto_count_left_flag, auto_count_left, delay_offset, delay_offset_label
+
+        selected_option = var.get()
+        if delay_entry.get():
+            try:
+                delay_offset = round(float(delay_entry.get()), 1)
+                delay_offset_label.config(text="Dealy offset : "+str(delay_offset))
+            except ValueError as e:
+                print(e)
+                delay_offset = 0
+        else:
+            delay_offset = 0
+            delay_offset_label.config(text="")
+        if selected_option == "timer":
+            auto_time_left_flag = True
+            auto_count_left_flag = False
+            try:
+                # 엔트리에서 입력된 시간을 분 단위에서 초 단위로 변환
+                auto_time_left = int(timer_entry.get()) * 60
+            except ValueError:
+                # 잘못된 입력 처리
+                print("Invalid input for timer. Please enter a number.")
+                return
+        elif selected_option == "count":
+            auto_count_left_flag = True
+            auto_time_left_flag = False
+            try:
+                auto_count_left = int(count_entry.get())
+            except ValueError:
+                # 잘못된 입력 처리
+                print("Invalid input for count. Please enter a number.")
+                return
+        else:
+            auto_time_left_flag = False
+            auto_count_left_flag = False
+        stop_auto_thread()  # 이전 스레드 종료
+        start_auto_thread()  # 새 스레드 시작
+        mac_var.set(1)
+        top.destroy()
+
+    apply_button = tk.Button(top, text="적용", command=lambda: on_apply(top))
+    stop_button = tk.Button(top, text="중단", command=on_stop)
+    close_button = tk.Button(top, text="닫기", command=top.destroy)
+
+    apply_button.grid(row=5, column=0, pady=5)
+    stop_button.grid(row=5, column=1, pady=5)
+    close_button.grid(row=5, column=2, pady=5)
+
+def auto_time_thread(stop_event):
+    global auto_time_left, auto_time_left_flag
+
+    def seconds_to_hms(seconds):
+        h = seconds // 3600
+        m = (seconds % 3600) // 60
+        s = seconds % 60
+        return f"{h:02d}:{m:02d}:{s:02d}"
+    
+    def update_label_for_time_finished():
+        automation_state.config(text="지정한 시간이 모두 소진되어 자동화가 종료되었습니다.")
+        auto_time_left_flag = False
+        mac_var.set(0)
+
+    while not stop_event.is_set() and auto_time_left > 0:
+        time.sleep(1)
+        auto_time_left -= 1
+        remaining_time = seconds_to_hms(auto_time_left)
+        window.after(0, lambda: automation_state.config(text=f"자동화 남은시간 : {remaining_time}"))
+
+    if auto_time_left <= 0:
+        window.after(0, update_label_for_time_finished)
+
+    if stop_event.is_set():
+        window.after(0, lambda: automation_state.config(text="Conditional automation not set (A.설정)"))
+        auto_time_left_flag = False
+
+def auto_count_thread(stop_event):
+    global auto_count_left, auto_count_left_flag
+
+    def update_label_for_count_finished():
+        mac_var.set(0)  # 자동화 종료 조건 충족
+        auto_count_left_flag = False
+        automation_state.config(text="지정한 횟수가 모두 소진되어 자동화가 종료되었습니다.")
+
+    while not stop_event.is_set():
+        time.sleep(1)  # 주기적으로 확인
+        window.after(0, lambda: automation_state.config(text=f"자동화 남은횟수 : {auto_count_left}"))
+        if auto_count_left <= 0:
+            window.after(0, update_label_for_count_finished)
+            break  # 루프 탈출
+    if stop_event.is_set():
+        window.after(0, lambda: automation_state.config(text="Conditional automation not set (A.설정)"))
+        mac_var.set(0)
+        auto_count_left_flag = False
+
+# 스레드 시작 및 종료 함수
+def start_auto_thread():
+    global auto_thread, stop_event
+    stop_event = threading.Event()
+    auto_thread = None
+
+    if auto_time_left_flag:
+        auto_thread = threading.Thread(target=auto_time_thread, args=(stop_event,))
+    elif auto_count_left_flag:
+        auto_thread = threading.Thread(target=auto_count_thread, args=(stop_event,))
+
+    if auto_thread is not None:
+        auto_thread.start()
+
+def stop_auto_thread():
+    global stop_event, auto_thread
+    if stop_event is not None:
+        stop_event.set()
+    #if auto_thread is not None and auto_thread.is_alive():
+    #    auto_thread.join()
+
+def on_stop():
+    global auto_thread, stop_event
+    stop_auto_thread()  # 스레드 종료
 
 def on_hold_seed_button_click():
     global last_generation_seed
-    last_generation_seed = entry_seed_value.get()  # entry_seed_value에서 값을 가져와 last_generation_seed에 저장
+    last_generation_seed = entry_seed_value.get()  
+
+def update_fullscreen_image(new_window, new_image_label):
+    last_updated_image = None
+    while True: 
+        time.sleep(1) 
+        if not new_window.winfo_exists():
+            break
+        try:
+            current_image = getattr(image_label, 'original_image', None)
+            if current_image and current_image != last_updated_image:
+                resized_image = resize_image_to_fit(current_image, new_window.winfo_height())
+                tk_image = ImageTk.PhotoImage(resized_image)
+                new_window.after(0, lambda img=tk_image: new_image_label.config(image=img))
+                new_image_label.image = tk_image
+                last_updated_image = current_image
+        except ValueError:
+            print(ValueError)
+
+def show_fullscreen_image():
+    new_window = tk.Toplevel()
+    new_window.attributes('-fullscreen', True)
+    new_window.configure(bg='black')
+
+    new_image_label = tk.Label(new_window, relief='solid', borderwidth=1, bg='black') 
+    new_image_label.pack(expand=True, fill='both')
+    new_window.bind("<Escape>", lambda e: new_window.destroy())
+
+    update_thread = threading.Thread(target=update_fullscreen_image, args=(new_window, new_image_label))
+    update_thread.daemon = True  
+    update_thread.start()
+
+def resize_image_to_fit(image, target_height):
+    original_width, original_height = image.size
+
+    ratio = target_height / float(original_height)
+    new_width = int(original_width * ratio)
+
+    resized_image = image.resize((new_width, target_height), Image.Resampling.LANCZOS)
+    return resized_image
+
+auto_time_left_flag = False
+auto_time_left = 0
+auto_count_left_flag = False
+auto_count_left = 0
+stop_event = None
+auto_thread = None
+delay_offset = 0
 
 start_time = datetime.now().strftime('%Y%m%d_%H%M')
 thread_controller = False
@@ -1684,6 +1977,8 @@ image_queue = []
 last_generation_seed = random.randint(0,9999999999)
 turbo_stop_bit = False
 task_finished = 0
+error_count = 0
+multi_token_enable = False
 
 whitelist = wlist.whitelist
 bag_of_tags = tagbag.bag_of_tags
@@ -1802,16 +2097,12 @@ button_stop.grid(row=0, column=5, sticky='ew')
 button_stop.config(state='disabled')
 
 # "설정" 버튼
-button_setting = tk.Button(button_frame, text="NAI 로그인 설정", command=lambda: NAI_setting(button_setting, button_generate))
+button_setting = tk.Button(button_frame, text="NAI 설정", command=lambda: NAI_setting(button_setting, button_generate))
 button_setting.grid(row=0, column=6, sticky='ew')
 
-# "터보" 버튼
-#button_turbo = tk.Button(button_frame, text="터보", command=lambda: NAI_setting_multi(button_turbo))
-#button_turbo.grid(row=0, column=7, sticky='ew')
-
 # '자동화'와 '터보on'을 포함할 서브 프레임 생성
 sub_frame = tk.Frame(button_frame)
-sub_frame.grid(row=0, column=8, sticky='ns')  # 서브 프레임을 열 8에 배치
+sub_frame.grid(row=0, column=7, sticky='ns')  # 서브 프레임을 열 8에 배치
 
 # '자동화' 체크박스를 서브 프레임에 배치
 mac_var = tk.IntVar()  
@@ -1823,6 +2114,10 @@ turbo_var = tk.IntVar()
 turbo_button = tk.Checkbutton(sub_frame, text="터보on", variable=turbo_var)
 turbo_button.grid(row=1, column=0, sticky='ew')  # 두 번째 행에 배치
 
+# "A.설정" 버튼
+button_auto_setting = tk.Button(button_frame, text="A.설정", command=show_automation_option)
+button_auto_setting.grid(row=0, column=8, sticky='ew')
+
 # 텍스트 및 고정 프롬프트 네거티브 등 프롬프트  창
 text_frame = tk.Frame(left_frame)
 text_frame.grid(row=6, column=0, padx=5, pady=5, sticky='w')
@@ -1991,6 +2286,10 @@ resolution3_frame.columnconfigure(2, weight=1)
 
 running_state = tk.Label(resolution3_frame, text="state : idle")
 running_state.grid(row = 0, column= 0, sticky='w')
+automation_state = tk.Label(resolution3_frame, text="Conditional automation not set (A.설정)")
+automation_state.grid(row = 1, column= 0, sticky='w')
+delay_offset_label = tk.Label(resolution3_frame, text="Dealy offset : "+str(delay_offset))
+delay_offset_label.grid(row = 1, column= 1, sticky='e')
 entry_rescale_value = tk.StringVar()
 entry_rescale_label = tk.Label(resolution3_frame, text="  Prompt Guidance Rescale : ", justify=tk.LEFT)
 entry_rescale_label.grid(row=0, column=1, sticky='e')
@@ -2016,6 +2315,9 @@ hold_seed_button.grid(row=0, column=2, sticky='ew')
 btn_add_low_freq = tk.Button(right_frame_area16, text="이미지를 클립보드에 복사", fg="blue", command=copy_image_to_clipboard)
 btn_add_low_freq.grid(row=0, column=3, padx=5, pady=5, sticky='ew')
 
+show_fullscreen_btn = tk.Button(right_frame_area16, text="전체화면 보기 (ESC로 닫기)", command=show_fullscreen_image)
+show_fullscreen_btn.grid(row=0, column=4, padx=5, pady=5, sticky='ew')
+
 # 하얀색 이미지 생성
 white_image = Image.new('RGB', (_size, _size), 'white')
 white_photo = ImageTk.PhotoImage(white_image)
@@ -2034,4 +2336,5 @@ listener_thread.start()
 
 Auto_login_check()
 
+window.protocol("WM_DELETE_WINDOW", exit_program)
 window.mainloop()