Create Dataset filtering for code completion

app.py

@@ -44,6 +44,19 @@ def get_leaderboard_for_task(task_pretty: str) -> gr.components.Dataframe:
     )
 
 
+def get_leaderboard_for_completion_task(dataset_name: str | None):
+    df = get_results_for_task(TASKS_PRETTY['project_code_completion'])
+    code_completion_dataset_names = df['Dataset'].unique()
+    if dataset_name is None:
+        dataset_name = code_completion_dataset_names[0]
+    filtered_df = df[df['Dataset']==dataset_name]
+    return gr.components.Dataframe(
+        value=filtered_df,
+        interactive=False,
+        datatype=get_types_per_task('project_code_completion'),
+    )
+
+
 with gr.Blocks() as demo:
     # intro
     gr.HTML(INTRODUCTION_TITLE)
@@ -58,15 +71,34 @@ with gr.Blocks() as demo:
                 with gr.Row():
                     gr.Markdown(TASKS_DESCRIPTIONS[TASKS_PRETTY_REVERSE[task_pretty]])
 
-                leaderboard_table = get_leaderboard_for_task(task_pretty)
-                task_input = gr.Text(value=task_pretty, visible=False)
+                if task_pretty == TASKS_PRETTY['project_code_completion']:
+                    leaderboard_table = get_leaderboard_for_completion_task(dataset_name=None)
+                    code_completion_dataset_names = get_results_for_task(TASKS_PRETTY['project_code_completion'])['Dataset'].unique().tolist()
+                else:
+                    leaderboard_table = get_leaderboard_for_task(task_pretty)
                 
-                refresh_button = gr.Button("🔄 Refresh", variant="secondary")
-                refresh_button.click(
-                    fn=get_leaderboard_for_task,
-                    inputs=task_input,
-                    outputs=leaderboard_table,
-                )
+                task_input = gr.Text(value=task_pretty, visible=False)
+
+                if task_pretty == TASKS_PRETTY['project_code_completion']:
+                    dataset_dropdown = gr.Dropdown(choices=code_completion_dataset_names, label="Select the Dataset")
+                    dataset_dropdown.change(
+                        fn=get_leaderboard_for_completion_task,
+                        inputs=dataset_dropdown,
+                        outputs=leaderboard_table
+                    )
+                    refresh_button = gr.Button("🔄 Refresh", variant="secondary")
+                    refresh_button.click(
+                        fn=get_leaderboard_for_completion_task,
+                        inputs=dataset_dropdown,
+                        outputs=leaderboard_table,
+                    )
+                else:
+                    refresh_button = gr.Button("🔄 Refresh", variant="secondary")
+                    refresh_button.click(
+                        fn=get_leaderboard_for_task,
+                        inputs=task_input,
+                        outputs=leaderboard_table,
+                    )
 
     # submission
     gr.HTML(SUBMISSION_TITLE)