weird update

app.py

@@ -79,6 +79,8 @@ def init_leaderboard(dataframe):
         dataframe = pd.DataFrame(columns=columns)
         logger.warning("Initializing empty leaderboard")
 
+    print("\n\n", "dataframe", dataframe, "--------------------------------\n\n")
+
     return Leaderboard(
         value=dataframe,
         datatype=[getattr(GUARDBENCH_COLUMN, col).type for col in DISPLAY_COLS],
@@ -93,6 +95,7 @@ def init_leaderboard(dataframe):
             ColumnFilter(GUARDBENCH_COLUMN.model_type.name, type="checkboxgroup", label="Model types"),
         ],
         interactive=False,
+        render=True,
     )
 
 
@@ -151,28 +154,24 @@ def submit_results(
 
 def refresh_data(version=CURRENT_VERSION):
     """
-    Refresh the leaderboard data from HuggingFace.
+    Refresh the leaderboard data and update all components.
     """
-    global LEADERBOARD_DF
     try:
-        logger.info(f"Performing scheduled refresh of leaderboard data for version {version}...")
-        new_df = get_leaderboard_df(version=version)
-        if new_df is not None and not new_df.empty:
-            LEADERBOARD_DF = new_df
-            logger.info("Scheduled refresh of leaderboard data completed")
-        else:
-            logger.warning("Refresh returned empty data, keeping existing data")
-            # If empty, create a dataframe with correct columns
-            if LEADERBOARD_DF is None or LEADERBOARD_DF.empty:
-                columns = [getattr(GUARDBENCH_COLUMN, col).name for col in DISPLAY_COLS]
-                LEADERBOARD_DF = pd.DataFrame(columns=columns)
+        logger.info(f"Performing scheduled refresh of leaderboard data...")
+        # Get new data
+        main_df = get_leaderboard_df(version=version)
+        category_dfs = [get_category_leaderboard_df(category, version=version) for category in CATEGORIES]
+
+        # For Leaderboard components, we need to return just the dataframes
+        # The component will handle the update internally
+        return dict(
+            value=main_df
+        ), *[dict(value=df) for df in category_dfs]
+
     except Exception as e:
         logger.error(f"Error in scheduled refresh: {e}")
-        # Ensure we have at least an empty dataframe with correct columns
-        if LEADERBOARD_DF is None or LEADERBOARD_DF.empty:
-            columns = [getattr(GUARDBENCH_COLUMN, col).name for col in DISPLAY_COLS]
-            LEADERBOARD_DF = pd.DataFrame(columns=columns)
-    return LEADERBOARD_DF
+        return dict(value=leaderboard.value), *[dict(value=tab.children[0].value)
+               for tab in category_tabs.children[1:]]
 
 
 def update_leaderboards(version):
@@ -300,7 +299,7 @@ with demo:
         with tabs:
             with gr.TabItem("🏅 Leaderboard", elem_id="guardbench-leaderboard-tab", id=0):
                 with gr.Row():
-                    refresh_button = gr.Button("Refresh Leaderboard", scale=3)
+                    refresh_button = gr.Button("Refresh Leaderboard")
                     version_selector = gr.Dropdown(
                         choices=BENCHMARK_VERSIONS,
                         label="Benchmark Version",
@@ -314,20 +313,20 @@ with demo:
                 with gr.Tabs(elem_classes="category-tabs") as category_tabs:
                     # First tab for average metrics across all categories
                     with gr.TabItem("📊 Overall Performance", elem_id="overall-tab"):
+                        print("LEADERBOARD_DF", LEADERBOARD_DF)
                         leaderboard = init_leaderboard(LEADERBOARD_DF)
 
                     # Create a tab for each category
                     for category in CATEGORIES:
                         with gr.TabItem(f"{category}", elem_id=f"category-{category.lower().replace(' ', '-')}-tab"):
+                            print("category DF", category)
                             category_df = get_category_leaderboard_df(category, version=CURRENT_VERSION)
+                            print("category DF", category_df)
                             category_leaderboard = init_leaderboard(category_df)
 
                 # Refresh button functionality
                 refresh_button.click(
-                    fn=lambda: [
-                        init_leaderboard(get_leaderboard_df(version=version_selector.value)),
-                        *[init_leaderboard(get_category_leaderboard_df(category, version=version_selector.value)) for category in CATEGORIES]
-                    ],
+                    fn=refresh_data,
                     inputs=[],
                     outputs=[leaderboard] + [category_tabs.children[i].children[0] for i in range(1, len(CATEGORIES) + 1)]
                 )
@@ -486,8 +485,9 @@ with demo:
             Last updated: {pd.Timestamp.now().strftime("%Y-%m-%d %H:%M:%S UTC")}
             """)
 
+# Set up the scheduler to refresh data periodically
 scheduler = BackgroundScheduler()
-scheduler.add_job(lambda: refresh_data(version=CURRENT_VERSION), 'interval', minutes=30)
+scheduler.add_job(refresh_data, 'interval', minutes=30)
 scheduler.start()
 
 # Launch the app

src/populate.py

@@ -192,7 +192,7 @@ def get_category_leaderboard_df(category: str, version="v0") -> pd.DataFrame:
         "last_updated": leaderboard_data.get("last_updated", datetime.now().isoformat()),
         "version": version
     }
-    print(filtered_leaderboard)
+    # print(filtered_leaderboard)
 
     # Convert to DataFrame
     return leaderboard_to_dataframe(filtered_leaderboard)