RTS Human Evaluation

.gitattributes

@@ -33,3 +33,8 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.jpeg filter=lfs diff=lfs merge=lfs -text
+*.tif filter=lfs diff=lfs merge=lfs -text
+*.tiff filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -15,9 +15,9 @@ from filelock import FileLock
 # Configuration
 # ----------------------
 
-SR_DIR = os.environ.get("SR_DIR", "./data/sr")
-GT_DIR = os.environ.get("GT_DIR", "./data/gt")
-RESULTS_DIR = os.environ.get("RESULTS_DIR", "./results")
+SR_DIR = os.environ.get("SR_DIR", "data/sr")
+GT_DIR = os.environ.get("GT_DIR", "data/gt")
+RESULTS_DIR = os.environ.get("RESULTS_DIR", "results")
 PROGRESS_PATH = os.path.join(RESULTS_DIR, "progress.json")
 ALL_RESULTS_JSONL = os.path.join(RESULTS_DIR, "all_results.jsonl")
 SAVE_PII = True  # set False to only store hashed user_id
@@ -213,8 +213,8 @@ def load_dataset(sr_dir: str, gt_dir: str) -> List[Sample]:
 # Progress & results I/O
 # ----------------------
 
-def hash_user_id(name: str, address: str) -> str:
-    norm = (name or "").strip().lower() + "|" + (address or "").strip().lower()
+def hash_user_id(name: str, email: str) -> str:
+    norm = (name or "").strip().lower() + "|" + (email or "").strip().lower()
     return hashlib.sha256(norm.encode("utf-8")).hexdigest()[:16]
 
 
@@ -363,14 +363,14 @@ def step_display_index(step: int, detect_choice: str | None) -> tuple[int, int]:
 # Gradio callbacks (core)
 # ----------------------
 
-def start_or_resume(name: str, address: str):
-    if not name or not address:
-        raise gr.Error("Please enter your name and address to begin.")
+def start_or_resume(name: str, email: str):
+    if not name or not email:
+        raise gr.Error("Please enter your name and email to begin.")
 
     ensure_paths()
     samples = load_dataset(SR_DIR, GT_DIR)
 
-    uid = hash_user_id(name, address)
+    uid = hash_user_id(name, email)
     progress = load_progress()
     if uid not in progress:
         progress[uid] = {"seen": []}
@@ -429,7 +429,7 @@ def update_overlay(test_img: Image.Image, selected_tiles: List[str] | None):
 
 def _save_record_and_progress(
     name: str,
-    address: str,
+    email: str,
     uid: str,
     samples: List[Sample],
     user_seen: List[str],
@@ -442,8 +442,8 @@ def _save_record_and_progress(
     reasoning: str,
 ):
     """Common save logic used by both 'finish' and 'next image' paths."""
-    if not name or not address:
-        raise gr.Error("Please enter your name and address.")
+    if not name or not email:
+        raise gr.Error("Please enter your name and email.")
     if detection_choice not in {"Yes", "No", "Unsure"}:
         detection_choice = "Unsure"
     valid_clarity = {
@@ -464,7 +464,7 @@ def _save_record_and_progress(
         "timestamp": time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime()),
         "user_id": uid,
         "name": name if SAVE_PII else None,
-        "address": address if SAVE_PII else None,
+        "email": email if SAVE_PII else None,
         "sample_id": sample.sample_id,
         "sr_path": sample.sr_path,
         "gt_path": sample.gt_path,
@@ -499,7 +499,7 @@ def _save_record_and_progress(
 
 def submit_finish(
     name: str,
-    address: str,
+    email: str,
     uid: str,
     samples: List[Sample],
     user_seen: List[str],
@@ -513,7 +513,7 @@ def submit_finish(
 ):
     """Save and show Thank You (no auto-next)."""
     progress = _save_record_and_progress(
-        name, address, uid, samples, user_seen, idx, a_is_sr,
+        name, email, uid, samples, user_seen, idx, a_is_sr,
         detection_choice, confidence, selected_tiles, clarity_choice, reasoning
     )
 
@@ -549,7 +549,7 @@ def submit_finish(
 
 def submit_next_image(
     name: str,
-    address: str,
+    email: str,
     uid: str,
     samples: List[Sample],
     user_seen: List[str],
@@ -563,7 +563,7 @@ def submit_next_image(
 ):
     """Save and immediately load the next sample."""
     progress = _save_record_and_progress(
-        name, address, uid, samples, user_seen, idx, a_is_sr,
+        name, email, uid, samples, user_seen, idx, a_is_sr,
         detection_choice, confidence, selected_tiles, clarity_choice, reasoning
     )
     # NEW — mark global done & release current reservation
@@ -728,7 +728,7 @@ with gr.Blocks(title="RTS Human Evaluation", theme=gr.themes.Soft()) as demo:
         # Retrogressive Thaw Slump (RTS) Human Evaluation
 
         **Instructions:**
-        1. Enter your name and address, then click **Start / Resume**.
+        1. Enter your name and email address, then click **Start / Resume**.
         2. Use **Next** and **Back** to go through the questions.
         3. On the **last step**, choose **Submit & Next Image** to continue, or **Submit & Finish** to end and see a Thank You screen (with **Start Again**).
         """
@@ -745,7 +745,7 @@ with gr.Blocks(title="RTS Human Evaluation", theme=gr.themes.Soft()) as demo:
     with gr.Group() as start_group:
         with gr.Row():
             name = gr.Textbox(label="Full name", placeholder="Jane Doe", autofocus=True)
-            address = gr.Textbox(label="Address", placeholder="123 Pine St, City / or Email")
+            email = gr.Textbox(label="Email address", placeholder="jane@example.com")
             start_btn = gr.Button("Start / Resume", variant="primary")
         status = gr.Markdown("\n")
 
@@ -823,7 +823,7 @@ with gr.Blocks(title="RTS Human Evaluation", theme=gr.themes.Soft()) as demo:
     # --- Wiring ---
     start_event = start_btn.click(
         start_or_resume,
-        inputs=[name, address],
+        inputs=[name, email],
         outputs=[
             state_uid, state_samples, state_seen, state_idx, state_a_is_sr,
             test_img, overlay_img, image_a, image_b, status, your_jsonl_path, eval_panel, tiles, state_step, intro_md, start_group
@@ -867,7 +867,7 @@ with gr.Blocks(title="RTS Human Evaluation", theme=gr.themes.Soft()) as demo:
         submit_finish,
         inputs=[
             name,
-            address,
+            email,
             state_uid,
             state_samples,
             state_seen,
@@ -906,7 +906,7 @@ with gr.Blocks(title="RTS Human Evaluation", theme=gr.themes.Soft()) as demo:
         submit_next_image,
         inputs=[
             name,
-            address,
+            email,
             state_uid,
             state_samples,
             state_seen,
@@ -947,10 +947,10 @@ with gr.Blocks(title="RTS Human Evaluation", theme=gr.themes.Soft()) as demo:
         outputs=[step_status, group_detection, group_confidence, group_tiles, group_clarity, group_reasoning, back_btn, next_btn],
     )
 
-    # Restart: start again using current name/address
+    # Restart: start again using current name/email
     restart_event = restart_btn.click(
         start_or_resume,
-        inputs=[name, address],
+        inputs=[name, email],
         outputs=[
             state_uid, state_samples, state_seen, state_idx, state_a_is_sr,
             test_img, overlay_img, image_a, image_b, status, your_jsonl_path, eval_panel, tiles, state_step, intro_md, start_group

requirements.txt

@@ -0,0 +1,4 @@
+gradio>=4.29
+pillow
+numpy
+filelock