character preview

.gitignore

@@ -0,0 +1 @@
+/scripts/__pycache__

app.py

@@ -0,0 +1,58 @@
+import gradio as gr
+import sys
+sys.path.append("scripts/")
+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()
+    
+    with gr.Blocks(js=JAVA_SCRIPT, css=CSS_SCRIPT, title=TITLE) as ui:
+        with gr.Row():
+            character1 = gr.Dropdown(
+                choices=character_list,
+                label=LANG["character1"],
+                value='none',
+                allow_custom_value=False,
+            )
+            
+            character2 = gr.Dropdown(
+                choices=character_list,
+                label=LANG["character2"],
+                value='none',
+                allow_custom_value=False,
+            )
+                            
+            character3 = gr.Dropdown(
+                choices=character_list,
+                label=LANG["character3"],
+                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'):
+                    thumb_image = gr.Gallery(type="pil", columns=3, show_download_button=False, object_fit='contain', height=512, label="Thumb")    
+                        
+        with gr.Row():
+            gr.Markdown(f"<a href='https://github.com/mirabarukaso/character_select_stand_alone_app'>Character Select SAA</a>")
+            
+        character1.change(fn=refresh_character_thumb_image,
+                          inputs=[character1,character2,character3],
+                          outputs=[thumb_image])
+        character2.change(fn=refresh_character_thumb_image,
+                          inputs=[character1,character2,character3],
+                          outputs=[thumb_image])
+        character3.change(fn=refresh_character_thumb_image,
+                          inputs=[character1,character2,character3],
+                          outputs=[thumb_image])
+        
+    ui.launch()

json/.gitattributes

@@ -0,0 +1 @@
+wai_character_thumbs.json filter=lfs diff=lfs merge=lfs -text

json/original_character.json

@@ -0,0 +1,9 @@
+{
+  "Fla": "two-tone fur, (light blue body fur:1.1), white cheek fur, cyan hair, white fur on cat tail front, orange eyes, slit pupils, furry, miqo'te, petite, ahoge, whiskers, long hair, hair between eyes, ear fluff, glowing eyes, (4 fingers:1.2), claws",
+  "Kaura": "two-tone fur, (silver grey body fur:1.1), (tiger striped:1.15), white cheek fur, grey hair, red eyes, slit pupils, furry, miqo'te, petite, ahoge, whiskers, long hair, hair between eyes, ear fluff, glowing eyes, (4 fingers:1.2), claws",
+  "Caelitis": "silver scales, dragon tail, fluffy tail, (blonde hair, long hair, blue eyes, dragon horn, curved horn, white horn:1.05), pointy ears, (feathered wings:1.05), petite, ahoge, fangs, slit pupils, glowing eyes, dragon humanoid, 5 fingers",
+  "Colus": "black scales, dragon tail, membranous tail, (darker blonde hair, short hair, gold eyes, dragon horn, curved horn, black horn:1.05), pointy ears, (membranous wings:1.05), petite, ahoge, fangs, slit pupils, glowing eyes, dragon humanoid, 5 fingers",
+  "Aura": "two-tone scales, (red scales:1.1), red hair, purple eyes, ahoge, short hair, dragon horn, dragon tail, pointy ears, membranous wings, spread wings, anthro dragon, dragon humanoid, petite, fangs, no pupils, (hair slicked back:1.1), (4 fingers:1.2), claws,  glowing eyes, white cheek",
+  "Caerulea": "two-tone scales, (blue scales:1.1), cyan hair, black sclera, gold eyes, shark tail, (skin-covered dragon horn:1.1), arm fins, ear fins, gills, (mohawk:0.8), sidelocks, anthro dragon, dragon humanoid, petite, fangs, no pupils, (hair slicked back:1.1), (4 fingers:1.2), claws,  glowing eyes, white cheek",
+  "Foxie": "two-tone fur, orange body fur, white cheek fur, red hair, green eyes, (multiple tails:1.1), fox tail,anthro kyuubi, petite, (fangs:0.7), whiskers, hair between eyes, long hair, glowing eyes, ear fluff, claws, (4 fingers:1.2)"
+}

json/wai_character_thumbs.json

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

scripts/__init__.py

@@ -0,0 +1 @@
+# This file is intentionally left blank to mark the directory as a Python package.

scripts/lib.py

@@ -0,0 +1,200 @@
+import datetime
+import gzip
+import hashlib
+import os
+import glob
+import textwrap
+import gradio as gr
+import numpy as np
+import requests
+import json
+import base64
+import webbrowser
+from io import BytesIO
+from PIL import Image
+from PIL import PngImagePlugin
+import random
+import argparse
+
+# Language
+LANG_EN = {
+    "character1": "Character list 1",
+    "character2": "Character list 2",
+    "character3": "Character list 3",
+    "action": "Action list",
+    "original_character": "Original Character",    
+}
+
+LANG = LANG_EN
+
+# JavaScript
+JAVA_SCRIPT = """
+function refresh() {
+    const url = new URL(window.location);
+
+    if (url.searchParams.get('__theme') !== 'dark') {
+        url.searchParams.set('__theme', 'dark');
+        window.location.href = url.href;
+    }
+}
+"""
+
+# CSS
+CSS_SCRIPT = """
+#custom_prompt_text textarea {color: darkorange}
+#positive_prompt_text textarea {color: greenyellow}
+#negative_prompt_text textarea {color: red}
+#ai_prompt_text textarea {color: hotpink}
+#prompt_ban_text textarea {color: Khaki}
+"""
+
+TITLE = "WAI Character Select Preview"
+CAT = "WAI_Character_Select"
+ENGLISH_CHARACTER_NAME = True
+
+current_dir = os.path.dirname(os.path.abspath(__file__))
+parent_dir = os.path.dirname(current_dir)
+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'},
+]
+def get_md5_hash(input_str):
+    md5_hash = hashlib.md5()
+    md5_hash.update(input_str.encode('utf-8'))
+    return md5_hash.hexdigest()
+
+def base64_to_image(base64_data):
+    compressed_data = base64.b64decode(base64_data)
+    webp_data = gzip.decompress(compressed_data)
+    image = Image.open(BytesIO(webp_data))  
+    return image
+            
+def load_jsons():
+    global character_list
+    global character_dict
+    global original_character_dict
+    global original_character_list
+    global wai_image_dict
+    
+    # download file
+    for item in wai_illustrious_character_select_files:
+        name = item['name']
+        file_path = item['file_path']
+        url = item['url']        
+            
+        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:
+                lines = file.readlines()
+                for line in lines:
+                    key, value = line.split(',')
+                    character_dict[key.strip()]=value.strip()
+            elif 'wai_image' == name:
+                wai_image_dict = json.load(file)                
+                                        
+    # 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 = ''
+    rnd_character = ''
+    
+    if 'none' == character:
+        return '', '', None
+    
+    if 'random' == character:
+        index = random_action_seed % len(character_list)
+        rnd_character = character_list[index]
+        if 'random' == rnd_character:
+            rnd_character = character_list[index+2]
+        elif 'none' == rnd_character:
+            rnd_character = character_list[index+1]
+    else:
+        rnd_character = character
+    if ENGLISH_CHARACTER_NAME:
+        chara = rnd_character
+    else:
+        chara = character_dict[rnd_character]
+        
+    md5_chara = get_md5_hash(chara.replace('(','\\(').replace(')','\\)'))                
+    thumb_image = Image.new('RGB', (128, 128), (128, 128, 128))
+    if wai_image_dict.keys().__contains__(md5_chara):
+        thumb_image = base64_to_image(wai_image_dict.get(md5_chara))
+    
+    opt_chara = chara
+    if optimise_tags:
+        opt_chara = opt_chara.replace('(', '\\(').replace(')', '\\)')  
+        #print(f'{CAT}:Optimise Tags:[{chara}]->[{opt_chara}]')
+    
+    if not opt_chara.endswith(','):
+        opt_chara = f'{opt_chara},'   
+            
+    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:
+        _, _, thumb_image1 = illustrious_character_select_ex(character = character1, random_action_seed=42)        
+        thumb_image.append(thumb_image1)
+    
+    if 'none' != character2 and 'random' != character2:
+        _, _, thumb_image2 = illustrious_character_select_ex(character = character2, random_action_seed=42)        
+        thumb_image.append(thumb_image2)
+    if 'none' != character3 and 'random' != character3:
+        _, _, thumb_image3 = illustrious_character_select_ex(character = character3, random_action_seed=42)        
+        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
+        
+    load_jsons()    
+    print(f'[{CAT}]:Starting...')
+    
+    return character_list, original_character_list, LANG