Update app.py

app.py

@@ -110,7 +110,8 @@ def update_leaderboard_dataset_parallel(hivex_env, path):
         row["User"] = user_id
         row["Model"] = model_id
         results = meta["model-index"][0]["results"][0]
-        row["Task"] = results["task"]["task-id"]
+        row["Task-ID"] = results["task"]["task-id"]
+        row["Task"] = results["task"]["name"]
 
         results_metrics = results["metrics"]
 
@@ -148,19 +149,19 @@ def run_update_dataset():
     )
 
 
-def get_data(rl_env, task, path) -> pd.DataFrame:
+def get_data(rl_env, task_id, path) -> pd.DataFrame:
     """
-    Get data from rl_env, filter by the given task, and drop the Task column.
-    :return: filtered data as a pandas DataFrame without the Task column
+    Get data from rl_env, filter by the given task_id, and drop the Task-ID column.
+    :return: filtered data as a pandas DataFrame without the Task-ID column
     """
     csv_path = path + "/" + rl_env + ".csv"
     data = pd.read_csv(csv_path)
 
-    # Filter the data to only include rows where the "Task" column matches the given task
-    filtered_data = data[data["Task"] == task]
+    # Filter the data to only include rows where the "Task-ID" column matches the given task_id
+    filtered_data = data[data["Task-ID"] == task_id]
 
-    # Drop the "Task" column
-    filtered_data = filtered_data.drop(columns=["Task"])
+    # Drop the "Task-ID" column
+    filtered_data = filtered_data.drop(columns=["Task-ID"])
 
     # Convert User and Model columns to clickable links
     for index, row in filtered_data.iterrows():
@@ -171,6 +172,23 @@ def get_data(rl_env, task, path) -> pd.DataFrame:
 
     return filtered_data
 
+def get_task(rl_env, task_id, path) -> str:
+    """
+    Get the task name from the leaderboard dataset based on the rl_env and task_id.
+    :return: The task name as a string
+    """
+    csv_path = path + "/" + rl_env + ".csv"
+    data = pd.read_csv(csv_path)
+
+    # Filter the data to find the row with the matching task_id
+    task_row = data[data["Task-ID"] == task_id]
+
+    # Check if the task exists and return the task name
+    if not task_row.empty:
+        task_name = task_row.iloc[0]["Task"]
+        return task_name
+    else:
+        return "Task not found"
 
 run_update_dataset()
 
@@ -204,8 +222,7 @@ with block:
             with gr.Tab(hivex_env["title"]) as env_tabs:
                 # TASK TABS
                 for j in range(0, hivex_env["task_count"]):
-                    task = "Task " + str(j + 1)
-                    with gr.TabItem(f"Task {j}"):
+                    with gr.TabItem(f"Task {j}: {get_task(hivex_env["hivex_env"], j, path_)}"):
                         with gr.Row():
                             gr_dataframe = gr.components.Dataframe(value=get_data(hivex_env["hivex_env"], j, path_), headers=["User", "Model"], datatype=["markdown", "markdown"], row_count=(100, 'fixed'))