first commit

app.py

@@ -2,19 +2,22 @@ import pathlib
 import tempfile
 from typing import BinaryIO, Literal
 import json
-from datasets import Dataset
 
 import gradio as gr
+from datasets import load_dataset, Dataset
+from huggingface_hub import upload_file
 from evaluation import evaluate_problem
 from datetime import datetime
 
 
 PROBLEM_TYPES = ["geometrical", "simple_to_build", "mhd_stable"]
+repo_id = "cgeorgiaw/constellaration-submissions"
 
-def evaluate_boundary(
+def submit_boundary(
     problem_type: Literal["geometrical", "simple_to_build", "mhd_stable"],
     boundary_file: BinaryIO,
 ) -> str:
+    
     file_path = boundary_file.name
     if not file_path:
         return "Error: Uploaded file object does not have a valid file path."
@@ -29,6 +32,24 @@ def evaluate_boundary(
         file_content = f_in.read()
         tmp.write(file_content)
         tmp_path = pathlib.Path(tmp.name)
+
+        # write to dataset
+        filename = f"data/{timestamp.replace(':', '-')}_{problem_type}.json"
+        record = {
+            "submission_time": timestamp,
+            "problem_type": problem_type,
+            "boundary_json": file_content.decode("utf-8"),  # Or store file path or URL
+        }
+        json.dump(record, tmp, indent=2)
+        tmp.flush()
+        upload_file(
+            path_or_fileobj=tmp.name,  # this is just a string path
+            path_in_repo=filename,
+            repo_id=repo_id,
+            repo_type="dataset",
+            commit_message=f"Add submission for {problem_type} at {timestamp}"
+        )
+
     try:
         result = evaluate_problem(problem_type, str(tmp_path))
         output = str(result)
@@ -37,24 +58,8 @@ def evaluate_boundary(
     finally:
         tmp_path.unlink()
 
-    save_to_dataset(problem_type, file_content, timestamp, output)
     return output
 
-
-def save_to_dataset(problem_type, file_content, timestamp, result):
-    # Example: convert to one-row dataframe
-    record = {
-        "submission_time": timestamp,
-        "problem_type": problem_type,
-        "boundary_json": file_content.decode("utf-8"),  # Or store file path or URL
-        "result": result,
-    }
-
-    # If this is the first entry, create dataset
-    dataset_file = "local_dataset.jsonl"
-    with open(dataset_file, "a") as f:
-        f.write(json.dumps(record) + "\n")
-
 def gradio_interface() -> gr.Blocks:
     with gr.Blocks() as demo:
         gr.Markdown(
@@ -73,7 +78,7 @@ def gradio_interface() -> gr.Blocks:
         output = gr.Textbox(label="Evaluation Result", lines=10)
         submit_btn = gr.Button("Evaluate")
         submit_btn.click(
-            evaluate_boundary,
+            submit_boundary,
             inputs=[problem_type, boundary_file],
             outputs=output,
         )