Update app.py

app.py

@@ -256,13 +256,12 @@ def get_difficulty_pattern_ids_and_key(rl_env, path):
         key = "Difficulty"
         difficulty_pattern_ids = data[key].unique()
     else:
-        # Handle the case where neither 'Pattern' nor 'Difficulty' columns exist
         key = None
         difficulty_pattern_ids = []
 
     return key, difficulty_pattern_ids
 
-def filter_checkbox_data(rl_env, task_id, selected_values, path):
+def filter_data(rl_env, task_id, selected_values, path):
     """
     Filters the data based on the selected difficulty/pattern values.
     """
@@ -274,13 +273,13 @@ def filter_checkbox_data(rl_env, task_id, selected_values, path):
         data = data[data[filter_column].isin(selected_values)]
 
     return data
+    
 
 run_update_dataset()
 
 block = gr.Blocks(css=custom_css)  # Attach the custom CSS here
 with block:
     with gr.Row(elem_id="header-row"):
-        # TITLE IMAGE
         gr.HTML(
             """
             <div style="width: 50%; margin: 0 auto; text-align: center;">
@@ -304,23 +303,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)):
@@ -366,4 +348,4 @@ scheduler = BackgroundScheduler()
 scheduler.add_job(restart, "interval", seconds=86400)
 scheduler.start()
 
-block.launch()
+block.launch()