Move thumb to remote

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, original_character_list, LANG = init()
+    character_list, LANG = init()
     
     with gr.Blocks(js=JAVA_SCRIPT, css=CSS_SCRIPT, title=TITLE) as ui:
         with gr.Row():
@@ -29,14 +29,6 @@ if __name__ == '__main__':
                 value='none',
                 allow_custom_value=False,
             )
-            '''
-            original_character = gr.Dropdown(
-                choices=original_character_list,
-                label=LANG["original_character"],
-                value='none',
-                allow_custom_value=False,
-            )
-            '''
             
         with gr.Row(elem_classes='main_row'):            
             with gr.Column(elem_classes='column_prompts'):

json/wai_character_thumbs.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:184e32e34ca9807614122651f9555ede639a19d7471ac0968ac0cbef22b98c62
-size 135382738
+oid sha256:e00bd43a69d59ff391b672a652f0e9b78c3fb4cc1fe146c6e08a9a425b4427cd
+size 135380826

scripts/lib.py

@@ -58,14 +58,11 @@ json_folder = os.path.join(parent_dir, 'json')
 
 character_list = ''
 character_dict = {}
-original_character_list = ''
-original_character_dict = {}
 wai_image_dict = {}
 
 wai_illustrious_character_select_files = [
-    {'name': 'original_character', 'file_path': os.path.join(json_folder, 'original_character.json'), 'url': 'local'},
-    {'name': 'wai_character', 'file_path': os.path.join(json_folder, 'wai_characters.csv'), 'url': 'local'},
-    {'name': 'wai_image', 'file_path': os.path.join(json_folder, 'wai_character_thumbs.json'), 'url': 'local'},
+    {'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'},
+    {'name': 'wai_image', 'file_path': os.path.join(json_folder, 'wai_character_thumbs.json'), 'url': 'https://huggingface.co/datasets/flagrantia/character_select_stand_alone_app/resolve/main/wai_character_thumbs.json'},
 ]
 def get_md5_hash(input_str):
     md5_hash = hashlib.md5()
@@ -77,12 +74,17 @@ def base64_to_image(base64_data):
     webp_data = gzip.decompress(compressed_data)
     image = Image.open(BytesIO(webp_data))  
     return image
+
+def download_file(url, file_path):   
+    response = requests.get(url)
+    response.raise_for_status() 
+    print(f'[{CAT}]:Downloading... {url}')
+    with open(file_path, 'wb') as file:
+        file.write(response.content)        
             
 def load_jsons():
     global character_list
     global character_dict
-    global original_character_dict
-    global original_character_list
     global wai_image_dict
     
     # download file
@@ -90,11 +92,12 @@ def load_jsons():
         name = item['name']
         file_path = item['file_path']
         url = item['url']        
+        
+        if not os.path.exists(file_path):
+            download_file(url, file_path)
             
-        with open(file_path, 'r', encoding='utf-8') as file:
-            if 'original_character' == name:
-                original_character_dict.update(json.load(file))                
-            elif 'wai_character' == name:
+        with open(file_path, 'r', encoding='utf-8') as file:        
+            if 'wai_character' == name:
                 lines = file.readlines()
                 for line in lines:
                     key, value = line.split(',')
@@ -102,14 +105,9 @@ def load_jsons():
             elif 'wai_image' == name:
                 wai_image_dict = json.load(file)                
                                         
-    # Create list
-        
+    # Create list        
     character_list = list(character_dict.values())   
-    
     character_list.insert(0, "none")
-    
-    original_character_list = list(original_character_dict.keys())    
-    original_character_list.insert(0, "none")
 
 def illustrious_character_select_ex(character = 'random', optimise_tags = True, random_action_seed = 1):
     chara = ''
@@ -147,30 +145,6 @@ def illustrious_character_select_ex(character = 'random', optimise_tags = True,
             
     return rnd_character, opt_chara, thumb_image
 
-def original_character_select_ex(character = 'random', random_action_seed = 1):
-    chara = ''
-    rnd_character = ''
-    
-    if 'none' == character:
-        return '', ''
-    
-    if 'random' == character:
-        index = random_action_seed % len(original_character_list)
-        rnd_character = original_character_list[index]
-        if 'random' == rnd_character:
-            rnd_character = original_character_list[index+2]
-        elif 'none' == rnd_character:
-            rnd_character = original_character_list[index+1]
-    else:
-        rnd_character = character
-    chara = original_character_dict[rnd_character]                                    
-    
-    opt_chara = chara
-    if not opt_chara.endswith(','):
-        opt_chara = f'{opt_chara},'   
-            
-    return rnd_character, opt_chara
-
 def refresh_character_thumb_image(character1, character2, character3):
     thumb_image = []
     if 'none' != character1 and 'random' != character1:
@@ -197,4 +171,4 @@ def init():
     load_jsons()    
     print(f'[{CAT}]:Starting...')
     
-    return character_list, original_character_list, LANG
+    return character_list, LANG