trying to write results

app.py

@@ -7,6 +7,7 @@ import pandas as pd
 import gradio as gr
 from datasets import load_dataset, Dataset
 from huggingface_hub import upload_file, hf_hub_download
+from gradio_leaderboard import ColumnFilter, Leaderboard, SelectColumns
 from evaluation import evaluate_problem
 from datetime import datetime
 import os
@@ -118,43 +119,55 @@ def get_leaderboard(problem_type: str):
     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
+    # leaderboard = df[["submission_time", "problem_type", score_field]].reset_index(drop=True)
+    return df
 
 def gradio_interface() -> gr.Blocks:
     with gr.Blocks() as demo:
-        gr.Markdown(
-            """
-        # Plasma Boundary Evaluation App
-        Upload your plasma boundary JSON and select the problem type to get your score.
-        """
-        )
-        with gr.Row():
-            problem_type = gr.Dropdown(
-                PROBLEM_TYPES, label="Problem Type", value="geometrical"
+        with gr.Tabs():
+            with gr.Tab("Submit"):
+                gr.Markdown(
+                    """
+                # Plasma Boundary Evaluation App
+                Upload your plasma boundary JSON and select the problem type to get your score.
+                """
+                )
+                with gr.Row():
+                    problem_type = gr.Dropdown(
+                        PROBLEM_TYPES, label="Problem Type", value="geometrical"
+                    )
+                    boundary_file = gr.File(label="Boundary JSON File (.json)")
+
+                boundary_file
+                output = gr.Textbox(label="Evaluation Result", lines=10)
+                submit_btn = gr.Button("Evaluate")
+                submit_btn.click(
+                    submit_boundary,
+                    inputs=[problem_type, boundary_file],
+                    outputs=output,
+                )
+        with gr.Tab("Leaderboards"):                
+            gr.Markdown("# Leaderboard")
+            leaderboard_type = gr.Dropdown(PROBLEM_TYPES, value="geometrical", label="Problem Type")
+            refresh_btn = gr.Button("Refresh Leaderboard")
+            leaderboard_df = gr.Dataframe(label="Leaderboard")
+
+            # Trigger refresh when button is clicked
+            refresh_btn.click(
+                lambda pt: get_leaderboard(pt).to_dict(orient="records"),
+                inputs=[leaderboard_type],
+                outputs=[leaderboard_df],
             )
-            boundary_file = gr.File(label="Boundary JSON File (.json)")
-
-        boundary_file
-        output = gr.Textbox(label="Evaluation Result", lines=10)
-        submit_btn = gr.Button("Evaluate")
-        submit_btn.click(
-            submit_boundary,
-            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(
+                value=leaderboard_df,
+                select_columns=["submission_time", "feasability", "score ⬆️", "objective"],
+                # search_columns=["model_name_for_query", "Type"],
+                # hide_columns=["model_name_for_query", "Model Size"],
+                # filter_columns=["T", "Precision", "Model Size"],
+            )
 
-        leaderboard_btn.click(
-            lambda pt: get_leaderboard(pt).to_dict(orient="records"),
-            inputs=[leaderboard_type],
-            outputs=[leaderboard_df]
-        )
+            
     return demo
 
 

requirements.txt

@@ -1,4 +1,5 @@
 constellaration==0.2.1
 gradio
 datasets
-huggingface_hub
+huggingface_hub
+gradio-leaderboard