Upload 2 files

Danbooru Prompt Selector/TEST2024/prompt_selector_auto_1212_testv8.exe

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:99a1fa3d70fb86abd05f3b1ab6b731597d1e81ff3a2c0e3e8668db60d81aa7b2
-size 150866140
+oid sha256:1d955af933c28910c437d6166dcb3a8c68ed22773991b53a81bdb1f9bd6f384a
+size 151425735

Danbooru Prompt Selector/TEST2024/prompt_selector_auto_1212_testv8.py

@@ -27,11 +27,14 @@ from IPython.display import display
 import win32clipboard
 from io import BytesIO
 from datetime import datetime
-import arti_list, tagbag, wlist
+import arti_list, tagbag, wlist, copyright_dict
 import character_dictionary as cd
 import pandas as pd
+import webbrowser
 
 BASE_URL="https://api.novelai.net"
+copyright_keys = list(copyright_dict.copyright_dict.keys())
+character_keys = list(cd.character_dictionary.keys())
 
 def argon_hash(email: str, password: str, size: int, domain: str) -> str:
     pre_salt = f"{password[:6]}{email}{domain}"
@@ -317,19 +320,48 @@ def random_function():
     include_keywords = [kw for kw in remaining_keywords if not kw.startswith('~') and not kw.startswith('*')]
 
     if current_deep_search_keywords != last_deep_search_keywords or not cached_rows:
-        with open('txt2img_temp_prompt.csv', 'r', newline='', encoding='utf-8') as f:
-            reader = csv.reader(f)
-            rows = []
-            for row in reader:
-                if all(re.search(r'(?:^|, )' + re.escape(kw) + r'(?:,|$)', ', '.join(row)) for kw in perfect_match_keywords) \
-                   and not any(exclude in cell for exclude in exclude_keywords for cell in row) \
-                   and all(include in cell for include in include_keywords for cell in row):
-                    # '{}' 안의 키워드 처리
-                    if all(any(re.search(r'(?:^|, )' + re.escape(kw) + r'(?:,|$)', ', '.join(cell)) for kw in perfect_group) or
-                           any(kw in cell for kw in regular_group) for perfect_group, regular_group in processed_keywords for cell in row):
-                        rows.append(row)
-            cached_rows = rows
-
+        if not os.path.exists('txt2img_temp_prompt.csv'):
+            text_output.delete('1.0', tk.END)
+            text_output.insert(tk.END, "txt2img_temp_prompt.csv가 없어 재검색을 수행합니다. 이 작업에는 매우 긴 시간이 소요됩니다. tags.csv 파일에 대해 초기화 키워드 및 .csv 내보내기 기능을 활용 해 주세요.")
+        try:
+            with open('txt2img_temp_prompt.csv', 'r', newline='', encoding='utf-8') as f:
+                reader = csv.reader(f)
+                rows = []
+                for row in reader:
+                    if all(re.search(r'(?:^|, )' + re.escape(kw) + r'(?:,|$)', ', '.join(row)) for kw in perfect_match_keywords) \
+                    and not any(exclude in cell for exclude in exclude_keywords for cell in row) \
+                    and all(include in cell for include in include_keywords for cell in row):
+                        # '{}' 안의 키워드 처리
+                        if all(any(re.search(r'(?:^|, )' + re.escape(kw) + r'(?:,|$)', ', '.join(cell)) for kw in perfect_group) or
+                            any(kw in cell for kw in regular_group) for perfect_group, regular_group in processed_keywords for cell in row):
+                            rows.append(row)
+                cached_rows = rows
+        except:
+            try:
+                search()
+                with open('txt2img_temp_prompt.csv', 'r', newline='', encoding='utf-8') as f:
+                    reader = csv.reader(f)
+                    rows = []
+                    for row in reader:
+                        if all(re.search(r'(?:^|, )' + re.escape(kw) + r'(?:,|$)', ', '.join(row)) for kw in perfect_match_keywords) \
+                        and not any(exclude in cell for exclude in exclude_keywords for cell in row) \
+                        and all(include in cell for include in include_keywords for cell in row):
+                            # '{}' 안의 키워드 처리
+                            if all(any(re.search(r'(?:^|, )' + re.escape(kw) + r'(?:,|$)', ', '.join(cell)) for kw in perfect_group) or
+                                any(kw in cell for kw in regular_group) for perfect_group, regular_group in processed_keywords for cell in row):
+                                rows.append(row)
+                cached_rows = rows
+            except:
+                def open_link(url):
+                    webbrowser.open_new(url)
+
+                text_output.delete('1.0', tk.END)
+                text_output.insert(tk.END, "tags.csv 파일을 [파일 열기] 버튼을 눌러 탑재하거나,")
+                text_output.insert(tk.END, "https://huggingface.co/baqu2213/PoemForSmallFThings/blob/main/Danbooru%20Prompt%20Selector/tags_2023.csv", "link")
+                text_output.insert(tk.END, " 에서 다운로드 하시기 바랍니다.")
+                text_output.tag_config("link", foreground="blue", underline=True)
+                text_output.tag_bind("link", "<Button-1>", lambda e: open_link("https://huggingface.co/baqu2213/PoemForSmallFThings/blob/main/Danbooru%20Prompt%20Selector/tags_2023.csv"))
+                return 
         last_deep_search_keywords = current_deep_search_keywords
     if not toggle_prompt_var.get(): text_output.delete('1.0', tk.END)
     
@@ -439,7 +471,7 @@ def random_function():
                 if kw in random_row_keywords:
                     random_row_keywords.remove(kw)
 
-            global previous_wildcard_artist, previous_wildcard_character
+            global previous_wildcard_artist, previous_wildcard_character, character_keys
             # temp_first_keywords에 boy와 girl 키워드를 적절한 위치에 추가
             if(girl_keywords and 'girl' in temp_first_keywords[0]): temp_first_keywords.pop(0)
             #print(wildcard_var.get())
@@ -493,7 +525,7 @@ def random_function():
             temp_rm = []
             if rm_characteristic_var.get():
                 for keyword in random_row_keywords:
-                    if keyword in bag_of_tags or "(" in keyword or "horns" in keyword:
+                    if keyword in bag_of_tags or keyword in character_keys or "(" in keyword or "horns" in keyword:
                         temp_rm.append(keyword)
                 for keyword in temp_rm:
                     random_row_keywords.remove(keyword)
@@ -538,6 +570,53 @@ def random_function():
 
         random_row_keywords = [keyword.strip(', ') for keyword in random_row_keywords]
         random_row_keywords = [keyword for keyword in random_row_keywords if keyword]
+
+        rm_copyright = []
+        global copyright_keys
+        if(rm_copyright_var.get() == 1):
+            for keyword in random_row_keywords:
+                if keyword in copyright_keys:
+                    rm_copyright.append(keyword)
+            if rm_copyright is not None:
+                for keyword in rm_copyright:
+                    random_row_keywords.remove(keyword)
+        else:
+            idx = 0
+            for keyword in random_row_keywords:
+                if keyword in copyright_keys:
+                    random_row_keywords[idx] = keyword+" (copyright)"
+                idx += 1
+        pop_character = []
+        for keyword in random_row_keywords:
+            if keyword in character_keys:
+                pop_character.append(keyword)
+        if pop_character is not None:
+            found_index = 0
+            # Search for 'girl' in the first four elements
+            for i in range(min(10, len(random_row_keywords))):
+                if 'girl' in random_row_keywords[i]:
+                    found_index = i
+                    break
+                elif 'boy' in random_row_keywords[i]:
+                    found_index = i
+                    break
+                elif 'other' in random_row_keywords[i]:
+                    found_index = i
+                    break
+            for keyword in pop_character:
+                found_index += 1
+                random_row_keywords.remove(keyword)
+                random_row_keywords.insert(found_index,keyword)
+            pop_artist = []
+            for keyword in random_row_keywords:
+                if keyword in afilter_30000:
+                    pop_artist.append(keyword)
+            if pop_artist is not None:
+                for keyword in pop_artist:
+                    found_index += 1
+                    random_row_keywords.remove(keyword)
+                    random_row_keywords.insert(found_index,"artist:"+keyword)
+
         #print(random_row_keywords)
         text_output.insert(tk.END, f"{', '.join(random_row_keywords)}")
         if auto_copy_var.get():
@@ -2342,6 +2421,10 @@ widldcard_button.grid(row=1, column=2, sticky='ew')
 folder_button = tk.Button(checkbox_frame, text="저장 폴더", command=open_file_explorer) #output_file_path
 folder_button.grid(row=1, column=3,  padx=5, sticky='ew')
 
+# "작품명 제거" 체크박스
+rm_copyright_var = tk.IntVar()
+rm_copyright_label = tk.Checkbutton(checkbox_frame, text="작품명 제거", variable=rm_copyright_var)
+rm_copyright_label.grid(row=2, column=0, pady=5, sticky='ew')
 
 window.bind('<Control-Return>', on_ctrl_enter)
 window.bind('<Shift-Return>', on_shift_enter)
@@ -2352,10 +2435,12 @@ if not (entry_fixed_prompt.get()): entry_fixed_prompt.insert(0, '1girl,')
 if not (entry_fixed_prompt_after.get()): entry_fixed_prompt_after.insert(0, 'great quality, aesthetic, absurdres')
 if not (entry_auto_hide.get()): entry_auto_hide.insert(0, 'monochrome, doujin cover, bad source, censored, bar censor')
 
+
+
 resolution1_frame = tk.Frame(left_frame)
-resolution1_frame.grid(row=15, column=0,  padx=5, pady=5, sticky='ew')
+resolution1_frame.grid(row=16, column=0,  padx=5, pady=5, sticky='ew')
 resolution2_frame = tk.Frame(left_frame)
-resolution2_frame.grid(row=16, column=0, padx=5, pady=5, sticky='ew')
+resolution2_frame.grid(row=17, column=0, padx=5, pady=5, sticky='ew')
 selected_resolution = tk.StringVar(value="1024 x 1024")
 
 tk.Radiobutton(resolution1_frame, text="1024 x 1024", variable=selected_resolution, value="1024 x 1024").grid(row=0, column=0, sticky='w')
@@ -2394,7 +2479,7 @@ option_menu = tk.OptionMenu(resolution2_frame, selected_option, *options, comman
 option_menu.grid(row=0, column=6,  padx=5, sticky='ew')
 
 resolution3_frame = tk.Frame(left_frame)
-resolution3_frame.grid(row=17, column=0, padx=5, pady=5, sticky='ew')
+resolution3_frame.grid(row=18, column=0, padx=5, pady=5, sticky='ew')
 resolution3_frame.columnconfigure(0, weight=2) 
 resolution3_frame.columnconfigure(1, weight=1) 
 resolution3_frame.columnconfigure(2, weight=1)