flagrantia/character_select_saa

1import os2import gradio as gr3import sys4sys.path.append(os.path.join(os.path.dirname(__file__), "scripts"))5from lib import init, refresh_character_thumb_image6from lib import TITLE7 8if __name__ == '__main__':9    character_list, character_list_cn, LANG, anima_character_list, anima_character_list_cn = init()10    11    with gr.Blocks(title=TITLE) as ui:12        with gr.Row():            13            character1 = gr.Dropdown(14                choices=anima_character_list,15                label=LANG["character1"],16                value='none',17                allow_custom_value=False,18            )    19                        20            character2 = gr.Dropdown(21                choices=character_list,22                label=LANG["character2"],23                value='none',24                allow_custom_value=False,25            )26            27            character3 = gr.Dropdown(28                choices=anima_character_list_cn,29                label=LANG["character3"],30                value='none',31                allow_custom_value=False,32            )33            34            character4 = gr.Dropdown(35                choices=character_list_cn,36                label=LANG["character4"],37                value='none',38                allow_custom_value=False,39            )40            41            dummy_dropdown = gr.Dropdown(visible=False, allow_custom_value=True)42            dummy_textbox = gr.Textbox(visible=False)43            44        with gr.Row(elem_classes='main_row'):            45            with gr.Column(elem_classes='column_prompts'):46                thumb_image = gr.Gallery(type="pil", allow_preview=True, columns=4, 47                                         show_share_button=False, show_fullscreen_button=False,48                                         object_fit='contain', label="Thumb", height=470)49                output_info = gr.Textbox(label="Characters' Info", lines=18)50        with gr.Row():            51            with gr.Column():52                gr.Markdown("<h2><span style=\"color:orange\">Thumb Image create by waiANIMA_v10Base10.safetensors and waiIllustriousSDXL_v160.safetensors with ComfyUI</span></h2>",)53            with gr.Column():                54                gr.Markdown("<a href='https://github.com/mirabarukaso/character_select_stand_alone_app'>Character Select SAA</a>")            55                gr.Markdown("<a href='https://github.com/mirabarukaso/character_select_stand_alone_app/blob/main/scripts/python/thumb-generator/README.md'>Thumb Generator</a>")            56            57        character1.change(fn=refresh_character_thumb_image,58                          inputs=[character1,character2,character3,character4],59                          outputs=[thumb_image, output_info])60        character2.change(fn=refresh_character_thumb_image,61                          inputs=[character1,character2,character3,character4],62                          outputs=[thumb_image, output_info])63        character3.change(fn=refresh_character_thumb_image,64                          inputs=[character1,character2,character3,character4],65                          outputs=[thumb_image, output_info])66        character4.change(fn=refresh_character_thumb_image,67                          inputs=[character1,character2,character3,character4],68                          outputs=[thumb_image, output_info])69        70    ui.launch(share=False, show_api=False)