Update app.py

app.py

@@ -260,19 +260,6 @@ 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
     
 
 run_update_dataset()
@@ -280,6 +267,7 @@ 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;">
@@ -303,6 +291,24 @@ 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"
+            if filter_column in data.columns:
+                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)):
@@ -335,10 +341,15 @@ with block:
                                 row_count=(row_count, "fixed"),
                             )
 
+                            # Use gr.State to hold environment and task information
+                            rl_env_state = gr.State(value=hivex_env["hivex_env"])
+                            task_id_state = gr.State(value=task_id)
+                            path_state = gr.State(value=path_)
+
                             # 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_)],
+                                inputs=[selected_checkboxes, rl_env_state, task_id_state, path_state],
                                 outputs=gr_dataframe,
                             )
                 else:
@@ -348,4 +359,4 @@ scheduler = BackgroundScheduler()
 scheduler.add_job(restart, "interval", seconds=86400)
 scheduler.start()
 
-block.launch()
+block.launch()