| import customtkinter |
| import tkinter as tk |
| from tkinter import font |
| from tkinter import Tk |
| from PIL import Image, ImageOps, ImageDraw, ImageFont,ImageTk |
| import os, sys |
| import ctypes |
| import pandas as pd |
| import numpy as np |
| import NAIA_search, NAIA_random_function_core, NAIA_Login, NAIA_generation |
| import json |
| import arti_list, tagbag, wlist, copyright_list_reformatted, remove_result_e, remove_result_qe, copyright_dict_0103 |
| import character_dictionary as cd |
| import time |
| from datetime import datetime |
| import random |
| import threading, pyperclip |
| from ctypes import windll |
| from CTkListbox import * |
| from tkinter import filedialog |
| import io |
| from openpyxl import Workbook |
| from openpyxl.styles import Alignment |
| from openpyxl.drawing.image import Image as OpenpyxlImage |
| from collections import Counter |
| from artist_dictionary import artist_dict |
| import re |
| import requests |
| import base64 |
| import queue |
| import webbrowser |
| import ctypes |
| from plyer import notification |
| import subprocess, copy |
| from tkinterdnd2 import TkinterDnD, DND_ALL |
| import gzip |
| from pathlib import Path |
|
|
| class Data: |
| def __init__(self, wlist, tagbag, arti_list, copyright_list_reformatted, cd, remove_result_e, remove_result_qe): |
| self.whitelist = wlist.whitelist |
| self.bag_of_tags = tagbag.bag_of_tags |
| self.afilter_30000 = arti_list.afilter_30000 |
| self.copyright_keys = copyright_list_reformatted.copyright_list |
| self.character_keys = list(cd.character_dictionary.keys()) |
| nsfw_word = list(remove_result_e.keyword_counts.keys()) |
| questionable_word = list(remove_result_qe.keyword_counts.keys()) |
| self.qe_word = nsfw_word + questionable_word |
|
|
| def get_qe_word(self): |
| return self.qe_word |
|
|
| class AccountSetting(customtkinter.CTkToplevel): |
| def __init__(self, *args, **kwargs): |
| super().__init__(*args, **kwargs) |
| self.title("NAI Account Setting") |
| self.attributes('-topmost', True) |
|
|
| def on_close(self): |
| self.withdraw() |
|
|
| def NAI_connect(self): |
| username = self.NAI_ID_entry.get().strip() |
| password = self.NAI_PW_entry.get().strip() |
| access_key = NAIA_Login.get_access_key(username, password) |
| try: |
| app.access_token = NAIA_Login.login(access_key) |
| app.NAI_ID = username[:4] |
| self.state_label.configure(text="(๋ก๊ทธ์ธ ์ฑ๊ณต, ๋ซ๊ธฐ ๋ฒํผ์ ๋๋ฌ์ฃผ์ธ์.)", font=my_font) |
| app.NAI_Account_Login.configure(state="disabled") |
| app.NAI_Token_Remove.configure(state="normal") |
| app.NAI_Account_State.configure(text="NAI Login : OK") |
| app.image_generation_button.configure(state="normal") |
| except Exception as e: |
| print(e) |
| self.state_label.configure(text="์์ด๋ ํน์ ๋น๋ฐ๋ฒํธ ์ค๋ฅ์
๋๋ค.", font=my_font) |
|
|
| self.button_frame = customtkinter.CTkFrame(self) |
| self.button_frame.grid(row=0, padx=5, pady=5, sticky="nsew") |
| self.button_frame.columnconfigure(0, weight=1) |
| self.button_frame.columnconfigure(1, weight=4) |
|
|
| my_font = customtkinter.CTkFont('Pretendard', 13) |
| self.NAI_ID_label = customtkinter.CTkLabel(self.button_frame, text="NAI ID:", font=my_font) |
| self.NAI_ID_label.grid(row = 0, column = 0, padx=5, pady=5, sticky="nsew") |
| self.NAI_PW_label = customtkinter.CTkLabel(self.button_frame, text="NAI PW:", font=my_font) |
| self.NAI_PW_label.grid(row = 1, column = 0, padx=5, pady=5, sticky="nsew") |
| self.NAI_ID_entry = customtkinter.CTkEntry(self.button_frame) |
| self.NAI_ID_entry.grid(row = 0, column = 1, padx=5, pady=5, sticky="nsew") |
| self.NAI_PW_entry = customtkinter.CTkEntry(self.button_frame, show="*") |
| self.NAI_PW_entry.grid(row = 1, column = 1, padx=5, pady=5, sticky="nsew") |
| self.connect_button = customtkinter.CTkButton(self.button_frame, text="Connect", command=lambda:NAI_connect(self)) |
| self.connect_button.grid(row = 2, column = 0, columnspan=2, padx=5, pady=5, sticky="n") |
| self.state_label = customtkinter.CTkLabel(self.button_frame, text="ํด๋น ์ ์๊ธฐ๋ฅ์ ์ ์์ ์ธ ์ ์ ํจํด์ด ์๋์ ์ฐธ๊ณ ๋ถํ๋๋ฆฝ๋๋ค.", font=my_font) |
| self.state_label.grid(row = 3, column = 0, columnspan=2, padx=5, pady=5, sticky="nsew") |
|
|
| self.protocol("WM_DELETE_WINDOW", lambda: on_close(self)) |
|
|
| class Advanced_setting(customtkinter.CTkToplevel): |
| def __init__(self, *args, **kwargs): |
| super().__init__(*args, **kwargs) |
| self.title("๊ธฐํ ์ค์ ") |
| self.resizable(width=False, height=False) |
| self.geometry("270x250") |
| self.attributes('-topmost', True) |
| my_font = customtkinter.CTkFont('Pretendard', 13) |
|
|
| def on_close(self): |
| self.withdraw() |
|
|
| def select_folder(self): |
| folder_selected = filedialog.askdirectory() |
| try: |
| self.save_path.configure(state="normal") |
| self.save_path.delete(0,"end") |
| self.save_path.insert(0, folder_selected) |
| self.save_path.configure(state="disabled") |
| app.output_file_path_personal = True |
| app.output_file_path = folder_selected |
| except: |
| pass |
|
|
| def deselect_folder(self): |
| self.save_path.configure(state="normal") |
| app.output_file_path = f"output_NAI\\{app.start_time}\\txt2img" |
| self.save_path.delete(0,"end") |
| self.save_path.insert(0, app.output_file_path) |
| self.save_path.configure(state="disabled") |
| app.output_file_path_personal = True |
|
|
| self.advanced_setting_frame = customtkinter.CTkFrame(self, width=250) |
| self.advanced_setting_frame.grid(row=0, column=0, padx=10, pady=5, sticky="nsew") |
| self.change_save_path_button = customtkinter.CTkButton(self.advanced_setting_frame, text="์ด๋ฏธ์ง ์ ์ฅ ๊ฒฝ๋ก ๋ณ๊ฒฝ", command= lambda: select_folder(self), font=my_font, width=250) |
| self.change_save_path_button.grid(row=0, column=0, pady=5,sticky="nsew") |
| self.save_path = customtkinter.CTkEntry(self.advanced_setting_frame, font=my_font, width=250) |
| self.save_path.insert(0, app.output_file_path) |
| self.save_path.configure(state="disabled") |
| self.save_path.grid(row=1, column=0, pady=5,sticky="nsew") |
| self.change_save_path_button2 = customtkinter.CTkButton(self.advanced_setting_frame, text="๊ฒฝ๋ก ์ด๊ธฐํ", command= lambda: deselect_folder(self), font=my_font, width=250) |
| self.change_save_path_button2.grid(row=2, column=0, pady=5,sticky="nsew") |
| self.file_name_label = customtkinter.CTkLabel(self.advanced_setting_frame, text=" -- ํ์ผ๋ช
.png ๊ท์น ์ค์ -- " , font=my_font, width=250) |
| self.file_name_label.grid(row=3, column=0, pady=5,sticky="n") |
| self.radio1 = customtkinter.CTkRadioButton(self.advanced_setting_frame, text="์์ฑ์๊ฐ.png : 20231230_123001.png", variable=app.name_var, value="time", font=my_font) |
| self.radio1.grid(row=4, column=0) |
| self.radio2 = customtkinter.CTkRadioButton(self.advanced_setting_frame, text="์์ฑ์์.png : 00001.png,00002.png", variable=app.name_var, value="count", font=my_font) |
| self.radio2.grid(row=5, column=0) |
| self.file_name_label2 = customtkinter.CTkLabel(self.advanced_setting_frame, text="*์์ฑ์์ ์ ์ฉ์ ๋ ์ง_์๊ฐ ํด๋์ ์ ์ฅ" , font=my_font, width=250) |
| self.file_name_label2.grid(row=6, column=0, pady=5,sticky="n") |
|
|
| self.protocol("WM_DELETE_WINDOW", lambda: on_close(self)) |
|
|
| class Automation_setting(customtkinter.CTkToplevel): |
| def __init__(self, *args, **kwargs): |
| super().__init__(*args, **kwargs) |
| self.title("Automation ์ข
๋ฃ/์ง์ฐ ์ค์ ") |
|
|
| self.var = customtkinter.StringVar() |
| self.attributes('-topmost', True) |
| my_font = customtkinter.CTkFont('Pretendard', 13) |
|
|
| def show_option(self): |
| if self.var.get() == 'timer': |
| self.timer_label.grid(row=4, column=0, pady=5) |
| self.timer_entry.grid(row=4, column=1, pady=5) |
| self.windows_shutdown_check.grid(row=5, column=0, columnspan=3, sticky="n", pady=5) |
| self.count_label.grid_forget() |
| self.count_entry.grid_forget() |
| elif self.var.get() == 'count': |
| self.count_label.grid(row=4, column=0, pady=5) |
| self.count_entry.grid(row=4, column=1, pady=5) |
| self.windows_shutdown_check.grid(row=5, column=0, columnspan=3, sticky="n", pady=5) |
| self.timer_label.grid_forget() |
| self.timer_entry.grid_forget() |
| else: |
| self.timer_label.grid_forget() |
| self.timer_entry.grid_forget() |
| self.count_label.grid_forget() |
| self.count_entry.grid_forget() |
| self.windows_shutdown_check.grid_forget() |
|
|
| def on_close(self): |
| self.withdraw() |
|
|
| self.upper_frame = customtkinter.CTkFrame(self, fg_color="#242424") |
| self.upper_frame.grid(row=0, column=0, columnspan=3, sticky="n") |
|
|
| self.label_time = customtkinter.CTkLabel(self.upper_frame, text="์ด๋ฏธ์ง ์์ฑ๋น ์ง์ฐ์๊ฐ ์ถ๊ฐ (์ด)", font = my_font) |
| self.label_time.grid(row=0, column=0, columnspan=3, sticky="n") |
|
|
| self.delay_entry = customtkinter.CTkEntry(self.upper_frame, font = my_font) |
| self.delay_entry.grid(row=1, column=0,columnspan=3, sticky="n") |
|
|
| self.repeat_frame = customtkinter.CTkFrame(self, fg_color="#242424") |
| self.repeat_frame.grid(row=2, column=0, columnspan=3, sticky="n") |
|
|
| self.label_repeat = customtkinter.CTkLabel(self.repeat_frame, text="๋์ผ ์ด๋ฏธ์ง๋ฅผ ๋ค์ ํ์๋งํผ ๋ฐ๋ณตํ์ฌ ์์ฑ (Default=1)", font = my_font) |
| self.label_repeat.grid(row=0, column=0, columnspan=3, sticky="n") |
|
|
| self.repeat_entry = customtkinter.CTkEntry(self.repeat_frame, font = my_font, width=70) |
| self.repeat_entry.grid(row=1, column=0,columnspan=3, sticky="n") |
|
|
| self.label_wildcard_repeat = customtkinter.CTkLabel(self.repeat_frame, text="์์ผ๋์นด๋ *์ฌ์ ๊ฐ๋ด*์ ๋ค์ ํ์๋งํผ ๋ฐ๋ณตํ์ฌ ์์ฑ (Default=1)", font = my_font) |
| self.label_wildcard_repeat.grid(row=2, column=0, columnspan=3, sticky="n") |
|
|
| self.wildcard_repeat_entry = customtkinter.CTkEntry(self.repeat_frame, font = my_font, width=70) |
| self.wildcard_repeat_entry.grid(row=3, column=0,columnspan=3, sticky="n") |
|
|
| self.label = customtkinter.CTkLabel(self.repeat_frame, text="์๋ํ ์ข
๋ฃ ์กฐ๊ฑด", font = my_font) |
| self.label.grid(row=4, column=0, columnspan=3, sticky="n") |
|
|
| self.radio1 = customtkinter.CTkRadioButton(self, text="๋ฌด์ ํ", variable=self.var, value="unlimited", command=lambda: show_option(self), font = my_font) |
| self.radio1.grid(row=3, column=0) |
| self.radio2 = customtkinter.CTkRadioButton(self, text="ํ์ด๋จธ", variable=self.var, value="timer", command=lambda: show_option(self), font = my_font) |
| self.radio2.grid(row=3, column=1) |
| self.radio3 = customtkinter.CTkRadioButton(self, text="์์ฑ์นด์ดํธ", variable=self.var, value="count", command=lambda: show_option(self), font = my_font) |
| self.radio3.grid(row=3, column=2) |
|
|
| self.timer_label = customtkinter.CTkLabel(self, text="์๋ํ ๋์ ์๊ฐ(๋ถ) : ", font = my_font) |
| self.timer_entry = customtkinter.CTkEntry(self) |
|
|
| self.count_label = customtkinter.CTkLabel(self, text="์๋ ์์ฑ ํ์ : ", font = my_font) |
| self.count_entry = customtkinter.CTkEntry(self) |
|
|
| self.windows_shutdown = customtkinter.IntVar() |
| self.windows_shutdown_check = customtkinter.CTkCheckBox(self, text="์๋ํ๊ฐ ์๋์ผ๋ก ๋๋๋ฉด ์๋์ฐ๋ฅผ ์ข
๋ฃํฉ๋๋ค.", variable=self.windows_shutdown, font = my_font) |
|
|
| self.apply_button = customtkinter.CTkButton(self, text="์ ์ฉ", font = my_font, command=lambda: on_apply(self)) |
| self.stop_button = customtkinter.CTkButton(self, text="์ค๋จ", font = my_font, command=lambda: on_stop(self)) |
| self.close_button = customtkinter.CTkButton(self, text="๋ซ๊ธฐ", font = my_font, command=lambda: on_close(self)) |
|
|
| self.apply_button.grid(row=6, column=0, pady=5) |
| self.stop_button.grid(row=6, column=1, pady=5) |
| self.close_button.grid(row=6, column=2, pady=5) |
|
|
| def on_apply(self): |
| selected_option = self.var.get() |
| if self.delay_entry.get(): |
| try: |
| app.delay_offset = round(float(self.delay_entry.get()), 1) |
| if app.delay_offset > 60: |
| app.delay_offset = 60 |
| elif app.delay_offset < -8: |
| app.delay_offset = -8 |
| app.automation_setting_button.configure(text=f"์๋ํ ์ค์ ({str(round(app.delay_offset, 1))}์ด)") |
| except ValueError as e: |
| print(e) |
| app.delay_offset = 0 |
| else: |
| app.delay_offset = 0 |
| app.automation_setting_button.configure(text=f"์๋ํ ์ค์ ") |
| if self.repeat_entry.get(): |
| try: |
| app.image_generation_repeat = int(self.repeat_entry.get()) |
| except: |
| self.repeat_entry.delete(0, "end") |
| app.image_generation_repeat = 1 |
| if self.wildcard_repeat_entry.get(): |
| try: |
| app.wildcard_preopen_repeat = int(self.wildcard_repeat_entry.get()) |
| app.wildcard_preopen_repeat_current = app.wildcard_preopen_repeat |
| except: |
| self.wildcard_repeat_entry.delete(0, "end") |
| app.image_generation_repeat = 1 |
| if selected_option == "timer": |
| app.auto_time_left_flag = True |
| app.auto_count_left_flag = False |
| try: |
| |
| app.auto_time_left = int(self.timer_entry.get()) * 60 |
| except ValueError: |
| |
| print("Invalid input for timer. Please enter a number.") |
| return |
| elif selected_option == "count": |
| app.auto_count_left_flag = True |
| app.auto_time_left_flag = False |
| try: |
| app.auto_count_left = int(self.count_entry.get()) |
| except ValueError: |
| |
| print("Invalid input for count. Please enter a number.") |
| return |
| else: |
| self.auto_time_left_flag = False |
| self.auto_count_left_flag = False |
| on_stop(self) |
| start_auto_thread() |
| |
| def show_shutdown_warning(): |
| def cancel_shutdown(): |
| os.system("shutdown -a") |
| warning_window.destroy() |
|
|
| def on_warning_window_close(): |
| if warning_window.after_id is not None: |
| app.after_cancel(warning_window.after_id) |
| warning_window.destroy() |
|
|
| app.deiconify() |
| app.lift() |
|
|
| warning_window = customtkinter.CTkToplevel(app) |
| warning_window.title("Shutdown Warning") |
| warning_window.attributes("-topmost", True) |
| warning_window.protocol("WM_DELETE_WINDOW", on_warning_window_close) |
| warning_window.after_id = None |
|
|
| label = customtkinter.CTkLabel(warning_window, text=" ํ๋ก๊ทธ๋จ ๋ฐ ์ปดํจํฐ๊ฐ 90/120์ด ํ์ ์ข
๋ฃ๋ฉ๋๋ค. ", font=my_font) |
| label.pack(pady=15) |
|
|
| cancel_button = customtkinter.CTkButton(warning_window, text="์ทจ์", command=cancel_shutdown, font=my_font) |
| cancel_button.pack(pady=5) |
|
|
| |
| warning_window.after_id = warning_window.after(90000, app.exit_program) |
|
|
| def shutdown_computer(): |
| os.system("shutdown /s /t 120") |
| show_shutdown_warning() |
|
|
| def auto_time_thread(stop_event): |
| def seconds_to_hms(seconds): |
| h = seconds // 3600 |
| m = (seconds % 3600) // 60 |
| s = seconds % 60 |
| if h != 0: |
| if (s%2) == 0: |
| return f"{h:02d}:{m:02d}" |
| else: |
| return f"{h:02d} {m:02d}" |
| else: |
| if (s%2) == 0: |
| return f"{m:02d}:{s:02d}" |
| else: |
| return f"{m:02d} {s:02d}" |
| |
| def update_label_for_time_finished(): |
| app.automation_button.configure(text="์๋ํ (์ข
๋ฃ๋จ)") |
| app.auto_time_left_flag = False |
| app.automation_button.deselect() |
| if self.windows_shutdown.get() == 1: |
| shutdown_computer() |
| else: |
| notification.notify( |
| title = '์๋์์ฑ์ด ์๋ฃ๋์์ด์.', |
| message = '๊ฒฐ๊ณผ๋ฅผ ํ์ธํด ๋ณด์ธ์.', |
| app_name = "NAIA", |
| app_icon = app.basedir+'/icoico.ico', |
| timeout = 5, |
| ) |
|
|
| while not stop_event.is_set() and app.auto_time_left > 0: |
| time.sleep(1) |
| app.auto_time_left -= 1 |
| remaining_time = seconds_to_hms(app.auto_time_left) |
| app.after(0, lambda: app.automation_button.configure(text=f"์๋ํ ({remaining_time})")) |
|
|
| if app.auto_time_left <= 0: |
| app.after(0, update_label_for_time_finished) |
|
|
| if app.stop_event.is_set(): |
| app.after(0, lambda: app.automation_button.configure(text="์๋ํ")) |
| app.auto_time_left_flag = False |
|
|
| def auto_count_thread(stop_event): |
| def update_label_for_count_finished(): |
| app.automation_button.configure(text="์๋ํ (์ข
๋ฃ๋จ)") |
| app.auto_time_left_flag = False |
| app.automation_button.deselect() |
| if self.windows_shutdown.get() == 1: |
| shutdown_computer() |
| else: |
| notification.notify( |
| title = '์๋์์ฑ์ด ์๋ฃ๋์์ด์.', |
| message = '๊ฒฐ๊ณผ๋ฅผ ํ์ธํด ๋ณด์ธ์.', |
| app_name = "NAIA", |
| app_icon = app.basedir+'/icoico.ico', |
| timeout = 5, |
| ) |
|
|
| while not stop_event.is_set(): |
| time.sleep(1) |
| app.after(0, lambda: app.automation_button.configure(text=f"์๋ํ ({app.auto_count_left})")) |
| if app.auto_count_left <= 0: |
| app.after(0, update_label_for_count_finished) |
| break |
| if app.stop_event.is_set(): |
| app.after(0, lambda: app.automation_button.configure(text="์๋ํ")) |
| app.auto_count_left_flag = False |
|
|
| |
| def start_auto_thread(): |
| app.stop_event = threading.Event() |
| app.auto_thread = None |
|
|
| if app.auto_time_left_flag: |
| app.auto_thread = threading.Thread(target=auto_time_thread, args=(app.stop_event,), daemon=True) |
| elif app.auto_count_left_flag: |
| app.auto_thread = threading.Thread(target=auto_count_thread, args=(app.stop_event,), daemon=True) |
|
|
| if app.auto_thread is not None: |
| app.auto_thread.start() |
|
|
| def stop_auto_thread(self): |
| if app.stop_event is not None: |
| app.stop_event.set() |
|
|
| def on_stop(self): |
| stop_auto_thread(self) |
|
|
| self.protocol("WM_DELETE_WINDOW", lambda: on_close(self)) |
|
|
| class Character_search(customtkinter.CTkToplevel): |
| def __init__(self, *args, **kwargs): |
| super().__init__(*args, **kwargs) |
| self.title("์บ๋ฆญํฐ ๊ฒ์ ๋ฐ ์ฌ์ ") |
| |
| self.app = app |
| self.attributes('-topmost', True) |
| my_font = customtkinter.CTkFont('Pretendard', 13) |
|
|
| character_book_dict = {} |
| try: |
| if os.path.exists('Character_book.json'): |
| with open('Character_book.json', 'r', encoding='utf-8') as f: |
| try: |
| character_book_dict = json.load(f) |
| self.dictionary_parity = True |
| except: |
| character_book_dict = {} |
| except FileNotFoundError: |
| character_book_dict = {} |
|
|
| def lost_top_most(self): |
| self.attributes('-topmost', False) |
|
|
| def on_close(self): |
| self.withdraw() |
|
|
| frame_head = customtkinter.CTkFrame(self, width=750, height=40) |
| frame_head.grid(row =0, column =0, columnspan=5, padx=5, pady=5, sticky="nsew") |
| frame_left = customtkinter.CTkFrame(self, width=300, height=520) |
| frame_left.grid(row =1, column =0, padx=5, pady=5, sticky="nsew") |
| frame_right = customtkinter.CTkFrame(self, width=450, height=520) |
| frame_right.grid(row =1, column =2, padx=5, pady=5, sticky="nsew") |
|
|
| search_word = customtkinter.StringVar() |
| search_entry = customtkinter.CTkEntry(frame_head, font=my_font, textvariable=search_word) |
|
|
| character_prompt_frame = customtkinter.CTkFrame(frame_right) |
| character_prompt_frame.grid(row=0, column=0, padx=10, pady=5, sticky="nsew") |
| prompt_label = customtkinter.CTkLabel(character_prompt_frame, text="์บ๋ฆญํฐ ํ๋กฌํํธ: ", font=my_font) |
| prompt_label.grid(row=0, column=0, padx=10, pady=5, sticky="w") |
|
|
| def generate_character(self): |
| prompt = character_prompt.get("0.0", "end-1c") |
| prompt_cos = cosplay_prompt.get("0.0", "end-1c") |
| concat_entry = generate_entry.get() + ", {{face focus, upper body}}" |
| keywords = [item.strip() for item in prompt.split(',')] |
| if len(keywords) >= 2: |
| keyword = keywords[1] |
| keywords[1] = '{{'+keyword+'}}' |
| keywords.insert(2, concat_entry) |
| else: |
| keyword = keywords[0] |
| keywords[0] = "{{" + keywords[0] + "}}" |
| request_prompt = ', '.join(keywords) |
| instant_image_generation_book(self, request_prompt, keyword, prompt, prompt_cos) |
|
|
| character_prompt_generation = customtkinter.CTkButton(character_prompt_frame, font=my_font, text="์ฆ์์์ฑ", fg_color="grey", hover_color="grey10", command=lambda: generate_character(self)) |
| character_prompt_generation.grid(row=0, column=1, padx=5, pady=5, sticky="w") |
| continuous_generation = customtkinter.IntVar() |
| continuous_generation_button = customtkinter.CTkCheckBox(character_prompt_frame, font=my_font, text="๋น ๋๊ฐ ์๋์ฑ์ฐ๊ธฐ", variable=continuous_generation) |
| continuous_generation_button.grid(row=0, column=2, padx=5, pady=5, sticky="w") |
|
|
| character_prompt = customtkinter.CTkTextbox(frame_right, font=customtkinter.CTkFont('Pretendard', 15), width=450) |
| character_prompt.grid(row=1, column=0, padx=10, pady=5, sticky="w") |
|
|
| cosplay_prompt_frame = customtkinter.CTkFrame(frame_right) |
| cosplay_prompt_frame.grid(row=2, column=0, padx=10, pady=5, sticky="nsew") |
| prompt_label = customtkinter.CTkLabel(cosplay_prompt_frame, text="์บ๋ฆญํฐ(Cosplay)):", font=my_font) |
| prompt_label.grid(row=0, column=0, padx=10, pady=5, sticky="w") |
| |
| |
| cosplay_prompt = customtkinter.CTkTextbox(frame_right, font=customtkinter.CTkFont('Pretendard', 15), width=450) |
| cosplay_prompt.grid(row=3, column=0, padx=10, pady=5, sticky="w") |
| image_label = customtkinter.CTkLabel(frame_right, text="") |
| image_label.grid(row=0, rowspan=4, column=1, padx=10, pady=5, sticky="w") |
| white_image = Image.new('RGB', (384, 512), 'white') |
| white_photo = customtkinter.CTkImage(white_image, size=(384,512)) |
| image_label.configure(image=white_photo) |
| generate_label = customtkinter.CTkLabel(self, text="Portrait ์์ฑ์ ์ ์ฉํ ์ ํ ๊ณ ์ ํ๋กฌํํธ : ", font=my_font) |
| generate_label.grid(row=2, column=0, padx=5, pady=5, sticky="w") |
| generate_entry = customtkinter.CTkEntry(self, font=my_font) |
| generate_entry.grid(row=2, column=1, columnspan=2, padx=5, pady=5, sticky="nsew") |
|
|
| if character_book_dict != None and 'user_fix_prompt' in character_book_dict and character_book_dict['user_fix_prompt'] != None: |
| generate_entry.delete(0, "end") |
| generate_entry.insert(0, character_book_dict['user_fix_prompt']) |
|
|
| def instant_image_generation_book(self, request_prompt, character_name, p1, p2): |
| if app.running_flag == False and app.automation_button.get() == 0: |
| character_prompt_generation.configure(state="disabled") |
| |
| app.running_flag == True |
| try: |
| scale_pre = float(scale_pre) |
| except: |
| scale_pre = 5.0 |
| app.cfg_scale_var.set("5.0") |
| rescale_pre = app.prompt_guidance_rescale_entry.get() |
| try: |
| rescale_pre = float(rescale_pre) |
| except: |
| rescale_pre = 0 |
| app.prompt_guidance_rescale_var.set("0") |
| uncond_pre = app.uncond_strength_entry.get() |
| try: |
| uncond_pre = float(uncond_pre) |
| uncond_pre = round(uncond_pre / 0.05) * 0.05 |
| if (uncond_pre > 1.5): |
| uncond_pre = 1.5 |
| app.uncond_strength_entry.delete(0, "end") |
| app.uncond_strength_entry.insert(0, "1.5") |
| except: |
| uncond_pre = 1.0 |
| app.uncond_strength_entry.delete(0, "end") |
| app.uncond_strength_entry.insert(0, "1.0") |
| gen_request = { |
| "width":896, |
| "height":1152, |
| "quality_toggle":app.auto_quality_toggle.get(), |
| "seed":random.randint(0,9999999999), |
| "sampler":app.sampler_button.get(), |
| "scale":scale_pre, |
| "sema":app.sema_button.get(), |
| "sema_dyn": app.dyn_button.get(), |
| "cfg_rescale": rescale_pre, |
| "uncond_scale":uncond_pre, |
| "prompt": request_prompt, |
| "negative":app.negative_prompt_input.get("0.0", "end-1c"), |
| "user_screen_size": app.get_max_size(), |
| "start_time": app.start_time, |
| "access_token": app.access_token, |
| "save_folder": app.output_file_path, |
| "png_rule": app.name_var.get(), |
| "type": "normal" |
| } |
| def run_generation(): |
| if gen_request["png_rule"] == "count": |
| app.generation_count += 1 |
| gen_request["count"] =app.generation_count |
| app.state_label.configure(text ="state : ๋๊ฐ ์ด๋ฏธ์ง ์์ฒญ๋จ ", text_color = "#FFFF97") |
| result_image, result_prompt, result_seed, info, filename = NAIA_generation.generate(gen_request) |
| app.running_flag == False |
| character_prompt_generation.configure(state="normal") |
| |
| app.state_label.configure(text ="state : ๋๊ฐ ์ด๋ฏธ์ง ์์ฒญ ๋ฐํ๋จ", text_color = "#DCE4EE") |
| if info: |
| temp = info.get('Comment', '') |
| temp = temp[temp.find("prompt")+10:temp.find("skip_cfg_below_sigma")-3].replace('"','') |
| else: |
| temp = result_prompt |
| app.image_label_report.configure(state="normal") |
| app.image_label_report.delete("0.0", "end") |
| app.image_label_report.configure(text_color="#DCE4EE") |
| app.image_label_report.insert("0.0", temp) |
| app.image_label_report.configure(state="disabled") |
| if result_image: |
| image_bytes = io.BytesIO() |
| book_image = Image.open(filename).resize((384,512)) |
| book_image = book_image.convert('RGB') |
| book_image.save(image_bytes, format='JPEG', quality=90) |
| image_bytes = base64.b64encode(image_bytes.getvalue()).decode('utf-8') |
| character_book_dict[character_name] = [0, p1, p2, image_bytes] |
| character_book_dict['user_fix_prompt'] = generate_entry.get() |
| with open('Character_book.json', 'w', encoding='utf-8') as f: |
| json.dump(character_book_dict, f, ensure_ascii=False, indent=4) |
| image_stream = io.BytesIO(base64.b64decode(image_bytes)) |
| reloaded_image = Image.open(image_stream) |
| image_label.configure(image=customtkinter.CTkImage(reloaded_image, size=(384,512))) |
| if app.state() != 'zoomed': |
| instant_result_image = customtkinter.CTkImage(result_image, size=(620,620)) |
| else: |
| current_image = Image.open(filename) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| app.image_label.configure(image=instant_result_image) |
| app.ext_set_image_to_queue(result_image, result_prompt, str(result_seed), filename) |
| if continuous_generation.get() == 1: |
| listbox.focus() |
| app.after(500, move_selection_down(self)) |
| else: |
| if continuous_generation.get() == 1: |
| app.after(500, generate_character(self)) |
| generation_thread = threading.Thread(target=run_generation, daemon=True) |
| generation_thread.start() |
|
|
| def move_selection_down(self): |
| selected_indices = listbox.curselection() |
| if selected_indices: |
| current_index = selected_indices[0] |
| next_index = current_index + 1 |
|
|
| |
| if next_index < listbox.size(): |
| listbox.selection_clear(current_index) |
| listbox.selection_set(next_index) |
| listbox.see(next_index) |
| else: |
| continuous_generation_button.deselect() |
| keyword_with_count = listbox.get(next_index).strip() |
| keyword = keyword_with_count.split(' - ')[0].strip() |
| if keyword not in character_book_dict and continuous_generation.get() == 1: |
| on_search() |
| app.after(500, generate_character(self)) |
| elif continuous_generation.get() == 1: |
| app.after(500, move_selection_down(self)) |
|
|
| def on_key_release(event): |
| search_keyword = search_entry.get() |
| update_listbox(search_keyword) |
|
|
| def on_search(): |
| |
| selected_indices = listbox.curselection() |
| if not selected_indices: |
| print("No selection made.") |
| return |
| |
| |
| selected_index = selected_indices[0] |
| |
| keyword_with_count = listbox.get(selected_index).strip() |
| keyword = keyword_with_count.split(' - ')[0].strip() |
| |
| |
| print(f"Analyzing keyword: {keyword}") |
| character, cosplay = analyze_keywords_in_data('csdataset.parquet', keyword) |
| character_prompt.delete("0.0", "end") |
| character_prompt.insert("0.0", character) |
| cosplay_prompt.delete("0.0", "end") |
| cosplay_prompt.insert("0.0", cosplay) |
|
|
| def analyze_keywords_in_data(parquet_file, keyword): |
| |
| df = pd.read_parquet(parquet_file) |
| |
| bag_of_tags = ['penis', 'character name', 'upper body','holding','full body', 'artist name', 'male focus', 'open mouth', 'mature male', 'muscular', 'muscular male', 'closed mouth','closed eyes', 'white background', 'solo', 'breasts', 'simple background', 'smile', 'looking at viewer', 'flat chest', 'small breasts', 'medium breasts', 'large breasts', 'huge breasts','aqua eyes', 'black eyes', 'blue eyes', 'brown eyes', 'green eyes', 'grey eyes', 'orange eyes', 'purple eyes', 'pink eyes', 'red eyes', 'white eyes', 'yellow eyes', 'amber eyes', 'heterochromia', 'multicolored eyes', 'aqua pupils', 'blue pupils', 'brown pupils', 'green pupils', 'grey pupils', 'orange pupils', 'pink pupils', 'purple pupils', 'red pupils', 'white pupils', 'yellow pupils', 'pointy ears', 'long pointy ears', 'aqua hair', 'black hair', 'blonde hair', 'blue hair', 'light blue hair', 'dark blue hair', 'brown hair', 'light brown hair', 'green hair', 'dark green hair', 'light green hair', 'grey hair', 'orange hair', 'pink hair', 'purple hair', 'light purple hair', 'red hair', 'white hair', 'multicolored hair', 'colored inner hair', 'colored tips', 'roots (hair)', 'gradient hair', 'print hair', 'rainbow hair', 'split-color hair', 'spotted hair', 'streaked hair', 'two-tone hair', 'very short hair', 'short hair', 'medium hair', 'long hair', 'very long hair', 'absurdly long hair', 'big hair', 'bald', 'bald girl', 'bob cut', 'inverted bob', 'bowl cut', 'buzz cut', 'chonmage', 'crew cut', 'flattop', 'okappa', 'pixie cut', 'undercut', 'flipped hair', 'wolf cut', 'cornrows', 'dreadlocks', 'hime cut', 'mullet', 'bow-shaped hair', 'braid', 'braided bangs', 'front braid', 'side braid', 'french braid', 'crown braid', 'single braid', 'multiple braids', 'twin braids', 'low twin braids', 'tri braids', 'quad braids', 'flower-shaped hair', 'hair bun', 'braided bun', 'single hair bun', 'double bun', 'cone hair bun', 'doughnut hair bun', 'heart hair bun', 'triple bun', 'cone hair bun', 'hair rings', 'single hair ring', 'half updo', 'one side up', 'two side up', 'low-braided long hair', 'low-tied long hair', 'mizura', 'multi-tied hair', 'nihongami', 'ponytail', 'folded ponytail', 'front ponytail', 'high ponytail', 'short ponytail', 'side ponytail', 'split ponytail', 'star-shaped hair', 'topknot', 'twintails', 'low twintails', 'short twintails', 'uneven twintails', 'tri tails', 'quad tails', 'quin tails', 'twisted hair', 'afro', 'huge afro', 'beehive hairdo', 'crested hair', 'pompadour', 'quiff', 'shouten pegasus mix mori', 'curly hair', 'drill hair', 'twin drills', 'tri drills', 'hair flaps', 'messy hair', 'pointy hair', 'ringlets', 'spiked hair', 'straight hair', 'wavy hair', 'bangs', 'arched bangs', 'asymmetrical bangs', 'bangs pinned back', 'blunt bangs', 'crossed bangs', 'diagonal bangs', 'dyed bangs', 'fanged bangs', 'hair over eyes', 'hair over one eye', 'long bangs', 'parted bangs', 'curtained hair', 'ribbon bangs', 'short bangs', 'swept bangs', 'hair between eyes', 'hair intakes', 'single hair intake', 'sidelocks', 'asymmetrical sidelocks', 'drill sidelocks', 'low-tied sidelocks', 'sidelocks tied back', 'single sidelock', 'ahoge', 'heart ahoge', 'huge ahoge', 'antenna hair', 'heart antenna hair', 'comb over', 'hair pulled back', 'hair slicked back', 'mohawk', 'oseledets', 'lone nape hair', 'hair bikini', 'hair censor', 'hair in own mouth', 'hair over breasts', 'hair over one breast', 'hair over crotch', 'hair over shoulder', 'hair scarf', 'alternate hairstyle', 'hair down', 'hair up', 'asymmetrical hair', 'sidecut', 'blunt ends', 'dark skin', 'dark-skinned female', 'pale skin', 'sun tatoo', 'black skin', 'blue skin', 'green skin', 'grey skin', 'orange skin', 'pink skin', 'purple skin', 'red skin', 'white skin', 'yellow skin', 'colored skin', 'multiple tails', 'demon tail', 'dragon tail', 'ghost tail', 'pikachu tail', 'snake head tail', 'fiery tail', 'bear tail', 'rabbit tail', 'cat tail', 'cow tail', 'deer tail', 'dog tail', 'ermine tail', 'fox tail', 'horse tail', 'leopard tail', 'lion tail', 'monkey tail', 'mouse tail', 'pig tail', 'sheep tail', 'squirrel tail', 'tiger tail', 'wolf tail', 'crocodilian tail', 'fish tail', 'scorpion tail', 'snake tail', 'tadpole tail'] |
| mini_bag_of_tags = ['penis', 'character name', 'upper body', 'full body', 'alternate costume','artist name', 'male focus', 'open mouth', 'mature male', 'muscular', 'muscular male', 'closed mouth','closed eyes','white background', 'solo', 'breasts', 'simple background', 'smile', 'looking at viewer'] |
| |
| filtered_df = df[df['character'] == keyword] |
| filtered_df = filtered_df[~filtered_df['character'].str.contains(',', na=False)] |
| if filtered_df.empty: |
| filtered_df = df[df['character'].str.contains(keyword, na=False, regex=False)] |
| filtered_df = filtered_df[~(filtered_df['rating'] == 'e')] |
| |
| |
| keyword_count_dict = {} |
| |
| |
| for general in filtered_df['general']: |
| |
| tags = [tag.strip() for tag in general.split(',')] |
| |
| |
| for tag in tags: |
| keyword_count_dict[tag] = keyword_count_dict.get(tag, 0) + 1 |
| |
| |
| exclude_keywords = bag_of_tags |
| keyword_count_dict1 = keyword_count_dict.copy() |
| keyword_count_dict1 = {k: v for k, v in keyword_count_dict1.items() if k not in mini_bag_of_tags} |
| keyword_count_dict = {k: v for k, v in keyword_count_dict.items() if k not in exclude_keywords} |
| |
| |
| highest_count = max(keyword_count_dict.values()) if keyword_count_dict else 0 |
| highest_count1 = max(keyword_count_dict1.values()) if keyword_count_dict1 else 0 |
| |
| if(len(filtered_df)) < 10: |
| weight = 0.6 |
| elif(len(filtered_df)) < 30: |
| weight = 0.5 |
| elif(len(filtered_df)) < 80: |
| weight = 0.45 |
| elif(len(filtered_df)) < 160: |
| weight = 0.35 |
| elif(len(filtered_df)) < 280: |
| weight = 0.3 |
| else: |
| weight = 0.2 |
| |
| result_list = [] |
| result_list1 = [] |
| |
| for k, v in keyword_count_dict.items(): |
| if '|' in k: continue |
| if v >= highest_count * weight: |
| result_list.append(k) |
| |
| for k, v in keyword_count_dict1.items(): |
| if '|' in k: continue |
| if v >= highest_count1 * weight: |
| result_list1.append(k) |
| |
| result_list1.insert(1, keyword) |
| result_list.insert(1, "alternative costume, ["+keyword+" (cosplay)]") |
| return ", ".join(result_list1),", ".join(result_list) |
|
|
| def update_listbox(search_keyword): |
| |
| if listbox.size() > 0: |
| listbox.delete(0, "end") |
|
|
| |
| all_matching_keywords = [] |
|
|
| |
| added_keywords = set() |
|
|
| |
| if search_keyword and len(search_keyword) >= 3: |
| |
| for key, keywords in copyright_dict_0103.copyright_dict.items(): |
| if search_keyword.lower() in key.lower(): |
| for keyword in keywords: |
| count = cd.character_dictionary.get(keyword, 0) |
| if count > 50 and keyword not in added_keywords: |
| all_matching_keywords.append((keyword, count)) |
| added_keywords.add(keyword) |
|
|
| |
| matching_keywords = [ |
| (k, v) for k, v in cd.character_dictionary.items() |
| if search_keyword.lower() in k.lower() and v > 50 and k not in added_keywords |
| ] |
| all_matching_keywords.extend(matching_keywords) |
|
|
| else: |
| |
| all_matching_keywords = [ |
| (k, v) for k, v in cd.character_dictionary.items() if v > 50 and k not in added_keywords |
| ] |
|
|
| |
| all_matching_keywords.sort(key=lambda item: item[1], reverse=True) |
| for keyword, count in all_matching_keywords: |
| listbox.insert("end", f"{keyword} - {count}") |
|
|
| search_entry.grid(row =0, column =0, padx=5, pady=5, sticky="nsew") |
| search_entry.bind("<KeyRelease>", on_key_release) |
| search_label = customtkinter.CTkLabel(frame_head, text="โ ๊ฒ์ ๋์ copyright/character ์
๋ ฅ (English) | ์ฑ๋ฅ ์ด์๋ก Danbooru 50์ฅ ์ด์๋ง ํ์", font=my_font) |
| search_label.grid(row =0, column =1, padx=5, pady=5, sticky="w") |
| self.after(2500, lambda: self.attributes('-topmost', False)) |
|
|
| def show_image(event): |
| highlight_tags = ["aqua","black","blonde","blue","brown","cyan","green","grey","orange","pink","purple","red","violet","white","yellow","mouth", "eyes", " cap", " ears", " girl", " ornament", " hat", "beret", " ear "] |
|
|
| def insert_with_color(cs_text_input, current_lookup): |
| words = [word.strip() for word in current_lookup.split(',')] |
| for index, word in enumerate(words): |
| highlight = False |
| start_index = cs_text_input.index("end-1c") |
| if index == len(words) - 1: |
| cs_text_input.insert("end", word) |
| else: |
| cs_text_input.insert("end", word + ", ") |
| end_index = cs_text_input.index("end-1c") |
|
|
| for tag in highlight_tags: |
| if tag in word: |
| highlight = True |
| if word == "hat": highlight = True |
| |
| if word == keyword or word in tagbag.bag_of_tags[5:] or 'tail' in word or 'pupil' in word: |
| cs_text_input.tag_add(word, start_index, end_index) |
| cs_text_input.tag_config(word, foreground="#FFFF97") |
| elif highlight: |
| cs_text_input.tag_add(word, start_index, end_index) |
| cs_text_input.tag_config(word, foreground="#AED395") |
| selected_indices = listbox.curselection() |
| selected_index = selected_indices[0] |
| keyword_with_count = listbox.get(selected_index).strip() |
| keyword = keyword_with_count.split(' - ')[0].strip() |
| if keyword in character_book_dict: |
| try: |
| image_stream = io.BytesIO(base64.b64decode(character_book_dict[keyword][3])) |
| reloaded_image = Image.open(image_stream) |
| image_label.configure(image=customtkinter.CTkImage(reloaded_image, size=(384,512))) |
| character_prompt.delete("0.0", "end") |
| insert_with_color(character_prompt, character_book_dict[keyword][1]) |
| cosplay_prompt.delete("0.0", "end") |
| insert_with_color(cosplay_prompt, character_book_dict[keyword][2]) |
| except: |
| return |
| else: |
| if character_prompt.get("0.0","end") != None: |
| character_prompt.delete("0.0", "end") |
| if cosplay_prompt.get("0.0","end") != None: |
| cosplay_prompt.delete("0.0", "end") |
| image_label.configure(image=white_photo) |
|
|
| listbox = tk.Listbox(frame_left, width=32, height=30, font = font.Font(family='Pretendard', size=13), bg='#2B2B2B', fg='#F8F8F8', borderwidth=2, highlightbackground='lightgrey') |
| listbox.bind('<<ListboxSelect>>', show_image) |
| listbox.grid(row = 0, column=0, padx=5, pady=5, sticky="nsew") |
|
|
| search_button = customtkinter.CTkButton(self, font=my_font, text="์กฐํ", fg_color="grey", hover_color="grey10", command=on_search) |
| search_button.grid(row = 1, column=1, padx=5, pady=150, sticky="w") |
|
|
| all_keywords = sorted( |
| ((k, v) for k, v in cd.character_dictionary.items() if v > 50), |
| key=lambda item: item[1], |
| reverse=True |
| ) |
| for keyword, count in all_keywords: |
| listbox.insert("end", f"{keyword} - {count}") |
| |
| self.protocol("WM_DELETE_WINDOW", lambda: on_close(self)) |
|
|
| collection_button = customtkinter.CTkButton(frame_head, text="Copyright ๋๊ฐ", font=my_font, fg_color="grey", hover_color="grey10", command=lambda: open_collection(self)) |
| collection_button.grid(row =0, column =2, padx=50, pady=5, sticky="e") |
|
|
| |
| |
|
|
| def open_collection(self): |
| self.attributes('-topmost', False) |
| collection_window = customtkinter.CTkToplevel() |
| collection_window.title("๋๊ฐ") |
| collection_window.attributes('-topmost', True) |
| collection_window.resizable(width=False, height=False) |
|
|
| _height = app.winfo_screenheight() |
| print(_height) |
| if _height < 1152: |
| yview = 660 |
| ylist = 46 |
| else: |
| yview = 700 |
| ylist = 50 |
|
|
| collection_frame_head = customtkinter.CTkFrame(collection_window, width=1220, height=40) |
| collection_frame_head.grid(row =0, column =0, columnspan=5, padx=5, pady=5, sticky="nsew") |
|
|
| collection_label1 = customtkinter.CTkLabel(collection_frame_head, font=my_font, text="์ด๋ฆ์ ๋๋ฌ ํน์ง์ ๋ณต์ฌํฉ๋๋ค.") |
| collection_label1.grid(row = 0, column=0, padx=5, pady=5, sticky="nsew") |
|
|
| collection_frame_left = customtkinter.CTkFrame(collection_window, width=300, height=yview) |
| collection_frame_left.grid(row =1, column =0, padx=5, pady=5, sticky="nsew") |
| collection_frame_right = customtkinter.CTkScrollableFrame(collection_window, width=900, height=yview) |
| collection_frame_right.grid(row =1, column =2, padx=5, pady=5, sticky="nsew") |
|
|
| show_window = [] |
|
|
| def copy_character_info(key): |
| text = character_book_dict[key][1].split(',') |
| text_list = [keyword.strip() for keyword in text] |
| charactersitics = [] |
| charactersitics.append("{"+key+"}") |
| count = 0 |
| for texts in text_list: |
| if count < 4 and texts in tagbag.bag_of_tags[5:] and 'tail' not in texts and 'pupil' not in texts and '1girl' not in texts and '1boy' not in texts and '1other' not in texts: |
| charactersitics.append(texts) |
| count += 1 |
| texts = ', '.join(charactersitics) |
| pyperclip.copy(texts) |
| collection_label1.configure(text=f"์ ํํ์ {texts}๊ฐ ํด๋ฆฝ๋ณด๋์ ๋ณต์ฌ๋์์ต๋๋ค.", text_color="#FFFF97") |
|
|
|
|
|
|
|
|
| def spread_image(event, show_window): |
| collection_frame_right._parent_canvas.yview_moveto(0) |
| selected_indices = collection_listbox.curselection() |
| selected_index = selected_indices[0] |
| collection_listbox.configure(state="disabled") |
| if show_window: |
| for i in show_window: |
| i.destroy() |
| keyword_with_count = collection_listbox.get(selected_index).strip() |
| keyword = ' '.join(keyword_with_count.split('%')[1].split()).strip() |
| try: |
| key_in_book = split_dict[keyword] |
| |
| except KeyError: |
| print(f"KeyError: '{keyword}' not found in the split dictionary.") |
| count = 0 |
| for key in key_in_book: |
| _row = count//3 |
| _column = count%3 |
| if key in character_book_dict: |
| f = customtkinter.CTkFrame(collection_frame_right, width = 298, height=394) |
| f.grid(row=_row, column=_column, padx=5, pady=5, sticky="nsew") |
| name = key |
| if len(key) > 45: |
| name = key[:44] |
| b = customtkinter.CTkButton(f, font=my_font,text=name, state="disabled" if len(key) > 45 else "normal", fg_color="grey10", hover_color="grey10", text_color_disabled="white", command=lambda name=name: copy_character_info(name)) |
| b.grid(row=0, column=0, padx=5, pady=5, sticky="nsew") |
| image_stream = io.BytesIO(base64.b64decode(character_book_dict[key][3])) |
| reloaded_image = Image.open(image_stream) |
| i = customtkinter.CTkLabel(f, text="", image=customtkinter.CTkImage(reloaded_image, size=(288,384))) |
| i.grid(row=1, column=0, padx=5, pady=5, sticky="nsew") |
| show_window.append(f) |
| count += 1 |
| collection_listbox.configure(state="normal") |
| collection_listbox.selection_clear(selected_index) |
|
|
| collection_listbox = tk.Listbox(collection_frame_left, width=32, height=ylist, font = font.Font(family='Pretendard', size=13), bg='#2B2B2B', fg='#F8F8F8', borderwidth=2, highlightbackground='lightgrey') |
| collection_listbox.bind('<<ListboxSelect>>',lambda event: spread_image(event, show_window)) |
| collection_listbox.grid(row = 0, column=0, padx=5, pady=5, sticky="nsew") |
|
|
| |
|
|
| all_collections = sorted( |
| ((k, v) for k, v in copyright_dict_0103.copyright_dict.items()), |
| key=lambda item: len(item[1]), |
| reverse=True |
| ) |
|
|
| split_dict = {} |
| for keyword, value in all_collections: |
| key_in_book = copyright_dict_0103.copyright_dict[keyword] |
| cover = sum(key in character_book_dict for key in key_in_book) |
| coverage = f"{round((cover / len(key_in_book)) * 100, 1) if cover else 0}%" |
| |
| |
| if len(key_in_book) > 96: |
| for i in range(0, len(key_in_book), 96): |
| part_key = f"{keyword} ({i//96 + 1})" |
| split_dict[part_key] = key_in_book[i:i+96] |
| cover_part = sum(key in character_book_dict for key in split_dict[part_key]) |
| coverage_part = f"{round((cover_part / len(split_dict[part_key])) * 100, 1) if cover_part else 0}%" |
| collection_listbox.insert("end", f"{coverage_part:<7s}{keyword} ({i//96 + 1})") |
| else: |
| split_dict[keyword] = key_in_book |
| collection_listbox.insert("end", f"{coverage:<7s}{keyword}") |
|
|
|
|
| class Preset_open(customtkinter.CTkToplevel): |
|
|
| def __init__(self, *args, **kwargs): |
| super().__init__(*args, **kwargs) |
| self.title("ํ๋กฌํํธ ๊ฒ์/์ค์ ํ๋ฆฌ์
") |
| self.attributes('-topmost', True) |
| self.resizable(width=False, height=False) |
|
|
| my_font = customtkinter.CTkFont('Pretendard', 13) |
|
|
| presets ={} |
| items = 0 |
| preset_frame_left = customtkinter.CTkFrame(self, width=250, height=520) |
| preset_frame_left.grid(row =1, rowspan=10, column =0, padx=5, pady=5, sticky="nsew") |
| preset_frame_right = customtkinter.CTkFrame(self, width=700, height=520) |
| preset_frame_right.grid(row =1, rowspan=10, column =1, columnspan = 2, padx=5, pady=5, sticky="nsew") |
| preset_insert_right_frame = customtkinter.CTkFrame(self) |
| preset_insert_right_frame.grid(row =0, column =0, columnspan=2, padx=5, pady=5, sticky="nsew") |
| preset_insert_button = customtkinter.CTkButton(preset_insert_right_frame, text="ํ๋ฆฌ์
์ถ๊ฐ", font=my_font, command=lambda: add_preset(self, presets)) |
| preset_insert_button.grid(row =0, column =0, padx=5, pady=5, sticky="nsew") |
| preset_insert_name = customtkinter.CTkLabel(preset_insert_right_frame, text="ํ๋ฆฌ์
๋ช
์นญ : ", font=my_font) |
| preset_insert_name.grid(row =0, column =1, padx=5, pady=5, sticky="w") |
| preset_insert_name_entry = customtkinter.CTkEntry(preset_insert_right_frame,width=200, font=my_font) |
| preset_insert_name_entry.grid(row =0, column =2, padx=5, pady=5, sticky="w") |
| preset_insert_description = customtkinter.CTkLabel(preset_insert_right_frame, text=" ์ค๋ช
: ", font=my_font) |
| preset_insert_description.grid(row =0, column =3, padx=5, pady=5, sticky="w") |
| preset_insert_description_entry = customtkinter.CTkEntry(preset_insert_right_frame, width=420, font=my_font) |
| preset_insert_description_entry.grid(row =0, column =4, padx=5, pady=5, sticky="w") |
| preset_frame_bottom = customtkinter.CTkFrame(self) |
| preset_frame_bottom.grid(row =11, column =0, columnspan=2, padx=5, pady=5, sticky="nsew") |
| preset_label = customtkinter.CTkLabel(preset_frame_bottom, text="์์ฑ๊ธฐ ๋ด ์ข์ธก ํจ๋์ ๊ฐ๋ค์ ํ๋ฆฌ์
์ผ๋ก ์ ์ฅํ ์ ์์ต๋๋ค.", font=my_font) |
| preset_label.grid(row =0, column=0,sticky="n" ) |
|
|
| def _yield(selected_option): |
| ypreset = self.presets[selected_option] |
| yield_preset_description.configure(text=ypreset['description']) |
| yield_preset_search.configure(text=ypreset['search']) |
| yield_preset_exclude.configure(text=ypreset['exclude']) |
| yield_preset_prompt.configure(state="normal") |
| yield_preset_prompt.delete("0.0", "end") |
| yield_preset_prompt.insert("0.0", ypreset['prompt']) |
| yield_preset_prompt.configure(state="disabled") |
| _text = "" |
| if ypreset['explicit'] == 1: |
| _text += "Explicit " |
| if ypreset['nsfw'] == 1: |
| _text += "NSFW " |
| if ypreset['sensitive'] == 1: |
| _text += "Sensitive " |
| if ypreset['general'] == 1: |
| _text += "General" |
| yield_activated_ratings.configure(text = _text) |
| yield_prefix.configure(state="normal") |
| yield_prefix.delete("0.0", "end") |
| yield_prefix.insert("0.0", ypreset['fix']) |
| yield_prefix.configure(state="disabled") |
| yield_postfix.configure(state="normal") |
| yield_postfix.delete("0.0", "end") |
| yield_postfix.insert("0.0", ypreset['fix_after']) |
| yield_postfix.configure(state="disabled") |
| yield_negative.configure(state="normal") |
| yield_negative.delete("0.0", "end") |
| yield_negative.insert("0.0", ypreset['negative']) |
| yield_negative.configure(state="disabled") |
| yield_autohide.configure(state="normal") |
| yield_autohide.delete("0.0", "end") |
| yield_autohide.insert("0.0", ypreset['auto_hide']) |
| yield_autohide.configure(state="disabled") |
| _text = "" |
| if ypreset['rm_artist'] == 1: |
| _text += "์๊ฐ๋ช
์ ๊ฑฐ " |
| if ypreset['rm_copyright'] == 1: |
| _text += "์ํ๋ช
์ ๊ฑฐ " |
| if ypreset['rm_character'] == 1: |
| _text += "์บ๋ฆญํฐํน์ง์ ๊ฑฐ" |
| yield_rm.configure(text = _text) |
|
|
|
|
|
|
| preset_list = CTkListbox(preset_frame_left, height=570, font=my_font, command= _yield) |
| preset_list.grid(row = 0, column=0, padx=5, pady=5, sticky="nsew") |
|
|
| show_preset_description = customtkinter.CTkButton(preset_frame_right, text="Description", font=my_font, width=150, fg_color="grey10", text_color_disabled="white",state="disabled") |
| show_preset_description.grid(row = 0, column = 0, padx=5, pady=5, sticky="w") |
| yield_preset_description = customtkinter.CTkLabel(preset_frame_right, text="", font=my_font, width=550) |
| yield_preset_description.grid(row = 0, column = 1, padx=5, pady=5, sticky="w") |
| |
| def insert_search(): |
| if preset_list.get(): |
| app.search_label_entry.delete(0, "end") |
| app.search_label_entry.insert(0, yield_preset_search.cget("text")) |
| show_preset_search = customtkinter.CTkButton(preset_frame_right, text="Search Keyword", font=my_font, width=150, fg_color="grey10", hover_color="grey", command=insert_search) |
| show_preset_search.grid(row = 1, column = 0, padx=5, pady=5, sticky="w") |
| yield_preset_search = customtkinter.CTkLabel(preset_frame_right, text="", font=my_font, width=550) |
| yield_preset_search.grid(row = 1, column = 1, padx=5, pady=5, sticky="w") |
| def insert_exclude(): |
| if preset_list.get(): |
| app.exclude_label_entry.delete(0, "end") |
| app.exclude_label_entry.insert(0, yield_preset_exclude.cget("text")) |
| show_preset_exclude = customtkinter.CTkButton(preset_frame_right, text="Exclude Keyword", font=my_font, width=150, fg_color="grey10", hover_color="grey", command=insert_exclude) |
| show_preset_exclude.grid(row = 2, column = 0, padx=5, pady=5, sticky="w") |
| yield_preset_exclude = customtkinter.CTkLabel(preset_frame_right, text="", font=my_font, width=550) |
| yield_preset_exclude.grid(row = 2, column = 1, padx=5, pady=5, sticky="w") |
| def insert_prompt(): |
| if preset_list.get(): |
| app.text_input.delete("0.0","end") |
| app.text_input.insert("0.0", yield_preset_prompt.get("0.0", "end")) |
| show_preset_prompt = customtkinter.CTkButton(preset_frame_right, text="Prompt", font=my_font, width=150, height=100, fg_color="grey10", hover_color="grey", command=insert_prompt) |
| show_preset_prompt.grid(row = 3, column = 0, padx=5, pady=5, sticky="w") |
| yield_preset_prompt = customtkinter.CTkTextbox(preset_frame_right, font=my_font, state="disabled",width=550, height=100, fg_color='#2B2B2B') |
| yield_preset_prompt.grid(row = 3, column = 1, padx=5, pady=5, sticky="w") |
| def insert_ratings(): |
| if preset_list.get(): |
| app.rating_select_explicit.deselect() |
| app.rating_select_nsfw.deselect() |
| app.rating_select_sensitive.deselect() |
| app.rating_select_general.deselect() |
| _text = yield_activated_ratings.cget("text") |
| if "Explicit" in _text: |
| app.rating_select_explicit.select() |
| if "NSFW" in _text: |
| app.rating_select_nsfw.select() |
| if "Sensitive" in _text: |
| app.rating_select_sensitive.select() |
| if "General" in _text: |
| app.rating_select_general.select() |
| show_activated_ratings = customtkinter.CTkButton(preset_frame_right, text="Ratings", font=my_font, width=150, fg_color="grey10", hover_color="grey", command=insert_ratings) |
| show_activated_ratings.grid(row = 4, column = 0, padx=5, pady=5, sticky="w") |
| yield_activated_ratings = customtkinter.CTkLabel(preset_frame_right, text="", font=my_font, width=550) |
| yield_activated_ratings.grid(row = 4, column = 1, padx=5, pady=5, sticky="w") |
|
|
| |
| def insert_prefix(): |
| if preset_list.get(): |
| app.fixed_prompt_input.delete("0.0","end") |
| app.fixed_prompt_input.insert("0.0", yield_prefix.get("0.0", "end")) |
| show_prefix = customtkinter.CTkButton(preset_frame_right, text="Pre-fix", font=my_font, width=150, height=50, fg_color="grey10", hover_color="grey", command=insert_prefix) |
| show_prefix.grid(row = 5, column = 0, padx=5, pady=5, sticky="w") |
| yield_prefix = customtkinter.CTkTextbox(preset_frame_right, font=my_font, state="disabled",width=550, height=50, fg_color='#2B2B2B') |
| yield_prefix.grid(row = 5, column = 1, padx=5, pady=5, sticky="w") |
| def insert_postfix(): |
| if preset_list.get(): |
| app.fixed_prompt_after_input.delete("0.0","end") |
| app.fixed_prompt_after_input.insert("0.0", yield_postfix.get("0.0", "end")) |
| show_postfix = customtkinter.CTkButton(preset_frame_right, text="Post-fix", font=my_font, width=150, height=50, fg_color="grey10", hover_color="grey", command=insert_postfix) |
| show_postfix.grid(row = 6, column = 0, padx=5, pady=5, sticky="w") |
| yield_postfix = customtkinter.CTkTextbox(preset_frame_right, font=my_font, state="disabled",width=550, height=50, fg_color='#2B2B2B') |
| yield_postfix.grid(row = 6, column = 1, padx=5, pady=5, sticky="w") |
| def insert_negative(): |
| if preset_list.get(): |
| app.negative_prompt_input.delete("0.0","end") |
| app.negative_prompt_input.insert("0.0", yield_negative.get("0.0", "end")) |
| show_negative = customtkinter.CTkButton(preset_frame_right, text="Negative prompt", font=my_font, width=150, height=70, fg_color="grey10", hover_color="grey", command=insert_negative) |
| show_negative.grid(row = 7, column = 0, padx=5, pady=5, sticky="w") |
| yield_negative = customtkinter.CTkTextbox(preset_frame_right, font=my_font, state="disabled",width=550, height=70, fg_color='#2B2B2B') |
| yield_negative.grid(row = 7, column = 1, padx=5, pady=5, sticky="w") |
| def insert_autohide(): |
| if preset_list.get(): |
| app.auto_hide_keyword_input.delete("0.0","end") |
| app.auto_hide_keyword_input.insert("0.0", yield_autohide.get("0.0", "end")) |
| show_autohide = customtkinter.CTkButton(preset_frame_right, text="Hidden keyword", font=my_font, width=150, height=70, fg_color="grey10", hover_color="grey", command=insert_autohide) |
| show_autohide.grid(row = 8, column = 0, padx=5, pady=5, sticky="w") |
| yield_autohide = customtkinter.CTkTextbox(preset_frame_right, font=my_font, state="disabled",width=550, height=70, fg_color='#2B2B2B') |
| yield_autohide.grid(row = 8, column = 1, padx=5, pady=5, sticky="w") |
| |
|
|
| def insert_rm(): |
| if preset_list.get(): |
| app.rm_artist_name_button.deselect() |
| app.rm_copyright_name_button.deselect() |
| app.rm_characteristic_button.deselect() |
| _text = yield_rm.cget("text") |
| if "์๊ฐ๋ช
์ ๊ฑฐ" in _text: |
| app.rm_artist_name_button.select() |
| if "์ํ๋ช
์ ๊ฑฐ" in _text: |
| app.rm_copyright_name_button.select() |
| if "์บ๋ฆญํฐํน์ง์ ๊ฑฐ" in _text: |
| app.rm_characteristic_button.select() |
| show_rm = customtkinter.CTkButton(preset_frame_right, text="Options", font=my_font, width=150, fg_color="grey10", hover_color="grey", command=insert_rm) |
| show_rm.grid(row = 9, column = 0, padx=5, pady=5, sticky="w") |
| yield_rm = customtkinter.CTkLabel(preset_frame_right, text="", font=my_font, width=550) |
| yield_rm.grid(row = 9, column = 1, padx=5, pady=5, sticky="w") |
| def apply_all(): |
| if preset_list.get(): |
| insert_search() |
| insert_exclude() |
| insert_prompt() |
| insert_ratings() |
| insert_prefix() |
| insert_postfix() |
| insert_negative() |
| insert_autohide() |
| insert_rm() |
| apply_all = customtkinter.CTkButton(preset_frame_right, text="์ผ๊ด์ค์ ", font=my_font, width=150, command=apply_all) |
| apply_all.grid(row = 10, column = 0, padx=5, pady=5, sticky="w") |
| def remove_preset(self): |
| presets = self.presets |
| preset_name = preset_list.get() |
| if preset_name in presets: |
| del presets[preset_name] |
| save_preset(presets) |
| self.presets = load_preset() |
| update_list(self, self.presets) |
| remove_preset_button = customtkinter.CTkButton(preset_frame_right, text="์ด ํ๋ฆฌ์
์ ์ญ์ ", font=my_font, fg_color="grey10", hover_color="grey", command=lambda: remove_preset(self)) |
| remove_preset_button.grid(row = 10, column =1, padx=5, pady=5, sticky="nsew") |
| |
|
|
|
|
| def add_preset(self, presets): |
| preset_name = preset_insert_name_entry.get() |
| if len(preset_name.strip()) < 1: |
| preset_label.configure(text="ํ๋ฆฌ์
์ด๋ฆ์ ์ ํํ ์
๋ ฅํด์ฃผ์ธ์.", text_color="#FFFF97") |
| else: |
| preset = { |
| 'preset_name':preset_name, |
| 'description':preset_insert_description_entry.get(), |
| 'search':app.search_label_entry.get(), |
| 'exclude':app.exclude_label_entry.get(), |
| 'prompt':app.text_input.get("0.0", "end-1c"), |
| 'fix':app.fixed_prompt_input.get("0.0", "end-1c"), |
| 'fix_after':app.fixed_prompt_after_input.get("0.0", "end-1c"), |
| 'negative':app.negative_prompt_input.get("0.0", "end-1c"), |
| 'auto_hide':app.auto_hide_keyword_input.get("0.0", "end-1c"), |
| 'rm_artist':app.rm_artist_name_button.get(), |
| 'rm_character':app.rm_characteristic_button.get(), |
| 'rm_copyright':app.rm_copyright_name_button.get(), |
| 'explicit':app.rating_select_explicit.get(), |
| 'nsfw':app.rating_select_nsfw.get(), |
| 'sensitive':app.rating_select_sensitive.get(), |
| 'general':app.rating_select_general.get() |
| } |
| updated_preset = {preset_name: preset} |
| if self.presets: |
| updated_preset.update(self.presets) |
| else: |
| self.presets = updated_preset |
| save_preset(updated_preset) |
| self.presets = load_preset() |
| update_list(self, self.presets) |
|
|
| def update_list(self, presets): |
| if preset_list.size() != 0: |
| preset_list.delete(0, "END") |
| if presets: |
| for i, names in enumerate(presets.keys()): |
| preset_list.insert(i, names) |
| self.presets = presets |
|
|
| def save_preset(presets): |
| with open('NAIA_preset.json', 'w', encoding='utf-8') as f: |
| json.dump(presets, f, ensure_ascii=False, indent=4) |
|
|
| def load_preset(): |
| try: |
| if os.path.exists('NAIA_preset.json'): |
| with open('NAIA_preset.json', 'r', encoding='utf-8') as f: |
| return json.load(f) |
| except FileNotFoundError: |
| return {} |
|
|
| def on_close(self): |
| self.withdraw() |
|
|
| presets = load_preset() |
| update_list(self, presets) |
| self.protocol("WM_DELETE_WINDOW", lambda: on_close(self)) |
|
|
|
|
|
|
|
|
| class App(customtkinter.CTk, TkinterDnD.DnDWrapper): |
| access_token = None |
| NAI_ID = None |
| data = Data(wlist, tagbag, arti_list, copyright_list_reformatted, cd, remove_result_e, remove_result_qe) |
|
|
| def __init__(self): |
| super().__init__() |
| self.TkdndVersion = TkinterDnD._require(self) |
|
|
| if getattr(sys, 'frozen', False): |
| basedir = sys._MEIPASS |
| else: |
| basedir = os.path.dirname(__file__) |
| self.basedir = basedir |
| ctypes.windll.gdi32.AddFontResourceW(os.path.abspath(os.path.join(basedir,"Pretendard-Bold.otf"))) |
| ctypes.windll.gdi32.AddFontResourceW(os.path.abspath(os.path.join(basedir,"Pretendard-Regular.otf"))) |
| my_font = customtkinter.CTkFont('Pretendard', 13) |
| large_font = customtkinter.CTkFont('Pretendard', 13) |
| v_large_font = customtkinter.CTkFont('Pretendard', 14, weight="normal") |
|
|
| |
| self.auto_time_left = 0 |
| self.auto_time_left_flag = False |
| self.auto_count_left = 0 |
| self.auto_count_left_flag = False |
| self.delay_offset = 0 |
| self.auto_thread = None |
| self.stop_event = None |
|
|
| |
| self.current_prompt_rating = None |
|
|
| |
| self.dictionary_parity = False |
|
|
| self.start_time_prime = datetime.now() |
| self.start_time = self.start_time_prime.strftime('%Y%m%d_%H%M') |
| self.xy_plot_count = 0 |
| self.import_image_negative = {} |
|
|
| self.title("NAIA Beta v1") |
| self.grid_rowconfigure(0, weight=1) |
| self.columnconfigure(2, weight=1) |
|
|
| windll.shcore.SetProcessDpiAwareness(1) |
| |
| self.left_frame = customtkinter.CTkScrollableFrame(self, width=510, height=960) |
| self.left_frame.grid(row=0, column=0, sticky="nsew") |
| self.right_frame = customtkinter.CTkScrollableFrame(self, width=768,height=960) |
| self.right_frame.grid(row=0, column=1, sticky="nsew") |
| self.hidden_frame = customtkinter.CTkFrame(self, width=250) |
| self.hidden_frame.grid(row=0, column=2, sticky="nsew") |
| self.image_label_report_sub = customtkinter.CTkTextbox(self.hidden_frame,width=250, height=500, font=large_font) |
| self.image_label_report_sub.grid(row=2, column=0, columnspan=2, pady=5, padx=5, sticky="nsew") |
| self.image_label_report_sub.configure(state="disabled") |
| self.state_label2 = customtkinter.CTkLabel(self.hidden_frame, text="state : idle", font=my_font) |
| self.state_label2.grid(row=8, column=0, columnspan=2, pady=5, padx=5, sticky="n") |
| self.hidden_frame.grid_forget() |
|
|
| def sync_text(): |
| content = self.image_label_report.get("0.0", "end-1c").rstrip() |
| label = self.state_label.cget("text").rstrip() |
| label_sub = self.state_label2.cget("text").rstrip() |
| content_sub = self.image_label_report_sub.get("0.0", "end-1c").rstrip() |
| if content != content_sub: |
| self.image_label_report_sub.configure(state="normal") |
| self.image_label_report_sub.delete("0.0", "end") |
| self.image_label_report_sub.insert("0.0", content) |
| self.image_label_report_sub.configure(text_color=self.image_label_report.cget("text_color"),state="disabled") |
| if label != label_sub: |
| self.state_label2.configure(text=label, text_color=self.image_label_report.cget("text_color")) |
| self.after(500, sync_text) |
|
|
| |
| self.search_frame = customtkinter.CTkFrame(self.left_frame) |
| self.search_frame.grid(row=0, column=0, sticky="nsew") |
|
|
| def open_AccountSetting(self): |
| if self.AccountSetting is None: |
| self.AccountSetting = AccountSetting(self) |
| else: |
| if self.AccountSetting.state() == 'withdrawn': |
| self.AccountSetting.deiconify() |
| else: |
| self.AccountSetting.NAI_ID_entry.focus() |
|
|
| def NAI_token_remove(self): |
| app.access_token = None |
| app.NAI_ID = None |
| if self.running_flag or self.automation_button.get()==1 or self.image_generation_button.cget("state") == "disabled": |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<UserAttention> NAI ํ ํฐ์ ์ ๊ฑฐ๊ฐ ๊ฐ๋ฅํ ์ํ๊ฐ ์๋๋๋ค. ๋ชจ๋ ์์
/์๋ํ ์ค์ ์ ๋ง์น๊ณ ์๋ ํด์ฃผ์ธ์.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| else: |
| app.NAI_Account_Login.configure(state="disabled") |
| app.NAI_Token_Remove.configure(state="disabled") |
| app.NAI_Account_State.configure(text="NAI Logout", text_color="red") |
| app.image_generation_button.configure(state="disabled") |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<UserAttention> NAI ํ ํฐ ์ ๊ฑฐ ์๋ฃ(Logout). ํ๋ก๊ทธ๋จ์ ์ ์์ ์ผ๋ก ์ข
๋ฃ ํ ์ฌ์คํํด์ผ ๋ฐ์๋ฉ๋๋ค.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
|
|
| |
| self.AccountSetting = None |
| |
| self.NAI_Account_Frame = customtkinter.CTkFrame(self.search_frame, width=440, height=50, bg_color="transparent") |
| self.NAI_Account_Frame.grid(row=0, column=0, padx=8, pady=5,sticky="nsew") |
| self.NAI_Account_Frame.columnconfigure(0, weight=6) |
| self.NAI_Account_Frame.columnconfigure(1, weight=4) |
| self.NAI_Account_Frame.columnconfigure(2, weight=4) |
|
|
| |
| self.NAI_Account_State = customtkinter.CTkLabel(self.NAI_Account_Frame, text="NAI Login : **** ", font=my_font) |
| |
| self.NAI_Account_State.grid(row=0, column=0, padx=20, sticky="w") |
| |
| self.NAI_Account_Login = customtkinter.CTkButton(self.NAI_Account_Frame, text="NAI ๊ณ์ ์ฐ๊ฒฐ", font=my_font, command=lambda: open_AccountSetting(self)) |
| self.NAI_Account_Login.grid(row=0, column=1, padx=20, sticky="ew") |
| |
| self.NAI_Token_Remove = customtkinter.CTkButton(self.NAI_Account_Frame, text="NAI ํ ํฐ์ ๊ฑฐ", state="disabled", fg_color="#848484", font=my_font, command=lambda: NAI_token_remove(self)) |
| self.NAI_Token_Remove.grid(row=0, column=2,padx=20, sticky="ew") |
|
|
| |
| self.search_label = customtkinter.CTkLabel(self.search_frame, text="๊ฒ์ ํค์๋ ์
๋ ฅ : keyword, *keyword, {keyword1|keyword2}", font=my_font) |
| self.search_label.grid(row=1, column=0, padx=8, sticky="w") |
| search_hyperlink = customtkinter.CTkLabel(self.search_frame, text="๊ฐ์ด๋ ์ด๊ธฐ (arca.live)", font=customtkinter.CTkFont('Pretendard', 13),text_color="lightblue", cursor="hand2") |
| search_hyperlink.grid(row=1, column=0, padx=8, sticky="e") |
| search_hyperlink.bind("<Button-1>", lambda e: webbrowser.open_new("https://arca.live/b/aiart/96006690")) |
| self.search_label_entry = customtkinter.CTkEntry(self.search_frame, width=490) |
| self.search_label_entry.grid(row=2, column=0, padx=8, sticky="w") |
| self.exclude_label = customtkinter.CTkLabel(self.search_frame, text="์ ์ธ ํค์๋ ์
๋ ฅ : keyword, ~keyword", font=my_font) |
| self.exclude_label.grid(row=3, column=0, padx=8, sticky="w") |
| self.exclude_label_entry = customtkinter.CTkEntry(self.search_frame, width=490) |
| self.exclude_label_entry.grid(row=4, column=0, padx=8, sticky="w") |
|
|
| |
| self.select_rating_frame = customtkinter.CTkFrame(self.search_frame, width=490) |
| self.select_rating_frame.grid(row=6, column=0, padx=8, pady=10, sticky="nsew") |
| |
| |
| self.rating_select_var_explicit = customtkinter.IntVar(value=1) |
| self.rating_select_var_nsfw = customtkinter.IntVar(value=1) |
| self.rating_select_var_sensitive = customtkinter.IntVar(value=1) |
| self.rating_select_var_general = customtkinter.IntVar(value=1) |
| self.rating_select_explicit = customtkinter.CTkCheckBox(self.select_rating_frame, width=85, text="Explicit", variable= self.rating_select_var_explicit, font=my_font) |
| self.rating_select_explicit.grid(row=0, column=0, sticky="ew") |
| self.rating_select_nsfw = customtkinter.CTkCheckBox(self.select_rating_frame, width=80,text="NSFW", variable= self.rating_select_var_nsfw, font=my_font) |
| self.rating_select_nsfw.grid(row=0, column=1, sticky="ew") |
| self.rating_select_sensitive = customtkinter.CTkCheckBox(self.select_rating_frame,width=90, text="Sensitive", variable= self.rating_select_var_sensitive, font=my_font) |
| self.rating_select_sensitive.grid(row=0, column=2, sticky="ew") |
| self.rating_select_general = customtkinter.CTkCheckBox(self.select_rating_frame,width=85, text="General", variable= self.rating_select_var_general, font=my_font) |
| self.rating_select_general.grid(row=0, column=3, sticky="e") |
|
|
| self.cached_rows = None |
| self.searching_flag = False |
| self.search_thread = None |
| self.freezed_cached_rows = None |
|
|
| def fav_check(id): |
| return id in self.fav_id_list |
|
|
| def null_prompt_attention(): |
| self.text_input.delete("0.0", "end") |
| self.text_input.insert("0.0", "ํ์ฌ ์กฐ๊ฑด์ ๋ง๋ ํ๋กฌํํธ๊ฐ ์์ต๋๋ค.") |
|
|
| def prompt_search(): |
| if not self.control_pressed: |
| self.searching_flag = True |
| self.search_button.configure(text="๊ฒ์ (์์
์ค)", state="disabled") |
| search_thread = threading.Thread(target=perform_search, daemon=True) |
| self.search_thread = search_thread |
| search_thread.start() |
| else: |
| depth_search() |
|
|
| def depth_search(): |
| depth_search_window = customtkinter.CTkToplevel() |
| depth_search_window.title("์ฌ์ธต๊ฒ์") |
| depth_search_window.attributes('-topmost', True) |
| depth_search_window.resizable(width=False, height=False) |
| self.freezed_cached_rows = self.cached_rows.copy() |
|
|
| def perform_depth_search(): |
| self.search_button.configure(text="์ฌ์ธต๊ฒ์์ค", state="disabled") |
| self.searching_flag = True |
| df = self.cached_rows |
| df = NAIA_search.search(df, search_label_entry.get(), exclude_label_entry.get()) |
| |
| if df is None: |
| null_prompt_attention() |
| else: |
| self.cached_rows = df |
| df.reset_index(drop=True, inplace=True) |
| update_labels(df) |
| self.searching_flag = False |
| depth_search_execute.configure(text="๋จ์ ํ๋กฌํํธ ํ์์ ์ฌ๊ฒ์ : "+str(len(self.cached_rows))) |
| prompt_rollback.configure(text="์ด์ ํ๋กฌํํธํ ๋ณต์ : "+str(len(self.freezed_cached_rows))) |
| self.search_button.configure(text="๊ฒ์", state="normal") |
| prompt_rollback.configure(state="normal") |
|
|
| def rollback(): |
| self.cached_rows = self.freezed_cached_rows.copy() |
| update_labels(self.cached_rows) |
| depth_search_execute.configure(text="๋จ์ ํ๋กฌํํธ ํ์์ ์ฌ๊ฒ์ : "+str(len(self.cached_rows))) |
| prompt_rollback.configure(state="disabled") |
|
|
| def depth_search_close(): |
| self.freezed_cached_rows = None |
| depth_search_window.destroy() |
|
|
| search_label = customtkinter.CTkLabel(depth_search_window, text="๊ฒ์ ํค์๋ ์
๋ ฅ : keyword, *keyword, {keyword1|keyword2}", font=my_font) |
| search_label.grid(row=1, column=0, columnspan=2, padx=8, pady=5, sticky="w") |
| search_label_entry = customtkinter.CTkEntry(depth_search_window, width=490) |
| search_label_entry.grid(row=2, column=0, columnspan=2, padx=8, pady=5, sticky="w") |
| exclude_label = customtkinter.CTkLabel(depth_search_window, text="์ ์ธ ํค์๋ ์
๋ ฅ : keyword, ~keyword", font=my_font) |
| exclude_label.grid(row=3, column=0, columnspan=2, padx=8, pady=5, sticky="w") |
| exclude_label_entry = customtkinter.CTkEntry(depth_search_window, width=490) |
| exclude_label_entry.grid(row=4, column=0, columnspan=2, padx=8, pady=5, sticky="w") |
| depth_search_execute = customtkinter.CTkButton(depth_search_window, text="๋จ์ ํ๋กฌํํธ ํ์์ ์ฌ๊ฒ์ : "+str(len(self.cached_rows)), font=my_font, command=perform_depth_search) |
| depth_search_execute.grid(row=5, column=0, padx=5,pady=5, sticky="n") |
| prompt_rollback = customtkinter.CTkButton(depth_search_window, text="์ด์ ํ๋กฌํํธํ ๋ณต์ : "+str(len(self.freezed_cached_rows)), font=my_font, fg_color="#7030A0", hover_color="#481F67", command=rollback, state="disabled") |
| prompt_rollback.grid(row=5, column=1, padx=5,pady=5, sticky="n") |
| depth_search_window.protocol("WM_DELETE_WINDOW", depth_search_close) |
|
|
| def perform_search(): |
| df = pd.read_parquet(os.path.join(basedir, "tags.parquet"), engine="pyarrow") |
| df = NAIA_search.search(df, self.search_label_entry.get(), self.exclude_label_entry.get(), self.rating_select_explicit.get(), self.rating_select_nsfw.get(), self.rating_select_sensitive.get(), self.rating_select_general.get()) |
| |
| if df is None: |
| null_prompt_attention() |
| else: |
| self.cached_rows = df |
| df.reset_index(drop=True, inplace=True) |
| update_labels(df) |
| self.searching_flag = False |
| self.search_button.configure(text="๊ฒ์", state="normal") |
|
|
| def update_labels(df): |
| self.searched_prompt_label.configure(text="๊ฒ์ ํ๋กฌํํธ ํ : " + str(len(self.cached_rows))) |
| self.cached_prompt_label.configure(text="๋จ์ ํ๋กฌํํธ ํ : " + str(len(self.cached_rows))) |
| print(df.info(memory_usage="deep")) |
|
|
| |
| self.search_button = customtkinter.CTkButton(self.select_rating_frame, text="๊ฒ์", font=my_font, command=prompt_search, width=130) |
| self.search_button.grid(row=0, column=4,padx=5, sticky="w") |
| |
| |
| self.searched_prompt_frame = customtkinter.CTkFrame(self.search_frame) |
| self.searched_prompt_frame.grid(row=7, column=0, padx=8, pady=5, sticky="nsew") |
| self.searched_prompt_frame.columnconfigure(0, weight= 1) |
| self.searched_prompt_frame.columnconfigure(1, weight= 1) |
| self.searched_prompt_frame.columnconfigure(2, weight= 1) |
| self.searched_prompt_label = customtkinter.CTkLabel(self.searched_prompt_frame, text="๊ฒ์ ํ๋กฌํํธ ํ : 0", font=my_font) |
| self.searched_prompt_label.grid(row=0, column=0, sticky="ew") |
| self.cached_prompt_label = customtkinter.CTkLabel(self.searched_prompt_frame, text="๋จ์ ํ๋กฌํํธ ํ : 0", font=my_font) |
| self.cached_prompt_label.grid(row=0, column=1, sticky="ew") |
| fdf = None |
|
|
| self.Preset_open = None |
| def open_Preset(self): |
| if self.Preset_open is None: |
| self.Preset_open = Preset_open(self) |
| else: |
| if self.Preset_open.state() == 'withdrawn': |
| self.Preset_open.deiconify() |
| else: |
| self.Preset_open.focus() |
|
|
| def open_favorite(): |
| favorite_window = customtkinter.CTkToplevel() |
| favorite_window.title("์ ํธ ํ๋กฌํํธ") |
| favorite_window.attributes('-topmost', True) |
| favorite_window.resizable(width=False, height=False) |
| self.freezed_cached_rows = self.cached_rows.copy() |
|
|
| global fdf |
| fdf = pd.read_parquet("favorite_prompt.parquet", engine="pyarrow") |
| odf = fdf.copy() |
| max_page = [1 + len(fdf) // 5] |
|
|
| def perform_favorite_search(): |
| global fdf |
| fdf = NAIA_search.search(odf, search_label_entry.get(), exclude_label_entry.get()) |
| |
| if type(fdf) == type(None): |
| assign_row.configure(text="๊ฒ์ ๊ฒฐ๊ณผ ์์") |
| return |
| |
| assign_row.configure(text="๋๋ค ํ๋กฌํํธ์ ์ฝ์
"+str(len(fdf))) |
| prompt_rollback.configure(text="์ด์ ํ๋กฌํํธํ ๋ณต์ : "+str(len(self.freezed_cached_rows))) |
| self.search_button.configure(text="๊ฒ์", state="normal") |
| max_page[0] = 1 + len(fdf) // 5 |
| current_page[0] = 0 |
| page.configure(text=f" 1 / {max_page[0]} ") |
| prev_button.configure(state="disabled") |
|
|
|
|
| if max_page[0] == 1: |
| next_button.configure(state="disabled") |
| else: |
| next_button.configure(state="normal") |
| update_ui_with_dataframe(current_page, favorite_window, headers_width, my_font) |
|
|
| def import_prompt(): |
| global fdf |
| if fdf is None: |
| null_prompt_attention() |
| else: |
| self.cached_rows = fdf.copy() |
| fdf.reset_index(drop=True, inplace=True) |
| update_labels(fdf) |
| prompt_rollback.configure(state="normal") |
|
|
| def rollback(): |
| global fdf |
| self.cached_rows = self.freezed_cached_rows.copy() |
| update_labels(self.cached_rows) |
| assign_row.configure(text="๋๋ค ํ๋กฌํํธ์ ์ฝ์
"+str(len(fdf))) |
| prompt_rollback.configure(state="disabled") |
|
|
| def favorite_search_close(): |
| self.freezed_cached_rows = None |
| favorite_window.destroy() |
|
|
| search_frame = customtkinter.CTkFrame(favorite_window, width=990) |
| search_frame.grid(row=0, column=0, padx=5,pady=5, sticky="nsew") |
| search_label = customtkinter.CTkLabel(search_frame, text="๊ฒ์ ํค์๋ ์
๋ ฅ : keyword, *keyword, {keyword1|keyword2}", font=my_font) |
| search_label.grid(row=0, column=0, columnspan=2, padx=8, pady=5, sticky="nsew") |
| search_label_entry = customtkinter.CTkEntry(search_frame, width=485) |
| search_label_entry.grid(row=1, column=0, columnspan=2, padx=8, pady=5, sticky="nsew") |
| exclude_label = customtkinter.CTkLabel(search_frame, text="์ ์ธ ํค์๋ ์
๋ ฅ : keyword, ~keyword", font=my_font) |
| exclude_label.grid(row=0, column=2, columnspan=2, padx=8, pady=5, sticky="nsew") |
| exclude_label_entry = customtkinter.CTkEntry(search_frame, width=485) |
| exclude_label_entry.grid(row=1, column=2, columnspan=2, padx=8, pady=5, sticky="nsew") |
|
|
| depth_search_execute = customtkinter.CTkButton(search_frame, text="์ ํธ ํ๋กฌํํธ๋ด ๊ฒ์", font=my_font, command=perform_favorite_search) |
| depth_search_execute.grid(row=2, column=0, padx=5,pady=5, sticky="n") |
|
|
| assign_row = customtkinter.CTkButton(search_frame, text="๋๋ค ํ๋กฌํํธ์ ์ฝ์
"+str(len(fdf)), font=my_font, fg_color="#CDCDCD", text_color="black", hover_color="#848484", command=import_prompt) |
| assign_row.grid(row=2, column=1, padx=5,pady=5, sticky="n") |
|
|
| prompt_rollback = customtkinter.CTkButton(search_frame, text="์ด์ ํ๋กฌํํธํ ๋ณต์ : "+str(len(self.freezed_cached_rows)), font=my_font, fg_color="#7030A0", hover_color="#481F67", command=rollback, state="disabled") |
| prompt_rollback.grid(row=2, column=2, padx=5,pady=5, sticky="n") |
|
|
| page_frame = customtkinter.CTkFrame(search_frame) |
| page_frame.grid(row=2, column=3, padx=5,pady=5 ,sticky="nsew") |
|
|
| def prev_pressed(): |
| current_page[0] -= 1 |
| if current_page[0] <= 0: |
| prev_button.configure(state="disabled") |
| if current_page[0] < max_page[0]: |
| next_button.configure(state="normal") |
| page.configure( text=f"{1+current_page[0]} / {max_page[0]} ") |
| update_ui_with_dataframe(current_page, favorite_window, headers_width, my_font) |
|
|
| def next_pressed(): |
| current_page[0] += 1 |
| if current_page[0] >= max_page[0]-1: |
| next_button.configure(state="disabled") |
| prev_button.configure(state="normal") |
| page.configure( text=f"{1+current_page[0]} / {max_page[0]} ") |
| update_ui_with_dataframe(current_page, favorite_window, headers_width, my_font) |
|
|
|
|
| page = customtkinter.CTkLabel(page_frame, text=f" 1 / {max_page[0]} ", font=my_font, width=70) |
| page.grid(row=0, column=1, padx=5, sticky="nsew") |
| prev_button = customtkinter.CTkButton(page_frame, text=" โ Prev ", font=my_font, width=70, state="disabled", command=prev_pressed) |
| prev_button.grid(row=0, column=0, padx=5 ,sticky="nsew") |
| next_button = customtkinter.CTkButton(page_frame, text=" Next โถ ", font=my_font, width=70, command=next_pressed) |
| next_button.grid(row=0, column=2, padx=5, sticky="nsew") |
| if max_page[0] == 1: |
| next_button.configure(state="disabled") |
|
|
|
|
|
|
| favorite_window.protocol("WM_DELETE_WINDOW", favorite_search_close) |
|
|
| headers = ['โ', 'character', 'copyright', 'artist', 'general prompts', 'image', 'delete'] |
| headers_width = [60, 120, 120, 120, 360, 120, 60] |
| header_frame = customtkinter.CTkFrame(favorite_window) |
| header_frame.grid(row=6, column=0, padx=5,pady=5, sticky="nsew") |
| empty_image = customtkinter.CTkImage(Image.new('RGB', size=(192,192), color="#2B2B2B"), size=(120, 120)) |
|
|
| for i, header in enumerate(headers): |
| label = customtkinter.CTkLabel(header_frame, text=header, width=headers_width[i], font=my_font) |
| label.grid(row=0, column=i, sticky="nsew", padx=5) |
|
|
| ui_elements = [] |
|
|
| for i in range(5): |
| row_elements = [] |
| if i < len(fdf): |
| row = fdf.iloc[i] |
| row_frame = customtkinter.CTkFrame(favorite_window) |
| row_frame.grid(row=i+7, column=0, sticky="nsew", padx=5, pady=5) |
|
|
| for j, header in enumerate(['โ', 'character', 'copyright', 'artist', 'general', 'image', 'delete']): |
| if header == 'โ': |
| button = customtkinter.CTkButton( |
| row_frame, width=headers_width[j], font=my_font, |
| text="โ", height=28, state="disabled" |
| ) |
| button.grid(row=0, padx=5, column=j, sticky="nsew") |
| element = button |
| elif header == 'delete': |
| delete_button = customtkinter.CTkButton( |
| row_frame, width=headers_width[j], font=my_font, text="โ", |
| height=28, fg_color="grey", hover_color="grey10", state="disabled" |
| ) |
| delete_button.grid(row=0, padx=5, column=j, sticky="nsew") |
| element = delete_button |
| elif header == 'image': |
| label = customtkinter.CTkLabel(row_frame, width=headers_width[j], font=my_font, text=" ", height=120) |
| label.grid(row=0, padx=5, column=j, sticky="nsew") |
| element = label |
| else: |
| text = customtkinter.CTkTextbox(row_frame, width=headers_width[j], font=my_font, height=120) |
| text.grid(row=0, padx=5, column=j, sticky="nsew") |
| element = text |
| row_elements.append(element) |
| |
| ui_elements.append(row_elements) |
|
|
| |
| def delete_image(image_path): |
| try: |
| os.remove(image_path) |
| except OSError as e: |
| print(f"Error deleting file {image_path}: {e.strerror}") |
|
|
| |
| def delete_row(fdf, idx, image_path, basedir): |
| |
| fdf.loc[fdf['id'] == idx, 'deleted'] = 'deleted' |
| |
| |
| fdf.drop(fdf[fdf['id'] == idx].index, inplace=True) |
| |
| |
| fdf.to_parquet("favorite_prompt.parquet", engine='pyarrow') |
|
|
| |
| delete_image(image_path) |
|
|
| update_ui_with_dataframe(current_page, favorite_window, headers_width, my_font) |
|
|
| |
| def make_delete_command(fdf, idx, image_path, basedir): |
| def command(): |
| delete_row(fdf, idx, image_path, basedir) |
| return command |
|
|
| def make_select_command(dataframe, row_id): |
| """ |
| ํน์ DataFrame ํ์ 'id' ๊ฐ์ ์ถ๋ ฅํ๋ ์ปค๋งจ๋๋ฅผ ์์ฑํ๋ ํจ์. |
| ์ด ํจ์๋ ๋๋ค ํจ์๋ฅผ ์ฌ์ฉํ์ฌ ๋ฒํผ์ ํ ๋น๋ ์ปค๋งจ๋๋ฅผ ๋ฐํํฉ๋๋ค. |
| """ |
| def command(): |
| popped_row = dataframe.loc[dataframe['id'] == row_id] |
| if not popped_row.empty: |
| popped_row_series = popped_row.iloc[0] |
| |
| random_function(popped_row_series) |
| return command |
|
|
| def update_ui_with_dataframe(current_page, favorite_window, headers_width, my_font): |
| global fdf |
| base_image_path = "favorite_prompts" |
| rows_per_page = 5 |
| start_index = current_page[0] * rows_per_page |
| end_index = min(start_index + rows_per_page, len(fdf)) |
|
|
| for i in range(5): |
| |
| if i < end_index - start_index: |
| row = fdf.iloc[start_index + i] |
| for j, header in enumerate(['โ', 'character', 'copyright', 'artist', 'general', 'image', 'delete']): |
| element = ui_elements[i][j] |
| if header == 'โ': |
| |
| element.configure(state="normal", command=make_select_command(fdf, row['id'])) |
| elif header == 'delete': |
| |
| image_path = f"{base_image_path}/{row['id']}.jpg" |
| element.configure(state="normal", command=make_delete_command(fdf, row['id'], image_path, basedir)) |
| elif header == 'image': |
| try: |
| image_path = f"{base_image_path}/{row['id']}.jpg" |
| if os.path.exists(image_path): |
| _image = customtkinter.CTkImage(Image.open(image_path), size=(120, 120)) |
| element.configure(image=_image) |
| else: |
| raise FileNotFoundError |
| except FileNotFoundError: |
| element.configure(text="Image Not Found") |
| else: |
| |
| text_value = row[header] if pd.notna(row[header]) else "" |
| element.delete("0.0", "end") |
| element.insert("0.0", text_value) |
| else: |
| |
| for j, element in enumerate(ui_elements[i]): |
| if isinstance(element, customtkinter.CTkLabel): |
| element.configure(text=" ", image=empty_image) |
| element.image = empty_image |
| elif isinstance(element, customtkinter.CTkTextbox): |
| element.delete("0.0", "end") |
| elif isinstance(element, customtkinter.CTkButton): |
| element.configure(state="disabled") |
|
|
|
|
|
|
| |
| current_page = [0] |
|
|
| |
| update_ui_with_dataframe(current_page, favorite_window, headers_width, my_font) |
| favorite_window.after(1500, lambda: favorite_window.attributes('-topmost', False)) |
|
|
|
|
|
|
| self.prompt_preset_button = customtkinter.CTkButton(self.searched_prompt_frame, width=80, text="ํ๋ฆฌ์
", fg_color="#848484", font=my_font, command= lambda: open_Preset(self)) |
| self.prompt_preset_button.grid(row=0, column=2, sticky="w") |
|
|
| self.fav_button = customtkinter.CTkButton(self.searched_prompt_frame, width=40, text=" โ
", font=my_font, fg_color="#7030A0", hover_color="#481F67", command= open_favorite) |
| self.fav_button.grid(row=0, column=2, padx=15, sticky="e") |
|
|
| try: |
| temp_df = pd.read_parquet("favorite_prompt.parquet", engine="pyarrow") |
| self.fav_id_list = [] |
| if 'id' in temp_df.columns: |
| self.fav_id_list.extend(temp_df['id'].tolist()) |
| del(temp_df) |
| except: |
| self.fav_id_list = [] |
| |
| self.fav_button.configure(state="disabled") |
|
|
| def on_enter_pressed(event): |
| if self.control_pressed: |
| NAIA_generate(self) |
| return "break" |
| else: |
| pass |
|
|
| def on_tab_pressed(event): |
| if self.toggle_prompt_fix_button.get() == 0: |
| time.sleep(0.2) |
| random_function() |
|
|
| |
| self.text_input_frame = customtkinter.CTkFrame(self.searched_prompt_frame) |
| self.text_input_frame.grid(row=1, column=0, columnspan=3, sticky="nsew") |
| self.text_input_label = customtkinter.CTkLabel(self.text_input_frame, text=" ----------------------------- ํ๋กฌํํธ ----------------------------- ", font=large_font) |
| self.text_input_label.grid(row = 0, sticky="n" ) |
| self.text_input = customtkinter.CTkTextbox(self.text_input_frame, width=490, font=v_large_font, undo=True) |
| self.text_input.grid(row=1, column=0, pady=5, sticky="nsew") |
| self.text_input.bind("<Return>", on_enter_pressed) |
| self.text_input.bind("<Tab>", on_tab_pressed) |
| self.current_popped_row = None |
|
|
| def random_function(edf = None): |
| if(self.random_function_pressed) == False: |
| self.random_function_pressed = True |
| self.random_function_button.configure(state="disabled") |
| if type(edf) == type(None): |
| if self.toggle_prompt_fix.get()==0 and type(self.cached_rows) == type(None): |
| self.text_input.delete("0.0", "end") |
| self.text_input.insert("0.0", "๋จผ์ ํค์๋ ๊ฒ์์ ํด ์ฃผ์ธ์.") |
| self.random_function_button.configure(state="normal") |
| self.random_function_pressed = False |
| return |
| if self.toggle_prompt_fix.get()==1 and type(self.cached_rows) == type(None): |
| self.random_function_pressed = False |
| return |
| elif self.toggle_prompt_fix.get()==0 and self.cached_rows.empty: |
| if (self.automation_button.get() == 0): |
| self.text_input.delete("0.0", "end") |
| self.text_input.insert("0.0", "๋จผ์ ํค์๋ ๊ฒ์์ ํด ์ฃผ์ธ์.") |
| self.random_function_button.configure(state="normal") |
| self.random_function_pressed = False |
| return |
| elif type(self.freezed_cached_rows) != type(None): |
| self.text_input.delete("0.0", "end") |
| self.text_input.insert("0.0", "๊ธฐ์กด ํ๋กฌํํธ๋ฅผ ๋ณต์ํฉ๋๋ค.") |
| self.cached_rows = self.freezed_cached_rows.copy() |
| time.sleep(1) |
| self.random_function_button.configure(state="normal") |
| self.random_function_pressed = False |
| app.after(0, random_function) |
| return |
| else: |
| self.text_input.delete("0.0", "end") |
| self.text_input.insert("0.0", "์ฌ๊ฒ์์ ์ํํฉ๋๋ค.") |
| prompt_search() |
| while(self.cached_rows.empty): |
| time.sleep(1) |
| self.random_function_button.configure(state="normal") |
| self.random_function_pressed = False |
| app.after(0, random_function) |
| return |
| if self.toggle_prompt_fix.get(): |
| self.random_function_pressed = False |
| return |
| else: |
| if self.random_artist_button.get() == 1: |
| try: |
| if self.random_artist_prefix.get() == "artist:": |
| random_artist_name = "artist:"+ random.choice(self.random_artist) |
| elif self.random_artist_prefix.get() == "(artist)": |
| random_artist_name = random.choice(self.random_artist)+" (artist)" |
| else: |
| random_artist_name = random.choice(self.random_artist) |
| except: |
| random_artist_name = "" |
| magic_word = { |
| "random_artist":True, |
| "random_artist_name": random_artist_name |
| } |
| else: |
| magic_word = { |
| "random_artist":False |
| } |
| if self.cond_prompt_button_var.get() == 1: |
| magic_word["cond"] = self.conditional_prompt_input.get("0.0", "end-1c") if len(self.conditional_prompt_input.get("0.0", "end-1c")) > 4 else None |
| if type(edf) == type(None): |
| random_index = np.random.choice(self.cached_rows.index) |
| popped_row = self.cached_rows.loc[random_index] |
| self.current_popped_row = popped_row.copy() |
| self.cached_rows.drop(random_index, inplace=True) |
| self.cached_prompt_label.configure(text = "๋จ์ ํ๋กฌํํธ ํ : "+str(len(self.cached_rows))) |
| else: |
| popped_row = edf.copy() |
| self.current_popped_row = popped_row.copy() |
| if(self.toggle_wildcard_preopen.get()==1): |
| if(self.wildcard_preopen_repeat_current >= self.wildcard_preopen_repeat): |
| self.wildcard_preopen_repeat_current = 0 |
| |
| temp_fixed = self.fixed_prompt_input.get("0.0", "end") |
| temp_fixed = [item.strip() for item in temp_fixed.split(',')] |
| global_temp = [] |
| if '<' in self.fixed_prompt_input.get("0.0", "end"): |
| wildcard_present = True |
| itc = 0 |
| while (wildcard_present and itc < 10): |
| wildcard_present = False |
| |
| for i, keyword in enumerate(temp_fixed): |
| if keyword.startswith('<') and keyword.endswith('>'): |
| wildcard_present = True |
| vbar_check = keyword[1:-1] |
| if '|' in vbar_check: |
| choices = vbar_check.split('|') |
| choice_dic = {} |
| for choice in choices: |
| match = re.match(r'(\d*\.?\d+):(.+)', choice) |
| if match: |
| value, keyword = float(match.group(1)), match.group(2).strip() |
| else: |
| value, keyword = 1, choice.strip() |
| choice_dic[keyword] = value |
| keywords = list(choice_dic.keys()) |
| weights = list(choice_dic.values()) |
| selected_instant_wildcard = random.choices(keywords, weights=weights, k=1)[0] |
| temp_fixed[i] = selected_instant_wildcard |
| |
| for i, keyword in enumerate(temp_fixed): |
| if "<" in keyword and ">" in keyword: |
| wildcard_present = True |
| input_str = keyword.strip('<>').strip() |
| if("__" in input_str): |
| adjectives = re.findall(r'__(.*?)__', input_str) |
| last_keyword = re.split(r'__.*?__', input_str)[-1] |
| adjective_string = "" |
| for adjective in adjectives: |
| adjective_string += (self.get_wildcard(adjective) + " ") |
| temp_fixed[i] = adjective_string + self.get_wildcard(last_keyword) |
| else: |
| out_wildcard = self.get_wildcard(input_str) |
| if ',' in out_wildcard: |
| split_wc = [item.strip() for item in out_wildcard.split(',')] |
| temp = [] |
| for _key in split_wc[1:]: |
| if "{" not in _key and "}" not in _key and "[" not in _key and "]" not in _key: |
| temp.append(_key) |
| if temp: |
| for _key in temp: |
| split_wc.remove(_key) |
| global_temp.append(_key) |
| if len(split_wc) == 2 and '{' not in split_wc[1] and '[' not in split_wc[1] and (split_wc[1].endswith('}') or split_wc[1].endswith(']')): |
| wc_post = split_wc[1].replace(']','').replace('}','') |
| global_temp.append(wc_post) |
| split_wc[0] += split_wc[1].replace(wc_post, '') |
| split_wc.pop() |
| out_wildcard = ', '.join(split_wc) |
| temp_fixed[i] = out_wildcard |
| itc += 1 |
| if not wildcard_present: |
| break |
| temp_fixed_result = ', '.join(temp_fixed) |
| if global_temp: temp_fixed_result += ', '+', '.join(global_temp) |
| self.previous_fix_prompt = temp_fixed_result |
| else: |
| temp_fixed_result = self.previous_fix_prompt |
| self.wildcard_preopen_repeat_current += 1 |
| prompt, self.current_prompt_rating = NAIA_random_function_core.RFP(popped_row, temp_fixed_result, self.fixed_prompt_after_input.get("0.0", "end"), self.auto_hide_keyword_input.get("0.0", "end") |
| ,self.rm_artist_name_button.get(), self.rm_copyright_name_button.get(),self.rm_characteristic_button.get(), self.rm_not_nsfw_button.get(), self.data, magic_word) |
| else: |
| prompt, self.current_prompt_rating = NAIA_random_function_core.RFP(popped_row, self.fixed_prompt_input.get("0.0", "end"), self.fixed_prompt_after_input.get("0.0", "end"), self.auto_hide_keyword_input.get("0.0", "end") |
| ,self.rm_artist_name_button.get(), self.rm_copyright_name_button.get(),self.rm_characteristic_button.get(), self.rm_not_nsfw_button.get(), self.data, magic_word) |
| self.text_input.delete("0.0", "end") |
| self.text_input.insert("0.0", prompt) |
| app.after(100, lambda: self.random_function_button.configure(state="normal")) |
| self.random_function_pressed = False |
|
|
| self.random_function_pressed = False |
| self.running_flag = False |
| self.error_count = 0 |
| self.turbo_fail_count = 0 |
| self.anlas_request = False |
| self.image_generation_repeat_prompt = "" |
|
|
| def NAIA_generate(self): |
| if self.seed_fix_button.get() == 0: |
| self.entry_seed_value.set(random.randint(0,9999999999)) |
| if self.random_resolution_button.get() == 1: |
| resolutions = ["1024 x 1024", "960 x 1088", "896 x 1152", "832 x 1216", "1088 x 960", "1152 x 896", "1216 x 832"] |
| random_resolution = random.choice(resolutions) |
| self.resolution_button.set(random_resolution) |
| if self.cached_rows is None and self.toggle_prompt_fix_button.get() == 0 and self.automation_button.get() == 1: |
| self.state_label.configure(text ="state : ์๋ํ ์ ํ๋กฌํํธ ๊ฒ์์ ์ํํ๋ ์ค ์
๋๋ค", text_color = "#FFFF97") |
| prompt_search() |
| if self.turbo_button.get() == 1: |
| origin = self.text_input.get("0.0", "end-1c") |
| pretest = [keyword.strip() for keyword in origin.split(',')] |
| if ('*(split nsfw)' in pretest or '*(split nsfw)-rand' in pretest): |
| pass |
| elif (('sex' not in pretest and 'group sex' not in pretest) or ('1girl' not in pretest and 'girls' not in origin[:20]) or ('1boy' not in pretest and 'boys' not in origin[:20])): |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<UserAttention> Explicit Turbo ๊ธฐ๋ฅ์ ํ๋กฌํํธ ๋ด์ 1girl/girls, 1boy/boys, sex/group sex๊ฐ ์กด์ฌํ ๋ ์ฌ์ฉ ๊ฐ๋ฅํฉ๋๋ค.") |
| self.turbo_fail_count += 1 |
| if self.turbo_fail_count >= 20: |
| self.turbo_button.deselect() |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| if self.automation_button.get() == 1 and self.toggle_prompt_fix_button.get() == 0: |
| random_function() |
| time.sleep(0.5) |
| if self.searching_flag == True: |
| self.search_thread.join() |
| return self.event_generate(GENERATE_EVENT, when="tail") |
| elif self.automation_button.get() == 1 and self.toggle_prompt_fix_button.get() == 1: |
| self.automation_button.deselect() |
| return |
| else: |
| return |
| self.turbo_fail_count = 0 |
|
|
| if not self.running_flag: |
| self.image_generation_button.configure(state="disabled") |
| self.running_flag = True |
| NAI_width, NAI_height = self.resolution_button.get().split(' x ') |
| NAI_width = str((int(NAI_width) // 64) * 64) |
| NAI_height = str((int(NAI_height) // 64) * 64) |
| if int(NAI_height) * int(NAI_width) > 1048576: |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<UserAttention> Anlas๊ฐ ์๋ชจ๋๋ ํด์๋ ์์ฒญ์
๋๋ค.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| self.anlas_request = True |
| scale_pre = self.cfg_scale_entry.get() |
| try: |
| scale_pre = float(scale_pre) |
| except: |
| scale_pre = 5.0 |
| self.cfg_scale_var.set("5.0") |
| rescale_pre = self.prompt_guidance_rescale_entry.get() |
| try: |
| rescale_pre = float(rescale_pre) |
| except: |
| rescale_pre = 0 |
| self.prompt_guidance_rescale_var.set("0") |
| uncond_pre = self.uncond_strength_entry.get() |
| try: |
| uncond_pre = float(uncond_pre) |
| uncond_pre = round(uncond_pre / 0.05) * 0.05 |
| if (uncond_pre > 1.5): |
| uncond_pre = 1.5 |
| self.uncond_strength_entry.delete(0, "end") |
| self.uncond_strength_entry.insert(0, "1.5") |
| except: |
| uncond_pre = 1.0 |
| self.uncond_strength_entry.delete(0, "end") |
| self.uncond_strength_entry.insert(0, "1.0") |
| |
| if self.image_generation_repeat_flag == False: |
| before_wildcard = self.text_input.get("0.0", "end-1c") |
| before_wildcard = [item.strip() for item in before_wildcard.split(',')] |
| if '<' in self.text_input.get("0.0", "end-1c"): |
| wildcard_present = True |
| itc = 0 |
| while (wildcard_present and itc < 10): |
| wildcard_present = False |
| |
| for i, keyword in enumerate(before_wildcard): |
| if keyword.startswith('<') and keyword.endswith('>'): |
| wildcard_present = True |
| vbar_check = keyword[1:-1] |
| if '|' in vbar_check: |
| choices = vbar_check.split('|') |
| choice_dic = {} |
| for choice in choices: |
| match = re.match(r'(\d*\.?\d+):(.+)', choice) |
| if match: |
| value, keyword = float(match.group(1)), match.group(2).strip() |
| else: |
| value, keyword = 1, choice.strip() |
| choice_dic[keyword] = value |
| keywords = list(choice_dic.keys()) |
| weights = list(choice_dic.values()) |
| selected_instant_wildcard = random.choices(keywords, weights=weights, k=1)[0] |
| before_wildcard[i] = selected_instant_wildcard |
| |
| for i, keyword in enumerate(before_wildcard): |
| if "<" in keyword and ">" in keyword: |
| wildcard_present = True |
| input_str = keyword.strip('<>').strip() |
| if("__" in input_str): |
| adjectives = re.findall(r'__(.*?)__', input_str) |
| last_keyword = re.split(r'__.*?__', input_str)[-1] |
| adjective_string = "" |
| for adjective in adjectives: |
| adjective_string += (self.get_wildcard(adjective) + " ") |
| before_wildcard[i] = adjective_string + self.get_wildcard(last_keyword) |
| else: |
| before_wildcard[i] = self.get_wildcard(input_str) |
| itc += 1 |
| if not wildcard_present: |
| break |
| else: |
| after_wildcard = self.text_input.get("0.0", "end-1c") |
| after_wildcard = ', '.join(before_wildcard) |
| if self.image_generation_repeat > 1: |
| if self.image_generation_repeat_flag == False: |
| self.image_generation_repeat_flag = True |
| self.image_generation_repeat_prompt = after_wildcard |
| self.image_generation_repeat_current = 1 |
| else: |
| after_wildcard = self.image_generation_repeat_prompt |
| self.image_generation_repeat_current += 1 |
| |
| gen_request = { |
| "width":NAI_width, |
| "height":NAI_height, |
| "quality_toggle":self.auto_quality_toggle.get(), |
| "seed":self.seed_entry.get(), |
| "sampler":self.sampler_button.get(), |
| "scale":scale_pre, |
| "uncond_scale":uncond_pre, |
| "sema":self.sema_button.get(), |
| "sema_dyn": self.dyn_button.get(), |
| "cfg_rescale": rescale_pre, |
| "prompt": after_wildcard, |
| "negative":self.negative_prompt_input.get("0.0", "end-1c"), |
| "user_screen_size": self.get_max_size(), |
| "start_time": self.start_time, |
| "access_token": self.access_token, |
| "save_folder": self.output_file_path, |
| "png_rule": self.name_var.get(), |
| "type": "normal", |
| "rating":self.current_prompt_rating |
| } |
| if self.cond_negative_button_var.get() == 1: |
| gen_request["cond_negative"] = self.conditional_negative_input.get("0.0", "end-1c") |
|
|
| if self.turbo_button.get() == 1 and "*(split nsfw)"in gen_request["prompt"]: |
| request_list = [] |
| pre_prompt = gen_request["prompt"] |
| pre_prompt = [keyword.strip() for keyword in pre_prompt.split(',')] |
| fix_words = self.fixed_prompt_input.get("0.0", "end-1c")+", "+self.fixed_prompt_after_input.get("0.0", "end-1c") |
| fix_words = [keyword.strip() for keyword in fix_words.split(',')] |
| if "*(split nsfw)-rand" in pre_prompt: |
| pre_prompt.remove("*(split nsfw)-rand") |
| random_seed = True |
| else: |
| pre_prompt.remove("*(split nsfw)") |
| random_seed = False |
| nsfw_word = [] |
| colors = ['black','blond','silver','gray','yellow','blue','purple','red','pink','brown','orange','green','aqua','gradient'] |
| qe_word = app.data.get_qe_word() + ['cross-section', 'x-ray'] |
| for keyword in pre_prompt: |
| if 'mouth' in keyword or 'eyes' in keyword: |
| if keyword not in fix_words: |
| pre_prompt.remove(keyword) |
| for keyword in pre_prompt: |
| if '{' not in keyword and '[' not in keyword and keyword in qe_word and any(color not in keyword for color in colors) : |
| nsfw_word.append(keyword) |
| if len(nsfw_word) != 0: |
| basket = [] |
| for keyword in nsfw_word: |
| if 'cross-section' in keyword or 'x-ray' in keyword: |
| if keyword not in basket: basket.append(keyword) |
| for keyword in nsfw_word: |
| if 'internal cumshot' in keyword or 'ejaculation' in keyword: |
| if keyword not in basket: basket.append(keyword) |
| for keyword in nsfw_word: |
| if 'cum' in keyword or 'ahegao' in keyword: |
| if keyword not in basket: basket.append(keyword) |
| for keyword in nsfw_word: |
| if 'after ' in keyword: |
| if keyword not in basket: basket.append(keyword) |
| for keyword in basket: |
| if keyword in nsfw_word: |
| nsfw_word.remove(keyword) |
| nsfw_word += basket |
| words = [] |
| if len(nsfw_word) > 13: |
| max_val = len(nsfw_word) // 4 |
| if len(nsfw_word) % 4 > 0: max_val += 1 |
| for i in range(max_val): |
| current_word = [] |
| if i < max_val - 1: |
| words.append(nsfw_word[i*4]) |
| words.append(nsfw_word[i*4+1]) |
| words.append(nsfw_word[i*4+2]) |
| words.append(nsfw_word[i*4+3]) |
| current_word.append(nsfw_word[i*4]) |
| current_word.append(nsfw_word[i*4+1]) |
| current_word.append(nsfw_word[i*4+2]) |
| current_word.append(nsfw_word[i*4+3]) |
| else: |
| if len(nsfw_word) % 4 == 0: |
| words.append(nsfw_word[i*4]) |
| words.append(nsfw_word[i*4+1]) |
| words.append(nsfw_word[i*4+2]) |
| words.append(nsfw_word[i*4+3]) |
| current_word.append(nsfw_word[i*4]) |
| current_word.append(nsfw_word[i*4+1]) |
| current_word.append(nsfw_word[i*4+2]) |
| current_word.append(nsfw_word[i*4+3]) |
| elif len(nsfw_word) % 4 == 3: |
| words.append(nsfw_word[i*4]) |
| words.append(nsfw_word[i*4+1]) |
| words.append(nsfw_word[i*4+2]) |
| current_word.append(nsfw_word[i*4]) |
| current_word.append(nsfw_word[i*4+1]) |
| current_word.append(nsfw_word[i*4+2]) |
| elif len(nsfw_word) % 4 == 2: |
| words.append(nsfw_word[i*4]) |
| words.append(nsfw_word[i*4+1]) |
| current_word.append(nsfw_word[i*4]) |
| current_word.append(nsfw_word[i*4+1]) |
| else: |
| words.append(nsfw_word[i*4]) |
| current_word.append(nsfw_word[i*4]) |
| gen_request_sub = copy.deepcopy(gen_request) |
| temp_prompt = []+pre_prompt |
| for keyword in nsfw_word: |
| if keyword not in words: |
| if random_seed == False: |
| temp_prompt[temp_prompt.index(keyword)] = " "*len(keyword) |
| else: |
| temp_prompt.remove(keyword) |
| elif keyword in current_word: |
| temp_prompt[temp_prompt.index(keyword)] = "{"+keyword+"}" |
| gen_request_sub["prompt"] = ', '.join(temp_prompt) |
| if random_seed: |
| gen_request_sub["seed" ] = random.randint(0,9999999999) |
| request_list.append(copy.deepcopy(gen_request_sub)) |
| elif len(nsfw_word) > 8: |
| max_val = len(nsfw_word) // 3 |
| if len(nsfw_word) % 3 > 0: max_val += 1 |
| for i in range(max_val): |
| current_word = [] |
| if i < max_val - 1: |
| words.append(nsfw_word[i*3]) |
| words.append(nsfw_word[i*3+1]) |
| words.append(nsfw_word[i*3+2]) |
| current_word.append(nsfw_word[i*3]) |
| current_word.append(nsfw_word[i*3+1]) |
| current_word.append(nsfw_word[i*3+2]) |
| else: |
| if len(nsfw_word) % 3 == 0: |
| words.append(nsfw_word[i*3]) |
| words.append(nsfw_word[i*3+1]) |
| words.append(nsfw_word[i*3+2]) |
| current_word.append(nsfw_word[i*3]) |
| current_word.append(nsfw_word[i*3+1]) |
| current_word.append(nsfw_word[i*3+2]) |
| elif len(nsfw_word) % 3 == 2: |
| words.append(nsfw_word[i*3]) |
| words.append(nsfw_word[i*3+1]) |
| current_word.append(nsfw_word[i*3]) |
| current_word.append(nsfw_word[i*3+1]) |
| else: |
| words.append(nsfw_word[i*3]) |
| current_word.append(nsfw_word[i*3]) |
| gen_request_sub = copy.deepcopy(gen_request) |
| temp_prompt = []+pre_prompt |
| for keyword in nsfw_word: |
| if keyword not in words: |
| if random_seed == False: |
| temp_prompt[temp_prompt.index(keyword)] = " "*len(keyword) |
| else: |
| temp_prompt.remove(keyword) |
| elif keyword in current_word: |
| temp_prompt[temp_prompt.index(keyword)] = "{"+keyword+"}" |
| gen_request_sub["prompt"] = ', '.join(temp_prompt) |
| if random_seed: |
| gen_request_sub["seed" ] = random.randint(0,9999999999) |
| request_list.append(copy.deepcopy(gen_request_sub)) |
| elif len(nsfw_word) > 3: |
| max_val = len(nsfw_word) // 2 + len(nsfw_word) % 2 |
| for i in range(max_val): |
| current_word = [] |
| if i < max_val - 1: |
| words.append(nsfw_word[i*2]) |
| words.append(nsfw_word[i*2+1]) |
| current_word.append(nsfw_word[i*2]) |
| current_word.append(nsfw_word[i*2+1]) |
| else: |
| if len(nsfw_word) % 2 == 0: |
| words.append(nsfw_word[i*2]) |
| words.append(nsfw_word[i*2+1]) |
| current_word.append(nsfw_word[i*2]) |
| current_word.append(nsfw_word[i*2+1]) |
| else: |
| words.append(nsfw_word[i*2]) |
| current_word.append(nsfw_word[i*2]) |
| gen_request_sub = copy.deepcopy(gen_request) |
| temp_prompt = []+pre_prompt |
| for keyword in nsfw_word: |
| if keyword not in words: |
| if random_seed == False: |
| temp_prompt[temp_prompt.index(keyword)] = " "*len(keyword) |
| else: |
| temp_prompt.remove(keyword) |
| elif keyword in current_word: |
| temp_prompt[temp_prompt.index(keyword)] = "{"+keyword+"}" |
| gen_request_sub["prompt"] = ', '.join(temp_prompt) |
| if random_seed: |
| gen_request_sub["seed" ] = random.randint(0,9999999999) |
| request_list.append(copy.deepcopy(gen_request_sub)) |
| else: |
| for word in nsfw_word: |
| words.append(word) |
| gen_request_sub = copy.deepcopy(gen_request) |
| temp_prompt = []+pre_prompt |
| for keyword in nsfw_word: |
| if keyword not in words: |
| if random_seed == False: |
| temp_prompt[temp_prompt.index(keyword)] = " "*len(keyword) |
| else: |
| temp_prompt.remove(keyword) |
| elif keyword == word: |
| temp_prompt[temp_prompt.index(keyword)] = "{"+word+"}" |
| gen_request_sub["prompt"] = ', '.join(temp_prompt) |
| if random_seed: |
| gen_request_sub["seed" ] = random.randint(0,9999999999) |
| request_list.append(copy.deepcopy(gen_request_sub)) |
| else: |
| request_list.append(gen_request) |
| elif self.turbo_button.get() == 1: |
| request_list = [] |
| try: |
| treq_0, treq_1, treq_2, treq_3 = NAIA_generation.make_turbo_prompt(gen_request) |
| request_list.append(gen_request) |
| request_list.append(treq_0) |
| request_list.append(treq_1) |
| request_list.append(treq_2) |
| request_list.append(treq_3) |
| except: |
| request_list.append(gen_request) |
| else: |
| pre_prompt = gen_request["prompt"] |
| pre_prompt = [keyword.strip() for keyword in pre_prompt.split(',')] |
| if "*(split nsfw)-rand" in pre_prompt: |
| pre_prompt.remove("*(split nsfw)-rand") |
| gen_request["prompt"] = ', '.join(pre_prompt) |
|
|
| def run_generation(): |
| if gen_request["png_rule"] == "count": |
| self.generation_count += 1 |
| gen_request["count"] =self.generation_count |
| if app.auto_count_left_flag == True: |
| app.auto_count_left -= 1 |
| if self.image_generation_repeat_flag == True: |
| gen_request["repeat"] = self.image_generation_repeat_current -1 |
| gen_request["repeat_max"] = self.image_generation_repeat |
| app.image_generation_button.configure(border_width = 2) |
| pass_bit = False |
| retry_count = 0 |
| while(not pass_bit): |
| self.state_label.configure(text ="state : NAI ์ด๋ฏธ์ง ์์ฑ ๋๊ธฐ์ค ... ", text_color = "#FFFF97") |
| result_image, result_prompt, result_seed, info, filename = NAIA_generation.generate(gen_request) |
| self.state_label.configure(text ="state : idle", text_color = "#DCE4EE") |
| app.image_generation_button.configure(border_width = 0) |
| if self.anlas_request: |
| self.anlas_request = False |
| self.get_anlas() |
| if info: |
| self.error_count = 0 |
| temp = info.get('Comment', '') |
| temp = temp[temp.find("prompt")+10:temp.find("skip_cfg_below_sigma")-3].replace('"','') |
| pass_bit = True |
| else: |
| self.error_count += 1 |
| retry_count += 1 |
| if self.error_count >= 10: |
| self.automation_button.deselect() |
| temp = result_prompt |
| if self.automation_button.get() == 1: |
| instant_wait = random.uniform(3.5, 5.5) + self.delay_offset |
| if instant_wait < 0 : instant_wait = 0.5 |
| time.sleep(instant_wait) |
| else: |
| time.sleep(1) |
| if self.automation_button.get() == 0 or retry_count >= 5: |
| break |
| self.running_flag = False |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.configure(text_color="#DCE4EE") |
| self.image_label_report.insert("0.0", temp) |
| self.image_label_report.configure(state="disabled") |
| if result_image: |
| if app.state() != 'zoomed': |
| instant_result_image = customtkinter.CTkImage(result_image, size=(620,620)) |
| else: |
| current_image = Image.open(filename) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| self.image_label.configure(image=instant_result_image) |
| set_image_to_queue(result_image, result_prompt, result_seed, filename) |
| |
| if self.automation_button.get() == 1: |
| instant_wait = random.uniform(3.5, 5.5) + self.delay_offset |
| if instant_wait < 0 : instant_wait = 0.5 |
| if self.image_generation_repeat_flag == False or (self.image_generation_repeat <= self.image_generation_repeat_current): |
| self.image_generation_repeat_current = 0 |
| self.image_generation_repeat_flag = False |
| if self.toggle_prompt_fix_button.get() == 0: |
| random_function() |
| |
| while(instant_wait > 0): |
| self.image_generation_button.configure(text=f"NAI ์ด๋ฏธ์ง ์์ฑ ({round(instant_wait)})") |
| if instant_wait >= 1: |
| time.sleep(1) |
| else: |
| time.sleep(instant_wait) |
| instant_wait -= 1 |
| self.image_generation_button.configure(text="NAI ์ด๋ฏธ์ง ์์ฑ") |
| if self.automation_button.get() == 1: |
| self.event_generate(GENERATE_EVENT, when="tail") |
| else: |
| self.image_generation_button.configure(state="normal") |
| else: |
| self.image_generation_repeat_current = 0 |
| self.image_generation_repeat_flag = False |
| self.image_generation_button.configure(state="normal") |
| def run_generation_turbo(gen_request): |
| if gen_request["png_rule"] == "count": |
| self.generation_count += 1 |
| gen_request["count"] =self.generation_count |
| if app.auto_count_left_flag == True: |
| app.auto_count_left -= 1 |
| app.image_generation_button.configure(border_width = 2) |
| pass_bit = False |
| retry_count = 0 |
| while(not pass_bit): |
| self.state_label.configure(text ="state : NAI ์ด๋ฏธ์ง ์์ฑ ๋๊ธฐ์ค (EXP.turbo) ... ", text_color = "#FFFF97") |
| result_image, result_prompt, result_seed, info, filename = NAIA_generation.generate(gen_request) |
| self.state_label.configure(text ="state : idle", text_color = "#DCE4EE") |
| app.image_generation_button.configure(border_width = 0) |
| if info: |
| self.error_count = 0 |
| temp = info.get('Comment', '') |
| temp = temp[temp.find("prompt")+10:temp.find("skip_cfg_below_sigma")-3].replace('"','') |
| pass_bit = True |
| else: |
| self.error_count += 1 |
| retry_count += 1 |
| if self.error_count >= 10: |
| self.automation_button.deselect() |
| self.turbo_button.deselect() |
| temp = result_prompt |
| if self.automation_button.get() == 1: |
| instant_wait = random.uniform(4.5, 8.5) + self.delay_offset |
| if instant_wait < 0 : instant_wait = 0.5 |
| time.sleep(instant_wait) |
| else: |
| time.sleep(1) |
| if self.turbo_button.get() == 0 or retry_count >= 5: |
| break |
| self.running_flag = False |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.configure(text_color="#DCE4EE") |
| self.image_label_report.insert("0.0", temp) |
| self.image_label_report.configure(state="disabled") |
| if result_image: |
| if app.state() != 'zoomed': |
| instant_result_image = customtkinter.CTkImage(result_image, size=(620,620)) |
| else: |
| current_image = Image.open(filename) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| self.image_label.configure(image=instant_result_image) |
| set_image_to_queue(result_image, result_prompt, str(result_seed), filename) |
| instant_wait = random.uniform(4.5, 8.5) + self.delay_offset |
| if instant_wait < 0 : instant_wait = 0.5 |
| while(instant_wait > 0): |
| self.image_generation_button.configure(text=f"NAI ์ด๋ฏธ์ง ์์ฑ ({round(instant_wait)})") |
| if instant_wait >= 1: |
| time.sleep(1) |
| else: |
| time.sleep(instant_wait) |
| instant_wait -= 1 |
| self.image_generation_button.configure(text="NAI ์ด๋ฏธ์ง ์์ฑ") |
|
|
| def turbo_process_request(): |
| turbo_count = 0 |
| while(request_list): |
| gen_request = request_list.pop(0) |
| generation_thread = threading.Thread(target=run_generation_turbo, args=(gen_request,), daemon=True) |
| generation_thread.start() |
| generation_thread.join() |
| turbo_count += 1 |
| if app.turbo_button.get() == 0: |
| app.state_label.configure(text =f"state : ์ฌ์ฉ์์ ์ํ ํฐ๋ณด์์ฒญ ์ค๋จ ({turbo_count})", text_color = "#FFFF97") |
| break |
| if self.automation_button.get() == 1: |
| if self.toggle_prompt_fix_button.get() == 0: |
| random_function() |
| self.event_generate(GENERATE_EVENT, when="tail") |
| else: |
| self.image_generation_button.configure(state="normal") |
|
|
| if self.turbo_button.get() == 0: |
| generation_thread = threading.Thread(target=run_generation, daemon=True) |
| generation_thread.start() |
| else: |
| turbo_process_request_thread = threading.Thread(target=turbo_process_request, daemon=True) |
| turbo_process_request_thread.start() |
|
|
| def open_AutomationSetting(self): |
| if self.Automation_setting is None: |
| self.Automation_setting = Automation_setting(self) |
| else: |
| if self.Automation_setting.state() == 'withdrawn': |
| self.Automation_setting.deiconify() |
| else: |
| self.Automation_setting.focus() |
|
|
| def show_advanced_settings(self): |
| if self.Advanced_setting is None: |
| self.Advanced_setting = Advanced_setting(self) |
| else: |
| if self.Advanced_setting.state() == 'withdrawn': |
| self.Advanced_setting.deiconify() |
| else: |
| self.Advanced_setting.focus() |
|
|
| def open_Character_search(self): |
| if self.Character_search is None: |
| if not os.path.exists("csdataset.parquet"): |
| make_parquet(self) |
| self.Character_search = Character_search(self) |
| else: |
| if self.Character_search.state() == 'withdrawn': |
| self.Character_search.deiconify() |
| else: |
| self.Character_search.focus() |
|
|
| def make_parquet(self): |
| filtered_dfs = [] |
| df = pd.read_parquet(os.path.join(basedir, "tags.parquet"),engine="pyarrow") |
| filtered_df = df[df['general'].str.contains(' solo,', na=False) & ~df['general'].str.contains('monochrome', na=False)] |
| filtered_dfs.append(filtered_df) |
| final_df = pd.concat(filtered_dfs, ignore_index=True) |
| final_df.to_parquet("csdataset.parquet") |
|
|
|
|
|
|
| self.Advanced_setting = None |
| self.Automation_setting = None |
| self.Character_search = None |
| |
| self.image_generation_frame = customtkinter.CTkFrame(self.text_input_frame) |
| self.image_generation_frame.grid(row=2, column=0, sticky="nsew") |
| self.image_generation_frame.columnconfigure(0, weight= 1) |
| self.image_generation_frame.columnconfigure(1, weight= 1) |
| self.image_generation_frame.columnconfigure(2, weight= 1) |
| self.random_function_button = customtkinter.CTkButton(self.image_generation_frame, text="๋๋ค/๋ค์ ํ๋กฌํํธ", font=my_font, command=random_function) |
| self.random_function_button.grid(row=0, column=0, pady=5, sticky="nsew") |
| self.automation_setting_button = customtkinter.CTkButton(self.image_generation_frame, text="์๋ํ ์ค์ ", fg_color="#CDCDCD", text_color="black", hover_color="#848484", font=my_font, command=lambda: open_AutomationSetting(self)) |
| self.automation_setting_button.grid(row=0, column=1, padx=15, sticky="ew") |
| self.image_generation_button = customtkinter.CTkButton(self.image_generation_frame, text="NAI ์ด๋ฏธ์ง ์์ฑ", fg_color="#ED7D31", hover_color="#CC5D12", font=my_font, state="disabled", command=lambda: NAIA_generate(self), text_color_disabled="black") |
| self.image_generation_button.grid(row=0, column=2, pady=5, sticky="nsew") |
|
|
| def hold_prompt(): |
| if self.toggle_prompt_fix_button.get() == 1: |
| if self.unlock_hold_prompt_var.get() == 0: |
| self.fixed_prompt_input.configure(state="disabled", text_color="#A2B8D2") |
| self.fixed_prompt_after_input.configure(state="disabled", text_color="#A2B8D2") |
| self.random_function_button.configure(state="disabled") |
| self.rm_not_nsfw_button.deselect() |
| self.rm_not_nsfw_button.configure(state="disabled") |
| self.random_artist_button.deselect() |
| self.random_artist_button.configure(state="disabled") |
| self.unlock_hold_prompt.configure(state="normal") |
| else: |
| self.fixed_prompt_input.configure(state="normal", text_color="#DCE4EE") |
| self.fixed_prompt_after_input.configure(state="normal", text_color="#DCE4EE") |
| self.random_function_button.configure(state="normal") |
| self.rm_not_nsfw_button.configure(state="normal") |
| self.random_artist_button.configure(state="normal") |
|
|
| def explicit_user_attention(): |
| if self.turbo_button.get() == 1: |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0","end") |
| self.image_label_report.insert("0.0", "<UserAttention> Explicit turbo ์์ฑ ๋์ค ์ค๋จ ํฌ๋ง์ ์ฒดํฌ๋ฐ์ค๋ฅผ ํด์ ํ์๊ธฐ ๋ฐ๋๋๋ค.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| else: |
| self.image_label_report.configure(text_color="#DCE4EE") |
|
|
| self.toggle_prompt_fix = customtkinter.IntVar() |
| self.toggle_prompt_fix_button = customtkinter.CTkCheckBox(self.image_generation_frame, text="ํ๋กฌํํธ ๊ณ ์ ", variable=self.toggle_prompt_fix, font=my_font, command=hold_prompt) |
| self.toggle_prompt_fix_button.grid(row=1, column=0, sticky="ew", padx=15, pady=5) |
| self.automation = customtkinter.IntVar() |
| self.automation_button = customtkinter.CTkCheckBox(self.image_generation_frame, text="์๋ํ", variable=self.automation, font=my_font) |
| self.automation_button.grid(row=1, column=1, sticky="ew", padx=15) |
| self.turbo = customtkinter.IntVar() |
| self.turbo_button = customtkinter.CTkCheckBox(self.image_generation_frame, text="์ฐ์ ์ด๋ฏธ์ง ์์ฑ(Explicit)", variable=self.turbo, font=my_font, command=explicit_user_attention) |
| self.turbo_button.grid(row=1, column=2, sticky="ew") |
| self.fixed_prompt_label = customtkinter.CTkLabel(self.text_input_frame, text=" ----------------------------- ์ ํ ๊ณ ์ ํ๋กฌํํธ ----------------------------- ", font=large_font) |
| self.fixed_prompt_label.grid(row = 3, column=0, sticky="n" ) |
| self.fixed_prompt_input = customtkinter.CTkTextbox(self.text_input_frame, width=490, height=100, font=v_large_font, undo=True) |
| self.fixed_prompt_input.grid(row=4, column=0, pady=5, sticky="nsew") |
| |
| |
| self.check_box_frame = customtkinter.CTkFrame(self.text_input_frame) |
| self.check_box_frame.grid(row=5, column=0, pady=5, columnspan=3, sticky="nsew") |
| self.rm_artist_name_var = customtkinter.IntVar() |
| self.rm_artist_name_button = customtkinter.CTkCheckBox(self.check_box_frame, text="์๊ฐ๋ช
์ ๊ฑฐ", variable=self.rm_artist_name_var, font=my_font) |
| self.rm_artist_name_button.grid(row=0, column=0, pady=5, padx=10, sticky="nsew") |
| self.rm_copyright_name_var = customtkinter.IntVar() |
| self.rm_copyright_name_button = customtkinter.CTkCheckBox(self.check_box_frame, text="์ํ๋ช
์ ๊ฑฐ", variable=self.rm_copyright_name_var, font=my_font) |
| self.rm_copyright_name_button.grid(row=0, column=1, pady=5, sticky="nsew") |
| self.rm_characteristic_var = customtkinter.IntVar() |
| self.rm_characteristic_button = customtkinter.CTkCheckBox(self.check_box_frame, text="์บ๋ฆญํฐ ํน์ง ์ ๊ฑฐ", variable=self.rm_characteristic_var, font=my_font) |
| self.rm_characteristic_button.grid(row=0, column=2, padx=10, pady=5, sticky="nsew") |
| self.rm_not_nsfw_var = customtkinter.IntVar() |
| self.rm_not_nsfw_button = customtkinter.CTkCheckBox(self.check_box_frame, text="NSFW Only", variable=self.rm_not_nsfw_var, font=my_font) |
| self.rm_not_nsfw_button.grid(row=0, column=3, padx=15, pady=5, sticky="nsew") |
|
|
| self.fixed_prompt_after_label = customtkinter.CTkLabel(self.text_input_frame, text=" ----------------------------- ํํ ๊ณ ์ ํ๋กฌํํธ ----------------------------- ", font=large_font) |
| self.fixed_prompt_after_label.grid(row = 6, column=0, sticky="n" ) |
| self.fixed_prompt_after_input = customtkinter.CTkTextbox(self.text_input_frame, width=490, height=60, font=v_large_font, undo=True) |
| self.fixed_prompt_after_input.grid(row=7, column=0, pady=5, sticky="nsew") |
| self.negative_prompt_label_frame = customtkinter.CTkFrame(self.text_input_frame, fg_color="#2B2B2B") |
| self.negative_prompt_label_frame.grid(row = 8, column=0, sticky="nsew" ) |
| self.negative_prompt_label = customtkinter.CTkLabel(self.negative_prompt_label_frame, text=" ----------------------------- ๋ค๊ฑฐํฐ๋ธ ํ๋กฌํํธ ------------- Strength : ", font=large_font) |
| self.negative_prompt_label.grid(row = 0, column=0, sticky="w" ) |
| self.uncond_strength_entry = customtkinter.CTkEntry(self.negative_prompt_label_frame, font=my_font, width=50) |
| self.uncond_strength_entry.grid(row = 0, column=1, sticky="e" ) |
| self.uncond_strength_entry.insert(0, "1.00") |
| self.negative_prompt_input = customtkinter.CTkTextbox(self.text_input_frame, width=490, height=100, font=v_large_font, undo=True) |
| self.negative_prompt_input.insert("0.0", "lowres, jpeg artifacts, worst quality, watermark, blurry, very displeasing") |
| self.negative_prompt_input.grid(row=9, column=0, pady=5, sticky="nsew") |
| self.auto_hide_label = customtkinter.CTkLabel(self.text_input_frame, text="----------------------------- ์๋์จ๊น ํค์๋ -----------------------------", font=my_font) |
| self.auto_hide_label.grid(row = 10, column=0, sticky="n" ) |
| self.auto_hide_keyword_input = customtkinter.CTkTextbox(self.text_input_frame, width=490, height=100, font=v_large_font, undo=True) |
| self.auto_hide_keyword_input.grid(row=11, column=0, pady=5, sticky="nsew") |
| self.auto_quality_toggle_var = customtkinter.IntVar() |
| self.auto_quality_toggle =customtkinter.CTkCheckBox(self.text_input_frame, text="Auto Quality Tag ํ์ฑํ (Undesired Content Preset์ ์ ์ธ)", font=my_font, variable=self.auto_quality_toggle_var) |
| self.auto_quality_toggle.grid(row = 12, column = 0, pady=5, sticky="w") |
| self.auto_quality_toggle.select() |
| self.unlock_hold_prompt_var = customtkinter.IntVar() |
|
|
| def unlock_hold(): |
| if self.unlock_hold_prompt_var.get() == 1: |
| self.fixed_prompt_input.configure(state="normal", text_color="#DCE4EE") |
| self.fixed_prompt_after_input.configure(state="normal", text_color="#DCE4EE") |
| elif self.unlock_hold_prompt_var.get() == 0 and self.toggle_prompt_fix_button.get() == 1: |
| self.fixed_prompt_input.configure(state="disabled", text_color="#A2B8D2") |
| self.fixed_prompt_after_input.configure(state="disabled", text_color="#A2B8D2") |
|
|
| def reset_wildcard_preopen(): |
| if self.toggle_wildcard_preopen_var.get() == 1: |
| self.wildcard_preopen_repeat_current = self.wildcard_preopen_repeat |
| |
| self.unlock_hold_prompt = customtkinter.CTkCheckBox(self.text_input_frame, text="ํ๋กฌํํธ ๊ณ ์ : ์ ํ/ํํ/๋๋ค ํ๋กฌํํธ ์ ๊ธ ๊ฐ์ ํด์ ", font=my_font, variable=self.unlock_hold_prompt_var, state="disabled", command=unlock_hold) |
| self.unlock_hold_prompt.grid(row = 13, column = 0, pady=5, sticky="w") |
| self.unlock_hold_prompt.deselect() |
| self.toggle_wildcard_preopen_var = customtkinter.IntVar() |
| self.toggle_wildcard_preopen = customtkinter.CTkCheckBox(self.text_input_frame, text="์ ํํ๋กฌ์ ์์ผ๋์นด๋๋ฅผ ๋๋ค ํ๋กฌํํธ์์ ๋จผ์ ์คํ", font=my_font, variable=self.toggle_wildcard_preopen_var, command=reset_wildcard_preopen) |
| self.toggle_wildcard_preopen.deselect() |
| self.toggle_wildcard_preopen.grid(row = 14, column = 0, pady=5, sticky="w") |
|
|
| self.conditional_frame = customtkinter.CTkFrame(self.text_input_frame, fg_color="#2B2B2B") |
| self.conditional_frame.grid(row = 15, column=0, sticky="w", pady=5) |
|
|
| self.cond_prompt_button_var = customtkinter.IntVar() |
| self.cond_prompt_button = customtkinter.CTkCheckBox(self.conditional_frame,text="์กฐ๊ฑด๋ถ ํ๋กฌํํธ ํ์ฑํ", variable=self.cond_prompt_button_var,font=my_font ) |
| self.cond_prompt_button.grid(row = 0, column = 0, sticky="w") |
|
|
| self.cond_negative_button_var = customtkinter.IntVar() |
| self.cond_negative_button = customtkinter.CTkCheckBox(self.conditional_frame,text="์กฐ๊ฑด๋ถ ๋ค๊ฑฐํฐ๋ธ ํ์ฑํ", variable=self.cond_negative_button_var, font=my_font ) |
| self.cond_negative_button.grid(row = 0, column = 1, sticky="w", padx=5) |
|
|
| self.conditional_prompt_label = customtkinter.CTkLabel(self.text_input_frame, text="------------------------ ์กฐ๊ฑด๋ถ ํ๋กฌํํธ (๋๋ค์ ์ ์ฉ) ------------------------", font=my_font) |
| self.conditional_prompt_label.grid(row = 16, column=0, sticky="n" ) |
| self.conditional_prompt_input = customtkinter.CTkTextbox(self.text_input_frame, width=490, height=100, font=v_large_font, undo=True) |
| self.conditional_prompt_input.grid(row=17, column=0, pady=5, sticky="nsew") |
|
|
| self.conditional_negative_label = customtkinter.CTkLabel(self.text_input_frame, text="------------------------ ์กฐ๊ฑด๋ถ ๋ค๊ฑฐํฐ๋ธ (์์ฑ์ ์ ์ฉ) ------------------------", font=my_font) |
| self.conditional_negative_label.grid(row = 18, column=0, sticky="n" ) |
| self.conditional_negative_input = customtkinter.CTkTextbox(self.text_input_frame, width=490, height=100, font=v_large_font, undo=True) |
| self.conditional_negative_input.grid(row=19, column=0, pady=5, sticky="nsew") |
|
|
| |
| self.image_history_frame = customtkinter.CTkFrame(self.right_frame) |
| self.image_history_frame.grid(row=0, column=0, columnspan=2, rowspan=15, padx=5, pady=5, sticky="n") |
|
|
| def show_context_menu(event): |
| if len(app.image_queue) != 0: |
| self.image_label_context_menu.tk_popup(event.x_root, event.y_root) |
|
|
| def copy_file(event=None): |
| focused_widget = self.focus_get() |
| if focused_widget == self: |
| file_path = self.image_queue[self.current_window][3] |
| command = f"powershell -command \"Get-Item '{file_path}' | Set-Clipboard\"" |
| subprocess.run(command, shell=True) |
| self.control_pressed = False |
|
|
| def copy_file_instant(event=None): |
| file_path = self.image_queue[self.current_window][3] |
| command = f"powershell -command \"Get-Item '{file_path}' | Set-Clipboard\"" |
| subprocess.run(command, shell=True) |
|
|
| |
| self.image_label = customtkinter.CTkLabel(self.image_history_frame, text="") |
| self.image_label.grid(row=1, column=0, rowspan=14, padx=10, pady=5, sticky="w") |
| self.image_label.bind("<Button-3>", show_context_menu) |
|
|
| self.image_label_context_menu = tk.Menu(self, tearoff=0, font=v_large_font) |
| self.image_label_context_menu.add_command(label="์ด๋ฏธ์ง๋ฅผ ํด๋ฆฝ๋ณด๋์ ๋ณต์ฌ", command=copy_file_instant) |
|
|
| |
| self.image_label_report = customtkinter.CTkTextbox(self.right_frame, width=740, height=100, font=large_font) |
| |
| self.image_label_under_frame = customtkinter.CTkFrame(self.image_history_frame) |
| self.image_label_under_frame.grid(row=0, column=0, padx=10, pady=5, sticky="ew") |
| self.open_save_folder = customtkinter.CTkButton(self.image_label_under_frame, text="ํด๋ ์ด๊ธฐ", font=my_font, fg_color="transparent", width=60,command=lambda: open_file_explorer(self)) |
| self.open_save_folder.grid(row=0, column=0,padx=5, pady=5, sticky="nsew") |
| self.window_label = customtkinter.CTkLabel(self.image_label_under_frame, text="0 / 0", font=my_font, width=60) |
| self.window_label.grid(row=0, column=4,padx=5, pady=5, sticky="nsew") |
| self.my_anlas = customtkinter.IntVar(value=0) |
| self.anlas_label = customtkinter.CTkLabel(self.image_label_under_frame, text="Anlas : 0", font=my_font) |
| self.anlas_label.grid(row=0, column=5,padx=5, pady=5, sticky="nsew") |
| self.request_upper_size_button = customtkinter.CTkButton(self.image_label_under_frame, text="img2img (trial)",fg_color="transparent", hover_color="grey10", text_color="#FFFF97", font=my_font,state="disabled", command=lambda: img2img(self), width=110) |
| self.request_upper_size_button.grid(row=0, column=6,padx=5, pady=5, sticky="nsew") |
| |
| |
| self.open_in_file_explorer = customtkinter.CTkButton(self.image_label_under_frame, text="ํ์ผ ์์น ์ด๊ธฐ", font=my_font, fg_color="transparent", width=80,command=lambda: open_in_file_explorer(self)) |
| self.open_in_file_explorer.grid(row=0, column=7,padx=5, pady=5, sticky="nsew") |
| self.open_save_folder.grid(row=0, column=0,padx=5, pady=5, sticky="nsew") |
| self.image_label_report.grid(row=16, column=0, padx=10, pady=5, sticky="ew") |
| self.image_label_report.configure(state="disabled") |
| white_image = Image.new('RGB', (768, 768), 'white') |
| |
| white_photo = customtkinter.CTkImage(white_image, size=(620,620)) |
| self.image_label.configure(image=white_photo) |
| self.image_label.image = white_photo |
|
|
| self.image_queue = [] |
| self.window = 0 |
|
|
| |
| self.image_history_sub_frame = customtkinter.CTkFrame(self.image_history_frame, width=100, height=550) |
| self.image_history_sub_frame.grid(row=1, rowspan=12, column=1, sticky="n") |
|
|
| self.output_file_path = f"output_NAI\\{self.start_time}\\txt2img" |
| self.import_image_count = 0 |
|
|
| def open_file_explorer(self): |
| if not os.path.exists(self.output_file_path): |
| os.makedirs(self.output_file_path) |
| os.startfile(self.output_file_path) |
|
|
| def open_in_file_explorer(self): |
| filename = self.image_queue[self.current_window][3] |
| explorer_command = f'explorer /select,"{os.path.abspath(filename)}"' |
| subprocess.run(explorer_command) |
|
|
| def up_button_pressed(): |
| if self.image_history_button_down.cget("state") == "disabled": |
| self.image_history_button_down.configure(state="normal") |
| if self.control_pressed: |
| self.window = len(self.image_queue) |
| else: |
| self.window+=1 |
| update() |
| show_text = str(self.window)+" / "+ str(len(self.image_queue)) |
| self.window_label.configure(text=show_text) |
| if self.window >= len(self.image_queue): |
| self.image_history_button_up.configure(state="disabled") |
| self.control_pressed = False |
| image_history_0_yield() |
|
|
| def down_button_pressed(): |
| if self.image_history_button_up.cget("state") == "disabled": |
| self.image_history_button_up.configure(state="normal") |
| if self.control_pressed: |
| if self.window >= 105: |
| self.window -= 100 |
| else: |
| self.window = 5 |
| else: |
| self.window-=1 |
| update() |
| show_text = str(self.window)+" / "+ str(len(self.image_queue)) |
| self.window_label.configure(text=show_text) |
| if self.window <= 5: |
| self.image_history_button_down.configure(state="disabled") |
| self.control_pressed = False |
| image_history_4_yield() |
|
|
| highlight_tags = ["aqua","black","blonde","blue","brown","cyan","green","grey","orange","pink","purple","red","violet","white","yellow","mouth", "eyes", " cap", " ears", " girl", " ornament", " hat", "beret", " ear "] |
|
|
| def insert_with_color(cs_text_input, current_lookup): |
| words = [word.strip() for word in current_lookup.split(',')] |
| for index, word in enumerate(words): |
| highlight = False |
| start_index = cs_text_input.index("end-1c") |
| if index == len(words) - 1: |
| cs_text_input.insert("end", word) |
| else: |
| cs_text_input.insert("end", word + ", ") |
| end_index = cs_text_input.index("end-1c") |
|
|
| for tag in highlight_tags: |
| if tag in word: |
| highlight = True |
| if word == "hat": highlight = True |
| |
| if word in cd.character_dictionary or word in tagbag.bag_of_tags[5:] or 'tail' in word or 'pupil' in word: |
| cs_text_input.tag_add(word, start_index, end_index) |
| cs_text_input.tag_config(word, foreground="#FFFF97") |
| elif highlight: |
| cs_text_input.tag_add(word, start_index, end_index) |
| cs_text_input.tag_config(word, foreground="#AED395") |
|
|
| self.current_window = None |
| def set_image_to_queue(imagen, prompt, seed, filename): |
| if type(self.current_popped_row) != type(None) and not self.current_popped_row.empty: |
| self.image_queue.append([imagen, prompt, seed, filename, self.current_popped_row.copy()]) |
| else: |
| self.image_queue.append([imagen, prompt, seed, filename]) |
| |
| self.request_upper_size_button.configure(state="normal") |
| self.current_window = self.window |
| if self.window == len(self.image_queue)-1: |
| self.window+=1 |
| update() |
| show_text = str(self.window)+" / "+ str(len(self.image_queue)) |
| |
| self.window_label.configure(text=show_text) |
| |
| def image_history_0_yield(event=None): |
| if len(self.image_queue) < 1: |
| return |
| if self.control_pressed: |
| if self.toggle_prompt_fix_button.get() == 0: |
| self.text_input.delete("0.0", "end") |
| insert_with_color(self.text_input, self.image_queue[self.window-1][1]) |
| else: |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<UserAttention> ํ๋กฌํํธ ๊ณ ์ ์ํ์์๋ ์ฌ์ฉ ํ ์ ์์ต๋๋ค.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| else: |
| self.current_window = self.window-1 |
| if app.state() != 'zoomed': |
| self.image_label.configure(image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(620,620))) |
| else: |
| current_image = Image.open(self.image_queue[self.window-1][3]) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| self.image_label.configure(image=instant_result_image) |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.configure(text_color="#DCE4EE") |
| self.image_label_report.insert("0.0", self.image_queue[self.window-1][1] + " seed : " + str(self.image_queue[self.window-1][2])) |
| self.image_label_report.configure(state="disabled") |
| if len(self.image_queue[self.window-1]) >= 5 and 'id' in self.image_queue[self.window-1][4].index: |
| if fav_check(self.image_queue[self.window-1][4]['id']): |
| self.history_export.configure(text="+ FAV (โ
)", state="disabled") |
| else: |
| self.history_export.configure(text="+ FAV (โ)", state="normal") |
| else: |
| self.history_export.configure(state="disabled") |
|
|
| def image_history_1_yield(event=None): |
| if len(self.image_queue) < 2: |
| return |
| if self.control_pressed: |
| if self.toggle_prompt_fix_button.get() == 0: |
| self.text_input.delete("0.0", "end") |
| insert_with_color(self.text_input, self.image_queue[self.window-2][1]) |
| else: |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<UserAttention> ํ๋กฌํํธ ๊ณ ์ ์ํ์์๋ ์ฌ์ฉ ํ ์ ์์ต๋๋ค.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| else: |
| self.current_window = self.window-2 |
| if app.state() != 'zoomed': |
| self.image_label.configure(image=customtkinter.CTkImage(self.image_queue[self.window-2][0], size=(620,620))) |
| else: |
| current_image = Image.open(self.image_queue[self.window-2][3]) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| self.image_label.configure(image=instant_result_image) |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.configure(text_color="#DCE4EE") |
| self.image_label_report.insert("0.0", self.image_queue[self.window-2][1] + " seed : " + str(self.image_queue[self.window-2][2])) |
| self.image_label_report.configure(state="disabled") |
| if len(self.image_queue[self.window-2]) >= 5 and 'id' in self.image_queue[self.window-2][4].index: |
| if fav_check(self.image_queue[self.window-2][4]['id']): |
| self.history_export.configure(text="+ FAV (โ
)", state="disabled") |
| else: |
| self.history_export.configure(text="+ FAV (โ)", state="normal") |
| else: |
| self.history_export.configure(state="disabled") |
| def image_history_2_yield(event=None): |
| if len(self.image_queue) < 3: |
| return |
| if self.control_pressed: |
| if self.toggle_prompt_fix_button.get() == 0: |
| self.text_input.delete("0.0", "end") |
| insert_with_color(self.text_input, self.image_queue[self.window-3][1]) |
| else: |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<UserAttention> ํ๋กฌํํธ ๊ณ ์ ์ํ์์๋ ์ฌ์ฉ ํ ์ ์์ต๋๋ค.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| else: |
| self.current_window = self.window-3 |
| if app.state() != 'zoomed': |
| self.image_label.configure(image=customtkinter.CTkImage(self.image_queue[self.window-3][0], size=(620,620))) |
| else: |
| current_image = Image.open(self.image_queue[self.window-3][3]) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| self.image_label.configure(image=instant_result_image) |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.configure(text_color="#DCE4EE") |
| self.image_label_report.insert("0.0", self.image_queue[self.window-3][1] + " seed : " + str(self.image_queue[self.window-3][2])) |
| self.image_label_report.configure(state="disabled") |
| if len(self.image_queue[self.window-3]) >= 5 and 'id' in self.image_queue[self.window-3][4].index: |
| if fav_check(self.image_queue[self.window-3][4]['id']): |
| self.history_export.configure(text="+ FAV (โ
)", state="disabled") |
| else: |
| self.history_export.configure(text="+ FAV (โ)", state="normal") |
| else: |
| self.history_export.configure(state="disabled") |
| def image_history_3_yield(event=None): |
| if len(self.image_queue) < 4: |
| return |
| if self.control_pressed: |
| if self.toggle_prompt_fix_button.get() == 0: |
| self.text_input.delete("0.0", "end") |
| insert_with_color(self.text_input, self.image_queue[self.window-4][1]) |
| else: |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<UserAttention> ํ๋กฌํํธ ๊ณ ์ ์ํ์์๋ ์ฌ์ฉ ํ ์ ์์ต๋๋ค.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| else: |
| self.current_window = self.window-4 |
| if app.state() != 'zoomed': |
| self.image_label.configure(image=customtkinter.CTkImage(self.image_queue[self.window-4][0], size=(620,620))) |
| else: |
| current_image = Image.open(self.image_queue[self.window-4][3]) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| self.image_label.configure(image=instant_result_image) |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.configure(text_color="#DCE4EE") |
| self.image_label_report.insert("0.0", self.image_queue[self.window-4][1] + " seed : " + str(self.image_queue[self.window-4][2])) |
| self.image_label_report.configure(state="disabled") |
| if len(self.image_queue[self.window-4]) >= 5 and 'id' in self.image_queue[self.window-4][4].index: |
| if fav_check(self.image_queue[self.window-4][4]['id']): |
| self.history_export.configure(text="+ FAV (โ
)", state="disabled") |
| else: |
| self.history_export.configure(text="+ FAV (โ)", state="normal") |
| else: |
| self.history_export.configure(state="disabled") |
| def image_history_4_yield(event=None): |
| if len(self.image_queue) < 5: |
| return |
| if self.control_pressed: |
| if self.toggle_prompt_fix_button.get() == 0: |
| self.text_input.delete("0.0", "end") |
| insert_with_color(self.text_input, self.image_queue[self.window-5][1]) |
| else: |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<UserAttention> ํ๋กฌํํธ ๊ณ ์ ์ํ์์๋ ์ฌ์ฉ ํ ์ ์์ต๋๋ค.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| else: |
| self.current_window = self.window-5 |
| if app.state() != 'zoomed': |
| self.image_label.configure(image=customtkinter.CTkImage(self.image_queue[self.window-5][0], size=(620,620))) |
| else: |
| current_image = Image.open(self.image_queue[self.window-5][3]) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| self.image_label.configure(image=instant_result_image) |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.configure(text_color="#DCE4EE") |
| self.image_label_report.insert("0.0", self.image_queue[self.window-5][1] + " seed : " + str(self.image_queue[self.window-5][2])) |
| self.image_label_report.configure(state="disabled") |
| if len(self.image_queue[self.window-5]) >= 5 and 'id' in self.image_queue[self.window-5][4].index: |
| if fav_check(self.image_queue[self.window-5][4]['id']): |
| self.history_export.configure(text="+ FAV (โ
)", state="disabled") |
| else: |
| self.history_export.configure(text="+ FAV (โ)", state="normal") |
| else: |
| self.history_export.configure(state="disabled") |
|
|
| def update(): |
| if len(self.image_queue) <= 5: |
| if len(self.image_queue) >= 1: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| if len(self.image_queue[self.window-1]) >= 5: |
| button_item_0 = self.image_queue[self.window-1][4].copy() |
| self.image_history_0.bind("<Button-3>", lambda event: on_right_click(event, button_item_0)) |
| if len(self.image_queue) >= 2: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| if len(self.image_queue[self.window-1]) >= 5: |
| button_item_0 = self.image_queue[self.window-1][4].copy() |
| self.image_history_0.unbind("<Button-3>") |
| self.image_history_0.bind("<Button-3>", lambda event: on_right_click(event, button_item_0)) |
| self.image_history_1.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-2][0], size=(100,100))) |
| if len(self.image_queue[self.window-2]) >= 5: |
| button_item_1 = self.image_queue[self.window-2][4].copy() |
| self.image_history_1.bind("<Button-3>", lambda event: on_right_click(event, button_item_1)) |
| if len(self.image_queue) >= 3: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| if len(self.image_queue[self.window-1]) >= 5: |
| button_item_0 = self.image_queue[self.window-1][4].copy() |
| self.image_history_0.unbind("<Button-3>") |
| self.image_history_0.bind("<Button-3>", lambda event: on_right_click(event, button_item_0)) |
| self.image_history_1.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-2][0], size=(100,100))) |
| if len(self.image_queue[self.window-2]) >= 5: |
| button_item_1 = self.image_queue[self.window-2][4].copy() |
| self.image_history_1.unbind("<Button-3>") |
| self.image_history_1.bind("<Button-3>", lambda event: on_right_click(event, button_item_1)) |
| self.image_history_2.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-3][0], size=(100,100))) |
| if len(self.image_queue[self.window-3]) >= 5: |
| button_item_2 = self.image_queue[self.window-3][4].copy() |
| self.image_history_2.bind("<Button-3>", lambda event: on_right_click(event, button_item_2)) |
| if len(self.image_queue) >= 4: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| if len(self.image_queue[self.window-1]) >= 5: |
| button_item_0 = self.image_queue[self.window-1][4].copy() |
| self.image_history_0.unbind("<Button-3>") |
| self.image_history_0.bind("<Button-3>", lambda event: on_right_click(event, button_item_0)) |
| self.image_history_1.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-2][0], size=(100,100))) |
| if len(self.image_queue[self.window-2]) >=5: |
| button_item_1 = self.image_queue[self.window-2][4].copy() |
| self.image_history_1.unbind("<Button-3>") |
| self.image_history_1.bind("<Button-3>", lambda event: on_right_click(event, button_item_1)) |
| self.image_history_2.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-3][0], size=(100,100))) |
| if len(self.image_queue[self.window-3]) >= 5: |
| button_item_2 = self.image_queue[self.window-3][4].copy() |
| self.image_history_2.unbind("<Button-3>") |
| self.image_history_2.bind("<Button-3>", lambda event: on_right_click(event, button_item_2)) |
| self.image_history_3.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-4][0], size=(100,100))) |
| if len(self.image_queue[self.window-4]) >= 5: |
| button_item_3 = self.image_queue[self.window-4][4].copy() |
| self.image_history_3.bind("<Button-3>", lambda event: on_right_click(event, button_item_3)) |
| if len(self.image_queue) == 5: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| if len(self.image_queue[self.window-1]) >= 5: |
| button_item_0 = self.image_queue[self.window-1][4].copy() |
| self.image_history_0.unbind("<Button-3>") |
| self.image_history_0.bind("<Button-3>", lambda event: on_right_click(event, button_item_0)) |
| self.image_history_1.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-2][0], size=(100,100))) |
| if len(self.image_queue[self.window-2]) >= 5: |
| button_item_1 = self.image_queue[self.window-2][4].copy() |
| self.image_history_1.unbind("<Button-3>") |
| self.image_history_1.bind("<Button-3>", lambda event: on_right_click(event, button_item_1)) |
| self.image_history_2.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-3][0], size=(100,100))) |
| if len(self.image_queue[self.window-3]) >= 5: |
| button_item_2 = self.image_queue[self.window-3][4].copy() |
| self.image_history_2.unbind("<Button-3>") |
| self.image_history_2.bind("<Button-3>", lambda event: on_right_click(event, button_item_2)) |
| self.image_history_3.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-4][0], size=(100,100))) |
| if len(self.image_queue[self.window-4]) >= 5: |
| button_item_3 = self.image_queue[self.window-4][4].copy() |
| self.image_history_3.unbind("<Button-3>") |
| self.image_history_3.bind("<Button-3>", lambda event: on_right_click(event, button_item_3)) |
| self.image_history_4.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-5][0], size=(100,100))) |
| if len(self.image_queue[self.window-5]) >= 5: |
| button_item_4 = self.image_queue[self.window-5][4].copy() |
| self.image_history_4.bind("<Button-3>", lambda event: on_right_click(event, button_item_4)) |
| else: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| if len(self.image_queue[self.window-1]) >= 5: |
| button_item_0 = self.image_queue[self.window-1][4].copy() |
| self.image_history_0.unbind("<Button-3>") |
| self.image_history_0.bind("<Button-3>", lambda event: on_right_click(event, button_item_0)) |
| self.image_history_1.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-2][0], size=(100,100))) |
| if len(self.image_queue[self.window-2]) >= 5: |
| button_item_1 = self.image_queue[self.window-2][4].copy() |
| self.image_history_1.unbind("<Button-3>") |
| self.image_history_1.bind("<Button-3>", lambda event: on_right_click(event, button_item_1)) |
| self.image_history_2.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-3][0], size=(100,100))) |
| if len(self.image_queue[self.window-3]) >= 5: |
| button_item_2 = self.image_queue[self.window-3][4].copy() |
| self.image_history_2.unbind("<Button-3>") |
| self.image_history_2.bind("<Button-3>", lambda event: on_right_click(event, button_item_2)) |
| self.image_history_3.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-4][0], size=(100,100))) |
| if len(self.image_queue[self.window-4]) >= 5: |
| button_item_3 = self.image_queue[self.window-4][4].copy() |
| self.image_history_3.unbind("<Button-3>") |
| self.image_history_3.bind("<Button-3>", lambda event: on_right_click(event, button_item_3)) |
| self.image_history_4.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-5][0], size=(100,100))) |
| if len(self.image_queue[self.window-5]) >= 5: |
| button_item_4 = self.image_queue[self.window-5][4].copy() |
| self.image_history_4.unbind("<Button-3>") |
| self.image_history_4.bind("<Button-3>", lambda event: on_right_click(event, button_item_4)) |
| if len(self.image_queue) == 6: |
| self.image_history_button_down.configure(state="normal") |
|
|
| def some_command(item): |
| print(item) |
| random_function(item) |
|
|
| def on_right_click(event, button_item): |
| context_menu.delete(0, tk.END) |
| context_menu.add_command(label="Requeue", command=lambda: some_command(button_item)) |
| context_menu.tk_popup(event.x_root, event.y_root) |
| |
| context_menu = tk.Menu(self, tearoff=0, font=v_large_font, fg="black") |
|
|
| black_image = Image.new('RGB', (100, 100), '#2B2B2B') |
| self.image_history_0 = customtkinter.CTkButton(self.image_history_sub_frame, width=100, height=100, corner_radius=8,text="", fg_color="transparent", image=customtkinter.CTkImage(black_image, size=(100,100)), command=image_history_0_yield) |
| self.image_history_0.grid(row=0, column=1, padx=5, pady=1, sticky="n") |
|
|
| self.image_history_1 = customtkinter.CTkButton(self.image_history_sub_frame, width=100, height=100, corner_radius=8,text="", fg_color="transparent", image=customtkinter.CTkImage(black_image, size=(100,100)), command=image_history_1_yield) |
| self.image_history_1.grid(row=1, column=1, padx=5, pady=1, sticky="n") |
| self.image_history_2 = customtkinter.CTkButton(self.image_history_sub_frame, width=100, height=100, corner_radius=8,text="", fg_color="transparent",image=customtkinter.CTkImage(black_image, size=(100,100)), command=image_history_2_yield) |
| self.image_history_2.grid(row=2, column=1, padx=5, pady=1, sticky="n") |
| self.image_history_3 = customtkinter.CTkButton(self.image_history_sub_frame, width=100, height=100, corner_radius=8,text="", fg_color="transparent", image=customtkinter.CTkImage(black_image, size=(100,100)), command=image_history_3_yield) |
| self.image_history_3.grid(row=3,column=1, padx=5, pady=1, sticky="n") |
| self.image_history_4 = customtkinter.CTkButton(self.image_history_sub_frame, width=100, height=100, corner_radius=8,text="", fg_color="transparent", image=customtkinter.CTkImage(black_image, size=(100,100)), command=image_history_4_yield) |
| self.image_history_4.grid(row=4, column=1, padx=5, pady=1, sticky="n") |
| self.image_history_button_up = customtkinter.CTkButton(self.image_history_frame, text="โฒ", width=100, font=my_font, state="disabled", command=up_button_pressed) |
| self.image_history_button_up.grid(row=0, column=1, pady=10, padx=5, sticky="n") |
| self.image_history_button_down = customtkinter.CTkButton(self.image_history_frame, text="โผ", width=100, font=my_font, state="disabled", command=down_button_pressed) |
| self.image_history_button_down.grid(row=13, column=1, pady=5, padx=5, sticky="n") |
|
|
| def save_to_excel(): |
| |
| |
| workbook = Workbook() |
| sheet = workbook.active |
|
|
| alignment = Alignment(wrapText=True) |
| sheet['A1'].alignment = alignment |
| sheet['A1'] = 'Prompt' |
| sheet['B1'] = 'Seed' |
| sheet['C1'] = 'Image' |
|
|
| image_objects = [] |
|
|
| |
| for index, (pil_image, prompt, seed, *_rest) in enumerate(self.image_queue): |
| |
| row_num = index + 1 |
|
|
| |
| prompt_cell = sheet.cell(row=row_num, column=1, value=prompt) |
| sheet.cell(row=row_num, column=2, value=seed) |
| prompt_cell.alignment = Alignment(wrap_text=True) |
|
|
| |
| pil_image = self.resize_image_to_fit(pil_image,192) |
| output = io.BytesIO() |
| pil_image.save(output, format='PNG') |
| output.seek(0) |
| image_objects.append(output) |
|
|
| |
| img = OpenpyxlImage(output) |
| img.anchor = f'C{row_num}' |
| sheet.add_image(img) |
|
|
| |
| scale_factor = 0.75 |
| sheet.row_dimensions[row_num].height = pil_image.height * scale_factor |
| sheet.column_dimensions['C'].width = pil_image.width * scale_factor / 7 |
|
|
| |
| sheet.column_dimensions['A'].width = 50 |
| sheet.column_dimensions['B'].width = 15 |
|
|
| |
| workbook.save(f"history_with_images_{self.start_time}.xlsx") |
|
|
| |
| for img_obj in image_objects: |
| img_obj.close() |
| if os.path.exists(f"history_with_images_{self.start_time}.xlsx"): |
| os.startfile(f"history_with_images_{self.start_time}.xlsx") |
|
|
| print("์์
ํ์ผ์ ์ ์ฅ๋จ") |
|
|
| def history_export(): |
| if self.control_pressed: |
| save_to_excel() |
| self.history_export.configure(text="+ FAV (โ)", fg_color="#F5F3C2", hover_color="#A69F40", text_color="black") |
| if self.image_queue and len(self.image_queue[self.current_window]) >= 5: |
| self.history_export.configure(state="normal") |
| else: |
| self.history_export.configure(state="disabled") |
| else: |
| """ |
| Description : @history_export ํ์์์๋ FAV ๊ธฐ๋ฅ์ผ๋ก ๋์ |
| """ |
| new_data = self.image_queue[self.current_window][4] |
| validation_prompt = self.image_queue[self.current_window][1] |
| prompt = new_data['general'] |
| validation_prompt = [key.strip() for key in validation_prompt.split(',')] |
| prompt = [key.strip() for key in prompt.split(',')] |
| autohide = self.auto_hide_keyword_input.get("0.0", "end") |
| autohide = [key.strip() for key in autohide.split(',')] + app.data.bag_of_tags |
| prompt = [key for key in prompt if key not in autohide] |
| not_validated = [key for key in prompt if key not in validation_prompt] |
| if len(not_validated) >= 3: |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", f"<ValidationError> ๋๋คํ๊ฒ ์์ฑ๋ ํ๋กฌํํธ๊ฐ ์๋๊ฑฐ๋, {len(not_validated)}๊ฐ์ ์ ๊ฑฐ๋ ํ๊ทธ๊ฐ ๊ฒ์ถ๋์ด Favorite ๋ฑ๋ก์ ์คํจํ์์ต๋๋ค. ํด๋น ๊ธฐ๋ฅ์ ๋๋ค ์์ฑ ํ๋กฌํํธ๋ฅผ DataFrame ๊ท๊ฒฉ์ ๋ง์ถฐ ๋ณต์ฌํ๋ ๊ธฐ๋ฅ์ด๊ธฐ ๋๋ฌธ์ ์์ ๋ ํ๋กฌํํธ์ ์ ์ฅ์ ์ง์ํ์ง ์์ต๋๋ค.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| return |
| try: |
| favorite_prompts = pd.read_parquet('favorite_prompt.parquet') |
| except FileNotFoundError: |
| dtype_dict = { |
| 'id': 'int64', |
| 'copyright': 'object', |
| 'character': 'object', |
| 'artist': 'object', |
| 'general': 'object', |
| 'meta': 'object', |
| 'rating': 'object', |
| 'score': 'object', |
| 'created_at': 'object', |
| } |
| favorite_prompts = pd.DataFrame(columns=dtype_dict.keys()) |
| self.fav_button.configure(state="normal") |
| try: |
| _temp = new_data['id'] |
| except: |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<TagsError> 0215 ๋ฒ์ ๋ถํฐ tags์ ๊ตฌ์ฑ์ด ๋ณ๊ฒฝ๋์์ต๋๋ค. favorite ๊ธฐ๋ฅ์ ์ฌ์ฉํ๊ธฐ ์ํด ๊ฒ์์ ๋ค์ ์ํ ํด ์ฃผ์ธ์.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| return |
| new_data_id = new_data['id'] |
| new_data = pd.DataFrame([new_data]) |
| success = False |
| if 'id' not in favorite_prompts.columns: |
| favorite_prompts = pd.concat([favorite_prompts, new_data], ignore_index=True) |
| success = True |
| elif not favorite_prompts['id'].isin([new_data_id]).any(): |
| favorite_prompts = pd.concat([favorite_prompts, new_data], ignore_index=True) |
| success = True |
| else: |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<AlreadyExist> ์ด๋ฏธ favorite์ ๋ฑ๋ก๋์ด์๋ ํ๋กฌํํธ์
๋๋ค.") |
| self.image_label_report.configure(text_color="#FFFF97") |
| self.image_label_report.configure(state="disabled") |
| if success: |
| pil_image = self.image_queue[self.current_window][0] |
| folder_path = 'favorite_prompts' |
| if not os.path.exists(folder_path): |
| os.makedirs(folder_path) |
| favorite_prompts.to_parquet('favorite_prompt.parquet') |
| image_path = os.path.join(folder_path, f"{_temp}.jpg") |
| pil_image.save(image_path, 'JPEG') |
| self.fav_id_list.append(_temp) |
| self.history_export.configure(text="FAV (โ
)", fg_color="#F5F3C2", hover_color="#A69F40", text_color="black", state="disabled") |
|
|
| self.history_export = customtkinter.CTkButton(self.image_history_frame, text="+ FAV (โ)", width=100, fg_color="#F5F3C2", hover_color="#A69F40", text_color="black", font=my_font, command=history_export, state="disabled") |
| self.history_export.grid(row=14, column=1, pady=10, padx=5, sticky="n") |
|
|
| |
| self.extended_right_frame = customtkinter.CTkFrame(self.right_frame) |
| self.extended_right_frame.grid(row=17, column=0, columnspan=2, padx=5, sticky="w") |
| self.random_artist_var = customtkinter.IntVar() |
| self.random_artist_button = customtkinter.CTkCheckBox(self.extended_right_frame, text="๋๋ค์๊ฐ ์ถ๊ฐ", variable=self.random_artist_var, font=my_font, state="disabled") |
| self.random_artist_button.grid(row=0, column=0,pady=5, padx=10, sticky="nsew") |
| self.random_artist_manage_button = customtkinter.CTkButton(self.extended_right_frame, text="๋๋ค์๊ฐ ๊ด๋ฆฌ", font=my_font, command=self.random_artist_window) |
| self.random_artist_manage_button.grid(row=0, column=1,pady=5, padx=5, sticky="nsew") |
| self.recommended_prompt_button = customtkinter.CTkButton(self.extended_right_frame, text="์ถ์ฒ ํ๋กฌํํธ", font=my_font, command= self.open_prompt_window) |
| self.recommended_prompt_button.grid(row=0, column=3,pady=5, padx=5, sticky="nsew") |
| self.wildcard_manager_button = customtkinter.CTkButton(self.extended_right_frame, text="์์ผ๋์นด๋ ๊ด๋ฆฌ", font=my_font, command=self.open_wildcard_window) |
| self.wildcard_manager_button.grid(row=0, column=2,pady=5, padx=5, sticky="nsew") |
| self.character_search_button = customtkinter.CTkButton(self.extended_right_frame, text="์บ๋ฆญํฐ ๊ฒ์", fg_color="#7030A0", hover_color="#481F67", font=my_font, command= lambda: open_Character_search(self)) |
| self.character_search_button.grid(row=0, column=4,pady=5, padx=5, sticky="nsew") |
| self.seed_fix_var = customtkinter.IntVar() |
| self.seed_fix_button = customtkinter.CTkCheckBox(self.extended_right_frame, text="์๋๊ณ ์ ", variable=self.seed_fix_var, font=my_font) |
| self.seed_fix_button.grid(row=1, column=0,pady=5, padx=10, sticky="nsew") |
| self.entry_seed_value = customtkinter.IntVar() |
| self.seed_entry = customtkinter.CTkEntry(self.extended_right_frame, textvariable=self.entry_seed_value) |
| self.seed_entry.grid(row=1, column=1,pady=5, padx=5, sticky="nsew") |
| self.seed_entry.insert(0, random.randint(0,9999999999)) |
| self.cfg_scale_label = customtkinter.CTkLabel(self.extended_right_frame, text="CFG Scale : ", font=my_font) |
| self.cfg_scale_label.grid(row=1, column=2,pady=5, padx=5, sticky="w") |
| self.cfg_scale_var = customtkinter.StringVar(value="5.0") |
| self.cfg_scale_entry = customtkinter.CTkEntry(self.extended_right_frame, width=65, textvariable=self.cfg_scale_var) |
| self.cfg_scale_entry.grid(row=1, column=2,pady=5, padx=5, sticky="e") |
| self.prompt_guidance_rescale_var = customtkinter.StringVar(value="0") |
| self.prompt_guidance_rescale_label = customtkinter.CTkLabel(self.extended_right_frame, text="Prompt Guidance Rescale :", font=my_font) |
| self.prompt_guidance_rescale_label.grid(row=1, column=3, pady=5, padx=5, sticky="w") |
| self.prompt_guidance_rescale_entry = customtkinter.CTkEntry(self.extended_right_frame, width=65) |
| self.prompt_guidance_rescale_entry.grid(row=1, column=4,pady=5, padx=10, sticky="w") |
| self.extra_setting_button = customtkinter.CTkButton(self.extended_right_frame, width=55, text="๊ธฐํ์ค์ ", font=my_font, fg_color="grey", hover_color="grey5", command=lambda: show_advanced_settings(self)) |
| self.extra_setting_button.grid(row=1, column=4,pady=5, padx=5, sticky="e") |
| self.random_resolution_var = customtkinter.IntVar() |
| self.random_resolution_button = customtkinter.CTkCheckBox(self.extended_right_frame, text="๋๋ค ํด์๋ ", variable=self.random_resolution_var, font=my_font) |
| self.random_resolution_button.grid(row=2, column=0,pady=5, padx=10, sticky="nsew") |
| self.resolution_var = customtkinter.StringVar(value="1024 x 1024") |
| self.resolution_button = customtkinter.CTkComboBox(self.extended_right_frame, width=160,values=["1024 x 1024", "960 x 1088", "896 x 1152", "832 x 1216", "1088 x 960", "1152 x 896", "1216 x 832"], variable=self.resolution_var, font=my_font) |
| self.resolution_button.grid(row=2, column=1, padx=5, sticky="w") |
| self.sampler_label = customtkinter.CTkLabel(self.extended_right_frame, text="Sampler :", font=my_font) |
| self.sampler_label.grid(row=2, column=2,pady=5, padx=15, sticky="w") |
| self.sampler_var = customtkinter.StringVar(value="k_euler_ancestral") |
| self.sampler_button = customtkinter.CTkComboBox(self.extended_right_frame, width=210,values=["k_euler", "k_euler_ancestral", "k_dpmpp_2s_ancestral", "k_dpmpp_sde", "k_dpmpp_2m"], variable=self.sampler_var, font=my_font) |
| self.sampler_button.grid(row=2, column=2, columnspan=2, pady=5, padx=15, sticky="e") |
| self.show_fullscreen_btn = customtkinter.CTkButton(self.extended_right_frame, text="์ ์ฒดํ๋ฉด(ESC๋ซ๊ธฐ)", font=my_font, command=self.show_fullscreen_image) |
| self.show_fullscreen_btn.grid(row=2, column=4,pady=5, padx=5, sticky="nsew") |
| self.sema_button_var = customtkinter.IntVar() |
|
|
| def sema_pressed(): |
| if self.sema_button_var.get() == 0: |
| self.dyn_button.deselect() |
|
|
| self.sema_button = customtkinter.CTkCheckBox(self.extended_right_frame, text="SMEA", variable=self.sema_button_var,font=my_font, command=sema_pressed) |
| self.sema_button.grid(row=3, column=0, pady=5, padx=10, sticky="w") |
| self.dyn_button_var = customtkinter.IntVar() |
| |
| def sema_dyn_pressed(): |
| if self.dyn_button_var.get() == 1: |
| self.sema_button.select() |
|
|
| self.dyn_button = customtkinter.CTkCheckBox(self.extended_right_frame, text="SMEA+DYN", variable=self.dyn_button_var, font=my_font, command=sema_dyn_pressed) |
| self.dyn_button.grid(row=3, column=1, pady=5, padx=10, sticky="w") |
| self.state_label = customtkinter.CTkLabel(self.extended_right_frame, text="state : idle", font=my_font) |
| self.state_label.grid(row=3, column=2, columnspan=2, pady=5, padx=10, sticky="n") |
| self.instant_row_button = customtkinter.CTkButton(self.extended_right_frame, text="์ธ์คํดํธ ์ด๋ฒคํธ", font=my_font, fg_color="grey10", command=self.open_instant_event) |
| self.instant_row_button.grid(row=3, column=4,pady=5, padx=5, sticky="nsew") |
| GENERATE_EVENT = "<<GenerateEvent>>" |
| sync_text() |
| self.bind(GENERATE_EVENT, lambda x=None: NAIA_generate(self)) |
| self.output_file_path_personal = False |
| self.png_name_rule = "time" |
| self.name_var = customtkinter.StringVar(value=self.png_name_rule) |
| self.window_resize_last_access = None |
| self.last_window_size_conf = None |
| self.generation_count = 0 |
| self.random_artist_list = None |
| self.random_artist = [] |
| self.random_artist_prefix = customtkinter.StringVar(value="none") |
|
|
| def on_window_resize(event): |
| if datetime.now() == app.last_window_size_conf: |
| return |
| time_difference = datetime.now() - app.start_time_prime |
| if (time_difference.total_seconds() < 3): |
| return |
| if app.last_window_size is None: |
| app.last_window_size = (app.winfo_width(), app.winfo_height()) |
| |
| current_size = (app.winfo_width(), app.winfo_height()) |
| if current_size != app.last_window_size: |
| app.last_window_size = current_size |
| if app.state() == 'zoomed': |
| print(app.winfo_width()) |
| self.right_frame.configure(width=1200) |
| self.image_label.grid_forget() |
| self.image_label_under_frame.grid_forget() |
| self.image_history_sub_frame.grid_forget() |
| self.image_history_button_down.grid_forget() |
| self.image_history_sub_frame.grid(row=0, pady=45, rowspan=12, column=1, sticky="n") |
| self.image_history_button_down.grid(row=9, column=1, pady=5, padx=5, sticky="n") |
| self.image_label.grid(row=0, column=0,rowspan=14, padx=10, pady=5, sticky="w") |
| self.hidden_frame.grid(row=0, column=2,rowspan=14, sticky="nsew") |
| self.history_export.grid(row=10, column=1, pady=1, padx=5, sticky="n") |
| self.random_artist_button = customtkinter.CTkCheckBox(self.hidden_frame, text="๋๋ค์๊ฐ ์ถ๊ฐ", variable=self.random_artist_var, font=my_font, state="disabled") |
| self.random_artist_button.grid(row=0, column=0,pady=5, padx=5, sticky="nsew") |
| self.seed_fix_button = customtkinter.CTkCheckBox(self.hidden_frame, text="์๋๊ณ ์ ", variable=self.seed_fix_var, font=my_font) |
| self.seed_fix_button.grid(row=1, column=0,pady=5, padx=5, sticky="nsew") |
| self.seed_entry = customtkinter.CTkEntry(self.hidden_frame, textvariable=self.entry_seed_value) |
| self.seed_entry.grid(row=1, column=1,pady=5, padx=5, sticky="nsew") |
| self.random_artist_manage_button = customtkinter.CTkButton(self.hidden_frame, text="๋๋ค์๊ฐ ๊ด๋ฆฌ", font=my_font, command=self.random_artist_window) |
| self.random_artist_manage_button.grid(row=0, column=1,pady=5, padx=5, sticky="nsew") |
| self.random_resolution_button2 = customtkinter.CTkCheckBox(self.hidden_frame, text="๋๋ค ํด์๋", variable=self.random_resolution_var, font=my_font) |
| self.random_resolution_button2.grid(row=3, column=0,pady=5, padx=5, sticky="nsew") |
| self.resolution_button2 = customtkinter.CTkComboBox(self.hidden_frame, width=160,values=["1024 x 1024", "960 x 1088", "896 x 1152", "832 x 1216", "1088 x 960", "1152 x 896", "1216 x 832"], variable=self.resolution_var, font=my_font) |
| self.resolution_button2.grid(row=3, column=1, padx=5, sticky="w") |
| self.sampler_label2 = customtkinter.CTkLabel(self.hidden_frame, text="Sampler :", font=my_font) |
| self.sampler_label2.grid(row=4, column=0,pady=5, padx=5, sticky="w") |
| self.sampler_button2 = customtkinter.CTkComboBox(self.hidden_frame, width=160,values=["k_euler", "k_euler_ancestral", "k_dpmpp_2s_ancestral", "k_dpmpp_sde", "k_dpmpp_2m"], variable=self.sampler_var, font=my_font) |
| self.sampler_button2.grid(row=4, column=1, columnspan=2, pady=5, padx=5, sticky="e") |
| self.cfg_scale_label2 = customtkinter.CTkLabel(self.hidden_frame, text="CFG Scale : ", font=my_font) |
| self.cfg_scale_label2.grid(row=5, column=0,pady=5, padx=5, sticky="w") |
| self.cfg_scale_entry2 = customtkinter.CTkEntry(self.hidden_frame, width=65, textvariable=self.cfg_scale_var) |
| self.cfg_scale_entry2.grid(row=5, column=1,pady=5, padx=5, sticky="w") |
| self.prompt_guidance_rescale_label2 = customtkinter.CTkLabel(self.hidden_frame, text="P.Guide Rescale :", font=my_font) |
| self.prompt_guidance_rescale_label2.grid(row=6, column=0, pady=5, padx=5, sticky="w") |
| self.prompt_guidance_rescale_entry2 = customtkinter.CTkEntry(self.hidden_frame, width=65, textvariable=self.prompt_guidance_rescale_var) |
| self.prompt_guidance_rescale_entry2.grid(row=6, column=1,pady=5, padx=5, sticky="w") |
| self.sema_button2 = customtkinter.CTkCheckBox(self.hidden_frame, text="SEMA", variable=self.sema_button_var,font=my_font) |
| self.sema_button2.grid(row=7, column=0, pady=5, padx=5, sticky="w") |
| self.dyn_button2 = customtkinter.CTkCheckBox(self.hidden_frame, text="SEMA+DYN", variable=self.dyn_button_var, font=my_font) |
| self.dyn_button2.grid(row=7, column=1, pady=5, padx=5, sticky="w") |
| self.open_save_folder2 = customtkinter.CTkButton(self.hidden_frame, text="ํด๋ ์ด๊ธฐ", font=my_font, fg_color="transparent", command=lambda: open_file_explorer(self), width=80) |
| self.open_save_folder2.grid(row=5, column=1,padx=5, pady=5, sticky="e") |
| self.extra_setting_button2 = customtkinter.CTkButton(self.hidden_frame, width=80, text="๊ธฐํ์ค์ ", font=my_font, fg_color="grey", hover_color="grey5", command=lambda: show_advanced_settings(self)) |
| self.extra_setting_button2.grid(row=6, column=1,pady=5, padx=5, sticky="e") |
| self.recommended_prompt_button2 = customtkinter.CTkButton(self.hidden_frame, text="์ถ์ฒ ํ๋กฌํํธ", font=my_font, command= self.open_prompt_window) |
| self.recommended_prompt_button2.grid(row=9, column=0, columnspan=2, pady=5, padx=5, sticky="nsew") |
| self.wildcard_manager_button2 = customtkinter.CTkButton(self.hidden_frame, text="์์ผ๋์นด๋ ๊ด๋ฆฌ", font=my_font, command=self.open_wildcard_window) |
| self.wildcard_manager_button2.grid(row=10, column=0,columnspan=2, pady=5, padx=5, sticky="nsew") |
| self.character_search_button2 = customtkinter.CTkButton(self.hidden_frame, text="์บ๋ฆญํฐ ๊ฒ์", fg_color="#7030A0", hover_color="#481F67", font=my_font, command= lambda: open_Character_search(self)) |
| self.character_search_button2.grid(row=11, column=0,columnspan=2, pady=5, padx=5, sticky="nsew") |
| self.window_label.grid_forget() |
| self.window_label = customtkinter.CTkLabel(self.hidden_frame, text=str(self.window)+" / "+ str(len(self.image_queue)), font=my_font) |
| self.window_label.grid(row=12, column=0,padx=5, pady=5, sticky="w") |
| self.request_upper_size_button2 = customtkinter.CTkButton(self.hidden_frame, text="img2img (trial)",fg_color="transparent", hover_color="grey10", text_color="#FFFF97", font=my_font, command=lambda: img2img(self), width=110) |
| self.request_upper_size_button2.grid(row=12, column=1,padx=5, pady=5, sticky="nsew") |
| |
| if self.image_queue: |
| current_image = Image.open(self.image_queue[self.window-1][3]) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| self.image_label.configure(image=instant_result_image) |
| else: |
| self.image_label.configure(image=customtkinter.CTkImage(white_image, size=(app.winfo_screenheight()-100, app.winfo_screenheight()-100))) |
| else: |
| self.right_frame.configure(width=768) |
| self.image_label_report.grid_forget() |
| self.image_label.grid_forget() |
| self.image_label_under_frame.grid_forget() |
| self.image_history_sub_frame.grid_forget() |
| self.image_history_button_up.grid_forget() |
| self.image_history_button_down.grid_forget() |
| self.image_label.grid(row=1, column=0, rowspan=14, padx=10, pady=5, sticky="w") |
| self.image_label_under_frame.grid(row=0, column=0, padx=10, pady=5, sticky="ew") |
| self.image_history_sub_frame.grid(row=1, rowspan=12, column=1, sticky="n") |
| self.image_label_report.grid(row=16, column=0, padx=10, pady=5, sticky="ew") |
| self.image_history_button_up.grid(row=0, column=1, pady=10, padx=5, sticky="n") |
| self.image_history_button_down.grid(row=13, column=1, pady=5, padx=5, sticky="n") |
| self.history_export.grid(row=14, column=1, pady=10, padx=5, sticky="n") |
| self.hidden_frame.grid_forget() |
| self.window_label.grid_forget() |
| self.window_label = customtkinter.CTkLabel(self.image_label_under_frame, text=str(self.window)+" / "+ str(len(self.image_queue)), font=my_font) |
| self.window_label.grid(row=0, column=4,padx=5, pady=5, sticky="w") |
|
|
|
|
| if self.image_queue: |
| self.image_label.configure(image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(620, 620))) |
| else: |
| self.image_label.configure(image=customtkinter.CTkImage(white_image, size=(620, 620))) |
| app.last_window_size_conf = datetime.now() |
| self.bind("<Configure>", on_window_resize) |
|
|
| def on_ctrl_press(event): |
| self.control_pressed = True |
| self.search_button.configure(text="์ฌ์ธต ๊ฒ์") |
| self.image_history_button_up.configure(text="โฒ (Top)") |
| self.image_history_button_down.configure(text="โผ (100)") |
| self.request_upper_size_button.configure(text="img2img (INST)", text_color="#FD8853") |
| self.history_export.configure(text=f"export ({len(self.image_queue)})", fg_color="#1F6AA5", hover_color="#144870", text_color="#DCE4EE") |
| if app.state() == 'zoomed': |
| self.request_upper_size_button2.configure(text="img2img (INST)", text_color="#FD8853") |
|
|
| def on_ctrl_release(event): |
| self.control_pressed = False |
| self.search_button.configure(text="๊ฒ์") |
| self.image_history_button_up.configure(text="โฒ") |
| self.image_history_button_down.configure(text="โผ") |
| self.request_upper_size_button.configure(text="img2img (trial)", text_color="#FFFF97") |
| self.history_export.configure(text="+ FAV (โ)", fg_color="#F5F3C2", hover_color="#A69F40", text_color="black") |
| if self.image_queue and len(self.image_queue[self.current_window]) >= 5: |
| self.history_export.configure(state="normal") |
| else: |
| self.history_export.configure(state="disabled") |
| if app.state() == 'zoomed': |
| self.request_upper_size_button2.configure(text="img2img (trial)", text_color="#FFFF97") |
|
|
| def on_escape(event=None): |
| self.focus_set() |
| self.bind('<Escape>', on_escape) |
|
|
| self.control_pressed = False |
| self.bind_all("<Control_L>", on_ctrl_press) |
| self.bind_all("<KeyRelease-Control_L>", on_ctrl_release) |
| self.bind('<Control-c>', copy_file) |
| self.last_window_size = None |
| self.random_artist_select = "global" |
| self.random_artist_list_length = 0 |
| self.protocol("WM_DELETE_WINDOW", lambda: (self.Automation_setting.destroy())) |
| self.wildcard_dict = None |
| self.wildacrds_dir = "" |
|
|
| def on_arrow_key(event): |
| key_pressed = event.keysym |
| |
| if key_pressed == "Up" and self.image_history_button_up.cget("state") == "normal": |
| up_button_pressed() |
| elif key_pressed == "Down" and self.image_history_button_down.cget("state") == "normal": |
| down_button_pressed() |
|
|
| def img2img(self): |
| if self.control_pressed: |
| instant_img2img(self) |
| self.control_pressed = False |
| else: |
| global img2img_window |
| if img2img_window is not None: |
| img2img_window.destroy() |
| img2img_window = None |
| _img2img(self) |
|
|
| def _img2img(self): |
| global iimg, img2img_window, inpaint_mode |
|
|
| def instant_image_generation(self, current_lookup): |
| try: |
| img2img_window.attributes('-topmost', False) |
| except: |
| pass |
| def image_to_base64(image): |
| image_bytesIO = io.BytesIO() |
| img.save(image_bytesIO, format="png") |
| return base64.b64encode(image_bytesIO.getvalue()).decode() |
| def mask_to_base64(image): |
| |
| i = np.array(image) |
| |
| if i.shape[-1] == 3: |
| alpha = np.sum(i, axis=-1) > 0 |
| alpha = np.uint8(alpha * 255) |
| rgba = np.dstack((i, alpha)) |
| elif i.shape[-1] == 4: |
| rgba = i |
| else: |
| raise ValueError("Unsupported image format") |
|
|
| |
| img = Image.fromarray(rgba) |
| |
| |
| image_bytesIO = io.BytesIO() |
| img.save(image_bytesIO, format="png") |
| return base64.b64encode(image_bytesIO.getvalue()).decode() |
| try: |
| i2i_generate_button.configure(state="disabled") |
| except: |
| pass |
| try: |
| request_prompt = i2i_text_input.get("0.0", "end") |
| except: |
| request_prompt = self.text_input.get("0.0", "end") |
| try: |
| request_seed = current_lookup[2] |
| filename = current_lookup[3] |
| except: |
| current_lookup = self.image_queue[self.current_window].copy() |
| request_seed = current_lookup[2] |
| filename = current_lookup[3] |
| img = Image.open(filename) |
| send_image = image_to_base64(img) |
| scale_pre = self.cfg_scale_entry.get() |
| try: |
| steps = request_upper_size_steps.get() |
| except: |
| steps = 28 |
| try: |
| scale_pre = float(scale_pre) |
| except: |
| scale_pre = 5.0 |
| self.cfg_scale_var.set("5.0") |
| rescale_pre = self.prompt_guidance_rescale_entry.get() |
| try: |
| rescale_pre = float(rescale_pre) |
| except: |
| rescale_pre = 0 |
| self.prompt_guidance_rescale_var.set("0") |
| try: |
| steps = int(steps) |
| except: |
| steps = 28 |
| self.cfg_scale_var.set("28") |
| uncond_pre = self.uncond_strength_entry.get() |
| try: |
| uncond_pre = float(uncond_pre) |
| uncond_pre = round(uncond_pre / 0.05) * 0.05 |
| if (uncond_pre > 1.5): |
| uncond_pre = 1.5 |
| self.uncond_strength_entry.delete(0, "end") |
| self.uncond_strength_entry.insert(0, "1.5") |
| except: |
| uncond_pre = 1.0 |
| self.uncond_strength_entry.delete(0, "end") |
| self.uncond_strength_entry.insert(0, "1.0") |
| try: |
| i2i_negative_text = i2i_negative_input.get("0.0", "end-1c") |
| except: |
| i2i_negative_text = self.negative_prompt_input.get("0.0", "end-1c") |
| inpaint_check = sum(sum(row) for row in mirror_image) >= 50 |
| gen_request = { |
| "width":self.i2i_width if not inpaint_check else cwidth , |
| "height":self.i2i_height if not inpaint_check else cheight, |
| "quality_toggle":self.auto_quality_toggle.get(), |
| "seed":request_seed if self.i2i_seed_hold_check == True else random.randint(0,9999999999), |
| "sampler":self.sampler_button.get(), |
| "scale":scale_pre, |
| "uncond_scale":uncond_pre, |
| "sema":self.sema_button.get(), |
| "sema_dyn": self.dyn_button.get(), |
| "cfg_rescale": rescale_pre, |
| "prompt": request_prompt, |
| "negative":i2i_negative_text, |
| "user_screen_size": self.get_max_size(), |
| "start_time": self.start_time, |
| "access_token": self.access_token, |
| "save_folder": self.output_file_path, |
| "png_rule": self.name_var.get(), |
| "type": "upper" if not inpaint_check else "inpaint", |
| "steps": steps if steps <= 50 else 50, |
| "image": send_image |
| } |
| if not inpaint_check: |
| gen_request["strength"] = round(self.i2i_value, 2) |
| gen_request["noise"] = 0 |
| if gen_request["type"] == "inpaint": |
| gen_request["mask"] = mask_to_base64(create_and_save_mask_image()) |
| gen_request["add_original_image"] = True if overlay_original_image_var.get()==1 else False |
| def run_generation(): |
| if gen_request["png_rule"] == "count": |
| self.generation_count += 1 |
| gen_request["count"] =self.generation_count |
| self.state_label.configure(text ="state : img2img ์์ฒญ๋จ ", text_color = "#FFFF97") |
| result_image, result_prompt, result_seed, info, filename = NAIA_generation.generate(gen_request) |
| self.state_label.configure(text ="state : img2img ์์ฒญ ๋ฐํ๋จ", text_color = "#DCE4EE") |
| self.get_anlas() |
| i2i_generate_button.configure(state="normal") |
| if info: |
| temp = info.get('Comment', '') |
| temp = temp[temp.find("prompt")+10:temp.find("skip_cfg_below_sigma")-3].replace('"','') |
| else: |
| temp = result_prompt |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.configure(text_color="#DCE4EE") |
| self.image_label_report.insert("0.0", temp) |
| self.image_label_report.configure(state="disabled") |
| if result_image: |
| if app.state() != 'zoomed': |
| instant_result_image = customtkinter.CTkImage(result_image, size=(620,620)) |
| else: |
| current_image = Image.open(filename) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| self.image_label.configure(image=instant_result_image) |
| img_label_result.grid(row=0, column=1) |
| img_label_result.configure(image=customtkinter.CTkImage(result_image, size=(576, 576))) |
| set_image_to_queue(result_image, result_prompt, str(result_seed), filename) |
| generation_thread = threading.Thread(target=run_generation, daemon=True) |
| generation_thread.start() |
| |
| def find_max_resolution(width, height, max_pixels=2166784, multiple_of=64): |
| ratio = width / height |
|
|
| max_width = int((max_pixels * ratio)**0.5) |
| max_height = int((max_pixels / ratio)**0.5) |
|
|
| max_width = (max_width // multiple_of) * multiple_of |
| max_height = (max_height // multiple_of) * multiple_of |
|
|
| while max_width * max_height > max_pixels: |
| max_width -= multiple_of |
| max_height = int(max_width / ratio) |
| max_height = (max_height // multiple_of) * multiple_of |
|
|
| return (max_width, max_height) |
|
|
| img2img_window = customtkinter.CTkToplevel() |
| img2img_window.title("img2img") |
| img2img_window.attributes('-topmost', True) |
| img2img_window.resizable(width=False, height=False) |
|
|
| current_lookup = copy.deepcopy(self.image_queue[self.current_window]) |
| filename = current_lookup[3] |
| img = Image.open(filename) |
| width, height = img.size |
|
|
| img2img_window_left = customtkinter.CTkFrame(img2img_window) |
| img2img_window_left.grid(row=0, column=0, pady=5, padx=5, sticky="nsew") |
| img2img_window_right = customtkinter.CTkFrame(img2img_window, width=586) |
| img2img_window_right.grid(row=0, column=1, pady=5, padx=5, sticky="nsew") |
| |
| |
|
|
| iimg = None |
| inpaint_mode = False |
| iimg = Image.open(filename).convert("RGB") |
| original_image = iimg.copy() |
| cwidth, cheight = iimg.size |
|
|
| |
| mirror_width = width // 8 |
| mirror_height = height // 8 |
| mirror_image = [[0 for _ in range(mirror_height)] for _ in range(mirror_width)] |
|
|
| img_label = customtkinter.CTkCanvas(img2img_window_left, width=cwidth, height=cheight) |
| img_label.grid(row=0, column=0) |
| img_label_result = customtkinter.CTkLabel(img2img_window_left, text="") |
|
|
| |
| def create_and_save_mask_image(): |
| |
| mask_image = Image.new('RGBA', (mirror_width, mirror_height), (0, 0, 0, 0)) |
| draw = ImageDraw.Draw(mask_image) |
| |
| |
| for x in range(mirror_width): |
| for y in range(mirror_height): |
| if mirror_image[x][y] > 0: |
| draw.point((x, y), fill=(255, 255, 255, 255)) |
| |
| |
| return mask_image |
|
|
| tk_image = ImageTk.PhotoImage(iimg) |
| canvas = img_label |
| canvas.create_image(0, 0, anchor="nw", image=tk_image) |
| canvas.image = tk_image |
|
|
| def create_custom_cursor(): |
| |
| cursor_image = Image.new("RGBA", (40, 40), (0, 0, 0, 0)) |
| draw = ImageDraw.Draw(cursor_image) |
| |
| draw.rectangle([0, 0, 39, 39], outline=(255, 255, 255, 255)) |
| return cursor_image |
|
|
| |
| def on_mouse_move(event): |
| |
| canvas.delete("custom_cursor") |
| |
| cursor_image = create_custom_cursor() |
| tk_cursor_image = ImageTk.PhotoImage(cursor_image) |
| |
| canvas.create_image(event.x, event.y, image=tk_cursor_image, anchor="center", tags="custom_cursor") |
| |
| canvas.cursor_image = tk_cursor_image |
|
|
| |
| canvas.bind("<Motion>", on_mouse_move) |
|
|
| def inpaint_check(): |
| global inpaint_mode |
| check = sum(sum(row) for row in mirror_image) >= 50 |
| if(check and not inpaint_mode): |
| inpaint_mode = True |
| i2i_slider_label.grid_forget() |
| i2i_slider.grid_forget() |
| i2i_resolution_label.grid_forget() |
| i2i_radio_frame.grid_forget() |
| overlay_original_image.grid(row=0, column=0, padx=30, sticky="n") |
| cancel_inpainting_button.grid(row=0, column=1, padx=30, sticky="n") |
| elif(inpaint_mode and not check): |
| inpaint_mode = False |
| overlay_original_image.grid_forget() |
| cancel_inpainting_button.grid_forget() |
| i2i_slider_label.grid(row=0, column=0, sticky="nsew") |
| i2i_slider.grid(row=0, column=1, sticky="nsew") |
| i2i_resolution_label.grid(row=1, column=0, sticky="nsew") |
| i2i_radio_frame.grid(row=1, column=1, pady=5, sticky="nsew") |
|
|
|
|
| def update_canvas_image(): |
| global tk_image |
| tk_image = ImageTk.PhotoImage(iimg) |
| canvas.create_image(0, 0, anchor="nw", image=tk_image, tags="original_image") |
| canvas.image = tk_image |
|
|
| def on_canvas_click_or_move(event): |
| x, y = event.x // 8, event.y // 8 |
| paint_on_image(x, y) |
| update_canvas_image() |
|
|
| def paint_on_image(x, y): |
| draw = ImageDraw.Draw(iimg, 'RGBA') |
| for i in range(max(0, x-2), min(mirror_width, x+3)): |
| for j in range(max(0, y-2), min(mirror_height, y+3)): |
| mirror_image[i][j] += 1 |
| if mirror_image[i][j] == 1: |
| draw.rectangle([(i*8, j*8), ((i+1)*8-1, (j+1)*8-1)], fill=(0, 0, 255, 128)) |
| inpaint_check() |
|
|
| def remove_paint_from_image(x, y): |
| global iimg |
| for i in range(max(0, x-2), min(mirror_width, x+3)): |
| for j in range(max(0, y-2), min(mirror_height, y+3)): |
| mirror_image[i][j] = 0 |
| |
| |
| original_pixels = original_image.load() |
| current_pixels = iimg.load() |
| |
| for xi in range(i*8, min((i+1)*8, width)): |
| for yi in range(j*8, min((j+1)*8, height)): |
| |
| current_pixels[xi, yi] = original_pixels[xi, yi] |
| inpaint_check() |
|
|
| def on_canvas_right_click_or_move(event): |
| |
| x, y = event.x // 8, event.y // 8 |
| |
| remove_paint_from_image(x, y) |
| |
| update_canvas_image() |
|
|
| |
| canvas.bind("<Button-3>", on_canvas_right_click_or_move) |
| canvas.bind("<B3-Motion>", on_canvas_right_click_or_move) |
| canvas.bind("<Button-1>", on_canvas_click_or_move) |
| canvas.bind("<B1-Motion>", on_canvas_click_or_move) |
|
|
| highlight_tags = ["aqua","black","blonde","blue","brown","cyan","green","grey","orange","pink","purple","red","violet","white","yellow","mouth", "eyes", " cap", " ears", " girl", " ornament", " hat", "beret", " ear "] |
|
|
| def insert_with_color(): |
| words = [word.strip() for word in current_lookup[1].split(',')] |
| for index, word in enumerate(words): |
| highlight = False |
| start_index = i2i_text_input.index("end-1c") |
| if index == len(words) - 1: |
| i2i_text_input.insert("end", word) |
| else: |
| i2i_text_input.insert("end", word + ", ") |
| end_index = i2i_text_input.index("end-1c") |
|
|
| for tag in highlight_tags: |
| if tag in word: |
| highlight = True |
| if word == "hat": highlight = True |
| |
| if word in cd.character_dictionary or word in tagbag.bag_of_tags[5:] or 'tail' in word or 'pupil' in word: |
| i2i_text_input.tag_add(word, start_index, end_index) |
| i2i_text_input.tag_config(word, foreground="#FFFF97") |
| elif highlight: |
| i2i_text_input.tag_add(word, start_index, end_index) |
| i2i_text_input.tag_config(word, foreground="#AED395") |
|
|
| i2i_text_input_label = customtkinter.CTkLabel(img2img_window_right, text=" ------------------- i2i ํ๋กฌํํธ ------------------- ", font=large_font) |
| i2i_text_input_label.grid(row = 0, sticky="w" ) |
| text_hyperlink = customtkinter.CTkLabel(img2img_window_right, text="๊ฐ์ด๋ ์ด๊ธฐ (arca.live)", font=customtkinter.CTkFont('Pretendard', 13), width=180, text_color="lightblue", cursor="hand2", state="disabled") |
| text_hyperlink.grid(row=0, column=0, padx=5, pady=5, sticky="e") |
| |
| i2i_text_input = customtkinter.CTkTextbox(img2img_window_right, width=490, height=240,font=v_large_font) |
| i2i_text_input.grid(row=1, column=0, pady=5, sticky="nsew") |
| insert_with_color() |
|
|
| def slider_event(value): |
| i2i_slider_label.configure(text=f"Strength : {value:.2f}") |
| self.i2i_value = value |
|
|
| def get_i2i_neagtive(filename): |
| with Image.open(filename) as img: |
| info = img.info.get('Comment', '') |
| start_pattern = '"uc": "' |
| end_pattern = '", "' |
| start_index = info.find(start_pattern) |
| if start_index != -1: |
| |
| start_index += len(start_pattern) |
| |
| |
| end_index = info.find(end_pattern, start_index) |
| if end_index != -1: |
| |
| extracted_string = info[start_index:end_index] |
| else: |
| if app.current_window in app.import_image_negative: |
| extracted_string = app.import_image_negative[app.current_window] |
| else: |
| extracted_string = "" |
| return extracted_string |
|
|
| i2i_offset_input_label = customtkinter.CTkLabel(img2img_window_right, text=" ------------------- i2i ๋ค๊ฑฐํฐ๋ธ ------------------- ", font=large_font) |
| i2i_offset_input_label.grid(row = 2, column=0,sticky="w" ) |
| i2i_character_search_button = customtkinter.CTkButton(img2img_window_right, text="์บ๋ฆญํฐ ๊ฒ์", fg_color="#7030A0", hover_color="#481F67", font=my_font, command= lambda: open_Character_search(self)) |
| i2i_character_search_button.grid(row=2, column=0, pady=5, padx=5, sticky="e") |
| i2i_negative_input = customtkinter.CTkTextbox(img2img_window_right, width=490, height=115,font=v_large_font) |
| i2i_negative_input.grid(row=3, column=0, pady=5, sticky="nsew") |
| i2i_negative = get_i2i_neagtive(filename) |
| i2i_negative_input.insert("0.0", i2i_negative) |
|
|
| i2i_slider_frame = customtkinter.CTkFrame(img2img_window_right, fg_color="#333333") |
| i2i_slider_frame.grid(row=4, column=0, pady=5, padx=5, sticky="nsew") |
|
|
| i2i_value = round(self.i2i_value, 2) |
| i2i_seed_hold_check = 1 if self.i2i_seed_hold_check else 0 |
| i2i_slider_label = customtkinter.CTkLabel(i2i_slider_frame, text="Strength : "+str(i2i_value), font=large_font, width=140) |
| i2i_slider_label.grid(row=0, column=0, sticky="nsew") |
| i2i_slider = customtkinter.CTkSlider(i2i_slider_frame, from_=0, to=0.99, number_of_steps=100, command=slider_event, width=340) |
| i2i_slider.grid(row=0, column=1, sticky="nsew") |
| i2i_slider.set(i2i_value) |
| i2i_resolution_label = customtkinter.CTkLabel(i2i_slider_frame, text="ํด์๋ ๋ฐฐ์จ", font=large_font, width=140) |
| i2i_resolution_label.grid(row=1, column=0, sticky="nsew") |
|
|
| def restore_original_image(): |
| global tk_image, iimg |
| iimg = Image.open(filename).convert("RGB") |
| tk_image = ImageTk.PhotoImage(iimg.copy()) |
| canvas.create_image(0, 0, anchor="nw", image=tk_image) |
| canvas.image = tk_image |
| for i in range(len(mirror_image)): |
| for j in range(len(mirror_image[i])): |
| mirror_image[i][j] = 0 |
| inpaint_check() |
|
|
| overlay_original_image_var = customtkinter.IntVar() |
| overlay_original_image = customtkinter.CTkCheckBox(i2i_slider_frame, text="Overlay Original Image", font=large_font, width=140, variable=overlay_original_image_var) |
| cancel_inpainting_button = customtkinter.CTkButton(i2i_slider_frame, text="์ธํ์ธํธ ์ทจ์", font=large_font, command=restore_original_image) |
|
|
| i2i_radio_frame = customtkinter.CTkFrame(i2i_slider_frame, fg_color="#333333") |
| i2i_radio_frame.grid(row=1, column=1, pady=5, sticky="nsew") |
|
|
| radio_var = customtkinter.IntVar(value=self.i2i_resolution) |
| i2i_request_width = customtkinter.IntVar(value=width) |
| i2i_request_height = customtkinter.IntVar(value=height) |
| img2img_window.after(2500, lambda: img2img_window.attributes('-topmost', False)) |
|
|
|
|
|
|
| def radiobutton_event(): |
| v = radio_var.get() |
| if v == 0: |
| request_width = width |
| request_height = height |
| self.i2i_resolution = 0 |
| elif v == 1: |
| request_width, request_height = find_max_resolution(width, height, 2166784) |
| self.i2i_resolution = 1 |
| elif v == 2: |
| request_width, request_height = find_max_resolution(width, height, 2662400) |
| self.i2i_resolution = 2 |
| i2i_resolution_label.configure(text=f"{request_width} x {request_height}") |
| i2i_request_width.set(request_width) |
| self.i2i_width = request_width |
| i2i_request_height.set(request_height) |
| self.i2i_height = request_height |
|
|
| radio_button_1 = customtkinter.CTkRadioButton(i2i_radio_frame, text="ํด์๋ ์ ์ง", command=radiobutton_event, variable= radio_var, value=0, font=large_font) |
| radio_button_1.grid(row=0, column=0, sticky="nsew") |
| radio_button_2 = customtkinter.CTkRadioButton(i2i_radio_frame, text="2.16 MP", command=radiobutton_event, variable= radio_var, value=1, font=large_font) |
| radio_button_2.grid(row=0, column=1, sticky="nsew") |
| radio_button_3 = customtkinter.CTkRadioButton(i2i_radio_frame, text="2.66 MP", command=radiobutton_event, variable= radio_var, value=2, font=large_font) |
| radio_button_3.grid(row=0, column=2, sticky="nsew") |
| radiobutton_event() |
|
|
| i2i_generate_frame = customtkinter.CTkFrame(img2img_window_right, fg_color="#2B2B2B") |
| i2i_generate_frame.grid(row=5, column=0, pady=5, sticky="n") |
|
|
| i2i_offset_input_label2 = customtkinter.CTkLabel(i2i_generate_frame, text=" ํด์๋์ Strength์ ๋ฐ๋ผ Anlas๊ฐ ์๋ชจ๋ฉ๋๋ค ", font=large_font, text_color="#FFFF97") |
| i2i_offset_input_label2.grid(row = 0, column=0, columnspan=3, sticky="n" ) |
|
|
| def seed_button_event(): |
| self.i2i_seed_hold_check = True if request_upper_size_seed_hold.get() == 1 else False |
|
|
| request_upper_size_seed_hold_var = customtkinter.IntVar(value=i2i_seed_hold_check) |
| request_upper_size_seed_hold = customtkinter.CTkCheckBox(i2i_generate_frame, text="์๋๊ณ ์ Step : ", font=my_font, variable=request_upper_size_seed_hold_var, command=seed_button_event) |
| request_upper_size_seed_hold.grid(row=1, column=0, padx=5, pady=5, sticky="nsew") |
| request_upper_size_seed_hold.select() if i2i_seed_hold_check == 1 else request_upper_size_seed_hold.deselect() |
| request_upper_size_steps = customtkinter.StringVar(value="28") |
| request_upper_size_steps_entry = customtkinter.CTkEntry(i2i_generate_frame, font=my_font, textvariable=request_upper_size_steps, width=60) |
| request_upper_size_steps_entry.grid(row=1, column=1,padx=5, pady=5, sticky="nsew") |
| i2i_generate_button = customtkinter.CTkButton(i2i_generate_frame, text="Request I2I", font=large_font, command=lambda: instant_image_generation(self, current_lookup)) |
| i2i_generate_button.grid(row = 1, column= 2, pady=5, sticky="n" ) |
| |
| def instant_img2img(self): |
| current_lookup = self.image_queue[self.current_window].copy() |
| def instant_image_generation(self): |
| def image_to_base64(image): |
| image_bytesIO = io.BytesIO() |
| img.save(image_bytesIO, format="png") |
| return base64.b64encode(image_bytesIO.getvalue()).decode() |
| request_prompt = current_lookup[1] |
| request_seed = current_lookup[2] |
| filename = current_lookup[3] |
| img = Image.open(filename) |
| send_image = image_to_base64(img) |
| scale_pre = self.cfg_scale_entry.get() |
| steps = 28 |
| try: |
| scale_pre = float(scale_pre) |
| except: |
| scale_pre = 5.0 |
| self.cfg_scale_var.set("5.0") |
| rescale_pre = self.prompt_guidance_rescale_entry.get() |
| try: |
| rescale_pre = float(rescale_pre) |
| except: |
| rescale_pre = 0 |
| self.prompt_guidance_rescale_var.set("0") |
| try: |
| steps = int(steps) |
| except: |
| steps = 28 |
| |
| uncond_pre = self.uncond_strength_entry.get() |
| try: |
| uncond_pre = float(uncond_pre) |
| uncond_pre = round(uncond_pre / 0.05) * 0.05 |
| if (uncond_pre > 1.5): |
| uncond_pre = 1.5 |
| self.uncond_strength_entry.delete(0, "end") |
| self.uncond_strength_entry.insert(0, "1.5") |
| except: |
| uncond_pre = 1.0 |
| self.uncond_strength_entry.delete(0, "end") |
| self.uncond_strength_entry.insert(0, "1.0") |
| i2i_negative_text = self.negative_prompt_input.get("0.0", "end-1c") |
| gen_request = { |
| "width":self.i2i_width, |
| "height":self.i2i_height, |
| "quality_toggle":self.auto_quality_toggle.get(), |
| "seed":request_seed if self.i2i_seed_hold_check == True else random.randint(0,9999999999), |
| "sampler":self.sampler_button.get(), |
| "scale":scale_pre, |
| "uncond_scale":uncond_pre, |
| "sema":self.sema_button.get(), |
| "sema_dyn": self.dyn_button.get(), |
| "cfg_rescale": rescale_pre, |
| "prompt": request_prompt, |
| "negative":i2i_negative_text, |
| "user_screen_size": self.get_max_size(), |
| "start_time": self.start_time, |
| "access_token": self.access_token, |
| "save_folder": self.output_file_path, |
| "png_rule": self.name_var.get(), |
| "type": "upper", |
| "steps": steps if steps <= 50 else 50, |
| "image": send_image, |
| "strength": round(self.i2i_value, 2), |
| "noise": 0 |
| } |
| def run_generation(): |
| if gen_request["png_rule"] == "count": |
| self.generation_count += 1 |
| gen_request["count"] =self.generation_count |
| self.state_label.configure(text ="state : img2img ์์ฒญ๋จ ", text_color = "#FFFF97") |
| result_image, result_prompt, result_seed, info, filename = NAIA_generation.generate(gen_request) |
| self.state_label.configure(text ="state : img2img ์์ฒญ ๋ฐํ๋จ", text_color = "#DCE4EE") |
| self.get_anlas() |
| if info: |
| temp = info.get('Comment', '') |
| temp = temp[temp.find("prompt")+10:temp.find("skip_cfg_below_sigma")-3].replace('"','') |
| else: |
| temp = result_prompt |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.configure(text_color="#DCE4EE") |
| self.image_label_report.insert("0.0", temp) |
| self.image_label_report.configure(state="disabled") |
| if result_image: |
| if app.state() != 'zoomed': |
| instant_result_image = customtkinter.CTkImage(result_image, size=(620,620)) |
| else: |
| current_image = Image.open(filename) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| self.image_label.configure(image=instant_result_image) |
| set_image_to_queue(result_image, result_prompt, str(result_seed), filename) |
| generation_thread = threading.Thread(target=run_generation, daemon=True) |
| generation_thread.start() |
| |
| def find_max_resolution(width, height, max_pixels=2166784, multiple_of=64): |
| ratio = width / height |
|
|
| max_width = int((max_pixels * ratio)**0.5) |
| max_height = int((max_pixels / ratio)**0.5) |
|
|
| max_width = (max_width // multiple_of) * multiple_of |
| max_height = (max_height // multiple_of) * multiple_of |
|
|
| while max_width * max_height > max_pixels: |
| max_width -= multiple_of |
| max_height = int(max_width / ratio) |
| max_height = (max_height // multiple_of) * multiple_of |
|
|
| return (max_width, max_height) |
|
|
|
|
| img = Image.open(current_lookup[3]) |
| width, height = img.size |
| v = self.i2i_resolution |
| if v == 0: |
| self.i2i_width = width |
| self.i2i_height = height |
| elif v == 1: |
| self.i2i_width, self.i2i_height = find_max_resolution(width, height, 2166784) |
| elif v == 2: |
| self.i2i_width, self.i2i_height = find_max_resolution(width, height, 2662400) |
| instant_image_generation(self) |
|
|
| self.bind("<Up>", on_arrow_key) |
| self.bind("<Down>", on_arrow_key) |
| self.image_generation_repeat = 1 |
| self.image_generation_repeat_flag = False |
| self.image_generation_repeat_current = 0 |
| self.wildcard_preopen_repeat = 1 |
| self.wildcard_preopen_repeat_flag = False |
| self.wildcard_preopen_repeat_current = 1 |
| self.previous_fix_prompt = None |
| self.i2i_value = 0.7 |
| self.i2i_seed_hold_check = True |
| self.i2i_resolution = 0 |
| self.i2i_width = 0 |
| self.i2i_height = 0 |
| self.i2i_steps = 28 |
| self.image_label.drop_target_register(DND_ALL) |
| self.image_label.dnd_bind("<<Drop>>", self.on_drop) |
| |
| |
| |
| def on_drop(self, event): |
| def read_info_from_image_stealth(image): |
| |
| width, height = image.size |
| pixels = image.load() |
|
|
| if image.mode != 'RGBA': |
| return {} |
|
|
| binary_data = '' |
| buffer_a = '' |
| index_a = 0 |
| confirming_signature = True |
| reading_param_len = False |
| reading_param = False |
| read_end = False |
|
|
| for x in range(width): |
| for y in range(height): |
|
|
| r, g, b, a = pixels[x, y] |
| buffer_a += str(a & 1) |
| index_a += 1 |
| |
| if confirming_signature and x == 0: |
| if y == 119: |
| decoded_sig = bytearray(int(buffer_a[i:i + 8], 2) for i in |
| range(0, len(buffer_a), 8)).decode('utf-8', errors='ignore') |
| if decoded_sig == 'stealth_pngcomp': |
| confirming_signature = False |
| reading_param_len = True |
| buffer_a = '' |
| index_a = 0 |
|
|
| elif reading_param_len: |
| if y == 151: |
| param_len = int(buffer_a, 2) |
| reading_param_len = False |
| reading_param = True |
| buffer_a = '' |
| index_a = 0 |
|
|
| elif reading_param: |
| if index_a == param_len: |
| binary_data = buffer_a |
| read_end = True |
| break |
| else: |
| read_end = True |
| break |
|
|
| if read_end: |
| break |
|
|
| if binary_data != '': |
| |
| byte_data = bytearray(int(binary_data[i:i + 8], 2) |
| for i in range(0, len(binary_data), 8)) |
| try: |
| decoded_data = gzip.decompress( |
| bytes(byte_data)).decode('utf-8') |
| return decoded_data |
| except Exception as e: |
| print(e) |
| pass |
|
|
| return {} |
|
|
| def download_image_from_url(url): |
| response = requests.get(url) |
| response.raise_for_status() |
| return Image.open(io.BytesIO(response.content)) |
| |
| def extract_keywords_from_image(image): |
| if image.info and image.info.get('dpi') is None: |
| result = image.info.get('Comment', '') |
| if result == '': |
| result = image.info.get('parameters', '') |
| if result != '': |
| try: |
| result = json.loads(result) |
| except: |
| result = image.info.get('parameters', '') |
| return result |
| else: |
| self.image_label_report.configure(state="normal", text_color = "#FFFF97") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<No Image Data Error> : ์ด๋ฏธ์ง์์ NAI ์ ๋ณด๋ฅผ ์ป๋๋ฐ ์คํจํ์์ต๋๋ค. ์ด๋ฏธ์ง์ ๋ฉํ๋ฐ์ดํฐ๊ฐ ์์ต๋๋ค.") |
| self.image_label_report.configure(state="disabled") |
| else: |
| return json.loads(result) |
| else: |
| stealth_info = read_info_from_image_stealth(image) |
| if isinstance(stealth_info, str): |
| |
| try: |
| result = stealth_info.replace("\\", "") |
| comment_data = json.loads(result) |
| return comment_data |
| except: |
| return stealth_info.replace("\\", "") |
| elif stealth_info and 'Comment' in stealth_info: |
| try: |
| |
| comment_data = json.loads(stealth_info['Comment']) |
| |
| if 'prompt' in comment_data: |
| return comment_data["prompt"].strip() |
| else: |
| self.image_label_report.configure(state="normal", text_color = "#FFFF97") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<Unsupported Image Type Error> : ์ด๋ฏธ์ง์์ NAI ์ ๋ณด๋ฅผ ์ป๋๋ฐ ์คํจํ์์ต๋๋ค. comment_data๊ฐ ์กด์ฌํ์ง๋ง NAI์ ํธํ๋๋ Stable Diffusion ์ด๋ฏธ์ง๊ฐ ์๋๋๋ค.") |
| self.image_label_report.configure(state="disabled") |
| except json.JSONDecodeError: |
| self.image_label_report.configure(state="normal", text_color = "#FFFF97") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<JSONDecodeError> : ์ด๋ฏธ์ง์์ NAI ์ ๋ณด๋ฅผ ์ป๋๋ฐ ์คํจํ์์ต๋๋ค.") |
| self.image_label_report.configure(state="disabled") |
| else: |
| self.image_label_report.configure(state="normal", text_color = "#FFFF97") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<No Image Data Error> : ์ด๋ฏธ์ง์์ NAI ์ ๋ณด๋ฅผ ์ป๋๋ฐ ์คํจํ์์ต๋๋ค. ์ด๋ฏธ์ง์ ๋ฉํ๋ฐ์ดํฐ๊ฐ ์์ต๋๋ค.") |
| self.image_label_report.configure(state="disabled") |
|
|
| file_path_or_url = event.data |
| file_path_or_url = file_path_or_url.strip("{}") |
| from_url = False |
| try: |
| |
| if file_path_or_url.startswith(('http://', 'https://')): |
| from_url = True |
| img = download_image_from_url(file_path_or_url) |
| else: |
| if file_path_or_url.startswith(('blob:')): |
| self.image_label_report.configure(state="normal", text_color = "#FFFF97") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", "<NAI Blob Image Error> : NAI ํํ์ด์ง์์ ์์ฑํ ์ด๋ฏธ์ง๋ NAIA๋ก ์ฆ์ ๋ณต์ฌํ ์ ์์ต๋๋ค. ์ด๋ฏธ์ง๋ฅผ ๋ค์ด๋ก๋ ํ ๋ค, ๋๋๊ทธ & ๋๋์ ํตํด NAIA์ ์ ๋ฌ ํด ์ฃผ์ธ์.") |
| self.image_label_report.configure(state="disabled") |
| return |
| try: |
| img = Image.open(file_path_or_url) |
| except: |
| try: |
| file_path_or_url = file_path_or_url.replace("\\", '') |
| img = Image.open(file_path_or_url) |
| except: |
| try: |
| file_path_or_url = file_path_or_url.replace("\\", '/') |
| img = Image.open(file_path_or_url) |
| except: |
| self.image_label_report.configure(state="normal", text_color = "#FFFF97") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", f"<Unhandled exception> : Tkdnd2๋ก๋ถํฐ ์ ํจํ์ง ์์ file path ๋๋ url : '{file_path_or_url}' ๋ฅผ ์ ๋ฌ ๋ฐ์์ต๋๋ค.") |
| self.image_label_report.configure(state="disabled") |
| return |
|
|
| |
| description = extract_keywords_from_image(img) |
| if description == '' or description == None: |
| return |
| elif type(description) == type(' '): |
| text_type, contents = 'str', description |
| if '"Software": "NovelAI"' in contents: |
| software = 'NAI' |
| else: |
| software = 'Local' |
| else: |
| text_type, contents = 'dict', description |
| if 'sm' in contents: |
| software = 'NAI' |
| else: |
| software = 'Local' |
|
|
| |
| import_window = customtkinter.CTkToplevel() |
| import_window.title("Image with Metadata found!") |
| import_window.attributes('-topmost', True) |
| import_window.resizable(width=False, height=False) |
|
|
| import_window_left = customtkinter.CTkFrame(import_window) |
| import_window_left.grid(row=0, column=0, pady=5, padx=5, sticky="nsew") |
| import_window_right = customtkinter.CTkFrame(import_window, width=596) |
| import_window_right.grid(row=0, column=1, pady=5, padx=5, sticky="nsew") |
|
|
| width, height = img.size |
| new_size = max(width, height) |
| new_image = Image.new("RGB", (new_size, new_size), 'black') |
| left = (new_size - width) // 2 |
| top = (new_size - height) // 2 |
| new_image.paste(img, (left, top)) |
| |
| import_window_img = customtkinter.CTkImage(new_image, size=(576, 576)) |
| import_window_img_label = customtkinter.CTkLabel(import_window_left, text="", image=import_window_img) |
| import_window_img_label.grid(row=0, column=0) |
|
|
| if software == 'Local': |
| local_prompt = customtkinter.CTkTextbox(import_window_right, height=576, width=586, font=customtkinter.CTkFont('Pretendard', 14)) |
| local_prompt.grid(row=0, column=0, pady=5, padx=5, sticky="nsew") |
| contents = 'Image Type : Local Generated \n\n' + str(contents).replace('\n', '\n\n') |
| local_prompt.insert("0.0", contents) |
| else: |
| if text_type == 'str': |
| try: |
| start_marker = '"Comment": "' |
| end_marker = '", "request_type"' |
| start_index = contents.find(start_marker) + len(start_marker) |
| end_index = contents.find(end_marker, start_index) |
| text = contents[start_index:end_index] |
| contents = {} |
| except: |
| local_prompt = customtkinter.CTkTextbox(import_window_right, height=766, width=766, font=customtkinter.CTkFont('Pretendard', 15)) |
| local_prompt.grid(row=0, column=0, pady=5, padx=5, sticky="nsew") |
| contents = 'Image Type : Local Generated \n\n' + str(contents).replace('\n', '\n\n') |
| local_prompt.insert("0.0", contents) |
| try: |
| start_marker = 'prompt": "' |
| end_marker = '", "steps' |
| start_index = text.find(start_marker) + len(start_marker) |
| end_index = text.find(end_marker, start_index) |
| contents['prompt'] = text[start_index:end_index] |
| text = text[end_index:] |
| except: |
| pass |
| try: |
| start_marker = '"uc": "' |
| end_marker = '", "request_type' |
| start_index = text.find(start_marker) + len(start_marker) |
| end_index = text.find(end_marker, start_index) |
| contents['uc'] = text[start_index:] |
| text = text[:start_index] |
| except: |
| pass |
| keys = ["steps", "scale", "uncond_scale", "cfg_rescale", "seed", "sampler", "sm", "sm_dyn"] |
| for key in keys: |
| |
| pattern = r'"{}":\s*([^,}}]+)'.format(key) |
| match = re.search(pattern, text) |
| |
| if match: |
| |
| value = match.group(1) |
| |
| |
| try: |
| |
| if '.' in value: |
| contents[key] = float(value) |
| else: |
| contents[key] = int(value) |
| except ValueError: |
| |
| if value.lower() == "true": |
| contents[key] = True |
| elif value.lower() == "false": |
| contents[key] = False |
| else: |
| |
| contents[key] = value.strip('"') |
|
|
| my_font = customtkinter.CTkFont('Pretendard', 13) |
| large_font = customtkinter.CTkFont('Pretendard', 14) |
| NAI_prompt_label = customtkinter.CTkLabel(import_window_right, width=586, font=my_font, text=" -------------------- ํ๋กฌํํธ -------------------- ") |
| NAI_prompt_label.grid(row=0, column=0, padx=5, sticky="nsew") |
| NAI_prompt = customtkinter.CTkTextbox(import_window_right, width=586, height=205, font=large_font) |
| NAI_prompt.grid(row=1, column=0, pady=5, padx=5, sticky="nsew") |
| NAI_negative_label = customtkinter.CTkLabel(import_window_right, width=586, font=my_font, text=" --------------- ๋ค๊ฑฐํฐ๋ธ ํ๋กฌํํธ --------------- ") |
| NAI_negative_label.grid(row=2, column=0, padx=5, sticky="nsew") |
| NAI_negative = customtkinter.CTkTextbox(import_window_right, width=586,height=130, font=large_font) |
| NAI_negative.grid(row=3, column=0, pady=5, padx=5, sticky="nsew") |
| NAI_prompt.insert("0.0", contents['prompt']) |
| NAI_prompt.configure(state="disabled") |
| NAI_negative.insert("0.0", contents['uc']) |
| NAI_negative.configure(state="disabled") |
|
|
| import_frame1 = customtkinter.CTkFrame(import_window_right, width=586, fg_color="#2B2B2B") |
| import_frame1.grid(row=4, column=0, pady=5, padx=5, sticky="nsew") |
|
|
| i_steps_label = customtkinter.CTkLabel(import_frame1, font=my_font, text="Steps : ") |
| i_steps_label.grid(row=0, column=0, padx=5, sticky="nsew") |
| i_steps = customtkinter.CTkEntry(import_frame1, font=my_font, width=35) |
| i_steps.grid(row=0, column=1, padx=5, sticky="nsew") |
| i_steps.insert(0, str(contents['steps'])) |
| i_steps.configure(state="disabled") |
|
|
| i_scale_label = customtkinter.CTkLabel(import_frame1, font=my_font, text="CFG Scale : ") |
| i_scale_label.grid(row=0, column=2, padx=5, sticky="nsew") |
| i_scale = customtkinter.CTkEntry(import_frame1, font=my_font, width=40) |
| i_scale.grid(row=0, column=3, padx=5, sticky="nsew") |
| i_scale.insert(0, str(contents['scale'])) |
| i_scale.configure(state="disabled") |
|
|
| i_uncond_label = customtkinter.CTkLabel(import_frame1, font=my_font, text="UC Strength : ") |
| i_uncond_label.grid(row=0, column=4, padx=5, sticky="nsew") |
| i_uncond = customtkinter.CTkEntry(import_frame1, font=my_font, width=40) |
| i_uncond.grid(row=0, column=5, padx=5, sticky="nsew") |
| i_uncond.insert(0, str(contents['uncond_scale'])) |
| i_uncond.configure(state="disabled") |
|
|
| i_rescale_label = customtkinter.CTkLabel(import_frame1, font=my_font, text="CFG Rescale : ") |
| i_rescale_label.grid(row=0, column=6, padx=5, sticky="nsew") |
| i_rescale = customtkinter.CTkEntry(import_frame1, font=my_font, width=40) |
| i_rescale.grid(row=0, column=7, padx=5, sticky="nsew") |
| i_rescale.insert(0, str(contents['cfg_rescale'])) |
| i_rescale.configure(state="disabled") |
|
|
| import_frame2 = customtkinter.CTkFrame(import_window_right, width=586, fg_color="#2B2B2B") |
| import_frame2.grid(row=5, column=0, pady=5, padx=5, sticky="nsew") |
|
|
| i_seed_label = customtkinter.CTkLabel(import_frame2, font=my_font, text="Steps : ") |
| i_seed_label.grid(row=0, column=0, padx=5, sticky="nsew") |
| i_seed = customtkinter.CTkEntry(import_frame2, font=my_font, width=100) |
| i_seed.grid(row=0, column=1, padx=5, sticky="nsew") |
| i_seed.insert(0, str(contents['seed'])) |
|
|
| i_sampler_label = customtkinter.CTkLabel(import_frame2, font=my_font, text="Steps : ") |
| i_sampler_label.grid(row=0, column=2, padx=5, sticky="nsew") |
| i_sampler = customtkinter.CTkEntry(import_frame2, font=my_font, width=100) |
| i_sampler.grid(row=0, column=3, padx=5, sticky="nsew") |
| i_sampler.insert(0, str(contents['sampler'])) |
|
|
| i_smea = customtkinter.CTkCheckBox(import_frame2, font=my_font, text="SMEA") |
| i_smea.grid(row=0, column=4, padx=5, sticky="nsew") |
| if contents['sm'] == True: |
| i_smea.select() |
| else: |
| i_smea.deselect() |
| i_dyn = customtkinter.CTkCheckBox(import_frame2, font=my_font, text="SMEA+DYN") |
| i_dyn.grid(row=0, column=5, padx=5, sticky="nsew") |
| if contents['sm_dyn'] == True: |
| i_dyn.select() |
| else: |
| i_dyn.deselect() |
| i_smea.configure(state="disabled") |
| i_dyn.configure(state="disabled") |
|
|
| import_frame3 = customtkinter.CTkFrame(import_window_right, width=586, fg_color="#2B2B2B") |
| import_frame3.grid(row=6, column=0, pady=5, padx=5, sticky="nsew") |
|
|
| i_resolution_label = customtkinter.CTkLabel(import_frame3, font=my_font, text="Resolution : ") |
| i_resolution_label.grid(row=0, column=0, padx=5, sticky="nsew") |
| i_resolution = customtkinter.CTkEntry(import_frame3, font=my_font, width=100) |
| i_resolution.grid(row=0, column=1, padx=5, sticky="nsew") |
| i_resolution.insert(0, f"{width} x {height}") |
|
|
| import_frame4 = customtkinter.CTkFrame(import_window_right, width=586, fg_color="#2B2B2B") |
| import_frame4.grid(row=7, column=0, pady=10, padx=5, sticky="n") |
|
|
| def image_to_base64(image): |
| image_bytesIO = io.BytesIO() |
| image.save(image_bytesIO, format="PNG") |
| return base64.b64encode(image_bytesIO.getvalue()).decode() |
|
|
| def save_base64_image(base64_string): |
| image_data = base64.b64decode(base64_string) |
| output_file_path = f"output_NAI\\{self.start_time}\\import" |
| d = Path(output_file_path) |
| d.mkdir(parents=True, exist_ok=True) |
| filename = (d / f"{app.import_image_count:05}.png" ) |
| filename.write_bytes(image_data) |
| return filename |
|
|
| def image_to_queue(): |
| if from_url == True: |
| base64_string = image_to_base64(img) |
| filename = save_base64_image(base64_string) |
| app.ext_set_image_to_queue(new_image.copy(), contents['prompt'], str(contents['seed']), filename) |
| app.import_image_count += 1 |
| else: |
| app.ext_set_image_to_queue(new_image.copy(), contents['prompt'], str(contents['seed']), file_path_or_url) |
| app.import_image_negative[len(app.image_queue)-1] = contents['uc'] |
|
|
| import_image_to_queue = customtkinter.CTkButton(import_frame4, font=my_font, text="์ด๋ฏธ์ง/ํ๋กฌํํธ๋ฅผ ํ์ ์ฝ์
", command=image_to_queue) |
| import_image_to_queue.grid(row=0, column=0, padx=5, sticky="nsew") |
| if text_type == 'str': |
| import_image_to_queue.configure(state="disabled") |
|
|
| def prompt_insert(): |
| app.text_input.delete("0.0", "end") |
| app.text_input.insert("0.0", contents['prompt']) |
| app.negative_prompt_input.delete("0.0", "end") |
| app.negative_prompt_input.insert("0.0", contents['uc']) |
| app.automation_button.deselect() |
|
|
| import_prompts = customtkinter.CTkButton(import_frame4, font=my_font, text="ํ๋กฌํํธ/๋ค๊ฑฐํฐ๋ธ ์ฝ์
", fg_color="grey", hover_color="grey10", command=prompt_insert) |
| import_prompts.grid(row=0, column=1, padx=5, sticky="nsew") |
|
|
| |
| def settings_insert(): |
| app.cfg_scale_var.set(str(contents['scale'])) |
| app.cfg_scale_entry.delete(0, "end") |
| app.cfg_scale_entry.insert(0, str(contents['scale'])) |
| app.uncond_strength_entry.delete(0, "end") |
| app.uncond_strength_entry.insert(0, str(contents['uncond_scale'])) |
| app.prompt_guidance_rescale_var.set(str(contents['cfg_rescale'])) |
| app.prompt_guidance_rescale_entry.delete(0, "end") |
| app.prompt_guidance_rescale_entry.insert(0, str(contents['cfg_rescale'])) |
| app.seed_fix_button.select() |
| app.seed_entry.delete(0, "end") |
| app.seed_entry.insert(0, str(contents['seed'])) |
| app.sampler_var.set(contents['sampler']) |
| app.sampler_button.set(contents['sampler']) |
| if contents['sm'] == True: |
| app.sema_button.select() |
| else: |
| app.sema_button.deselect() |
| if contents['sm_dyn'] == True: |
| app.dyn_button.select() |
| else: |
| app.dyn_button.deselect() |
| app.random_resolution_button.deselect() |
| app.resolution_button.set(f"{width} x {height}") |
| app.resolution_var.set(f"{width} x {height}") |
| app.cond_negative_button.deselect() |
| app.automation_button.deselect() |
|
|
| import_settings = customtkinter.CTkButton(import_frame4, font=my_font, text="์ค์ ๊ฐ ์ผ๊ด ์ฝ์
", fg_color="#F5F3C2", hover_color="#A69F40", text_color="black", command=settings_insert) |
| import_settings.grid(row=0, column=2, padx=5, sticky="nsew") |
|
|
|
|
|
|
|
|
| import_window.after(1500, lambda: import_window.attributes('-topmost', False)) |
|
|
| except Exception as e: |
| self.image_label_report.configure(state="normal", text_color = "#FFFF97") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.insert("0.0", f"Error: {e}") |
| self.image_label_report.configure(state="disabled") |
|
|
|
|
|
|
| def ext_set_image_to_queue(self, imagen, prompt, seed, filename): |
| self.image_queue.append([imagen, prompt, seed, filename]) |
| |
| self.request_upper_size_button.configure(state="normal") |
| self.current_window = self.window |
| if self.window == len(self.image_queue)-1: |
| self.window+=1 |
| self.update() |
| show_text = str(self.window)+" / "+ str(len(self.image_queue)) |
| |
| self.window_label.configure(text=show_text) |
|
|
| def update(self): |
| if len(self.image_queue) <= 5: |
| if len(self.image_queue) >= 1: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| if len(self.image_queue) >= 2: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| self.image_history_1.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-2][0], size=(100,100))) |
| if len(self.image_queue) >= 3: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| self.image_history_1.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-2][0], size=(100,100))) |
| self.image_history_2.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-3][0], size=(100,100))) |
| if len(self.image_queue) >= 4: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| self.image_history_1.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-2][0], size=(100,100))) |
| self.image_history_2.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-3][0], size=(100,100))) |
| self.image_history_3.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-4][0], size=(100,100))) |
| if len(self.image_queue) == 5: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| self.image_history_1.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-2][0], size=(100,100))) |
| self.image_history_2.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-3][0], size=(100,100))) |
| self.image_history_3.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-4][0], size=(100,100))) |
| self.image_history_4.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-5][0], size=(100,100))) |
| else: |
| self.image_history_0.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-1][0], size=(100,100))) |
| self.image_history_1.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-2][0], size=(100,100))) |
| self.image_history_2.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-3][0], size=(100,100))) |
| self.image_history_3.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-4][0], size=(100,100))) |
| self.image_history_4.configure(self.image_history_sub_frame, image=customtkinter.CTkImage(self.image_queue[self.window-5][0], size=(100,100))) |
| if len(self.image_queue) == 6: |
| self.image_history_button_down.configure(state="normal") |
|
|
| |
|
|
| def save_settings(self): |
| settings = { |
| "NAI_ID": self.NAI_ID if self.NAI_ID else " ", |
| "access_token": self.access_token if self.access_token else " ", |
| "searach_keyword": self.search_label_entry.get() if len(self.search_label_entry.get()) > 3 else " ", |
| "exclude_keyword": self.exclude_label_entry.get() if len(self.exclude_label_entry.get()) > 3 else " ", |
| "explicit": self.rating_select_explicit.get(), |
| "nsfw": self.rating_select_nsfw.get(), |
| "sensitive": self.rating_select_sensitive.get(), |
| "general": self.rating_select_general.get(), |
| "rm_artist": self.rm_artist_name_button.get(), |
| "rm_copyright": self.rm_copyright_name_button.get(), |
| "rm_character": self.rm_characteristic_button.get(), |
| "prompt": self.text_input.get("0.0", "end-1c") if len(self.text_input.get("0.0", "end")) > 4 else " ", |
| "negative": self.negative_prompt_input.get("0.0", "end-1c") if len(self.negative_prompt_input.get("0.0", "end")) > 4 else " ", |
| "fix": self.fixed_prompt_input.get("0.0", "end-1c") if len(self.fixed_prompt_input.get("0.0", "end")) > 4 else " ", |
| "after": self.fixed_prompt_after_input.get("0.0", "end-1c") if len(self.fixed_prompt_after_input.get("0.0", "end")) > 4 else " ", |
| "auto_hide": self.auto_hide_keyword_input.get("0.0", "end-1c") if len(self.auto_hide_keyword_input.get("0.0", "end")) > 4 else " ", |
| "cfg_scale": str(self.cfg_scale_entry.get()), |
| "uncond_scale": str(self.uncond_strength_entry.get()), |
| "guidance": str(self.prompt_guidance_rescale_entry.get()), |
| "sampler": self.sampler_var.get(), |
| "random": self.random_resolution_button.get(), |
| "sema": self.sema_button.get(), |
| "dyn": self.dyn_button.get(), |
| "personal_folder":self.output_file_path_personal, |
| "save_folder":self.output_file_path if self.output_file_path_personal else " ", |
| "png_rule":self.name_var.get(), |
| "quality_toggle":self.auto_quality_toggle.get(), |
| "cond_prompt": self.conditional_prompt_input.get("0.0", "end-1c") if len(self.conditional_prompt_input.get("0.0", "end")) > 4 else " ", |
| "cond_negative": self.conditional_negative_input.get("0.0", "end-1c") if len(self.conditional_negative_input.get("0.0", "end")) > 4 else " ", |
| "activate_cond_prompt": self.cond_prompt_button.get(), |
| "activate_cond_negative": self.cond_negative_button.get(), |
| "prompt_rating": self.current_prompt_rating, |
| "resolution": self.resolution_button.get(), |
| "wildcard_preopen": self.toggle_wildcard_preopen.get(), |
| "i2i": [self.i2i_value, self.i2i_seed_hold_check, self.i2i_resolution] |
| } |
| with open('app_settings.json', 'w', encoding='utf-8') as f: |
| json.dump(settings, f, ensure_ascii=False, indent=4) |
| if type(self.cached_rows) != type(None) and not self.cached_rows.empty: |
| self.cached_rows.to_parquet('txt2img_temp_prompt.parquet') |
| |
|
|
| def load_settings(self): |
| if os.path.exists('app_settings.json'): |
| with open('app_settings.json', 'r', encoding='utf-8') as f: |
| try: |
| settings = json.load(f) |
| try: |
| self.NAI_ID = settings["NAI_ID"] if len(settings["NAI_ID"]) > 2 else None |
| except: pass |
| try: |
| self.access_token = settings["access_token"] if len(settings["access_token"]) > 2 else None |
| if self.access_token: |
| self.NAI_Account_Login.configure(state="disabled") |
| self.NAI_Token_Remove.configure(state="normal") |
| self.NAI_Account_State.configure(text="NAI Login : OK") |
| self.image_generation_button.configure(state="normal") |
| except: pass |
| try: |
| self.search_label_entry.insert(0, settings["searach_keyword"]) |
| except: pass |
| try: |
| self.exclude_label_entry.insert(0, settings["exclude_keyword"]) |
| except: pass |
| try: |
| self.rating_select_var_explicit.set(settings["explicit"]) |
| if settings["explicit"] == 1: self.rating_select_explicit.select() |
| else: self.rating_select_explicit.deselect() |
| except: pass |
| try: |
| self.rating_select_var_nsfw.set(settings[ "nsfw"]) |
| if settings["nsfw"] == 1: self.rating_select_nsfw.select() |
| else: self.rating_select_nsfw.deselect() |
| except: pass |
| try: |
| self.rating_select_var_sensitive.set(settings["sensitive"]) |
| if settings["sensitive"] == 1: self.rating_select_sensitive.select() |
| else: self.rating_select_sensitive.deselect() |
| except: pass |
| try: |
| self.rating_select_var_general.set(settings["general"]) |
| if settings["general"] == 1: self.rating_select_general.select() |
| else: self.rating_select_general.deselect() |
| except: pass |
| try: |
| self.rm_artist_name_var = settings["rm_artist"] |
| if settings["rm_artist"] == 1: self.rm_artist_name_button.select() |
| else: self.rm_artist_name_button.deselect() |
| except: pass |
| try: |
| self.rm_copyright_name_var = settings["rm_copyright"] |
| if settings["rm_copyright"] == 1: self.rm_copyright_name_button.select() |
| else: self.rm_copyright_name_button.deselect() |
| except: pass |
| try: |
| self.rm_characteristic_var = settings["rm_character"] |
| if settings["rm_character"] == 1: self.rm_characteristic_button.select() |
| else: self.rm_characteristic_button.deselect() |
| except: pass |
| try: |
| self.text_input.insert("0.0", settings["prompt"]) |
| except: pass |
| try: |
| self.negative_prompt_input.delete("0.0", "end") |
| self.negative_prompt_input.insert("0.0", settings["negative"]) |
| except: pass |
| try: |
| self.fixed_prompt_input.insert("0.0", settings["fix"]) |
| except: pass |
| try: |
| self.fixed_prompt_after_input.insert("0.0", settings["after"]) |
| except: pass |
| try: |
| self.auto_hide_keyword_input.insert("0.0", settings[ "auto_hide"]) |
| except: pass |
| try: |
| self.cfg_scale_entry.delete(0, "end") |
| self.cfg_scale_entry.insert(0, str(settings["cfg_scale"])) |
| self.cfg_scale_var.set(settings["cfg_scale"]) |
| except: pass |
| try: |
| self.prompt_guidance_rescale_entry.delete(0, "end") |
| self.prompt_guidance_rescale_entry.insert(0, str(settings["guidance"])) |
| self.prompt_guidance_rescale_var.set(settings["guidance"]) |
| except: pass |
| try: |
| self.sampler_var.set(settings["sampler"]) |
| self.sampler_button.set(settings["sampler"]) |
| except: pass |
| try: |
| self.random_resolution_var.set(settings["random"]) |
| if settings["random"] == 1: self.random_resolution_button.select() |
| else: self.random_resolution_button.deselect() |
| except: pass |
| try: |
| self.sema_button_var.set(settings["sema"]) |
| if settings["sema"] == 1: self.sema_button.select() |
| else: self.sema_button.deselect() |
| except: pass |
| try: |
| self.dyn_button_var.set(settings["dyn"]) |
| if settings["dyn"] == 1: self.dyn_button.select() |
| else: self.dyn_button.deselect() |
| except: pass |
| try: |
| if settings["personal_folder"] == True: |
| self.output_file_path_personal =True |
| self.output_file_path = settings["save_folder"] |
| else: self.output_file_path_personal =False |
| except: pass |
| try: |
| self.name_var.set(settings["png_rule"]) |
| except: pass |
| try: |
| self.auto_quality_toggle_var=settings["quality_toggle"] |
| if settings["quality_toggle"] == 1: self.auto_quality_toggle.select() |
| else: self.auto_quality_toggle.deselect() |
| except: pass |
| try: |
| self.conditional_prompt_input.insert("0.0", settings[ "cond_prompt"]) |
| except: pass |
| try: |
| self.conditional_negative_input.insert("0.0", settings[ "cond_negative"]) |
| except: pass |
| try: |
| self.cond_prompt_button_var.set(settings["activate_cond_prompt"]) |
| if settings["activate_cond_prompt"] == 1: self.cond_prompt_button.select() |
| else: self.cond_prompt_button.deselect() |
| except: pass |
| try: |
| self.cond_prompt_button_var.set(settings["activate_cond_negative"]) |
| if settings["activate_cond_negative"] == 1: self.cond_negative_button.select() |
| else: self.cond_negative_button.deselect() |
| except: pass |
| try: |
| self.current_prompt_rating = settings["prompt_rating"] |
| except: pass |
| try: |
| self.resolution_var.set(settings["resolution"]) |
| self.resolution_button.set(settings["resolution"]) |
| except: pass |
| try: |
| self.toggle_wildcard_preopen_var.set(settings["wildcard_preopen"]) |
| if settings["wildcard_preopen"] == 1: self.toggle_wildcard_preopen.select() |
| else: self.toggle_wildcard_preopen.deselect() |
| except: pass |
| try: |
| self.uncond_strength_entry.delete(0, "end") |
| self.uncond_strength_entry.insert(0, settings["uncond_scale"]) |
| except: pass |
| try: |
| self.i2i_value, self.i2i_seed_hold_check, self.i2i_resolution = settings["i2i"][0], settings["i2i"][1],settings["i2i"][2] |
| except: pass |
| except json.JSONDecodeError as e: |
| self.image_label_report.configure(state="normal", text_color = "#FFFF97") |
| self.image_label_report.insert("0.0", "์ธ์ด๋ธ ํ์ผ์ ๋ถ๋ฌ์ค๋๋ฐ ์คํจํ์์ต๋๋ค : " + str(e)) |
| self.image_label_report.configure(state="disabled") |
| if os.path.exists('txt2img_temp_prompt.parquet'): |
| try: |
| self.cached_rows = pd.read_parquet('txt2img_temp_prompt.parquet') |
| self.cached_prompt_label.configure(text = "๋จ์ ํ๋กฌํํธ ํ : "+str(len(self.cached_rows))) |
| if 'id' not in self.cached_rows.columns: |
| self.image_label_report.configure(state="normal", text_color = "#FFFF97") |
| self.image_label_report.insert("0.0", "<TagsError> 0215 ๋ฒ์ ๋ถํฐ tags์ ๊ตฌ์ฑ์ด ๋ณ๊ฒฝ๋์์ต๋๋ค. favorite ๊ธฐ๋ฅ์ ์ฌ์ฉํ๊ธฐ ์ํด ๊ฒ์์ ๋ค์ ์ํ ํด ์ฃผ์ธ์.") |
| self.image_label_report.configure(state="disabled") |
| except: pass |
|
|
| def exit_program(self): |
| self.save_settings() |
| app.destroy() |
|
|
| def get_max_size(self): |
| width = self.winfo_screenheight() |
| return 768 if width < 1440 else 768 |
|
|
| def update_fullscreen_image(self,new_window, new_image_label): |
| last_updated_image = None |
| while True: |
| time.sleep(1) |
| if not new_window.winfo_exists(): |
| break |
| if self.image_queue: |
| current_image = Image.open(self.image_queue[self.window-1][3]) |
| if current_image and current_image != last_updated_image: |
| |
| original_width, original_height = current_image.size |
| ratio = new_window.winfo_screenheight() / float(original_height) |
| new_width = int(original_width * ratio) |
| tk_image = customtkinter.CTkImage(current_image, size=(new_width,new_window.winfo_screenheight())) |
| new_window.after(0, lambda img=tk_image: new_image_label.configure(image=img)) |
| new_image_label.image = tk_image |
| last_updated_image = current_image |
|
|
| def show_fullscreen_image(self): |
| new_window = customtkinter.CTkToplevel() |
| new_window.attributes('-fullscreen', True) |
| new_window.state('zoomed') |
| new_window.configure(bg='black') |
|
|
| new_image_label = customtkinter.CTkLabel(new_window, text=" ") |
| new_image_label.pack(expand=True, fill='both') |
| new_window.bind("<Escape>", lambda e: new_window.destroy()) |
|
|
| update_thread = threading.Thread(target=self.update_fullscreen_image, args=(new_window, new_image_label), daemon=True) |
| update_thread.daemon = True |
| update_thread.start() |
|
|
| def resize_image_to_fit(self, 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 |
| |
| def open_prompt_window(self): |
| prompt_window = customtkinter.CTkToplevel() |
| prompt_window.title("์ถ์ฒ ํ๋กฌํํธ") |
| prompt_window.attributes('-topmost', True) |
| prompt_window.resizable(width=False, height=False) |
|
|
| text_label1 = customtkinter.CTkLabel(prompt_window, text="๊ฒ์ ๊ฒฐ๊ณผ ๋ด ๊ณ ๋น๋ ํค์๋", font=customtkinter.CTkFont('Pretendard', 13), width=500) |
| text_label1.grid(row=0, column=0, padx=5, pady=5, sticky="nsew") |
| text_output1 = customtkinter.CTkTextbox(prompt_window, height=250, width=350, font=customtkinter.CTkFont('Pretendard', 15)) |
| text_output1.grid(row=1, column=0, padx=5, pady=5, sticky="nsew") |
|
|
| text_label2 = customtkinter.CTkLabel(prompt_window, text="๊ฒ์ ๊ฒฐ๊ณผ ๋ด ์๊ฐ๋ช
์์", font=customtkinter.CTkFont('Pretendard', 13), width=500) |
| text_label2.grid(row=2, column=0, padx=5, pady=5, sticky="nsew") |
| text_output2 = customtkinter.CTkTextbox(prompt_window, height=250, width=350, font=customtkinter.CTkFont('Pretendard', 15)) |
| text_output2.grid(row=3, column=0, padx=5, pady=5, sticky="nsew") |
|
|
| text_label3 = customtkinter.CTkLabel(prompt_window, text="๋ง์ด ์ฌ์ฉ๋ ์บ๋ฆญํฐ ์์", font=customtkinter.CTkFont('Pretendard', 13), width=500) |
| text_label3.grid(row=4, column=0, padx=5, pady=5, sticky="nsew") |
| text_output3 = customtkinter.CTkTextbox(prompt_window, height=250, width=350, font=customtkinter.CTkFont('Pretendard', 15)) |
| text_output3.grid(row=5, column=0, padx=5, pady=5, sticky="nsew") |
|
|
| if type(self.cached_rows) != type(None) and not self.cached_rows.empty: |
| counts1 = Counter() |
| ndf = self.cached_rows[self.cached_rows['general'].notnull()] |
| for row in ndf['general']: |
| if row != None: |
| substrings = [substring.strip() for substring in row.split(',') if substring.strip()] |
| counts1.update(substrings) |
| top_200_keywords = counts1.most_common(200) |
| formatted_keywords = [f"{keyword}: {count}" for keyword, count in top_200_keywords] |
| formatted_text = "\n".join(formatted_keywords) |
| text_output1.insert("0.0", formatted_text) |
| del[ndf] |
|
|
| counts2 = Counter() |
| ndf = self.cached_rows[self.cached_rows['artist'].notnull()] |
| for row in ndf['artist']: |
| if row != None: |
| substrings = [substring.strip() for substring in row.split(',') if substring.strip()] |
| counts2.update(substrings) |
| top_200_keywords = [(keyword, count) for keyword, count in counts2.most_common(200) if keyword in artist_dict] |
| formatted_keywords = [f"{keyword}: {count}" for keyword, count in top_200_keywords] |
| formatted_text = "\n".join(formatted_keywords) |
| text_output2.insert("0.0", formatted_text) |
| del[ndf] |
|
|
| counts3 = Counter() |
| ndf = self.cached_rows[self.cached_rows['character'].notnull()] |
| for row in ndf['character']: |
| if row != None: |
| substrings = [substring.strip() for substring in row.split(',') if substring.strip()] |
| counts3.update(substrings) |
| top_200_keywords = counts3.most_common(200) |
| formatted_keywords = [f"{keyword}: {count}" for keyword, count in top_200_keywords] |
| formatted_text = "\n".join(formatted_keywords) |
| text_output3.insert("0.0", formatted_text) |
| del[ndf] |
|
|
| |
| def random_artist_window(self): |
| rartist_window = customtkinter.CTkToplevel() |
| rartist_window.title("๋๋ค ์๊ฐ๋ช
์ค์ ") |
| rartist_window.attributes('-topmost', True) |
| rartist_window.resizable(width=False, height=False) |
|
|
| rartist_window_left = customtkinter.CTkFrame(rartist_window, width=250) |
| rartist_window_left.grid(row=0, column=0, padx=5, pady=5, sticky="nsew") |
| |
| |
| text_label1 = customtkinter.CTkLabel(rartist_window_left, text="๋๋ค ์๊ฐ ์ฝ์
๋ฐฉ๋ฒ ์ ํ", font=customtkinter.CTkFont('Pretendard', 13), width=250) |
| text_label1.grid(row=0, column=0, padx=5, pady=5, sticky="nsew") |
| self.random_artist_select = customtkinter.StringVar() |
| ramdom_artist_global = customtkinter.CTkRadioButton(rartist_window_left, text="์ ์ฒด ์๊ฐ๋ช
๋ฆฌ์คํธ์์", variable=self.random_artist_select, value="global", font=customtkinter.CTkFont('Pretendard', 13)) |
| ramdom_artist_global.grid(row=1, column=0, padx=5, pady=5, sticky="w") |
| ramdom_artist_local = customtkinter.CTkRadioButton(rartist_window_left, text="๊ฒ์ ํ๋กฌํํธ ์์์", variable=self.random_artist_select, value="local", font=customtkinter.CTkFont('Pretendard', 13)) |
| ramdom_artist_local.grid(row=2, column=0, padx=5, pady=5, sticky="w") |
| text_label2 = customtkinter.CTkLabel(rartist_window_left, text="๋จ๋ถ๋ฃจ ๋ด ์ต์ ์ด๋ฏธ์ง ์ ์ค์ (์ต์: 80)", font=customtkinter.CTkFont('Pretendard', 13), width=250) |
| text_label2.grid(row=3, column=0, padx=5, pady=5, sticky="nsew") |
| self.danbooru_minimum = customtkinter.IntVar(value=80) |
| danbooru_entry = customtkinter.CTkEntry(rartist_window_left, textvariable=self.danbooru_minimum, font=customtkinter.CTkFont('Pretendard', 13), width=50) |
| danbooru_entry.grid(row=4, column=0, padx=5, pady=5, sticky="n") |
| danbooru_set = customtkinter.CTkButton(rartist_window_left, text="์๋ก ํ์ฌ",font=customtkinter.CTkFont('Pretendard', 13), width=50, command=self.get_artist) |
| danbooru_set.grid(row=5, column=0, padx=5, pady=5, sticky="n") |
| self.random_artist_list_length = customtkinter.CTkLabel(rartist_window_left, text="total length : 0", font=customtkinter.CTkFont('Pretendard', 13), width=250) |
| self.random_artist_list_length.grid(row=6, column=0, padx=5, pady=5, sticky="nsew") |
| text_label3 = customtkinter.CTkLabel(rartist_window_left, text="์๊ฐ๋ช
์ฝ์
๋ฐฉ์", font=customtkinter.CTkFont('Pretendard', 13), width=250) |
| text_label3.grid(row=7, column=0, padx=5, pady=5, sticky="nsew") |
| random_artist_radio1 = customtkinter.CTkRadioButton(rartist_window_left, text="์์ artist: ๋ถ์ด๊ธฐ", variable=self.random_artist_prefix, value="artist:", font=customtkinter.CTkFont('Pretendard', 13)) |
| random_artist_radio1.grid(row=8, column=0, padx=5, pady=5, sticky="w") |
| random_artist_radio2 = customtkinter.CTkRadioButton(rartist_window_left, text="๋ค์ (artist) ๋ถ์ด๊ธฐ", variable=self.random_artist_prefix, value="(artist)", font=customtkinter.CTkFont('Pretendard', 13)) |
| random_artist_radio2.grid(row=9, column=0, padx=5, pady=5, sticky="w") |
| random_artist_radio3 = customtkinter.CTkRadioButton(rartist_window_left, text="๊ธฐ๋ณธ ์คํ์ผ๋ก", variable=self.random_artist_prefix, value="none", font=customtkinter.CTkFont('Pretendard', 13)) |
| random_artist_radio3.grid(row=10, column=0, padx=5, pady=5, sticky="w") |
| self.random_artist_list = customtkinter.CTkTextbox(rartist_window_left, height=350, width=250, font=customtkinter.CTkFont('Pretendard', 15)) |
| self.random_artist_list.grid(row=11, column=0, padx=5, pady=5, sticky="nsew") |
| text_label4 = customtkinter.CTkLabel(rartist_window_left, text="๊ณ ์ ํ๋กฌํํธ์์๋ <random_artist>๋ก ํธ์ถ", font=customtkinter.CTkFont('Pretendard', 13), width=250) |
| text_label4.grid(row=12, column=0, padx=5, pady=5, sticky="nsew") |
|
|
| |
| |
| |
| |
| |
|
|
| def reactivate_wildcards(self): |
| self.activate_wildcards() |
| pretty_string = "" |
| for key, value in self.wildcard_dict.items(): |
| pretty_string += f"{key}: {value}\n" |
| self.wildcard_text.configure(state="normal") |
| self.wildcard_text.delete('0.0', "end") |
| self.wildcard_text.insert('0.0', pretty_string) |
| self.wildcard_text.configure(state="disabled") |
|
|
| def open_wildcard_folder(self): |
| if not os.path.exists(self.wildcards_dir): |
| os.makedirs(self.wildcards_dir) |
| os.startfile(self.wildcards_dir) |
|
|
| def open_wildcard_window(self): |
| wildcard_window = customtkinter.CTkToplevel() |
| wildcard_window.title("์์ผ๋์นด๋ ๊ด๋ฆฌ") |
| wildcard_window.attributes('-topmost', True) |
| wildcard_window.resizable(width=False, height=False) |
|
|
| wildcard_window_left = customtkinter.CTkFrame(wildcard_window) |
| wildcard_window_left.grid(row=0, column=0, sticky="nsew") |
| wildcard_window_right = customtkinter.CTkFrame(wildcard_window) |
| wildcard_window_right.grid(row=0, column=1, sticky="nsew") |
|
|
| text_label_frame = customtkinter.CTkFrame(wildcard_window_right, fg_color="#2B2B2B") |
| text_label_frame.grid(row=0, column=0, padx=5, pady=5, sticky="nsew") |
| text_label3 = customtkinter.CTkLabel(text_label_frame, text="์์ผ๋์นด๋ ๊ด๋ฆฌ", font=customtkinter.CTkFont('Pretendard', 13), width=300) |
| text_label3.grid(row=0, column=0, padx=5, pady=5, sticky="nsew") |
| text_hyperlink = customtkinter.CTkLabel(text_label_frame, text="๊ฐ์ด๋ ์ด๊ธฐ (arca.live)", font=customtkinter.CTkFont('Pretendard', 13), width=180, text_color="lightblue", cursor="hand2") |
| text_hyperlink.grid(row=0, column=1, padx=5, pady=5, sticky="nsew") |
| text_hyperlink.bind("<Button-1>", lambda e: webbrowser.open_new("https://arca.live/b/aiart/95877361")) |
|
|
| def wildcard_folder_callback(choice): |
| if choice != '': |
| if choice != '(+) ํด๋ ์ถ๊ฐ': |
| value = list(self.wildcard_dict[wildcard_folder_select_var.get()])+['(+) ์ ์์ผ๋์นด๋'] |
| wildcard_folder_select_var.set(choice) |
| wildcard_file_select.configure(values=value) |
| wildcard_file_select.set('') |
| else: |
| value = list(self.wildcard_dict.keys()) |
| dialog = customtkinter.CTkInputDialog(text="์ ์์ผ๋์นด๋ ํด๋๋ช
์ ์
๋ ฅํ์ธ์ (๊ฐ์ ์ด๋ฆ ๋ถ๊ฐ):", title="ํด๋ ์์ฑ") |
| input_value = dialog.get_input() |
| if input_value and input_value not in value: |
| os.makedirs(self.wildcards_dir+'/'+input_value) |
| self.reactivate_wildcards() |
| wildcard_folder_select.configure(values=list(self.wildcard_dict.keys())+['(+) ํด๋ ์ถ๊ฐ']) |
| wildcard_folder_select.set(input_value) |
| wildcard_file_select.set('') |
| else: |
| wildcard_folder_select.set('') |
|
|
|
|
| def wildcard_select_callback(choice): |
| if choice != '': |
| if choice == '(+) ์ ์์ผ๋์นด๋': |
| value = list(self.wildcard_dict[wildcard_folder_select_var.get()]) |
| dialog = customtkinter.CTkInputDialog(text="์ ์์ผ๋์นด๋์ ์ด๋ฆ์ ์
๋ ฅํ์ธ์ (๊ฐ์ ์ด๋ฆ ๋ถ๊ฐ):", title="์์ผ๋์นด๋ ์์ฑ") |
| input_value = dialog.get_input() |
| if input_value and input_value not in value: |
| filepath = self.wildcards_dir+'/'+input_value if wildcard_folder_select_var.get() == "none" else self.wildcards_dir+'/'+wildcard_folder_select_var.get()+'/'+input_value |
| with open(filepath+'.txt', 'w') as file: |
| pass |
| self.reactivate_wildcards() |
| value2 = list(self.wildcard_dict[wildcard_folder_select_var.get()])+['(+) ์ ์์ผ๋์นด๋'] |
| wildcard_file_select.configure(values=value2) |
| wildcard_file_select.set('') |
| else: |
| self.wildcard_file_select_var = choice if wildcard_folder_select_var.get() == "none" else wildcard_folder_select_var.get()+'/'+choice |
| filepath = self.wildcards_dir+'/'+choice if wildcard_folder_select_var.get() == "none" else self.wildcards_dir+'/'+wildcard_folder_select_var.get()+'/'+choice |
| if os.path.exists(filepath): |
| with open(filepath, 'r', encoding='utf-8') as file: |
| lines = file.readlines() |
| updated_lines = [] |
| for line in lines: |
| line = line.strip() |
| updated_lines.append(line) |
| wildcard_select_text.configure(state="normal") |
| wildcard_select_text.delete("0.0","end") |
| wildcard_select_text.insert("0.0", '\n'.join(updated_lines)) |
| wildcard_select_text.configure(state="disabled") |
| self.current_wildcard_path = filepath |
|
|
| def fix_wildcard(): |
| if fix_button.cget("text") == "์์ ": |
| save_button.configure(state="normal") |
| wildcard_select_text.configure(state="normal") |
| fix_button.configure(text="์ทจ์") |
| wildcard_folder_select.configure(state="disabled") |
| wildcard_file_select.configure(state="disabled") |
| coy_button.configure(state="disabled") |
| elif fix_button.cget("text") == "์ทจ์": |
| save_button.configure(state="disabled") |
| wildcard_select_text.configure(state="disabled") |
| fix_button.configure(text="์์ ") |
| wildcard_folder_select.configure(state="normal") |
| wildcard_file_select.configure(state="normal") |
| coy_button.configure(state="normal") |
|
|
| def save_wildcard(): |
| text = wildcard_select_text.get("1.0", "end-1c") |
| with open(self.current_wildcard_path, "w", encoding="utf-8") as file: |
| file.write(text) |
| wildcard_select_text.configure(state="disabled") |
| fix_button.configure(text="์์ ") |
| wildcard_folder_select.configure(state="normal") |
| wildcard_file_select.configure(state="normal") |
| coy_button.configure(state="normal") |
| save_button.configure(state="disabled") |
|
|
| def copy_wildcard(): |
| text = self.wildcard_file_select_var |
| text = text.replace(".txt","") |
| pyperclip.copy('<'+text+'>') |
| |
| wildcard_window_select_frame = customtkinter.CTkFrame(wildcard_window_right) |
| wildcard_window_select_frame.grid(row=1, column=0, sticky="n") |
| text_label4 = customtkinter.CTkLabel(wildcard_window_select_frame, text="ํด๋ ์ ํ:", font=customtkinter.CTkFont('Pretendard', 13)) |
| text_label4.grid(row=0, column=0, padx=15, pady=5, sticky="nsew") |
| text_label5 = customtkinter.CTkLabel(wildcard_window_select_frame, text="ํ์ผ ์ ํ:", font=customtkinter.CTkFont('Pretendard', 13)) |
| text_label5.grid(row=0, column=2, padx=15, pady=5, sticky="nsew") |
| wildcard_folder_select_var = customtkinter.StringVar(value="none") |
| wildcard_folder_select = customtkinter.CTkComboBox(wildcard_window_select_frame,values=list(self.wildcard_dict.keys())+['(+) ํด๋ ์ถ๊ฐ'],command=wildcard_folder_callback, variable=wildcard_folder_select_var, font=customtkinter.CTkFont('Pretendard', 13)) |
| wildcard_folder_select.grid(row=0, column=1, padx=15, pady=5, sticky="nsew") |
| self.wildcard_file_select_var = "" |
| self.current_wildcard_path = "" |
| wildcard_file_select = customtkinter.CTkComboBox(wildcard_window_select_frame,values=list(self.wildcard_dict[wildcard_folder_select_var.get()])+['(+) ์ ์์ผ๋์นด๋'],command=wildcard_select_callback, font=customtkinter.CTkFont('Pretendard', 13)) |
| wildcard_file_select.grid(row=0, column=3, padx=15, pady=5, sticky="nsew") |
| wildcard_file_select.set("") |
| text_label6 = customtkinter.CTkLabel(wildcard_window_select_frame, text="ํด๋ ๋ฐ ํ์ผ์ ์ญ์ ๊ธฐ๋ฅ์ ์ ๊ณต๋์ง ์์ต๋๋ค. ์์ผ๋์นด๋ ํด๋์์ ์ง์ ์ญ์ ํ์ธ์.", font=customtkinter.CTkFont('Pretendard', 13)) |
| text_label6.grid(row=1, column=0, columnspan=4, padx=5, pady=5, sticky="n") |
| fix_button = customtkinter.CTkButton(wildcard_window_select_frame, text="์์ ", font=customtkinter.CTkFont('Pretendard', 13), width=100, hover_color="grey10", command=fix_wildcard) |
| fix_button.grid(row=2, column=0, padx=15, pady=5, sticky="n") |
| save_button = customtkinter.CTkButton(wildcard_window_select_frame, text="์ ์ฅ", font=customtkinter.CTkFont('Pretendard', 13), width=100, fg_color="grey", hover_color="grey10", command=save_wildcard, state="disabled") |
| save_button.grid(row=2, column=1, padx=15, pady=5, sticky="n") |
| coy_button = customtkinter.CTkButton(wildcard_window_select_frame, text="<์์ผ๋์นด๋>๋ฅผ ํด๋ฆฝ๋ณด๋์ ๋ณต์ฌ", font=customtkinter.CTkFont('Pretendard', 13), fg_color="#7030A0", hover_color="#481F67", command=copy_wildcard) |
| coy_button.grid(row=2, column=2, columnspan=2, padx=5, pady=5, sticky="n") |
| text_label7 = customtkinter.CTkLabel(wildcard_window_right, text="<UserAttention> Enter ์
๋ ฅ์ ๋ค๋ฅธ ํ์ผ๋ก ์ทจ๊ธ๋ฉ๋๋ค. ์์ ํ ๊ผญ ์ ์ฅ ๋ฒํผ์ ๋๋ฅด์ธ์.", font=customtkinter.CTkFont('Pretendard', 13), text_color="#FFFF97") |
| text_label7.grid(row=3, column=0, columnspan=4, padx=5, pady=5, sticky="n") |
| wildcard_select_text = customtkinter.CTkTextbox(wildcard_window_right, font=customtkinter.CTkFont('Pretendard', 14), width=490, height=280, state="disabled") |
| wildcard_select_text.grid(row=2, column=0, columnspan=4, padx=5, pady=5, sticky="nsew") |
|
|
| |
| |
| text_label1 = customtkinter.CTkLabel(wildcard_window_left, text="์์ผ๋์นด๋ ์ ๋ณด", font=customtkinter.CTkFont('Pretendard', 13), width=500) |
| text_label1.grid(row=0, column=0, padx=5, pady=5, sticky="nsew") |
| self.wildcard_text = customtkinter.CTkTextbox(wildcard_window_left, font=customtkinter.CTkFont('Pretendard', 13), width=500) |
| self.wildcard_text.grid(row=1, column=0, padx=5, pady=5, sticky="nsew") |
| text_label2 = customtkinter.CTkLabel(wildcard_window_left, text="txtํ์ผ ๋ด ๊ฐ์ค์น ๋ฌธ๋ฒ(100: keyword)์ ๊ณ์ ์ ํจํฉ๋๋ค. default=100", font=customtkinter.CTkFont('Pretendard', 13), width=500) |
| text_label2.grid(row=2, column=0, padx=5, pady=5, sticky="nsew") |
| self.wildcard_button_frame = customtkinter.CTkFrame(wildcard_window_left) |
| self.wildcard_button_frame.grid(row=3, column=0, padx=5, pady=5, sticky="n") |
| self.reactive_button = customtkinter.CTkButton(self.wildcard_button_frame, font=customtkinter.CTkFont('Pretendard', 13), text="์์ผ๋์นด๋ ์
๋ฐ์ดํธ", command= self.reactivate_wildcards) |
| self.reactive_button.grid(row=0, column=0, padx=5, pady=5, sticky="n") |
| self.open_wildcard_button = customtkinter.CTkButton(self.wildcard_button_frame, font=customtkinter.CTkFont('Pretendard', 13), text="์์ผ๋์นด๋ ํด๋ ์ด๊ธฐ", command= self.open_wildcard_folder) |
| self.open_wildcard_button.grid(row=0, column=1, padx=15, pady=5, sticky="n") |
| self.wildcard_text_sample = customtkinter.CTkTextbox(wildcard_window_left, font=customtkinter.CTkFont('Pretendard', 13), width=500, height=100, text_color="#FFFF97") |
| self.wildcard_text_sample.grid(row=4, column=0, padx=5, pady=5, sticky="nsew") |
| self.wildcard_text_sample.insert("0.0", "Example : 1girl, <koharu (blue archive)|noa (blue archive)|unicorn (azur lane)|<characters>>, <random_artist>, <happy|surprised|depressed>, <<__colors__clothes>|<{{clothes}}>>, looking at viewer") |
| self.wildcard_text_sample.configure(state="disabled") |
| text_label3 = customtkinter.CTkLabel(wildcard_window_left, text="<keyword1|keyword2> ํํ ํ๊ธฐ๋ '|'๋ก ๊ตฌ๋ถ๋๋ ์ธ์คํดํธ ์์ผ๋์นด๋์
๋๋ค.", font=customtkinter.CTkFont('Pretendard', 13), width=500) |
| text_label3.grid(row=5, column=0, padx=5, pady=5, sticky="nsew") |
| self.reactivate_wildcards() |
|
|
| def get_artist(self): |
| self.random_artist_list.delete("0.0", "end") |
| danbooru_minimum = self.danbooru_minimum.get() |
| if self.random_artist_select.get() == "local": |
| counts = Counter() |
| ndf = self.cached_rows[self.cached_rows['artist'].notnull()] |
| for row in ndf['artist']: |
| if row != None: |
| substrings = [substring.strip() for substring in row.split(',') if substring.strip()] |
| counts.update(substrings) |
| top_500_keywords = [(keyword, count) for keyword, count in counts.most_common(500) if keyword in artist_dict] |
| filtered_keywords = [(keyword, count) for keyword, count in top_500_keywords if artist_dict.get(keyword, 0) > danbooru_minimum] |
| formatted_keywords = [f"{keyword}: {count}" for keyword, count in filtered_keywords] |
| keywords_only = [item.split(": ")[0] for item in formatted_keywords] |
| self.random_artist_list_length.configure(text='total length : ' + str(len(filtered_keywords))) |
| self.random_artist = keywords_only |
| try: |
| with open("wildcards/random_artist.txt", 'w') as file: |
| for keyword in keywords_only: |
| if self.random_artist_prefix.get() == "artist:": |
| file.write("artist:"+ keyword + '\n') |
| elif self.random_artist_prefix.get() == "(artist)": |
| file.write(keyword + ' (artist)\n') |
| else: |
| file.write(keyword + '\n') |
| except: pass |
| formatted_text = "\n".join(formatted_keywords) |
| self.random_artist_list.insert("0.0", formatted_text) |
| else: |
| filtered_keywords = [keyword for keyword, count in artist_dict.items() if count > danbooru_minimum] |
| formatted_keywords = [f"{keyword}: {count}" for keyword, count in artist_dict.items() if count > danbooru_minimum] |
| self.random_artist_list_length.configure(text='total length : ' + str(len(filtered_keywords))) |
| formatted_text = "\n".join(formatted_keywords) |
| self.random_artist_list.insert("0.0", formatted_text) |
| self.random_artist = filtered_keywords |
| try: |
| with open("wildcards/random_artist.txt", 'w') as file: |
| for keyword in filtered_keywords: |
| if self.random_artist_prefix.get() == "artist:": |
| file.write("artist:"+ keyword + '\n') |
| elif self.random_artist_prefix.get() == "(artist)": |
| file.write(keyword + ' (artist)\n') |
| else: |
| file.write(keyword + '\n') |
| except: pass |
| self.random_artist_button.configure(state="normal") |
|
|
| def activate_wildcards(self): |
| wildcards_dir = os.path.join(os.getcwd(), 'wildcards') |
|
|
| |
| if not os.path.exists(wildcards_dir): |
| os.makedirs(wildcards_dir) |
|
|
| |
| wildcard_dict = {'none': []} |
|
|
| |
| for item in os.listdir(wildcards_dir): |
| item_path = os.path.join(wildcards_dir, item) |
| |
| if os.path.isfile(item_path) and item.endswith('.txt'): |
| wildcard_dict['none'].append(item) |
| |
| elif os.path.isdir(item_path): |
| wildcard_dict[item] = [] |
| |
| for subitem in os.listdir(item_path): |
| subitem_path = os.path.join(item_path, subitem) |
| if os.path.isfile(subitem_path) and subitem.endswith('.txt'): |
| wildcard_dict[item].append(subitem) |
| self.wildcard_dict = wildcard_dict |
| self.wildcards_dir = wildcards_dir |
|
|
| def get_wildcard(self, input_str): |
| def read_file_with_fallback(file_path, encodings=['utf-8', 'cp949']): |
| for encoding in encodings: |
| try: |
| with open(file_path, 'r', encoding=encoding, errors='ignore') as file: |
| return file.readlines() |
| except UnicodeDecodeError: |
| continue |
| return None |
| |
| bracket_count = input_str.count('[') |
| brace_count = input_str.count('{') |
|
|
| modified_input_str = input_str.replace('[', '').replace(']', '').replace('{', '').replace('}', '') |
|
|
| file_path = "" |
| if '/' in modified_input_str: |
| folder, filename = modified_input_str.split('/', 1) |
| file_path = os.path.join(self.wildcards_dir, folder, filename + '.txt') |
| valid_file = folder in self.wildcard_dict and filename + ".txt" in self.wildcard_dict[folder] and os.path.exists(file_path) |
| else: |
| file_path = os.path.join(self.wildcards_dir, modified_input_str + ".txt") |
| valid_file = modified_input_str + ".txt" in self.wildcard_dict['none'] and os.path.exists(file_path) |
| |
| if valid_file: |
| lines = read_file_with_fallback(file_path) |
| |
| if lines: |
| choice_dic = {} |
| for line in lines: |
| match = re.match(r'(\d*\.?\d+):(.+)', line) |
| if match: |
| value, keyword = float(match.group(1)), match.group(2).strip() |
| else: |
| value, keyword = 100, line.strip() |
| choice_dic[keyword] = value |
|
|
| keywords = list(choice_dic.keys()) |
| weights = list(choice_dic.values()) |
| result = random.choices(keywords, weights=weights, k=1)[0].strip() |
| else: |
| result = modified_input_str |
|
|
| if brace_count > 0: |
| return '{' * brace_count + result + '}' * brace_count |
| elif bracket_count > 0: |
| return '[' * bracket_count + result + ']' * bracket_count |
| else: |
| return result |
|
|
| |
| |
| def get_anlas(self): |
| try: |
| response = requests.get("https://api.novelai.net/user/subscription", headers={ |
| "Authorization": f"Bearer {self.access_token}"}) |
| data_dict = json.loads(response.content) |
| trainingStepsLeft = data_dict['trainingStepsLeft'] |
| anlas = int(trainingStepsLeft['fixedTrainingStepsLeft']) + \ |
| int(trainingStepsLeft['purchasedTrainingSteps']) |
| app.my_anlas.set(anlas) |
| self.anlas_label.configure(text="Anlas : "+str(app.my_anlas.get()), text_color="#FFFF97") |
| except Exception as e: |
| print(e) |
|
|
| return None |
| |
| def open_instant_event(self): |
| my_font = customtkinter.CTkFont('Pretendard', 13) |
| instant_event_window = customtkinter.CTkToplevel() |
| instant_event_window.title("์ธ์คํดํธ ์ด๋ฒคํธ (์์
๋์ค ์ข
๋ฃํ์ง ๋ง์ธ์)") |
| instant_event_window.attributes('-topmost', True) |
| instant_event_window.resizable(width=False, height=False) |
| tabview = customtkinter.CTkTabview(instant_event_window) |
| tabview.grid(row=0, column=0, padx=5, pady=5) |
| generation_frame = customtkinter.CTkFrame(instant_event_window, height=300) |
| generation_frame.grid(row=0, column=1, padx=5, pady=5) |
|
|
| generation_label = customtkinter.CTkLabel(generation_frame, text="0 / 0", font=my_font) |
| generation_label.grid(row=0, column=0, padx=5, pady=5, sticky="n") |
| self.instant_stop_bit = False |
|
|
| def stop_gen(): |
| self.instant_stop_bit = True |
| generation_stop_button.configure(state = "disabled") |
| xyz_execute_button.configure(state="normal") |
|
|
| def close_window(): |
| stop_gen() |
| instant_event_window.destroy() |
| |
| generation_stop_button = customtkinter.CTkButton(generation_frame, text="Stop", font=my_font, state="disabled", command=stop_gen) |
| generation_stop_button.grid(row=1, column=0, padx=5, pady=25, sticky="n") |
| instant_event_window.protocol("WM_DELETE_WINDOW", close_window) |
|
|
| |
| tabview.add("X/Y plot") |
|
|
| def xyz_excute(): |
| comboboxes = [xyz_box1, xyz_box2] |
| prompts = [xyz_prompt1, xyz_prompt2] |
| xyz_list = [] |
| prompt_list = [] |
| prompt_len = 0 |
| for i, combobox in enumerate(comboboxes): |
| value = combobox.get() |
| |
| if value == "None": |
| continue |
| |
| elif value == "CFG Scale": |
| xyz_list.append("CFG Scale") |
| input_string = prompts[i].get("0.0", "end-1c") |
| split_strings = [s.strip() for s in input_string.split(',')] |
| float_values = [float(s) for s in split_strings] |
| cfg_scale_list = [] |
| while float_values[0] <= float_values[1]: |
| cfg_scale_list.append(float_values[0]) |
| float_values[0] += float_values[2] |
| cfg_scale_list = [round(v, 1) for v in cfg_scale_list] |
| |
| elif value == "PG.Rescale": |
| xyz_list.append("PG.Rescale") |
| input_string = prompts[i].get("0.0", "end-1c") |
| split_strings = [s.strip() for s in input_string.split(',')] |
| float_values = [float(s) for s in split_strings] |
| pg_rescale_list = [] |
| while float_values[0] <= float_values[1]: |
| pg_rescale_list.append(float_values[0]) |
| float_values[0] += float_values[2] |
| pg_rescale_list = [round(v, 1) for v in pg_rescale_list] |
| |
| elif value == "ํ๋กฌํํธ ๊ฐ์กฐ": |
| xyz_list.append("ํ๋กฌํํธ ๊ฐ์กฐ") |
| input_string = prompts[i].get("0.0", "end-1c") |
| split_strings = [s.strip() for s in input_string.split(',')] |
| split_strings[0] = str(split_strings[0]) |
| split_strings[1] = int(split_strings[1]) |
| split_strings[2] = int(split_strings[2]) |
| split_strings[3] = int(split_strings[3]) |
| keyword = split_strings[0] |
| |
| if prompt_len == 0: |
| before_wildcard = self.text_input.get("0.0", "end-1c") |
| before_wildcard = [item.strip() for item in before_wildcard.split(',')] |
| if '<' in self.text_input.get("0.0", "end-1c"): |
| |
| for i, keyword in enumerate(before_wildcard): |
| if keyword.startswith('<') and keyword.endswith('>'): |
| vbar_check = keyword[1:-1] |
| if '|' in vbar_check: |
| choices = vbar_check.split('|') |
| choice_dic = {} |
| for choice in choices: |
| match = re.match(r'(\d*\.?\d+):(.+)', choice) |
| if match: |
| value, keyword = float(match.group(1)), match.group(2).strip() |
| else: |
| value, keyword = 1, choice.strip() |
| choice_dic[keyword] = value |
| keywords = list(choice_dic.keys()) |
| weights = list(choice_dic.values()) |
| selected_instant_wildcard = random.choices(keywords, weights=weights, k=1)[0] |
| before_wildcard[i] = selected_instant_wildcard |
| |
| for i, keyword in enumerate(before_wildcard): |
| if "<" in keyword: |
| input_str = keyword.strip('<>').strip() |
| if("__" in input_str): |
| adjectives = re.findall(r'__(.*?)__', input_str) |
| last_keyword = re.split(r'__.*?__', input_str)[-1] |
| adjective_string = "" |
| for adjective in adjectives: |
| adjective_string += (self.get_wildcard(adjective) + " ") |
| before_wildcard[i] = adjective_string + self.get_wildcard(last_keyword) |
| else: |
| before_wildcard[i] = self.get_wildcard(input_str) |
| keyword_index = before_wildcard.index(keyword) |
| start = split_strings[1] |
| end = split_strings[2] |
| step = split_strings[3] |
| prompt_list.append([]) |
| first_prompt = [] |
| for i in range(start, end + 1, step): |
| if i < 0: |
| before_wildcard[keyword_index] = '[' * abs(i) + keyword + ']' * abs(i) |
| first_prompt.append('[' * abs(i) + keyword + ']' * abs(i)) |
| elif i == 0: |
| before_wildcard[keyword_index] = keyword |
| first_prompt.append(keyword) |
| else: |
| before_wildcard[keyword_index] = '{' * i + keyword + '}' * i |
| first_prompt.append('{' * i + keyword + '}' * i) |
| prompt_list[prompt_len].append(before_wildcard.copy()) |
| prompt_len+=1 |
| else: |
| start = split_strings[1] |
| end = split_strings[2] |
| step = split_strings[3] |
| second_prompt = [] |
| for i in range(start, end + 1, step): |
| if i < 0: |
| second_prompt.append('[' * abs(i) + keyword + ']' * abs(i)) |
| elif i == 0: |
| second_prompt.append(keyword) |
| else: |
| second_prompt.append('{' * i + keyword + '}' * i) |
| second_prompt_keyword = keyword |
| |
| elif value == "ํ๋กฌํํธ ์ค์": |
| xyz_list.append("ํ๋กฌํํธ ์ค์") |
| input_string = prompts[i].get("0.0", "end-1c") |
| split_strings = [s.strip() for s in input_string.split(',')] |
| keyword = split_strings[0] |
| |
| if prompt_len == 0: |
| before_wildcard = self.text_input.get("0.0", "end-1c") |
| before_wildcard = [item.strip() for item in before_wildcard.split(',')] |
| if '<' in self.text_input.get("0.0", "end-1c"): |
| |
| for i, keyword in enumerate(before_wildcard): |
| if keyword.startswith('<') and keyword.endswith('>'): |
| vbar_check = keyword[1:-1] |
| if '|' in vbar_check: |
| choices = vbar_check.split('|') |
| choice_dic = {} |
| for choice in choices: |
| match = re.match(r'(\d*\.?\d+):(.+)', choice) |
| if match: |
| value, keyword = float(match.group(1)), match.group(2).strip() |
| else: |
| value, keyword = 1, choice.strip() |
| choice_dic[keyword] = value |
| keywords = list(choice_dic.keys()) |
| weights = list(choice_dic.values()) |
| selected_instant_wildcard = random.choices(keywords, weights=weights, k=1)[0] |
| before_wildcard[i] = selected_instant_wildcard |
| |
| for i, keyword in enumerate(before_wildcard): |
| if "<" in keyword: |
| input_str = keyword.strip('<>').strip() |
| if("__" in input_str): |
| adjectives = re.findall(r'__(.*?)__', input_str) |
| last_keyword = re.split(r'__.*?__', input_str)[-1] |
| adjective_string = "" |
| for adjective in adjectives: |
| adjective_string += (self.get_wildcard(adjective) + " ") |
| before_wildcard[i] = adjective_string + self.get_wildcard(last_keyword) |
| else: |
| before_wildcard[i] = self.get_wildcard(input_str) |
| keyword_index = before_wildcard.index(keyword) |
| prompt_list.append([]) |
| first_prompt = [] |
| for i in split_strings: |
| before_wildcard[keyword_index] = i |
| first_prompt.append(i) |
| prompt_list[prompt_len].append(before_wildcard.copy()) |
| prompt_len+=1 |
| else: |
| second_prompt = [] |
| for i in split_strings: |
| second_prompt.append(i) |
| second_prompt_keyword = keyword |
| elif value == "SMEA 0/1/dyn": |
| xyz_list.append("SMEA") |
| smea_list = [(False, False), (True, False), (True, True)] |
| elif value == "Samplers": |
| xyz_list.append("Samplers") |
| sampler_list = ["k_euler", "k_euler_ancestral", "k_dpmpp_2s_ancestral", "k_dpmpp_sde", "k_dpmpp_2m"] |
| |
| |
| NAI_width, NAI_height = self.resolution_button.get().split(' x ') |
| scale_pre = self.cfg_scale_entry.get() |
| try: |
| scale_pre = float(scale_pre) |
| except: |
| scale_pre = 5.0 |
| app.cfg_scale_var.set("5.0") |
| rescale_pre = app.prompt_guidance_rescale_entry.get() |
| try: |
| rescale_pre = float(rescale_pre) |
| except: |
| rescale_pre = 0 |
| app.prompt_guidance_rescale_var.set("0") |
| uncond_pre = self.uncond_strength_entry.get() |
| try: |
| uncond_pre = float(uncond_pre) |
| uncond_pre = round(uncond_pre / 0.05) * 0.05 |
| if (uncond_pre > 1.5): |
| uncond_pre = 1.5 |
| self.uncond_strength_entry.delete(0, "end") |
| self.uncond_strength_entry.insert(0, "1.5") |
| except: |
| uncond_pre = 1.0 |
| self.uncond_strength_entry.delete(0, "end") |
| self.uncond_strength_entry.insert(0, "1.0") |
| _gen_request = { |
| "width":NAI_width, |
| "height":NAI_height, |
| "quality_toggle":self.auto_quality_toggle.get(), |
| "seed":self.seed_entry.get(), |
| "sampler":self.sampler_button.get(), |
| "scale":scale_pre, |
| "uncond_scale":uncond_pre, |
| "sema":self.sema_button.get(), |
| "sema_dyn": self.dyn_button.get(), |
| "cfg_rescale": rescale_pre, |
| "prompt": self.text_input.get("0.0", "end-1c"), |
| "negative":self.negative_prompt_input.get("0.0", "end-1c"), |
| "user_screen_size": self.get_max_size(), |
| "start_time": self.start_time, |
| "access_token": self.access_token, |
| "save_folder": self.output_file_path, |
| "png_rule": self.name_var.get(), |
| "type": "normal", |
| "rating":self.current_prompt_rating |
| } |
| x = [] |
| y = [] |
| z = [] |
| xyzs = [x, y, z] |
| prompt_len = 0 |
| ybar = None |
| for i, xyz in enumerate(xyz_list): |
| if i == 0: |
| if xyz_list[i] == "CFG Scale": |
| for cfg in cfg_scale_list: |
| gen_request = _gen_request.copy() |
| gen_request["scale"] = cfg |
| xyzs[i].append(gen_request.copy()) |
| xbar = [str("CFG: {:.1f}".format(x)) for x in cfg_scale_list] |
| elif xyz_list[i] == "PG.Rescale": |
| for pgr in pg_rescale_list: |
| gen_request = _gen_request.copy() |
| gen_request["cfg_rescale"] = pgr |
| xyzs[i].append(gen_request.copy()) |
| xbar = [str("PGR: {:.1f}".format(x)) for x in pg_rescale_list] |
| elif xyz_list[i] == "ํ๋กฌํํธ ๊ฐ์กฐ": |
| for prompt in prompt_list[prompt_len]: |
| gen_request = _gen_request.copy() |
| gen_request["prompt"] = ', '.join(prompt) |
| xyzs[i].append(gen_request.copy()) |
| xbar = first_prompt |
| prompt_len+=1 |
| elif xyz_list[i] == "ํ๋กฌํํธ ์ค์": |
| for prompt in prompt_list[prompt_len]: |
| gen_request = _gen_request.copy() |
| gen_request["prompt"] = ', '.join(prompt) |
| xyzs[i].append(gen_request.copy()) |
| xbar = first_prompt |
| prompt_len+=1 |
| elif xyz_list[i] == "SMEA": |
| for tp in smea_list: |
| gen_request = _gen_request.copy() |
| gen_request["sema"], gen_request["sema_dyn"] = tp |
| xyzs[i].append(gen_request.copy()) |
| xbar = ['None', "SMEA", "SMEA+DYN"] |
| elif xyz_list[i] == "Samplers": |
| for sam in sampler_list: |
| gen_request = _gen_request.copy() |
| gen_request["sampler"] = sam |
| xyzs[i].append(gen_request.copy()) |
| xbar = ["k_euler", "k_euler_ancestral", "k_dpmpp_2s_ancestral", "k_dpmpp_sde", "k_dpmpp_2m"] |
| elif i == 1: |
| if xyz_list[i] == "CFG Scale": |
| for cfg in cfg_scale_list: |
| _x = copy.deepcopy(x) |
| for value in _x: |
| value["scale"] = cfg |
| y.append(_x) |
| ybar = [str("CFG: {:.1f}".format(x)) for x in cfg_scale_list] |
| elif xyz_list[i] == "PG.Rescale": |
| for pgr in pg_rescale_list: |
| _x = copy.deepcopy(x) |
| for value in _x: |
| value["cfg_rescale"] = pgr |
| y.append(_x) |
| ybar = [str("PGR: {:.1f}".format(x)) for x in pg_rescale_list] |
| elif xyz_list[i] == "ํ๋กฌํํธ ๊ฐ์กฐ": |
| if prompt_len == 0: |
| for prompt in prompt_list[prompt_len]: |
| _x = copy.deepcopy(x) |
| for value in _x: |
| value["prompt"] = prompt |
| y.append(_x) |
| ybar = prompt_list[prompt_len] |
| else: |
| for keyword in second_prompt: |
| _x = copy.deepcopy(x) |
| for value in _x: |
| prompt = value["prompt"] |
| prompt = [item.strip() for item in prompt.split(',')] |
| idx = prompt.index(second_prompt_keyword) |
| prompt[idx] = keyword |
| value["prompt"] = ', '.join(prompt) |
| y.append(_x) |
| ybar = second_prompt |
| elif xyz_list[i] == "ํ๋กฌํํธ ์ค์": |
| if prompt_len == 0: |
| for prompt in prompt_list[prompt_len]: |
| _x = copy.deepcopy(x) |
| for value in _x: |
| value["prompt"] = prompt |
| y.append(_x) |
| ybar = prompt_list[prompt_len] |
| else: |
| for keyword in second_prompt: |
| _x = copy.deepcopy(x) |
| for value in _x: |
| prompt = value["prompt"] |
| prompt = [item.strip() for item in prompt.split(',')] |
| idx = prompt.index(second_prompt_keyword) |
| prompt[idx] = keyword |
| value["prompt"] = ', '.join(prompt) |
| y.append(_x) |
| ybar = second_prompt |
| elif xyz_list[i] == "SMEA": |
| for tp in smea_list: |
| _x = copy.deepcopy(x) |
| for value in _x: |
| value["sema"], value["sema_dyn"] = tp |
| y.append(_x) |
| ybar = ['None', "SMEA", "SMEA+DYN"] |
| elif xyz_list[i] == "Samplers": |
| for sam in sampler_list: |
| _x = copy.deepcopy(x) |
| for value in _x: |
| value["sampler"] = sam |
| y.append(_x) |
| ybar = ["k_euler", "k_euler_ancestral", "k_dpmpp_2s_ancestral", "k_dpmpp_sde", "k_dpmpp_2m"] |
| xyz_list = [] |
| prompt_list = [] |
| prompt_len = 0 |
|
|
| def run_generation(xy, xbar, ybar=None): |
| if isinstance(xy[0], list): |
| ydim = len(xy) |
| xdim = len(xy[0]) |
| else: |
| ydim = 1 |
| xdim = len(xy) |
|
|
| max_count = str(ydim * xdim) |
| generation_label.configure(text = "0" +' / '+ max_count) |
| if not ybar: ybar = " " |
|
|
| label_font_size = 40 |
| label_space = 500 |
| image_width = int(NAI_width) |
| image_height = int(NAI_height) |
| canvas_width = label_space + image_width * xdim |
| canvas_height = image_height * ydim + label_space |
| final_image = Image.new("RGB", (canvas_width, canvas_height), "white") |
| font = ImageFont.truetype("arial.ttf", label_font_size) |
| draw = ImageDraw.Draw(final_image) |
| draw.text( |
| (label_space +((image_width * xdim) / 2), label_space // 3), |
| f"seed : {self.seed_entry.get()}", |
| fill="black", |
| font=font, |
| anchor="mm" |
| ) |
|
|
| xyz_execute_button.configure(state = "disabled") |
| generation_stop_button.configure(state = "normal") |
| pass_counter = 0 |
|
|
| for yloop in range(ydim): |
| if self.instant_stop_bit: |
| break |
| for xloop in range(xdim): |
| if ydim > 1: |
| gen_request = xy[yloop][xloop] |
| else: |
| gen_request = xy[xloop] |
| app.image_generation_button.configure(border_width = 2) |
| generation_pass = False |
| while_counter = 0 |
| while (not self.instant_stop_bit and not generation_pass): |
| self.state_label.configure(text ="state : NAI ์ด๋ฏธ์ง ์์ฑ ๋๊ธฐ์ค ... ", text_color = "#FFFF97") |
| if while_counter >= 15: |
| self.instant_stop_bit = True |
| if gen_request["png_rule"] == "count": |
| app.generation_count += 1 |
| gen_request["count"] =app.generation_count |
| result_image, result_prompt, result_seed, info, filename = NAIA_generation.generate(gen_request) |
| self.state_label.configure(text ="state : idle", text_color = "#DCE4EE") |
| app.image_generation_button.configure(border_width = 0) |
| if info: |
| temp = info.get('Comment', '') |
| temp = temp[temp.find("prompt")+10:temp.find("skip_cfg_below_sigma")-3].replace('"','') |
| generation_pass = True |
| pass_counter += 1 |
| else: |
| temp = result_prompt |
| while_counter += 1 |
| generation_pass = False |
| self.running_flag = False |
| generation_label.configure(text = str(pass_counter) +' / '+ max_count) |
| self.image_label_report.configure(state="normal") |
| self.image_label_report.delete("0.0", "end") |
| self.image_label_report.configure(text_color="#DCE4EE") |
| self.image_label_report.insert("0.0", temp) |
| self.image_label_report.configure(state="disabled") |
| if result_image: |
| if app.state() != 'zoomed': |
| instant_result_image = customtkinter.CTkImage(result_image, size=(620,620)) |
| else: |
| current_image = Image.open(filename) |
| original_width, original_height = current_image.size |
| max_size = app.winfo_screenheight()-100 |
| if original_width > max_size or original_height > max_size: |
| new_image = Image.new("RGB", (max_size, max_size), "black") |
| new_image.paste(current_image, ((max_size - original_width) // 2, (max_size - original_height) // 2)) |
| instant_result_image = customtkinter.CTkImage(new_image, size=(max_size, max_size)) |
| else: |
| instant_result_image = customtkinter.CTkImage(current_image, size=(max_size, max_size)) |
| self.image_label.configure(image=instant_result_image) |
| app.ext_set_image_to_queue(result_image, result_prompt, str(result_seed), filename) |
| x_position = label_space + (xloop * image_width) |
| y_position = label_space + (yloop * image_height) |
| final_image.paste(Image.open(filename), (x_position, y_position)) |
| draw = ImageDraw.Draw(final_image) |
| if yloop == 0: |
| draw.text( |
| (x_position + image_width / 2, label_space / 2), |
| xbar[xloop], |
| fill="black", |
| font=font, |
| anchor="mm" |
| ) |
| if xloop == 0: |
| draw.text( |
| (label_space // 2, y_position + image_height // 2), |
| ybar[yloop], |
| fill="black", |
| font=font, |
| anchor="mm" |
| ) |
| if self.instant_stop_bit: |
| break |
| time.sleep(2) |
| image_path = f"xy_plot_{self.start_time}_{self.xy_plot_count}.jpg" |
| final_image.save(image_path) |
| os.startfile(image_path) |
| self.xy_plot_count += 1 |
| generation_stop_button.configure(state = "disabled") |
| xyz_execute_button.configure(state="normal") |
| self.instant_stop_bit = False |
| pass_counter = 0 |
| |
|
|
| if y: |
| generation_thread = threading.Thread(target=run_generation, args=(y, xbar, ybar), daemon=True) |
| generation_thread.start() |
| else: |
| generation_thread = threading.Thread(target=run_generation, args=(x, xbar), daemon=True) |
| generation_thread.start() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
| xyz_frame = customtkinter.CTkFrame(tabview.tab("X/Y plot")) |
| xyz_frame.grid(row=1, column=0, padx=5, pady=5) |
| xyz_label1 = customtkinter.CTkLabel(xyz_frame, text="X value : ", font=my_font) |
| xyz_label1.grid(row=0, column=0, padx=5, pady=5) |
| xyz_label2 = customtkinter.CTkLabel(xyz_frame, text="Y value : ", font=my_font) |
| xyz_label2.grid(row=1, column=0, padx=5, pady=5) |
| xyz_label3 = customtkinter.CTkLabel(xyz_frame, text="Z value : ", font=my_font) |
| |
| xyz_values = ['None','CFG Scale', 'ํ๋กฌํํธ ๊ฐ์กฐ', 'PG.Rescale', 'SMEA 0/1/dyn', 'Samplers', 'ํ๋กฌํํธ ์ค์'] |
| xyz_box1 = customtkinter.CTkComboBox(xyz_frame, values=xyz_values, font=my_font) |
| xyz_box2 = customtkinter.CTkComboBox(xyz_frame, values=xyz_values, font=my_font) |
| xyz_box3 = customtkinter.CTkComboBox(xyz_frame, values=xyz_values, font=my_font, state="disabled") |
| xyz_box1.grid(row=0, column=1, padx=5, pady=5) |
| xyz_box2.grid(row=1, column=1, padx=5, pady=5) |
| |
| xyz_prompt1 = customtkinter.CTkTextbox(xyz_frame, width= 300, height=80, font=my_font) |
| xyz_prompt2 = customtkinter.CTkTextbox(xyz_frame, width= 300, height=80, font=my_font) |
| xyz_prompt3 = customtkinter.CTkTextbox(xyz_frame, width= 300, height=80, font=my_font, state="disabled") |
| xyz_prompt1.grid(row=0, column=2, padx=5, pady=5) |
| xyz_prompt2.grid(row=1, column=2, padx=5, pady=5) |
| |
| xyz_execute_button = customtkinter.CTkButton(xyz_frame, text="์์
์์", font=my_font, command=xyz_excute) |
| xyz_execute_button.grid(row=3, columnspan=3, padx=5, pady=5, sticky="n") |
| |
| ''' |
| self.resolution_button2 = customtkinter.CTkComboBox(self.hidden_frame, width=160,values=["1024 x 1024", "960 x 1088", "896 x 1152", "832 x 1216", "1088 x 960", "1152 x 896", "1216 x 832"], variable=self.resolution_var, font=my_font) |
| self.resolution_button2.grid(row=3, column=1, padx=5, sticky="w")' |
| ''' |
|
|
|
|
|
|
|
|
| if __name__ == "__main__": |
| customtkinter.set_appearance_mode("dark") |
|
|
| |
| iimg = None |
| img2img_window = None |
| inpaint_mode = False |
| |
|
|
| app = App() |
| app.load_settings() |
| app.protocol("WM_DELETE_WINDOW", app.exit_program) |
| app.activate_wildcards() |
| app.get_anlas() |
| app.after(1000, lambda: setattr(app, 'last_window_size', (app.winfo_width(), app.winfo_height()))) |
| app.mainloop() |
|
|
