Leaderboard path fix to persistent storage

app.py

@@ -3,11 +3,13 @@ import os
 import gradio as gr
 from gradio_modal import Modal
 
-from content import HEADER_MARKDOWN, LEADERBOARD_TAB_TITLE_MARKDOWN, SUBMISSION_TAB_TITLE_MARKDOWN, ADDITIONAL_NOTES_MARKDOWN, LEADERBOARD_CSS
+from content import HEADER_MARKDOWN, LEADERBOARD_TAB_TITLE_MARKDOWN, SUBMISSION_TAB_TITLE_MARKDOWN, \
+    ADDITIONAL_NOTES_MARKDOWN, LEADERBOARD_CSS
 from leaderboard_server import LeaderboardServer
 
 # Initialize server and task list
-server = LeaderboardServer()
+DATA_ROOT = os.environ.get("DATA_ROOT")
+server = LeaderboardServer(local_leaderboard_path=DATA_ROOT)
 TASKS = list(server.tasks_metadata.keys())
 EXPECTED_TOKEN = os.environ.get("SUBMISSION_TOKEN")
 
@@ -45,7 +47,7 @@ def submit_model(task, datasets, hyp_file, submitted_by, model_id, model_link_in
         existing = leaderboard_df[
             (leaderboard_df["Submitted by"] == submitted_by.strip()) &
             (leaderboard_df["Model ID"] == model_id.strip())
-        ]
+            ]
         if not existing.empty:
             return gr.update(value="❌ A submission with this name and model ID already exists.", visible=True)
 
@@ -58,9 +60,11 @@ def submit_model(task, datasets, hyp_file, submitted_by, model_id, model_link_in
         print(e)
         return gr.update(visible=True, value=f"❌ Error: {str(e)}")
 
+
 def get_leaderboard_df(task):
     return server.get_leaderboard(task)
 
+
 # Gradio UI
 with gr.Blocks(css=LEADERBOARD_CSS) as demo:
     gr.Markdown(HEADER_MARKDOWN)
@@ -83,7 +87,8 @@ with gr.Blocks(css=LEADERBOARD_CSS) as demo:
 
             with gr.Row():
                 task_dropdown = gr.Dropdown(choices=TASKS, value=TASKS[0], label="Select Task")
-                dataset_checkboxes = gr.CheckboxGroup(choices=get_datasets_for_task(TASKS[0]), label="Select Datasets", value=get_datasets_for_task(TASKS[0]))
+                dataset_checkboxes = gr.CheckboxGroup(choices=get_datasets_for_task(TASKS[0]), label="Select Datasets",
+                                                      value=get_datasets_for_task(TASKS[0]))
             task_dropdown.change(fn=update_datasets, inputs=task_dropdown, outputs=dataset_checkboxes)
 
             with gr.Row():
@@ -117,4 +122,4 @@ with gr.Blocks(css=LEADERBOARD_CSS) as demo:
             )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()