Update app.py

app.py

@@ -262,7 +262,18 @@ def get_difficulty_pattern_ids_and_key(rl_env, path):
 
     return key, difficulty_pattern_ids
 
+def filter_checkbox_data(rl_env, task_id, selected_values, path):
+    """
+    Filters the data based on the selected difficulty/pattern values.
+    """
+    data = get_data(rl_env, task_id, path)
+    
+    # If there are selected values, filter the DataFrame
+    if selected_values:
+        filter_column = "Pattern" if "Pattern" in data.columns else "Difficulty"
+        data = data[data[filter_column].isin(selected_values)]
 
+    return data
 
 run_update_dataset()
 
@@ -292,43 +303,66 @@ with block:
         )
 
     path_ = download_leaderboard_dataset()
-    # gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
+
+    def filter_data(rl_env, task_id, selected_values, path):
+        """
+        Filters the data based on the selected difficulty/pattern values.
+        """
+        data = get_data(rl_env, task_id, path)
+        
+        # If there are selected values, filter the DataFrame
+        if selected_values:
+            filter_column = "Pattern" if "Pattern" in data.columns else "Difficulty"
+            data = data[data[filter_column].isin(selected_values)]
+
+        return data
+
+    def update_filtered_data(selected_values, rl_env, task_id, path):
+        filtered_data = filter_data(rl_env, task_id, selected_values, path)
+        return filtered_data
+
     # ENVIRONMENT TABS
-    with gr.Tabs() as tabs:  # elem_classes="tab-buttons"
+    with gr.Tabs() as tabs:
         for env_index in range(0, len(hivex_envs)):
             hivex_env = hivex_envs[env_index]
             with gr.Tab(f"{hivex_env['title']}") as env_tabs:
-
-                # Call the function to get the actual values
                 dp_key, difficulty_pattern_ids = get_difficulty_pattern_ids_and_key(
                     hivex_env["hivex_env"], path_
                 )
-                if dp_key is not None:
-                    gr.CheckboxGroup([str(dp_id) for dp_id in difficulty_pattern_ids], label=dp_key)
 
-                # TASK TABS
-                for task_id in range(0, hivex_env["task_count"]):
-                    task_title = convert_to_title_case(
-                        get_task(hivex_env["hivex_env"], task_id, path_)
+                if dp_key is not None and difficulty_pattern_ids:
+                    selected_checkboxes = gr.CheckboxGroup(
+                        [str(dp_id) for dp_id in difficulty_pattern_ids], label=dp_key
                     )
-                    with gr.TabItem(f"Task {task_id}: {task_title}"):
-                        with gr.Row():
+
+                    for task_id in range(0, hivex_env["task_count"]):
+                        task_title = convert_to_title_case(
+                            get_task(hivex_env["hivex_env"], task_id, path_)
+                        )
+                        with gr.TabItem(f"Task {task_id}: {task_title}"):
+
+                            # Display initial data
                             data = get_data(hivex_env["hivex_env"], task_id, path_)
-                            row_count = len(data)  # Number of rows in the data
+                            row_count = len(data)
 
-                            gr_dataframe = gr.components.Dataframe(
+                            gr_dataframe = gr.DataFrame(
                                 value=data,
                                 headers=["User", "Model"],
                                 datatype=["markdown", "markdown"],
-                                row_count=(
-                                    row_count,
-                                    "fixed",
-                                ),  # Set to the exact number of rows in the data
+                                row_count=(row_count, "fixed"),
                             )
 
+                            # Add a callback to update the DataFrame when checkboxes are changed
+                            selected_checkboxes.change(
+                                fn=update_filtered_data,
+                                inputs=[selected_checkboxes, gr.Textbox(hivex_env["hivex_env"]), gr.Number(task_id), gr.Textbox(path_)],
+                                outputs=gr_dataframe,
+                            )
+                else:
+                    gr.HTML("<p>No difficulty or pattern data available for this environment.</p>")
 
 scheduler = BackgroundScheduler()
 scheduler.add_job(restart, "interval", seconds=86400)
 scheduler.start()
 
-block.launch()
+block.launch()