Upload 3 files

.gitattributes

@@ -118,3 +118,4 @@ Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/WEBUIA_0119.exe filter=lfs
 Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/WEBUIA_0124.exe filter=lfs diff=lfs merge=lfs -text
 Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0201_testv1.exe filter=lfs diff=lfs merge=lfs -text
 Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0203_testv2.exe filter=lfs diff=lfs merge=lfs -text
+Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/WEBUIA_0211.exe filter=lfs diff=lfs merge=lfs -text

Danbooru Prompt Selector/TEST2024/NAIA_random_function_core (WEBUIA_0211).py

@@ -0,0 +1,118 @@
+import pandas
+import numpy as np
+
+def find_keyword_index(general):
+    # boys와 girls 리스트의 원소가 있는지 확인 및 인덱스 저장
+    boys = ["1boy", "2boys", "3boys", "4boys", "5boys", "6+boys"]
+    girls = ["1girl", "2girls", "3girls", "4girls", "5girls", "6+girls"]
+    #others = ["1other", "2others", "3others", "4others", "5others", "6+others"]
+    boys_indices = [i for i, item in enumerate(general[:6]) if item in boys]
+    girls_indices = [i for i, item in enumerate(general[:6]) if item in girls]
+
+    # case 1과 case 2: girls 리스트의 원소 찾기
+    if girls_indices:
+        return girls_indices[0]+1
+    
+    # case 3: boys 리스트의 원소 찾기
+    if boys_indices:
+        return boys_indices[0]+1
+
+    # case 4: 해당 사항 없음
+    return 2
+
+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"]
+    general = [item.strip() for item in popped_row['general'].split(',')]
+    if nsfw == 1:
+        nsfw_word = []
+        for keyword in general:
+            if keyword in data.qe_word or keyword in data.bag_of_tags or "horns" in keyword or "(" in keyword or keyword in boys or keyword in girls:
+                nsfw_word.append(keyword)
+        general = nsfw_word
+    if rm_c == 1:
+        temp_general = []
+        for keyword in general:
+            if keyword in data.bag_of_tags:
+                temp_general.append(keyword)
+        for keyword in temp_general:
+            general.remove(keyword)
+    fix = [item.strip() for item in fix_prompt[:-1].split(',')]
+    at_first = []
+    for fp in fix:
+        if fp.startswith('*'):
+            at_first.append(fp[1:])
+    if at_first:
+        for af in at_first:
+            if '*'+af in fix:
+                fix.remove('*'+af)
+    if rm_a == 0:
+        if popped_row['artist']:
+            artists = [item.strip() for item in popped_row['artist'].split(',')]
+            #for i, keyword in enumerate(artists):
+            #    if '(' in keyword and ')' in keyword:
+            #        artists[i] = keyword.replace('(', '\\(').replace(')', '\\)')
+            artist = ["artist:" + _artist for _artist in artists]
+            fix = fix + artist
+    if rm_s == 0:
+        if popped_row['copyright']:
+            series = [item.strip() for item in popped_row['copyright'].split(',')]
+            #for i, keyword in enumerate(series):
+            #    if '(' in keyword and ')' in keyword:
+            #        series[i] = keyword.replace('(', '\\(').replace(')', '\\)')
+        else:
+            series = []
+        fix = fix + series
+    after = [item.strip() for item in after_prompt[:-1].split(',')]
+    auto_hide = [item.strip() for item in auto_hide_prompt[:-1].split(',')] + ["| |", ":|", "\||/", "<|> <|>", "|| ||", ";|"]
+    fix_index = find_keyword_index(general)
+    processed = general.copy()
+    temp_hide_prompt = []
+    processed[fix_index:fix_index] = fix
+    processed += after
+    for keyword in processed:
+        if keyword in auto_hide:
+            temp_hide_prompt.append(keyword)
+    for keyword in temp_hide_prompt:
+        processed.remove(keyword)
+
+    if rm_c == 0:
+        if popped_row['character']:
+            character = [item.strip() for item in popped_row['character'].split(',')]
+            #for i, keyword in enumerate(character):
+            #    if '(' in keyword and ')' in keyword:
+            #        character[i] = keyword.replace('(', '\\(').replace(')', '\\)')
+            processed[fix_index:fix_index] = character
+            fix_index+=len(character)
+    if magic_word["random_artist"] == True:
+        processed.insert(fix_index, magic_word["random_artist_name"])
+        
+    boy_in_processed = girl_in_processed = None
+    for boy in boys:
+        if boy in processed:
+            boy_in_processed = boy
+            break
+
+    for girl in girls:
+        if girl in processed:
+            girl_in_processed = girl
+            break
+        
+    if boy_in_processed and girl_in_processed:
+        boy_index = processed.index(boy_in_processed)
+        girl_index = processed.index(girl_in_processed)
+        if boy_index > girl_index:
+            processed.pop(boy_index)
+            processed.insert(girl_index, boy_in_processed)
+    if(popped_row['rating'] == 'e'):
+        rating_text = ', rating: explicit, nsfw'
+    elif(popped_row['rating'] == 'q'):
+        rating_text = ', rating: questionable, nsfw'
+    elif(popped_row['rating'] == 's'):
+        rating_text = ', rating: sensitive'
+    else:
+        rating_text = ', rating: general'
+
+    if at_first:
+        processed = at_first + processed
+    return ', '.join(processed)+rating_text

Danbooru Prompt Selector/TEST2024/WEBUIA_0211.exe

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:934e862dfac3c748a8baecc068269578151770ec14d68f8ceeac03b331bdbb2f
+size 837840755