first commit

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    cmake \
+    libnetcdf-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY . .
+RUN pip install -r requirements.txt
+
+EXPOSE 7860
+ENV GRADIO_SERVER_NAME="0.0.0.0"
+
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,84 @@
+import pathlib
+import tempfile
+from typing import BinaryIO, Literal
+import json
+from datasets import Dataset
+
+import gradio as gr
+from evaluation import evaluate_problem
+from datetime import datetime
+
+
+PROBLEM_TYPES = ["geometrical", "simple_to_build", "mhd_stable"]
+
+def evaluate_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."
+    
+    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,
+    ):
+        file_content = f_in.read()
+        tmp.write(file_content)
+        tmp_path = pathlib.Path(tmp.name)
+    try:
+        result = evaluate_problem(problem_type, str(tmp_path))
+        output = str(result)
+    except Exception as e:
+        output = f"Error during evaluation:\n{e}"
+    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(
+            """
+        # 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(
+            evaluate_boundary,
+            inputs=[problem_type, boundary_file],
+            outputs=output,
+        )
+    return demo
+
+
+if __name__ == "__main__":
+    gradio_interface().launch()

requirements.txt

@@ -0,0 +1,3 @@
+constellaration==0.2.1
+gradio
+datasets