dropdown with submission type updating example download

app.py

@@ -4,7 +4,13 @@ import gradio as gr
 from gradio_leaderboard import Leaderboard
 
 from utils import fetch_hf_results, show_output_box
-from constants import ASSAY_LIST, ASSAY_RENAME, ASSAY_EMOJIS, ASSAY_DESCRIPTION
+from constants import (
+    ASSAY_LIST,
+    ASSAY_RENAME,
+    ASSAY_EMOJIS,
+    ASSAY_DESCRIPTION,
+    EXAMPLE_FILE_DICT,
+)
 from about import ABOUT_TEXT, FAQS
 from submit import make_submission
 
@@ -22,6 +28,17 @@ def format_leaderboard_table(df_results: pd.DataFrame, assay: str | None = None)
     return df.sort_values(by="spearman", ascending=False)
 
 
+# Cache the results to avoid multiple downloads
+_cached_results = None
+
+
+def get_cached_results():
+    global _cached_results
+    if _cached_results is None:
+        _cached_results = fetch_hf_results()
+    return _cached_results
+
+
 def get_leaderboard_object(assay: str | None = None):
     filter_columns = ["model"]
     if assay is None:
@@ -29,7 +46,7 @@ def get_leaderboard_object(assay: str | None = None):
     # TODO how to sort filter columns alphabetically?
     Leaderboard(
         # TODO(Lood) check that this actually refreshes using the function
-        value=format_leaderboard_table(df_results=fetch_hf_results(), assay=assay),
+        value=format_leaderboard_table(df_results=get_cached_results(), assay=assay),
         datatype=["str", "str", "str", "number"],
         select_columns=["model", "property", "spearman", "spearman_cross_val"],
         search_columns=["model"],
@@ -90,18 +107,13 @@ with gr.Blocks() as demo:
             filename = gr.State(value=None)
             eval_state = gr.State(value=None)
             user_state = gr.State(value=None)
-            anonymous_state = gr.State(value=False)
+            submission_type_state = gr.State(value="GDPa1")
+            download_file_state = gr.State(value=EXAMPLE_FILE_DICT["GDPa1"])
 
             login_button = gr.LoginButton(
                 value="Sign in with Hugging Face to see account name"
             )  # Note(Lood): Is this mandatory?
 
-            gr.DownloadButton(
-                label="📥 Download example submission CSV",
-                value="data/example-predictions.csv",
-                variant="secondary",
-            )
-
             with gr.Row():
                 with gr.Column():
                     username_input = gr.Textbox(
@@ -109,11 +121,17 @@ with gr.Blocks() as demo:
                         placeholder="Enter your Hugging Face username",
                         info="This will be displayed on the leaderboard.",
                     )
-
-                    anonymous_checkbox = gr.Checkbox(
-                        label="Would you like to keep your submission anonymous?"
-                    )  # Can make this ticked by default
                 with gr.Column():
+                    submission_type_dropdown = gr.Dropdown(
+                        choices=["GDPa1", "GDPa1_cross_validation"],
+                        value="GDPa1",
+                        label="Submission Type",
+                    )
+                    download_button = gr.DownloadButton(
+                        label="📥 Download example submission CSV",
+                        value=EXAMPLE_FILE_DICT["GDPa1"],
+                        variant="secondary",
+                    )
                     submission_file = gr.File(label="Submission CSV")
 
             # TODO(Lood): How do we get the username from the login button instead? Do we want this?
@@ -123,6 +141,27 @@ with gr.Blocks() as demo:
                 outputs=user_state,
             )
 
+            def update_submission_type_and_file(submission_type):
+                download_file = EXAMPLE_FILE_DICT.get(
+                    submission_type, EXAMPLE_FILE_DICT["GDPa1"]
+                )
+                return (
+                    submission_type,
+                    download_file,
+                    gr.DownloadButton(
+                        label="📥 Download example submission CSV",
+                        value=download_file,
+                        variant="secondary",
+                    ),
+                )
+
+            # Update submission type state and download button when dropdown changes
+            submission_type_dropdown.change(
+                fn=update_submission_type_and_file,
+                inputs=submission_type_dropdown,
+                outputs=[submission_type_state, download_file_state, download_button],
+            )
+
             submit_btn = gr.Button("Evaluate")
             message = gr.Textbox(label="Status", lines=1, visible=False)
             # help message
@@ -132,7 +171,7 @@ with gr.Blocks() as demo:
 
             submit_btn.click(
                 make_submission,
-                inputs=[submission_file, user_state, anonymous_state],
+                inputs=[submission_file, user_state, submission_type_state],
                 outputs=[message],
             ).then(
                 fn=show_output_box,