import os import uuid 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.htr_tool import htr_tool_tab from tabs.overview_tab import overview from tabs.stepwise_htr_tool import stepwise_htr_tool_tab session_uuid = str(uuid.uuid1()) with gr.Blocks(title="Riksarkivet", theme=theme, css=css) as demo: with gr.Row(): with gr.Column(scale=1): text_ip_output = gr.Markdown(TextApp.demo_version) with gr.Column(scale=2): 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("Overview"): overview.render() SECRET_KEY = os.environ.get("AM_I_IN_A_DOCKER_CONTAINER", False) if SECRET_KEY: demo.load(fn=TrafficDataHandler.onload_store_metric_data, inputs=None, outputs=None) 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)