refresh on load

app.py

@@ -1,5 +1,3 @@
-from functools import partial
-
 import gradio as gr
 from gradio_leaderboard import Leaderboard, ColumnFilter, SelectColumns
 import pandas as pd
@@ -43,11 +41,28 @@ SKIP_VALIDATION = True  # TODO temp
 
 
 def restart_space():
+    logger.info("Restarting space")
     API.restart_space(repo_id=REPO_ID)
 
 
+def refresh_leaderboard_data():
+    """Refresh the leaderboard data from the latest results"""
+    try:
+        logger.info("Loading latest leaderboard data...")
+        new_leaderboard_df = get_leaderboard_df(RESULTS_REPO)
+
+        if new_leaderboard_df is not None and not new_leaderboard_df.empty:
+            logger.info("Leaderboard data refreshed successfully")
+            return new_leaderboard_df
+        else:
+            logger.warning("No new leaderboard data found")
+            return None
+    except Exception as e:
+        logger.error(f"Error refreshing leaderboard data: {e}")
+        return None
+
+
 lbdb = F1Data(cp_ds_name=CODE_PROBLEMS_REPO, sub_ds_name=SUBMISSIONS_REPO, res_ds_name=RESULTS_REPO, split=SPLIT)
-leaderboard_df = get_leaderboard_df(RESULTS_REPO)
 
 
 logger.info("Initialized LBDB")
@@ -129,7 +144,8 @@ with demo:
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.TabItem("🏅 FormulaOne Leaderboard", elem_id="formulaone-leaderboar-tab-table", id=0):
-            leaderboard = init_leaderboard(leaderboard_df)
+            leaderboard_df = get_leaderboard_df(RESULTS_REPO)
+            leaderboard_component = init_leaderboard(leaderboard_df)
 
         # with gr.TabItem("📝 About", elem_id="llm-benchmark-tab-table", id=1):
         #     logger.info("Tab about")
@@ -245,6 +261,9 @@ with demo:
             #     show_copy_button=True,
             # )
 
+    demo.load(refresh_leaderboard_data, inputs=[], outputs=[leaderboard_component])
+
+
 logger.info("Scheduler")
 scheduler = BackgroundScheduler()
 scheduler.add_job(restart_space, "interval", seconds=1800)