ui/run_detail.py

"""The Runs page for the Trackio UI."""

import gradio as gr

try:
    import trackio.utils as utils
    from trackio.sqlite_storage import SQLiteStorage
    from trackio.ui import fns
except ImportError:
    import utils
    from sqlite_storage import SQLiteStorage
    from ui import fns

RUN_DETAILS_TEMPLATE = """
## Run Details
* **Run Name:** `{run_name}`
* **Group:** `{group}`
* **Created:** {created} by {username}
"""

with gr.Blocks() as run_detail_page:
    with gr.Sidebar() as sidebar:
        logo_urls = utils.get_logo_urls()
        logo = gr.Markdown(
            f"""
                <img src='{logo_urls["light"]}' width='80%' class='logo-light'>
                <img src='{logo_urls["dark"]}' width='80%' class='logo-dark'>            
            """
        )
        project_dd = gr.Dropdown(
            label="Project", allow_custom_value=True, interactive=False
        )
        run_dd = gr.Dropdown(label="Run")

    navbar = gr.Navbar(value=[("Metrics", ""), ("Runs", "/runs")], main_page_name=False)

    run_details = gr.Markdown(RUN_DETAILS_TEMPLATE)

    run_config = gr.JSON(label="Run Config")

    def configure(request: gr.Request):
        project = request.query_params.get("selected_project")
        run = request.query_params.get("selected_run")
        runs = SQLiteStorage.get_runs(project)
        return project, gr.Dropdown(choices=runs, value=run)

    def update_run_details(project, run):
        config = SQLiteStorage.get_run_config(project, run)
        if not config:
            return gr.Markdown("No run details available"), {}

        group = config.get("_Group", "None")

        created = config.get("_Created", "Unknown")
        if created != "Unknown":
            created = utils.format_timestamp(created)

        username = config.get("_Username", "Unknown")
        if username and username != "None" and username != "Unknown":
            username = f"[{username}](https://huggingface.co/{username})"

        details_md = RUN_DETAILS_TEMPLATE.format(
            run_name=run, group=group, created=created, username=username
        )

        config_display = {k: v for k, v in config.items() if not k.startswith("_")}

        return gr.Markdown(details_md), config_display

    gr.on(
        [run_detail_page.load],
        fn=configure,
        outputs=[project_dd, run_dd],
        show_progress="hidden",
        queue=False,
        api_name=False,
    ).then(
        fns.update_navbar_value,
        inputs=[project_dd],
        outputs=[navbar],
        show_progress="hidden",
        api_name=False,
        queue=False,
    )

    gr.on(
        [run_dd.change],
        update_run_details,
        inputs=[project_dd, run_dd],
        outputs=[run_details, run_config],
        show_progress="hidden",
        api_name=False,
        queue=False,
    )