Update app.py

app.py

@@ -43,12 +43,18 @@ def get_leaderboard_for_task(task_pretty: str) -> gr.components.Dataframe:
         datatype=get_types_per_task(TASKS_PRETTY_REVERSE[task_pretty]),
     )
 
+code_completion_dataset_names = (
+    get_results_for_task(TASKS_PRETTY['project_code_completion'])['Dataset'].unique().tolist()
+)
+cc_ds_names_html = {gr.HTML(ds_name): ds_name for ds_name in code_completion_dataset_names}
 
-def get_leaderboard_for_completion_task(dataset_name: str | None):
+def get_leaderboard_for_completion_task(dataset_name: str | None | gr.HTML):
     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]
+    if isinstance(dataset_name, gr.HTML):
+        dataset_name = cc_ds_names_html[dataset_name]
     filtered_df = df[df['Dataset']==dataset_name]
     return gr.components.Dataframe(
         value=filtered_df,
@@ -73,14 +79,14 @@ with gr.Blocks() as demo:
 
                 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)
                 
                 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 = gr.Dropdown(choices=list(cc_ds_names_html), label="Select the Dataset")
                     dataset_dropdown.change(
                         fn=get_leaderboard_for_completion_task,
                         inputs=dataset_dropdown,