forestcalled/text-generation-webui

1import gradio as gr2 3from modules import chat, presets, shared, ui, utils4from modules.utils import gradio5 6 7def create_ui():8    mu = shared.args.multi_user9 10    # Text file saver11    with gr.Group(visible=False, elem_classes='file-saver') as shared.gradio['file_saver']:12        shared.gradio['save_filename'] = gr.Textbox(lines=1, label='File name')13        shared.gradio['save_root'] = gr.Textbox(lines=1, label='File folder', info='For reference. Unchangeable.', interactive=False)14        shared.gradio['save_contents'] = gr.Textbox(lines=10, label='File contents')15        with gr.Row():16            shared.gradio['save_confirm'] = gr.Button('Save', elem_classes="small-button", variant='primary', interactive=not mu)17            shared.gradio['save_cancel'] = gr.Button('Cancel', elem_classes="small-button")18 19    # Text file deleter20    with gr.Group(visible=False, elem_classes='file-saver') as shared.gradio['file_deleter']:21        shared.gradio['delete_filename'] = gr.Textbox(lines=1, label='File name')22        shared.gradio['delete_root'] = gr.Textbox(lines=1, label='File folder', info='For reference. Unchangeable.', interactive=False)23        with gr.Row():24            shared.gradio['delete_confirm'] = gr.Button('Delete', elem_classes="small-button", variant='stop', interactive=not mu)25            shared.gradio['delete_cancel'] = gr.Button('Cancel', elem_classes="small-button")26 27    # Character saver/deleter28    with gr.Group(visible=False, elem_classes='file-saver') as shared.gradio['character_saver']:29        shared.gradio['save_character_filename'] = gr.Textbox(lines=1, label='File name', info='The character will be saved to your characters/ folder with this base filename.')30        with gr.Row():31            shared.gradio['save_character_confirm'] = gr.Button('Save', elem_classes="small-button", variant='primary', interactive=not mu)32            shared.gradio['save_character_cancel'] = gr.Button('Cancel', elem_classes="small-button")33 34    with gr.Group(visible=False, elem_classes='file-saver') as shared.gradio['character_deleter']:35        gr.Markdown('Confirm the character deletion?')36        with gr.Row():37            shared.gradio['delete_character_confirm'] = gr.Button('Delete', elem_classes="small-button", variant='stop', interactive=not mu)38            shared.gradio['delete_character_cancel'] = gr.Button('Cancel', elem_classes="small-button")39 40    # Preset saver41    with gr.Group(visible=False, elem_classes='file-saver') as shared.gradio['preset_saver']:42        shared.gradio['save_preset_filename'] = gr.Textbox(lines=1, label='File name', info='The preset will be saved to your presets/ folder with this base filename.')43        shared.gradio['save_preset_contents'] = gr.Textbox(lines=10, label='File contents')44        with gr.Row():45            shared.gradio['save_preset_confirm'] = gr.Button('Save', elem_classes="small-button", variant='primary', interactive=not mu)46            shared.gradio['save_preset_cancel'] = gr.Button('Cancel', elem_classes="small-button")47 48 49def create_event_handlers():50    shared.gradio['save_confirm'].click(51        lambda x, y, z: utils.save_file(x + y, z), gradio('save_root', 'save_filename', 'save_contents'), None).then(52        lambda: gr.update(visible=False), None, gradio('file_saver'))53 54    shared.gradio['delete_confirm'].click(55        lambda x, y: utils.delete_file(x + y), gradio('delete_root', 'delete_filename'), None).then(56        lambda: gr.update(visible=False), None, gradio('file_deleter'))57 58    shared.gradio['delete_cancel'].click(lambda: gr.update(visible=False), None, gradio('file_deleter'))59    shared.gradio['save_cancel'].click(lambda: gr.update(visible=False), None, gradio('file_saver'))60 61    shared.gradio['save_character_confirm'].click(62        chat.save_character, gradio('name2', 'greeting', 'context', 'character_picture', 'save_character_filename'), None).then(63        lambda: gr.update(visible=False), None, gradio('character_saver')).then(64        lambda x: gr.update(choices=utils.get_available_characters(), value=x), gradio('save_character_filename'), gradio('character_menu'))65 66    shared.gradio['delete_character_confirm'].click(67        chat.delete_character, gradio('character_menu'), None).then(68        lambda: gr.update(visible=False), None, gradio('character_deleter')).then(69        lambda: gr.update(choices=(characters := utils.get_available_characters()), value=characters[0]), None, gradio('character_menu'))70 71    shared.gradio['save_character_cancel'].click(lambda: gr.update(visible=False), None, gradio('character_saver'))72    shared.gradio['delete_character_cancel'].click(lambda: gr.update(visible=False), None, gradio('character_deleter'))73 74    shared.gradio['save_preset'].click(75        ui.gather_interface_values, gradio(shared.input_elements), gradio('interface_state')).then(76        presets.generate_preset_yaml, gradio('interface_state'), gradio('save_preset_contents')).then(77        lambda: 'My Preset', None, gradio('save_preset_filename')).then(78        lambda: gr.update(visible=True), None, gradio('preset_saver'))79 80    shared.gradio['save_preset_confirm'].click(81        lambda x, y: utils.save_file(f'presets/{x}.yaml', y), gradio('save_preset_filename', 'save_preset_contents'), None).then(82        lambda: gr.update(visible=False), None, gradio('preset_saver')).then(83        lambda x: gr.update(choices=utils.get_available_presets(), value=x), gradio('save_preset_filename'), gradio('preset_menu'))84 85    shared.gradio['save_preset_cancel'].click(lambda: gr.update(visible=False), None, gradio('preset_saver'))86 87    shared.gradio['delete_preset'].click(88        lambda x: f'{x}.yaml', gradio('preset_menu'), gradio('delete_filename')).then(89        lambda: 'presets/', None, gradio('delete_root')).then(90        lambda: gr.update(visible=True), None, gradio('file_deleter'))91 92    shared.gradio['save_grammar'].click(93        ui.gather_interface_values, gradio(shared.input_elements), gradio('interface_state')).then(94        lambda x: x, gradio('grammar_string'), gradio('save_contents')).then(95        lambda: 'grammars/', None, gradio('save_root')).then(96        lambda: 'My Fancy Grammar.gbnf', None, gradio('save_filename')).then(97        lambda: gr.update(visible=True), None, gradio('file_saver'))98 99    shared.gradio['delete_grammar'].click(100        lambda x: x, gradio('grammar_file'), gradio('delete_filename')).then(101        lambda: 'grammars/', None, gradio('delete_root')).then(102        lambda: gr.update(visible=True), None, gradio('file_deleter'))103