Show CN name

.gitignore

@@ -1 +1,3 @@
 /scripts/__pycache__
+json/wai_character_thumbs.json
+json/wai_characters.csv

app.py

@@ -5,7 +5,7 @@ from lib import init, refresh_character_thumb_image
 from lib import JAVA_SCRIPT, CSS_SCRIPT, TITLE
 
 if __name__ == '__main__':
-    character_list, LANG = init()
+    character_list, character_list_cn, LANG = init()
     
     with gr.Blocks(js=JAVA_SCRIPT, css=CSS_SCRIPT, title=TITLE) as ui:
         with gr.Row():
@@ -24,7 +24,7 @@ if __name__ == '__main__':
             )
                             
             character3 = gr.Dropdown(
-                choices=character_list,
+                choices=character_list_cn,
                 label=LANG["character3"],
                 value='none',
                 allow_custom_value=False,

scripts/lib.py

@@ -18,9 +18,9 @@ import argparse
 
 # Language
 LANG_EN = {
-    "character1": "Character list 1",
-    "character2": "Character list 2",
-    "character3": "Character list 3",
+    "character1": "Character list EN",
+    "character2": "Character list EN",
+    "character3": "Character list CN",
     "action": "Action list",
     "original_character": "Original Character",    
 }
@@ -59,6 +59,7 @@ json_folder = os.path.join(parent_dir, 'json')
 character_list = ''
 character_dict = {}
 wai_image_dict = {}
+character_list_cn = ''
 
 wai_illustrious_character_select_files = [
     {'name': 'wai_character', 'file_path': os.path.join(json_folder, 'wai_characters.csv'), 'url':'https://raw.githubusercontent.com/mirabarukaso/character_select_stand_alone_app/refs/heads/main/json/wai_characters.csv'},
@@ -86,6 +87,7 @@ def load_jsons():
     global character_list
     global character_dict
     global wai_image_dict
+    global character_list_cn
     
     # download file
     for item in wai_illustrious_character_select_files:
@@ -108,8 +110,10 @@ def load_jsons():
     # Create list        
     character_list = list(character_dict.values())   
     character_list.insert(0, "none")
+    character_list_cn = list(character_dict.keys())   
+    character_list_cn.insert(0, "none")
 
-def illustrious_character_select_ex(character = 'random', optimise_tags = True, random_action_seed = 1):
+def illustrious_character_select_ex(character = 'random', optimise_tags = True, random_action_seed = 1, use_cn=False):
     chara = ''
     rnd_character = ''
     
@@ -125,7 +129,7 @@ def illustrious_character_select_ex(character = 'random', optimise_tags = True,
             rnd_character = character_list[index+1]
     else:
         rnd_character = character
-    if ENGLISH_CHARACTER_NAME:
+    if not use_cn:
         chara = rnd_character
     else:
         chara = character_dict[rnd_character]
@@ -148,27 +152,19 @@ def illustrious_character_select_ex(character = 'random', optimise_tags = True,
 def refresh_character_thumb_image(character1, character2, character3):
     thumb_image = []
     if 'none' != character1 and 'random' != character1:
-        _, _, thumb_image1 = illustrious_character_select_ex(character = character1, random_action_seed=42)        
+        _, _, thumb_image1 = illustrious_character_select_ex(character = character1)        
         thumb_image.append(thumb_image1)
     
     if 'none' != character2 and 'random' != character2:
-        _, _, thumb_image2 = illustrious_character_select_ex(character = character2, random_action_seed=42)        
+        _, _, thumb_image2 = illustrious_character_select_ex(character = character2)        
         thumb_image.append(thumb_image2)
     if 'none' != character3 and 'random' != character3:
-        _, _, thumb_image3 = illustrious_character_select_ex(character = character3, random_action_seed=42)        
+        _, _, thumb_image3 = illustrious_character_select_ex(character = character3, use_cn=True)        
         thumb_image.append(thumb_image3)
     return thumb_image
 
-def init():
-    global ENGLISH_CHARACTER_NAME
-    global LANG
-        
-    ENGLISH_CHARACTER_NAME = True
-    if ENGLISH_CHARACTER_NAME:
-        print(f'[{CAT}]:Use tags as Character Name')
-        LANG = LANG_EN
-        
+def init():        
     load_jsons()    
     print(f'[{CAT}]:Starting...')
     
-    return character_list, LANG
+    return character_list, character_list_cn, LANG