fix of init loading bug of the space chatgpt

app.py

@@ -1,35 +1,70 @@
 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
+
+# --------------------
+# Init
+# --------------------
 DATA_ROOT = os.environ.get("DATA_ROOT")
+EXPECTED_TOKEN = os.environ.get("SUBMISSION_TOKEN")
+
 server = LeaderboardServer(local_leaderboard_path=DATA_ROOT)
 TASKS = list(server.tasks_metadata.keys())
-EXPECTED_TOKEN = os.environ.get("SUBMISSION_TOKEN")
+FIRST_TASK = TASKS[0] if TASKS else None
 
 
+# --------------------
+# Helpers
+# --------------------
 def get_datasets_for_task(task):
     path = os.path.join("references", task)
-    if not os.path.exists(path):
+    if not task or not os.path.exists(path):
         return []
     return [f.replace(".json", "") for f in os.listdir(path) if f.endswith(".json")]
 
 
 def update_datasets(task):
-    return gr.CheckboxGroup(choices=get_datasets_for_task(task), value=get_datasets_for_task(task))
+    datasets = get_datasets_for_task(task)
+    return gr.CheckboxGroup(choices=datasets, value=datasets)
+
+
+def get_leaderboard_df(task):
+    if not task:
+        return None
+    return server.get_leaderboard(task)
 
 
-def submit_model(task, datasets, hyp_file, submitted_by, model_id, model_link_input, token, normalize):
+def submit_model(
+    task,
+    datasets,
+    hyp_file,
+    submitted_by,
+    model_id,
+    model_link_input,
+    token,
+    normalize,
+):
     if not hyp_file:
         return gr.update(visible=True, value="⚠️ Please upload a hypothesis file.")
 
-    if not submitted_by.strip() or not model_id.strip() or not model_link_input.strip() or not token.strip():
+    if not all(
+        [
+            submitted_by.strip(),
+            model_id.strip(),
+            model_link_input.strip(),
+            token.strip(),
+        ]
+    ):
         return gr.update(visible=True, value="⚠️ All fields are required.")
 
     if token.strip() != EXPECTED_TOKEN:
@@ -39,22 +74,16 @@ def submit_model(task, datasets, hyp_file, submitted_by, model_id, model_link_in
         "submitted_by": submitted_by.strip(),
         "model_id": model_id.strip(),
         "model_link": model_link_input.strip(),
-        "normalize": normalize  # Include normalization info in metadata if needed
+        "normalize": normalize,
     }
 
-    leaderboard_df = server.get_leaderboard(task)
-    if len(leaderboard_df) > 0:
-        existing = leaderboard_df[
-            (leaderboard_df["Submitted by"] == submitted_by.strip()) &
-            (leaderboard_df["Model ID"] == model_id.strip())
-            ]
-        # TODO: Uncomment this after testing done
-        # if not existing.empty:
-        #     return gr.update(value="❌ A submission with this name and model ID already exists.", visible=True)
-
     try:
         server.prepare_model_for_submission(
-            hyp_file.name, metadata, task, datasets, normalize=normalize
+            hyp_file.name,
+            metadata,
+            task,
+            datasets,
+            normalize=normalize,
         )
         return gr.update(visible=True, value="✅ Submission successful!")
     except Exception as e:
@@ -62,85 +91,122 @@ def submit_model(task, datasets, hyp_file, submitted_by, model_id, model_link_in
         return gr.update(visible=True, value=f"❌ Error: {str(e)}")
 
 
-def get_leaderboard_df(task):
-    return server.get_leaderboard(task)
-
-
-# Gradio UI
+# --------------------
+# UI
+# --------------------
 with gr.Blocks(css=LEADERBOARD_CSS) as demo:
     gr.Markdown(HEADER_MARKDOWN)
 
-    with gr.Tabs(selected=0) as tabs:
+    with gr.Tabs() as tabs:
+        # -------- Leaderboard Tab --------
         with gr.Tab("📈 Leaderboard"):
             gr.Markdown(LEADERBOARD_TAB_TITLE_MARKDOWN)
 
-            first_task = TASKS[0] if len(TASKS) > 0 else None
-
             leaderboard_task_dropdown = gr.Dropdown(
                 choices=TASKS,
-                value=first_task,
-                label="Select Task for Leaderboard"
+                value=FIRST_TASK,
+                label="Select Task for Leaderboard",
             )
 
-            leaderboard_output = gr.components.Dataframe(
-                datatype=["markdown", "markdown", "float", "float", "float", "float", "float", "float"],
-                value=None,
+            leaderboard_output = gr.Dataframe(
+                value=get_leaderboard_df(FIRST_TASK),
+                datatype=[
+                    "markdown",
+                    "markdown",
+                    "float",
+                    "float",
+                    "float",
+                    "float",
+                    "float",
+                    "float",
+                ],
                 interactive=False,
-                label="Leaderboard"
+                label="Leaderboard",
             )
 
             leaderboard_task_dropdown.change(
                 fn=get_leaderboard_df,
                 inputs=leaderboard_task_dropdown,
-                outputs=leaderboard_output
+                outputs=leaderboard_output,
             )
 
+        # -------- Submit Tab --------
         with gr.Tab("📤 Submit"):
             gr.Markdown(SUBMISSION_TAB_TITLE_MARKDOWN)
 
             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]))
-            task_dropdown.change(fn=update_datasets, inputs=task_dropdown, outputs=dataset_checkboxes)
+                task_dropdown = gr.Dropdown(
+                    choices=TASKS,
+                    value=FIRST_TASK,
+                    label="Select Task",
+                )
+
+                dataset_checkboxes = gr.CheckboxGroup(
+                    choices=get_datasets_for_task(FIRST_TASK),
+                    value=get_datasets_for_task(FIRST_TASK),
+                    label="Select Datasets",
+                )
+
+            task_dropdown.change(
+                fn=update_datasets,
+                inputs=task_dropdown,
+                outputs=dataset_checkboxes,
+            )
 
             with gr.Row():
                 submitted_by_input = gr.Text(label="Submitted by")
                 model_id_input = gr.Text(label="Model Identifier")
-                model_link_input = gr.Text(label="Model Link", placeholder="Link to model or code repository")
-                token_input = gr.Text(label="Submission Token", type="password")
-
-            hyp_file_upload = gr.File(label="Upload Hypothesis JSON", file_types=[".json"])
+                model_link_input = gr.Text(
+                    label="Model Link",
+                    placeholder="Link to model or code repository",
+                )
+                token_input = gr.Text(
+                    label="Submission Token",
+                    type="password",
+                )
+
+            hyp_file_upload = gr.File(
+                label="Upload Hypothesis JSON",
+                file_types=[".json"],
+            )
 
-            normalize_checkbox = gr.Checkbox(label="Apply Chime-8 Normalization", value=True)
+            normalize_checkbox = gr.Checkbox(
+                label="Apply Chime-8 Normalization",
+                value=True,
+            )
 
             submit_btn = gr.Button("Submit")
 
-            with Modal("Submission Feedback", visible=False) as loading_msg:
-                feedback_text = gr.Text(visible=True, label="⏳ Processing your submission...")
+            with Modal("Submission Feedback", visible=False) as modal:
+                feedback_text = gr.Text(
+                    visible=True,
+                    label="Submission status",
+                )
+
             gr.Markdown(ADDITIONAL_NOTES_MARKDOWN)
 
             submit_btn.click(
-                lambda: gr.update(visible=True),  # Show loading
-                outputs=loading_msg
+                fn=lambda: gr.update(visible=True),
+                outputs=modal,
             ).then(
                 fn=submit_model,
-                inputs=[task_dropdown, dataset_checkboxes, hyp_file_upload,
-                        submitted_by_input, model_id_input, model_link_input, token_input, normalize_checkbox],
-                outputs=[feedback_text],
+                inputs=[
+                    task_dropdown,
+                    dataset_checkboxes,
+                    hyp_file_upload,
+                    submitted_by_input,
+                    model_id_input,
+                    model_link_input,
+                    token_input,
+                    normalize_checkbox,
+                ],
+                outputs=feedback_text,
             ).then(
-                fn=lambda task: get_leaderboard_df(task),
+                fn=get_leaderboard_df,
                 inputs=task_dropdown,
-                outputs=leaderboard_output
+                outputs=leaderboard_output,
             )
-    demo.load(
-        fn=lambda: gr.Tabs(selected=1),
-        outputs=[tabs]
-    ).then(
-        fn=get_leaderboard_df,
-        inputs=[leaderboard_task_dropdown],
-        outputs=[leaderboard_output]
-    )
+
 
 if __name__ == "__main__":
     demo.launch()