File size: 4,189 Bytes
bb7f1f4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 | import gradio as gr
import scripts.mo.ui_navigation as nav
import scripts.mo.ui_styled_html as styled
from scripts.mo.environment import env
from scripts.mo.ui_debug import debug_ui_block
from scripts.mo.ui_details import details_ui_block
from scripts.mo.ui_download import download_ui_block
from scripts.mo.ui_edit import edit_ui_block
from scripts.mo.ui_home import home_ui_block
from scripts.mo.ui_import_export import import_export_ui_block
from scripts.mo.ui_remove import remove_ui_block
from scripts.mo.utils import get_json_record_data
def on_json_box_change(json_state, home_refresh_token):
state = nav.get_nav_state(json_state)
is_home_visible = state['is_home_visible']
if is_home_visible:
home_refresh_token = nav.generate_ui_token()
return [
gr.Column.update(visible=is_home_visible),
gr.Column.update(visible=state['is_details_visible']),
gr.Column.update(visible=state['is_edit_visible']),
gr.Column.update(visible=state['is_remove_visible']),
gr.Column.update(visible=state['is_download_visible']),
gr.Column.update(visible=state['is_import_export_visible']),
gr.Column.update(visible=state['is_debug_visible']),
gr.Textbox.update(value=home_refresh_token),
gr.Textbox.update(value=state['details_record_id']),
gr.Textbox.update(value=state['edit_data']),
gr.Textbox.update(value=state['remove_record_id']),
gr.Textbox.update(value=state['download_info']),
gr.Textbox.update(value=state['filter_state']),
gr.Textbox.update(value=get_json_record_data(state['details_record_info_id']))
]
def main_ui_block():
with gr.Blocks(elem_id='model_organizer_tab') as main_block:
if env.is_storage_has_errors():
gr.HTML(styled.alert_danger(env.storage_error))
return main_block
elif not env.is_storage_initialized():
gr.HTML(styled.alert_danger('Storage not initialized'))
return main_block
_json_nav_box = gr.Textbox(value=nav.navigate_home(), label='mo_json_nav_box', elem_id='mo_json_nav_box',
elem_classes='mo-alert-warning', visible=False)
with gr.Column(visible=True, elem_id='mo_home_tab') as home_block:
home_refresh_box = home_ui_block()
with gr.Column(visible=False) as record_details_block:
details_id_box = details_ui_block()
with gr.Column(visible=False) as edit_record_block:
edit_id_box = edit_ui_block()
with gr.Column(visible=False) as remove_record_block:
remove_id_box = remove_ui_block()
with gr.Column(visible=False) as download_block:
download_id_box = download_ui_block()
with gr.Column(visible=False) as import_export_block:
filter_state_box = import_export_ui_block()
with gr.Column(visible=False) as debug_block:
if env.is_debug_mode_enabled():
debug_ui_block()
else:
gr.Row()
details_data_box = gr.Textbox(value='\{\}', label='mo_record_info_nav_box', elem_id='mo_record_info_nav_box', elem_classes='mo-alert-warning', visible=False)
_json_nav_box.change(on_json_box_change,
inputs=[_json_nav_box, home_refresh_box],
outputs=[home_block,
record_details_block,
edit_record_block,
remove_record_block,
download_block,
import_export_block,
debug_block,
home_refresh_box,
details_id_box,
edit_id_box,
remove_id_box,
download_id_box,
filter_state_box,
details_data_box
])
return main_block
|