import os

import gradio as gr

from helper.gradio_config import css, theme
from helper.text.text_app import TextApp
from helper.utils import TrafficDataHandler
from tabs.about_tab import about_tab
from tabs.help_tab import help_tab
from tabs.htr_tool import htr_tool_tab
from tabs.stepwise_htr_tool import stepwise_htr_tool_tab

handler = TrafficDataHandler()

VERSION = "Demo version 0.0.2"

with gr.Blocks(title="Riksarkivet", theme=theme, css=css) as demo:
    with gr.Row():
        with gr.Column(scale=1):
            text_ip_output = gr.Markdown(VERSION)
        with gr.Column(scale=1):
            gr.Markdown(TextApp.title_markdown)
        with gr.Column(scale=1):
            gr.Markdown(TextApp.title_markdown_img)

    with gr.Tabs():
        with gr.Tab("Fast track"):
            htr_tool_tab.render()

        with gr.Tab("Stepwise"):
            stepwise_htr_tool_tab.render()

        with gr.Tab("Help"):
            help_tab.render()

        with gr.Tab("About"):
            about_tab.render()

    SECRET_KEY = os.environ.get("AM_I_IN_A_DOCKER_CONTAINER", False)
    if SECRET_KEY:
        demo.load(handler.onload_store_metric_data)


demo.queue(concurrency_count=2, max_size=2)


if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860, show_api=False, show_error=True)