import gradio as gr

from app.gradio_config import css, theme
from app.tabs.adv_htrflow_tab import adv_htrflow_pipeline
from app.tabs.submit import submit
from app.tabs.examples_tab import examples
from app.tabs.templating import (
    templating_block,
    template_image_folder,
    template_yaml_folder,
)
from app.utils.md_helper import load_markdown

gr.set_static_paths(paths=[template_image_folder])
gr.set_static_paths(paths=[template_yaml_folder])


with gr.Blocks(title="HTRflow", theme=theme, css=css) as demo:
    with gr.Row():
        with gr.Column(scale=1):
            pass
        with gr.Column(scale=2):
            gr.Markdown(load_markdown(None, "main_title"))
        with gr.Column(scale=1):
            gr.Markdown(load_markdown(None, "main_sub_title"))

    with gr.Tabs(elem_classes="top-navbar") as navbar:
        with gr.Tab(label="Templating") as tab_templating:
            templating_block.render()

        with gr.Tab(label="Submit Job") as tab_submit:
            submit.render()

        with gr.Tab(label="Output & Visualize") as tab_examples:
            examples.render()


demo.queue()

if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7862, enable_monitoring=False)