app.py

@@ -24,18 +24,23 @@ api = HfApi(token=HF_TOKEN)
 hivex_envs = [
     {
         "hivex_env": "hivex-wind-farm-control",
+        "task_count": 2,
     },
     {
         "hivex_env": "hivex-wildfire-resource-management",
+        "task_count": 3,
     },
     {
         "hivex_env": "hivex-drone-based-reforestation",
+        "task_count": 7,
     },
     {
         "hivex_env": "hivex-ocean-plastic-collection",
+        "task_count": 4,
     },
     {
         "hivex_env": "hivex-aerial-wildfire-suppression",
+        "task_count": 9,
     },
 ]
 
@@ -196,18 +201,32 @@ main_block = gr.Blocks()
 with main_block:
     with gr.Row(elem_id="header-row"):
         # TITLE + "<p>Total models: " + str(len(HARD_LEADERBOARD_DF))+ "</p>"
-        gr.HTML("<h1>Leaderboard</h1>")
+        gr.HTML("<h1>HIVEX-Leaderboard</h1>")
 
     # gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
-        with gr.Tab("💎 Hard Set") as hard_tabs:
-            with gr.TabItem(
-                "🏅 Benchmark", elem_id="llm-benchmark-tab-table", id="hard_bench"
-            ):
-                gr.DataTable(
-                    get_data(
-                        "hivex-wind-farm-control", "datasets/hivex-leaderboard-data"
-                    ),
-                    elem_id="hard_benchmark_table",
-                    elem_classes="table",
-                )
+        for i in range(0, len(hivex_envs)):
+            hivex_env = hivex_envs[i]
+            with gr.Tab(hivex_env["hivex_env"]) as tab:
+                for j in range(0, hivex_env["task_count"]):
+                    task = "Task " + str(j + 1)
+                    with gr.TabItem(task):
+                        gr.DataTable(
+                            get_data(
+                                hivex_env["hivex_env"],
+                                "datasets/hivex-leaderboard-data",
+                            ),
+                            elem_id=hivex_env["hivex_env"] + "_" + task,
+                            elem_classes="table",
+                        )
+        # with gr.Tab("💎 Hard Set") as hard_tabs:
+        #     with gr.TabItem(
+        #         "🏅 Benchmark", elem_id="llm-benchmark-tab-table", id="hard_bench"
+        #     ):
+        #         gr.DataTable(
+        #             get_data(
+        #                 "hivex-wind-farm-control", "datasets/hivex-leaderboard-data"
+        #             ),
+        #             elem_id="hard_benchmark_table",
+        #             elem_classes="table",
+        #         )