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