Update app.py

app.py

@@ -260,7 +260,23 @@ def get_difficulty_pattern_ids_and_key(rl_env, path):
         difficulty_pattern_ids = []
 
     return key, difficulty_pattern_ids
+
+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
 
 run_update_dataset()
 
@@ -291,23 +307,6 @@ with block:
 
     path_ = download_leaderboard_dataset()
 
-    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:
         for env_index in range(0, len(hivex_envs)):
@@ -318,38 +317,39 @@ with block:
                 )
 
                 # Check if dp_key is defined and difficulty_pattern_ids is not empty
-                if dp_key is not None and len(difficulty_pattern_ids) > 0:
+                if dp_key is not None and difficulty_pattern_ids.size > 0:
                     selected_checkboxes = gr.CheckboxGroup(
                         [str(dp_id) for dp_id in difficulty_pattern_ids], label=dp_key
                     )
 
-                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}"):
+                    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)
+                            # Display initial data
+                            data = get_data(hivex_env["hivex_env"], task_id, path_)
+                            row_count = len(data)
 
-                        gr_dataframe = gr.DataFrame(
-                            value=data,
-                            headers=["User", "Model"],
-                            datatype=["markdown", "markdown"],
-                            row_count=(row_count, "fixed"),
-                        )
+                            gr_dataframe = gr.DataFrame(
+                                value=data,
+                                headers=["User", "Model"],
+                                datatype=["markdown", "markdown"],
+                                row_count=(row_count, "fixed"),
+                            )
 
-                        if dp_key is not None and len(difficulty_pattern_ids) > 0:
                             # Add a callback to update the DataFrame when checkboxes are changed
                             selected_checkboxes.change(
                                 fn=update_filtered_data,
-                                inputs=[selected_checkboxes, hivex_env["hivex_env"], task_id, path_],
+                                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()