Merge commit '1345efbcdfb9dd66c56616c233d2669ce3f84ded'

app.py

@@ -1,4 +1,4 @@
-from functools import partial
+import random
 
 import gradio as gr
 from gradio_leaderboard import Leaderboard, ColumnFilter, SelectColumns
@@ -43,20 +43,26 @@ SKIP_VALIDATION = True  # TODO temp
 
 
 def restart_space():
+    logger.info("Restarting space")
     API.restart_space(repo_id=REPO_ID)
 
 
-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")
+def refresh_leaderboard_data():
+    """Refresh the leaderboard data from the latest results"""
+    global leaderboard_df
+    try:
+        logger.info("Loading leaderboard data...")
+        new_leaderboard_df = get_leaderboard_df(RESULTS_REPO)
 
-# (
-#     finished_eval_queue_df,
-#     running_eval_queue_df,
-#     pending_eval_queue_df,
-# ) = get_evaluation_queue_df(EVAL_REQUESTS_PATH, EVAL_COLS)
+        if new_leaderboard_df is not None and not new_leaderboard_df.empty:
+            logger.info("Leaderboard data refreshed successfully")
+            leaderboard_df = 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
 
 
 def init_leaderboard(dataframe):
@@ -89,6 +95,19 @@ def init_leaderboard(dataframe):
     )
 
 
+lbdb = F1Data(cp_ds_name=CODE_PROBLEMS_REPO, sub_ds_name=SUBMISSIONS_REPO, res_ds_name=RESULTS_REPO, split=SPLIT)
+
+leaderboard_df = None
+
+logger.info("Initialized LBDB")
+
+# (
+#     finished_eval_queue_df,
+#     running_eval_queue_df,
+#     pending_eval_queue_df,
+# ) = get_evaluation_queue_df(EVAL_REQUESTS_PATH, EVAL_COLS)
+
+
 # Display image using Markdown
 # banner = "![Leaderboard Banner](file/assets/banner.png)"
 
@@ -129,7 +148,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)
+            refresh_leaderboard_data()  # updates leaderboard_df
+            leaderboard_component = init_leaderboard(leaderboard_df)
 
         # with gr.TabItem("📝 About", elem_id="llm-benchmark-tab-table", id=1):
         #     logger.info("Tab about")
@@ -245,9 +265,13 @@ with demo:
             #     show_copy_button=True,
             # )
 
+    demo.load(lambda: leaderboard_df, inputs=[], outputs=[leaderboard_component])
+
+
 logger.info("Scheduler")
 scheduler = BackgroundScheduler()
 scheduler.add_job(restart_space, "interval", seconds=1800)
+scheduler.add_job(refresh_leaderboard_data, "interval", seconds=120)
 scheduler.start()
 logger.info("Launch")
 demo.queue(default_concurrency_limit=40).launch()