1313 from trackio .sqlite_storage import SQLiteStorage
1414 from trackio .utils import (
1515 RESERVED_KEYS ,
16- TRACKIO_LOGO_PATH ,
16+ TRACKIO_LOGO_DIR ,
1717 downsample ,
1818 get_color_mapping ,
1919 )
2020except : # noqa: E722
2121 from sqlite_storage import SQLiteStorage
22- from utils import RESERVED_KEYS , TRACKIO_LOGO_PATH , downsample , get_color_mapping
22+ from utils import RESERVED_KEYS , TRACKIO_LOGO_DIR , downsample , get_color_mapping
2323
2424css = """
2525#run-cb .wrap {
@@ -307,6 +307,7 @@ def sort_metrics_by_prefix(metrics: list[str]) -> list[str]:
307307
308308def configure (request : gr .Request ):
309309 sidebar_param = request .query_params .get ("sidebar" )
310+ dark_mode = request .query_params .get ("__theme" ) == "dark"
310311 match sidebar_param :
311312 case "collapsed" :
312313 sidebar = gr .Sidebar (open = False , visible = True )
@@ -315,16 +316,25 @@ def configure(request: gr.Request):
315316 case _:
316317 sidebar = gr .Sidebar (open = True , visible = True )
317318
319+ if dark_mode :
320+ logo = gr .Markdown (
321+ f"<img src='/gradio_api/file={ TRACKIO_LOGO_DIR } /trackio_logo_type_dark_transparent.png' width='80%'>"
322+ )
323+ else :
324+ logo = gr .Markdown (
325+ f"<img src='/gradio_api/file={ TRACKIO_LOGO_DIR } /trackio_logo_type_light_transparent.png' width='80%'>"
326+ )
327+
318328 if metrics := request .query_params .get ("metrics" ):
319- return metrics .split ("," ), sidebar
329+ return metrics .split ("," ), sidebar , logo
320330 else :
321- return [], sidebar
331+ return [], sidebar , logo
322332
323333
324334with gr .Blocks (theme = "citrus" , title = "Trackio Dashboard" , css = css ) as demo :
325335 with gr .Sidebar (open = False ) as sidebar :
326- gr .Markdown (
327- f"<div style='display: flex; align-items: center; gap: 8px;'>< img src='/gradio_api/file={ TRACKIO_LOGO_PATH } ' width='32' height='32'><span style='font-size: 2em; font-weight: bold;'>Trackio</span></div >"
336+ logo = gr .Markdown (
337+ f"<img src='/gradio_api/file={ TRACKIO_LOGO_DIR } /trackio_logo_type_light_transparent.png ' width='80%' >"
328338 )
329339 project_dd = gr .Dropdown (label = "Project" , allow_custom_value = True )
330340 run_tb = gr .Textbox (label = "Runs" , placeholder = "Type to filter..." )
@@ -350,7 +360,7 @@ def configure(request: gr.Request):
350360 metrics_subset = gr .State ([])
351361 user_interacted_with_run_cb = gr .State (False )
352362
353- gr .on ([demo .load ], fn = configure , outputs = [metrics_subset , sidebar ])
363+ gr .on ([demo .load ], fn = configure , outputs = [metrics_subset , sidebar , logo ])
354364 gr .on (
355365 [demo .load ],
356366 fn = get_projects ,
@@ -506,4 +516,4 @@ def update_dashboard(
506516
507517
508518if __name__ == "__main__" :
509- demo .launch (allowed_paths = [TRACKIO_LOGO_PATH ], show_api = False , show_error = True )
519+ demo .launch (allowed_paths = [TRACKIO_LOGO_DIR ], show_api = False , show_error = True )
0 commit comments