Upload 3 files

Danbooru Prompt Selector/TEST2024/NAIA_0217_testv1.exe

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b86106cdd1e97f072bea374d18a7b19112c2a49ccd21a826052cbc8c52fb8023
-size 888962451
+oid sha256:df93f967cae2277ce50bd3b520c453a62489f4e88d599fb0e6f7a31fea1cac9d
+size 888962659

Danbooru Prompt Selector/TEST2024/NAIA_0217_testv1.py

@@ -1448,7 +1448,7 @@ class App(customtkinter.CTk, TkinterDnD.DnDWrapper):
                 null_prompt_attention() # 문자열 비었다고 경고
             else:
                 self.cached_rows = df
-                #df.reset_index(drop=True, inplace=True)
+                df.reset_index(drop=True, inplace=True)
                 update_labels(df)
             self.searching_flag = False
             self.search_button.configure(text="검색", state="normal")
@@ -1845,7 +1845,6 @@ class App(customtkinter.CTk, TkinterDnD.DnDWrapper):
                         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)
-                        #print(random_index)
                         popped_row = self.cached_rows.loc[random_index]
                         self.current_popped_row = popped_row.copy()
                         self.cached_rows.drop(random_index, inplace=True)

Danbooru Prompt Selector/TEST2024/NAIA_random_function_core.py

@@ -1,4 +1,4 @@
-import pandas
+import pandas as pd
 import numpy as np
 import re
 
@@ -96,6 +96,8 @@ def find_keyword_index(general):
 def RFP(popped_row, fix_prompt, after_prompt, auto_hide_prompt, rm_a, rm_s, rm_c, nsfw, data, magic_word):
     boys = ["1boy", "2boys", "3boys", "4boys", "5boys", "6+boys"]
     girls = ["1girl", "2girls", "3girls", "4girls", "5girls", "6+girls"]
+    if isinstance(popped_row, pd.DataFrame):
+        popped_row = popped_row.iloc[0]
     general = [item.strip() for item in popped_row['general'].split(',')]
     special_word_check = fix_prompt+', '+after_prompt
     special_word_check = [item.strip() for item in special_word_check.split(',')]