mirabarukaso
Update tag complete method
e60fe6a
raw
history blame
2.78 kB
import gradio as gr
import sys
sys.path.append("scripts/")
from lib import init, refresh_character_thumb_image, get_prompt_manager
from lib import JAVA_SCRIPT, CSS_SCRIPT, TITLE
if __name__ == '__main__':
character_list, character_list_cn, 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_cn,
label=LANG["character3"],
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', label="Thumb")
with gr.Row():
with gr.Column():
custom_prompt = gr.Textbox(value='', label='Semi-auto tag complete test. Try tag* *tag *tag* (e621_sfw.csv@DominikDoom)')
suggestions = gr.Dataset(components=[custom_prompt], samples_per_page=30, samples=[])
with gr.Column():
gr.Markdown("<h2><span style=\"color:orangered\">Thumb Image create by waiNSFWIllustrious_v120.safetensors with ComfyUI</span></h2>",)
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])
custom_prompt.change(fn=get_prompt_manager().update_suggestions, inputs=[custom_prompt], outputs=[suggestions])
suggestions.select(fn=get_prompt_manager().apply_suggestion, inputs=[suggestions, custom_prompt], outputs=[custom_prompt, suggestions])
ui.launch()