Update app.py

app.py

@@ -33,6 +33,9 @@ custom_css = """
 }
 """
 
+# Pattern: 0 Default, 1 Grid, 2 Chain, 3 Circle, 4 Square, 5 Cross, 6 Two_Rows, 7 Field, 8 Random
+pattern_map = {0: "Default", 1: "Grid", 2: "Chain", 3: "Circle", 4: "Square", 5: "Cross", 6: "Two Rows", 7: "Field", 8: "Random" }
+
 hivex_envs = [
     {
         "title": "Wind Farm Control",
@@ -112,6 +115,8 @@ def update_leaderboard_dataset_parallel(hivex_env, path):
         results = meta["model-index"][0]["results"][0]
         row["Task-ID"] = results["task"]["task-id"]
         row["Task"] = results["task"]["name"]
+        key = "Pattern" if "pattern-id" in results["task"] else "Difficulty"
+        row[key] = pattern_map[results["task"]["pattern-id"]] if "pattern-id" in results["task"] else results["task"]["difficulty-id"]
 
         results_metrics = results["metrics"]
 
@@ -163,6 +168,10 @@ def get_data(rl_env, task_id, path) -> pd.DataFrame:
     # Drop the "Task-ID" column
     filtered_data = filtered_data.drop(columns=["Task-ID"])
 
+    # Drop the "Task" column
+    filtered_data = filtered_data.drop(columns=["Task"])
+    
+
     # Convert User and Model columns to clickable links
     for index, row in filtered_data.iterrows():
         user_id = row["User"]