trying to write results

app.py

@@ -2,6 +2,7 @@ import pathlib
 import tempfile
 from typing import BinaryIO, Literal
 import json
+import pandas as pd
 
 import gradio as gr
 from datasets import load_dataset, Dataset
@@ -86,6 +87,8 @@ def read_boundary(filename):
 def write_results(record, result):
     record.update(result)
     record['result_filename'] = record['submission_filename'].strip('.json') + '_results.json'
+    record['evaluated'] = True
+
     with tempfile.NamedTemporaryFile(mode="w", suffix=".json", delete=False) as tmp:
         json.dump(record, tmp, indent=2)
         tmp.flush()
@@ -103,6 +106,21 @@ def write_results(record, result):
     return
 
 
+def get_leaderboard(problem_type: str):
+    ds = load_dataset(results_repo, split=problem_type)
+
+    filtered = ds.filter(lambda x: x["evaluated"])
+
+    if len(filtered) == 0:
+        return pd.DataFrame(columns=["submission_time", "problem_type", "score"])
+
+    df = pd.DataFrame(filtered)
+    score_field = "score" if "score" in df.columns else "objective"  # fallback
+
+    df = df.sort_values(by=score_field, ascending=True)
+    leaderboard = df[["submission_time", "problem_type", score_field]].reset_index(drop=True)
+    return leaderboard
+
 def gradio_interface() -> gr.Blocks:
     with gr.Blocks() as demo:
         gr.Markdown(
@@ -125,6 +143,18 @@ def gradio_interface() -> gr.Blocks:
             inputs=[problem_type, boundary_file],
             outputs=output,
         )
+
+        with gr.Row():
+            leaderboard_type = gr.Dropdown(PROBLEM_TYPES, label="Leaderboard Problem Type", value="geometrical")
+            leaderboard_btn = gr.Button("Load Leaderboard")
+
+        leaderboard_df = gr.Dataframe(label="Leaderboard")
+
+        leaderboard_btn.click(
+            lambda pt: get_leaderboard(pt).to_dict(orient="records"),
+            inputs=[leaderboard_type],
+            outputs=[leaderboard_df]
+        )
     return demo