Upload 3 files

.gitattributes

@@ -126,3 +126,4 @@ Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0214_testv1.exe filte
 Danbooru[[:space:]]Prompt[[:space:]]Selector/NAIA_0214_testv2.exe filter=lfs diff=lfs merge=lfs -text
 Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0216_testv1.exe filter=lfs diff=lfs merge=lfs -text
 Danbooru[[:space:]]Prompt[[:space:]]Selector/TEST2024/NAIA_0217_testv1.exe filter=lfs diff=lfs merge=lfs -text
+NAIA_0217_testv1.exe filter=lfs diff=lfs merge=lfs -text

NAIA_0217_testv1.exe

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8803a6e4f5776eff4333f0733ca9ac2c27befa04f810906d52da3cf6ed4095bc
+size 888962570

NAIA_random_function_core.py

@@ -0,0 +1,206 @@
+import pandas as pd
+import numpy as np
+import re
+
+def parse_and_execute_commands(processed, user_input, fix, after, popped_row):
+    commands = [cmd.strip() for cmd in user_input.split(',')]
+    for command in commands:
+        condition, cmd = parse_conditional_command(command)
+        if check_condition(processed, condition, popped_row):
+            processed = execute_command(processed, cmd, fix, after)
+    return processed
+
+def parse_conditional_command(command):
+    match = re.match(r"\((.*?)\)\:(.*)", command)
+    if match:
+        return match.groups()
+    return '', command
+
+def check_condition(processed, condition, popped_row):
+    if not condition:
+        return True
+    sub_conditions = re.split(r'\)\s*&\s*\(', condition)
+    sub_conditions = [re.sub(r'^\(|\)$', '', cond) for cond in sub_conditions]
+
+    results = []
+    for sub_cond in sub_conditions:
+        if '&' in sub_cond:
+            results.append(all(check_condition(processed, cond, popped_row) for cond in sub_cond.split('&')))
+        elif '|' in sub_cond:
+            results.append(any(check_condition(processed, cond, popped_row) for cond in sub_cond.split('|')))
+        else:
+            if sub_cond in ['e', 'q', 's', 'g']:
+                results.append(sub_cond == popped_row['rating'])
+            elif sub_cond in ['~e', '~q', '~s', '~g']:
+                results.append(sub_cond != popped_row['rating'])
+            # PM
+            elif sub_cond.startswith('*'):
+                results.append(sub_cond[1:] in processed)
+            # NOT IN
+            elif sub_cond.startswith('~!'):
+                results.append(sub_cond[2:] not in processed)
+            elif sub_cond.startswith('~'):
+                results.append(any(sub_cond[1:] not in element for element in processed))
+            # CONTAIN
+            else:
+                results.append(any(sub_cond in element for element in processed))
+    return all(results)
+
+def execute_command(processed, command, fix, after):
+    if '+=' in command:
+        keyword, addition = command.split('+=', 1)
+        addition = addition.replace('^', ', ')
+        #addiction split 해서 프롬프트 존재하는지 체크 필요
+        return insert_text_after_keyword(processed, keyword, addition, fix, after)
+    elif '=' in command:
+        keyword, replacement = command.split('=', 1)
+        if keyword in processed:
+            replacement = replacement.replace('^', ', ')
+            #replacement split 해서 프롬프트 존재하는지 체크 필요
+            index = processed.index(keyword)
+            processed[index] = replacement
+    return processed
+
+def insert_text_after_keyword(processed, user_keyword, user_additional_keyword, fix, after):
+    if user_keyword == "prompt":
+        processed.append(user_additional_keyword)
+    elif user_keyword == "prefix":
+        fix.append(user_additional_keyword)
+    elif user_keyword == "postfix":
+        after.append(user_additional_keyword)
+    elif user_keyword in processed:
+        index = processed.index(user_keyword) + 1
+        processed.insert(index, user_additional_keyword)
+    return processed
+
+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: 해당 사항 없음
+    # 2월 12일에 해당 부분 return 2 -> return 0 으로 수정하였으며 이 부분 트래킹 필요
+    return 0
+
+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(',')]
+    special_word_check = fix_prompt+', '+after_prompt
+    special_word_check = [item.strip() for item in special_word_check.split(',')]
+    special_word_check = [item for item in special_word_check if item.startswith('*')]
+    exc = []
+    for keyword in general:
+        if keyword == '!' or keyword == '!?' or keyword == '!!':
+            exc.append(keyword)
+    if exc:
+        for keyword in exc:
+            general.remove(keyword)
+        general[3:3] = exc
+
+    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.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(',')]
+            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(',')]
+        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 = []
+    for keyword in processed:
+        if keyword in auto_hide:
+            temp_hide_prompt.append(keyword)
+    for keyword in temp_hide_prompt:
+        processed.remove(keyword)
+    #여기에서 문제의 magic word 처리
+    if "cond" in magic_word and magic_word["cond"]:
+        user_input = magic_word["cond"]
+        processed = parse_and_execute_commands(processed, user_input, fix, after, popped_row)
+    
+    processed[fix_index:fix_index] = fix
+    processed += after
+
+    if rm_c == 0 and '*(remove character name)' not in special_word_check and '*(set character name after prefix)' not in special_word_check and '*(set character name before prefix)' not in special_word_check:
+        if popped_row['character']:
+            character = [item.strip() for item in popped_row['character'].split(',')]
+            processed[fix_index:fix_index] = character
+            fix_index+=len(character)
+    elif rm_c == 0 and '*(set character name after prefix)' in special_word_check:
+        if popped_row['character']:
+            character = [item.strip() for item in popped_row['character'].split(',')]
+            processed[fix_index+len(fix):fix_index+len(fix)] = character
+            #fix_index+=len(character)
+    elif rm_c == 0 and '*(set character name before prefix)' in special_word_check:
+        if popped_row['character']:
+            character = [item.strip() for item in popped_row['character'].split(',')]
+            processed[fix_index+1:fix_index+1] = 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)
+
+    s_word = ['*(remove character name)', '*(set character name after prefix)', '*(set character name before prefix)']
+    for keyword in s_word:
+        if keyword in processed:
+            processed.remove(keyword)
+            
+    if at_first:
+        processed = at_first + processed
+    return ', '.join(processed), popped_row['rating']