still trying to make the leaderboard

about.py

@@ -0,0 +1,9 @@
+import os
+from huggingface_hub import HfApi
+
+PROBLEM_TYPES = ["geometrical", "simple_to_build", "mhd_stable"]
+TOKEN = os.environ.get("HF_TOKEN")
+CACHE_PATH=os.getenv("HF_HOME", ".")
+API = HfApi(token=TOKEN)
+submissions_repo = "cgeorgiaw/constellaration-submissions"
+results_repo = "cgeorgiaw/constellaration-results"

app.py

@@ -12,77 +12,9 @@ from gradio_leaderboard import ColumnFilter, Leaderboard, SelectColumns
 from evaluation import evaluate_problem
 from datetime import datetime
 import os
-from huggingface_hub import HfApi
-
-PROBLEM_TYPES = ["geometrical", "simple_to_build", "mhd_stable"]
-TOKEN = os.environ.get("HF_TOKEN")
-CACHE_PATH=os.getenv("HF_HOME", ".")
-API = HfApi(token=TOKEN)
-submissions_repo = "cgeorgiaw/constellaration-submissions"
-results_repo = "cgeorgiaw/constellaration-results"
-
-def submit_boundary(
-    problem_type: Literal["geometrical", "simple_to_build", "mhd_stable"],
-    boundary_file: BinaryIO,
-    user_state,
-) -> str:
-    
-    # error handling
-    if user_state is None:
-        raise gr.Error("You must be logged in to submit a file.")
-    
-    file_path = boundary_file.name
-
-    if not file_path:
-        raise gr.Error("Uploaded file object does not have a valid file path.")
-    
-    path_obj = pathlib.Path(file_path)
-    timestamp = datetime.utcnow().isoformat()
-
-    with (
-        path_obj.open("rb") as f_in,
-        tempfile.NamedTemporaryFile(delete=False, suffix=".json") as tmp_boundary,
-    ):
-        file_content = f_in.read()
-        tmp_boundary.write(file_content)
-        tmp_boundary_path = pathlib.Path(tmp_boundary.name)
-
-        # write to dataset
-        filename = f"{problem_type}/{timestamp.replace(':', '-')}_{problem_type}.json"
-        record = {
-            "submission_filename": filename,
-            "submission_time": timestamp,
-            "problem_type": problem_type,
-            "boundary_json": file_content.decode("utf-8"),
-            "evaluated": False,
-            "user": user_state,
-        }
-        with tempfile.NamedTemporaryFile(mode="w", suffix=".json", delete=False) as tmp:
-            json.dump(record, tmp, indent=2)
-            tmp.flush()
-            tmp_name = tmp.name
-
-        API.upload_file(
-            path_or_fileobj=tmp_name, 
-            path_in_repo=filename,
-            repo_id=submissions_repo,
-            repo_type="dataset",
-            commit_message=f"Add submission for {problem_type} at {timestamp}"
-        )
-        pathlib.Path(tmp_name).unlink()
-
-    '''# then do eval
-    local_path = read_boundary(filename)
-
-    try:
-        result = evaluate_problem(problem_type, local_path)
-        write_results(record, result)
-    except Exception as e:
-        raise gr.Error(f"Error during file write:\n{e}")
-    finally:'''
-    tmp_boundary_path.unlink()
 
-    return "✅ Your submission has been received! Sit tight and your scores will appear on the leaderboard shortly.", filename
+from submit import submit_boundary
+from about import PROBLEM_TYPES, TOKEN, CACHE_PATH, API, submissions_repo, results_repo
 
 def read_boundary(filename):
     local_path = hf_hub_download(
@@ -131,7 +63,7 @@ def get_user(profile: gr.OAuthProfile | None) -> str:
         return "Please login to submit a boundary for evaluation."
     return profile.username
 
-def get_leaderboard(problem_type: str):
+def get_leaderboard():
     ds = load_dataset(results_repo, split='train')
     df = pd.DataFrame(ds)
     
@@ -148,15 +80,14 @@ def gradio_interface() -> gr.Blocks:
         with gr.Tabs(elem_classes="tab-buttons"):
             with gr.TabItem("Leaderboard", elem_id="boundary-benchmark-tab-table"):                
                 gr.Markdown("# Boundary Design Leaderboard")
-                leaderboard_type = gr.Dropdown(PROBLEM_TYPES, value="simple_to_build", label="Problem Type")
-                leaderboard_df = get_leaderboard(leaderboard_type)
+                leaderboard_df = get_leaderboard()
 
                 Leaderboard(
                     value=leaderboard_df,
                     select_columns=["submission_time", "feasibility", "score", "objective", "user"],
-                    search_columns=["submission_time", "score"],
+                    search_columns=["submission_time", "score", "user"],
                     hide_columns=["result_filename", "submission_filename", "minimize_objective", "boundary_json", "evaluated"],
-                    # filter_columns=["T", "Precision", "Model Size"],
+                    filter_columns=["problem_type", "submission_time"],
                 )
 
                 # def update_leaderboard(problem_type):

submit.py

@@ -0,0 +1,69 @@
+import pathlib
+from pathlib import Path
+import tempfile
+from typing import BinaryIO, Literal
+import json
+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
+from huggingface_hub import HfApi
+
+def submit_boundary(
+    problem_type: Literal["geometrical", "simple_to_build", "mhd_stable"],
+    boundary_file: BinaryIO,
+    user_state,
+) -> str:
+    
+    # error handling
+    if user_state is None:
+        raise gr.Error("You must be logged in to submit a file.")
+    
+    file_path = boundary_file.name
+
+    if not file_path:
+        raise gr.Error("Uploaded file object does not have a valid file path.")
+    
+    path_obj = pathlib.Path(file_path)
+    timestamp = datetime.utcnow().isoformat()
+
+    with (
+        path_obj.open("rb") as f_in,
+        tempfile.NamedTemporaryFile(delete=False, suffix=".json") as tmp_boundary,
+    ):
+        file_content = f_in.read()
+        tmp_boundary.write(file_content)
+        tmp_boundary_path = pathlib.Path(tmp_boundary.name)
+
+        # write to dataset
+        filename = f"{problem_type}/{timestamp.replace(':', '-')}_{problem_type}.json"
+        record = {
+            "submission_filename": filename,
+            "submission_time": timestamp,
+            "problem_type": problem_type,
+            "boundary_json": file_content.decode("utf-8"),
+            "evaluated": False,
+            "user": user_state,
+        }
+        with tempfile.NamedTemporaryFile(mode="w", suffix=".json", delete=False) as tmp:
+            json.dump(record, tmp, indent=2)
+            tmp.flush()
+            tmp_name = tmp.name
+
+        API.upload_file(
+            path_or_fileobj=tmp_name, 
+            path_in_repo=filename,
+            repo_id=submissions_repo,
+            repo_type="dataset",
+            commit_message=f"Add submission for {problem_type} at {timestamp}"
+        )
+        pathlib.Path(tmp_name).unlink()
+
+    tmp_boundary_path.unlink()
+
+    return "✅ Your submission has been received! Sit tight and your scores will appear on the leaderboard shortly.", filename

visualize.py

@@ -0,0 +1,6 @@
+from constellaration.utils import (
+    file_exporter,
+    visualization,
+    visualization_utils,
+)
+