mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-12-26 14:20:40 +01:00
74 lines
4.0 KiB
Python
74 lines
4.0 KiB
Python
import gradio as gr
|
|
|
|
from modules import shared, ui, utils
|
|
from modules.github import clone_or_pull_repository
|
|
from modules.utils import gradio
|
|
|
|
|
|
def create_ui():
|
|
with gr.Tab("Session", elem_id="session-tab"):
|
|
with gr.Row():
|
|
with gr.Column():
|
|
shared.gradio['reset_interface'] = gr.Button("Apply flags/extensions and restart")
|
|
with gr.Row():
|
|
shared.gradio['toggle_dark_mode'] = gr.Button('Toggle 💡')
|
|
shared.gradio['save_settings'] = gr.Button('Save UI defaults to settings.yaml')
|
|
|
|
with gr.Row():
|
|
with gr.Column():
|
|
shared.gradio['extensions_menu'] = gr.CheckboxGroup(choices=utils.get_available_extensions(), value=shared.args.extensions, label="Available extensions", info='Note that some of these extensions may require manually installing Python requirements through the command: pip install -r extensions/extension_name/requirements.txt', elem_classes='checkboxgroup-table')
|
|
|
|
with gr.Column():
|
|
shared.gradio['bool_menu'] = gr.CheckboxGroup(choices=get_boolean_arguments(), value=get_boolean_arguments(active=True), label="Boolean command-line flags", elem_classes='checkboxgroup-table')
|
|
|
|
with gr.Column():
|
|
if not shared.args.multi_user:
|
|
shared.gradio['save_session'] = gr.Button('Save session')
|
|
shared.gradio['load_session'] = gr.File(type='binary', file_types=['.json'], label="Upload Session JSON")
|
|
|
|
extension_name = gr.Textbox(lines=1, label='Install or update an extension', info='Enter the GitHub URL below and press Enter. For a list of extensions, see: https://github.com/oobabooga/text-generation-webui-extensions ⚠️ WARNING ⚠️ : extensions can execute arbitrary code. Make sure to inspect their source code before activating them.')
|
|
extension_status = gr.Markdown()
|
|
|
|
extension_name.submit(
|
|
clone_or_pull_repository, extension_name, extension_status, show_progress=False).then(
|
|
lambda: gr.update(choices=utils.get_available_extensions(), value=shared.args.extensions), None, gradio('extensions_menu'))
|
|
|
|
# Reset interface event
|
|
shared.gradio['reset_interface'].click(
|
|
set_interface_arguments, gradio('extensions_menu', 'bool_menu'), None).then(
|
|
lambda: None, None, None, _js='() => {document.body.innerHTML=\'<h1 style="font-family:monospace;padding-top:20%;margin:0;height:100vh;color:lightgray;text-align:center;background:var(--body-background-fill)">Reloading...</h1>\'; setTimeout(function(){location.reload()},2500); return []}')
|
|
|
|
shared.gradio['toggle_dark_mode'].click(lambda: None, None, None, _js='() => {document.getElementsByTagName("body")[0].classList.toggle("dark")}')
|
|
shared.gradio['save_settings'].click(
|
|
ui.gather_interface_values, gradio(shared.input_elements), gradio('interface_state')).then(
|
|
ui.save_settings, gradio('interface_state', 'preset_menu', 'instruction_template', 'extensions_menu', 'show_controls'), gradio('save_contents')).then(
|
|
lambda: './', None, gradio('save_root')).then(
|
|
lambda: 'settings.yaml', None, gradio('save_filename')).then(
|
|
lambda: gr.update(visible=True), None, gradio('file_saver'))
|
|
|
|
|
|
def set_interface_arguments(extensions, bool_active):
|
|
shared.args.extensions = extensions
|
|
|
|
bool_list = get_boolean_arguments()
|
|
|
|
for k in bool_list:
|
|
setattr(shared.args, k, False)
|
|
for k in bool_active:
|
|
setattr(shared.args, k, True)
|
|
|
|
shared.need_restart = True
|
|
|
|
|
|
def get_boolean_arguments(active=False):
|
|
exclude = ["default", "notebook", "chat"]
|
|
|
|
cmd_list = vars(shared.args)
|
|
bool_list = sorted([k for k in cmd_list if type(cmd_list[k]) is bool and k not in exclude + ui.list_model_elements()])
|
|
bool_active = [k for k in bool_list if vars(shared.args)[k]]
|
|
|
|
if active:
|
|
return bool_active
|
|
else:
|
|
return bool_list
|