fix: refresh button

app.py

@@ -9,12 +9,7 @@ from dabstep_benchmark.leaderboard import *
 def restart_space():
     HF_API.restart_space(repo_id=HF_LEADERBOARD)
     
-# Helper function to update both tables
-def update_tables():
-    leaderboard_df = generate_leaderboard_df()
-    validated = leaderboard_df[leaderboard_df["validated"] == True].drop(columns=["validated"])
-    unvalidated = leaderboard_df[leaderboard_df["validated"] == False].drop(columns=["validated"])
-    return validated, unvalidated
+
 
 
 
@@ -28,11 +23,11 @@ if __name__ == "__main__":
         gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
         
         # Generate initial leaderboard data
-        validated, unvalidated = update_tables()
-        
+        validated_lb, unvalidated_lb = generate_leaderboard_df()
+
         with gr.Tab("Validated"):
             verified_table = gr.Dataframe(
-                value=validated,
+                value=validated_lb,
                 datatype=["markdown", "str", "str", "str", "markdown", "str", "str", "str"],
                 interactive=False,
                 column_widths=["20%"],
@@ -41,14 +36,14 @@ if __name__ == "__main__":
         
         with gr.Tab("Unvalidated"):
             unverified_table = gr.Dataframe(
-                value=unvalidated,
+                value=unvalidated_lb,
                 datatype=["markdown", "str", "str", "str", "markdown", "str", "str", "str"],
                 interactive=False,
                 column_widths=["20%"],
                 wrap=True,
             )
         # create a Gradio event listener that runs when the page is loaded to populate the dataframe
-        demo.load(update_tables, inputs=None, outputs=[verified_table, unverified_table])
+        demo.load(generate_leaderboard_df, inputs=None, outputs=[verified_table, unverified_table])
 
         refresh_button = gr.Button("Refresh")
         refresh_button.click(

dabstep_benchmark/leaderboard.py

@@ -1,4 +1,6 @@
 import re
+from typing import Tuple
+
 import gradio as gr
 import json
 import datetime
@@ -228,7 +230,7 @@ def process_submission(
         Please refresh the leaderboard to see your score displayed.
 """)
 
-def generate_leaderboard_df() -> pd.DataFrame:
+def generate_leaderboard_df() -> Tuple[pd.DataFrame, pd.DataFrame]:
     task_scores_df = DATASETS["task_scores"].to_pandas()
     submissions_df = DATASETS["submissions"].to_pandas()
 
@@ -311,4 +313,7 @@ def generate_leaderboard_df() -> pd.DataFrame:
     # sort-by best score
     df.sort_values(by="Hard Level Accuracy (%)", ascending=False, inplace=True)
 
-    return df
+    validated = leaderboard_df[leaderboard_df["validated"] == True].drop(columns=["validated"])
+    unvalidated = leaderboard_df[leaderboard_df["validated"] == False].drop(columns=["validated"])
+
+    return validated, unvalidated