Refactor init_leaderboard function to handle multiple subsets and improve column selection and hiding

app.py

@@ -64,26 +64,39 @@ except Exception:
     restart_space()
 
 
-LEADERBOARD_DF = get_leaderboard_df(RESULTS_REPO, EVAL_RESULTS_PATH, "2024-06")
+LEADERBOARD_DF = get_leaderboard_df(RESULTS_REPO, EVAL_RESULTS_PATH)
 
 
-def init_leaderboard(dataframe):
-    if dataframe is None or dataframe.empty:
-        raise ValueError("Leaderboard DataFrame is empty or None.")
-    return Leaderboard(
-        value=dataframe,
-        datatype=[c.type for c in fields(AutoEvalColumn)],
-        select_columns=SelectColumns(
-            default_selection=[c.name for c in fields(AutoEvalColumn) if c.displayed_by_default],
-            cant_deselect=[c.name for c in fields(AutoEvalColumn) if c.never_hidden],
-            label="Select Columns to Display:",
-        ),
-        search_columns=[AutoEvalColumn.model.name],
-        hide_columns=[c.name for c in fields(AutoEvalColumn) if c.hidden],
-        filter_columns=[],
-        interactive=False,
+def init_leaderboard(dataframes):
+    if dataframes is None or not dataframes:
+        raise ValueError("Leaderboard data is empty or None.")
+    
+    def create_leaderboard(df):
+        return Leaderboard(
+            value=df,
+            datatype=[c.type for c in fields(AutoEvalColumn)],
+            select_columns=SelectColumns(
+                default_selection=[c.name for c in fields(AutoEvalColumn) if c.displayed_by_default],
+                cant_deselect=[c.name for c in fields(AutoEvalColumn) if c.never_hidden],
+                label="Select Columns to Display:",
+            ),
+            search_columns=[AutoEvalColumn.model.name],
+            hide_columns=[c.name for c in fields(AutoEvalColumn) if c.hidden],
+            filter_columns=[],
+            interactive=False,
+        )
+    
+    subset_names = list(dataframes.keys())
+    selected_subset = gr.Dropdown(choices=subset_names, label="Select Dataset Subset", value=subset_names[0])
+    
+    leaderboard = gr.Dynamic(create_leaderboard, inputs=[selected_subset], outputs="output")
+    selected_subset.change(
+        fn=lambda x: create_leaderboard(dataframes[x]),
+        inputs=[selected_subset],
+        outputs=leaderboard
     )
-
+    
+    return leaderboard
 
 demo = gr.Blocks(css=custom_css)
 with demo: