Refactor init_leaderboard function to handle multiple subsets, improve column selection and hiding, and include Dataset Version in filter_columns

app.py

@@ -73,28 +73,32 @@ def init_leaderboard(dataframes, subsets):
     
     with gr.Row():
         selected_subset = gr.Dropdown(choices=subsets, label="Select Dataset Subset", value=subsets[-1])
-        research_textbox = gr.Textbox(placeholder="🔍 Search Models... [press enter]", label="Filter Models by Name", )
     
     with gr.Row():
         datatype = [c.type for c in fields(AutoEvalColumn)]
         dataframe = gr.Dataframe(dataframes, datatype=datatype, type="pandas")
 
 
-    # return Leaderboard(
-    #     value=dataframes,
-    #     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=[
-    #         "Dataset Version",
-    #     ],
-    #     interactive=False,
-    # )
+    return Leaderboard(
+        value=dataframes,
+        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=[
+            ColumnFilter(
+                column=AutoEvalColumn.dataset_version.name,
+                choices=subsets,
+                default=subsets[-1],
+            )
+            # gr.Dropdown(choices=subsets, label="Select Dataset Subset", value=subsets[-1])
+        ],
+        interactive=False,
+    )
     
 demo = gr.Blocks(css=custom_css)
 with demo: