Update app.py

app.py

@@ -251,32 +251,32 @@ def convert_to_title_case(text: str) -> str:
     return title_case_text
 
 
-def get_difficulty_pattern_ids_and_key(rl_env, path):
+def get_elevation_pattern_ids_and_key(rl_env, path):
     csv_path = path + "/" + rl_env + ".csv"
     data = pd.read_csv(csv_path)
 
     if "Pattern" in data.columns:
         key = "Pattern"
-        difficulty_pattern_ids = sorted(data[key].unique())
-    elif "Difficulty" in data.columns:
-        key = "Difficulty"
-        difficulty_pattern_ids = sorted(data[key].unique())
+        elevation_pattern_ids = sorted(data[key].unique())
+    elif "Terrain Elevation Levels" in data.columns:
+        key = "Terrain Elevation Levels"
+        elevation_pattern_ids = sorted(data[key].unique())
     else:
         key = None
-        difficulty_pattern_ids = []
+        elevation_pattern_ids = []
 
-    return key, difficulty_pattern_ids
+    return key, elevation_pattern_ids
 
 def filter_data(rl_env, task_id, selected_values, path):
     """
-    Filters the data based on the selected difficulty/pattern values.
+    Filters the data based on the selected elevation/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 == "Difficulty":
+        filter_column = "Pattern" if "Pattern" in data.columns else "Terrain Elevation Levels"
+        if filter_column == "Terrain Elevation Levels":
             selected_values = [np.int64(sv) for sv in selected_values] 
         data = data[data[filter_column].isin(selected_values)]
 
@@ -320,14 +320,14 @@ with block:
         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:
-                dp_key, difficulty_pattern_ids = get_difficulty_pattern_ids_and_key(
+                dp_key, elevation_pattern_ids = get_elevation_pattern_ids_and_key(
                     hivex_env["hivex_env"], path_
                 )
 
-                # 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:
+                # Check if dp_key is defined and elevation_pattern_ids is not empty
+                if dp_key is not None and len(elevation_pattern_ids) > 0:
                     selected_checkboxes = gr.CheckboxGroup(
-                        [str(dp_id) for dp_id in difficulty_pattern_ids], label=dp_key
+                        [str(dp_id) for dp_id in elevation_pattern_ids], label=dp_key
                     )
                 
                 for task_id in range(0, hivex_env["task_count"]):