Add download option

app.py

@@ -1,6 +1,7 @@
 import json
 import math
 import subprocess
+import tempfile
 from pathlib import Path
 
 import numpy as np
@@ -44,6 +45,13 @@ raw_data, original_df = get_leaderboard_df(DATA_PATH, COLS, BENCHMARK_COLS)
 leaderboard_df = original_df.copy()
 
 
+
+def export_csv(df):
+    csv_filename = Path(tempfile._get_default_tempdir()) / f"scandeval_leaderboard_{next(tempfile._get_candidate_names())}.csv"
+    df.to_csv(csv_filename)
+    return str(csv_filename)
+    
+    
 def plot_stats(data_path, plotting_library="plotly", columns=None, table=None):
     plots = {}
     files = Path(data_path).rglob("*.jsonl")
@@ -163,13 +171,14 @@ def update_table_and_plot(
     filtered_df = filter_models(hidden_df, type_query, size_query, precision_query, show_deleted)
     filtered_df = filter_queries(query, filtered_df)
     df = select_columns(filtered_df, columns)
+    export_filename = export_csv(df)
     df = (df.style
           .format(precision=2, thousands=",", decimal=".")
           .highlight_max(props="background-color: lightgreen; color: black;", axis=0, subset=df.columns[1:])
           .highlight_between(props="color: red;", axis=0, subset=df.columns[1:], left=-np.inf, right=-np.inf)
           )
     fig = plot_stats(DATA_PATH, columns=columns, table=df)
-    return df, fig
+    return df, fig, export_filename
 
 
 def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
@@ -305,6 +314,8 @@ with demo:
                 interactive=False,
                 visible=True,
             )
+            leaderboard_file = gr.File(interactive=False, value=export_csv(leaderboard_df[cols_to_show]), visible=True)
+
         with gr.TabItem("📊 LLM Plots", elem_id="llm-benchmark-tab-plot", id=1):
             leaderboard_plot = gr.components.Plot(plot_stats(DATA_PATH))
 
@@ -329,6 +340,7 @@ with demo:
         [
             leaderboard_table,
             leaderboard_plot,
+            leaderboard_file,
         ],
     )
     for selector in [shown_columns, filter_columns_type, filter_columns_precision, filter_columns_size,
@@ -347,6 +359,7 @@ with demo:
             [
                 leaderboard_table,
                 leaderboard_plot,
+                leaderboard_file,
             ],
             queue=True,
         )