Added automatic saving and made UI/UX simpler

app.py

@@ -11,10 +11,17 @@ RESULTS_DIR = "results"
 
 def load_pairs():
     """Load matching image pairs from both folders, keyed by filename."""
-    ro_files = {f for f in os.listdir(ROENTGEN_DIR)
-                if os.path.isfile(os.path.join(ROENTGEN_DIR, f))}
-    pe_files = {f for f in os.listdir(PEDISYNTH_DIR)
-                if os.path.isfile(os.path.join(PEDISYNTH_DIR, f))}
+    if not os.path.isdir(ROENTGEN_DIR) or not os.path.isdir(PEDISYNTH_DIR):
+        return []
+
+    ro_files = {
+        f for f in os.listdir(ROENTGEN_DIR)
+        if os.path.isfile(os.path.join(ROENTGEN_DIR, f))
+    }
+    pe_files = {
+        f for f in os.listdir(PEDISYNTH_DIR)
+        if os.path.isfile(os.path.join(PEDISYNTH_DIR, f))
+    }
 
     common = sorted(ro_files & pe_files)
     pairs = []
@@ -40,24 +47,31 @@ ALL_PAIRS = load_pairs()
 os.makedirs(RESULTS_DIR, exist_ok=True)
 
 
-def start_session(doctor_id):
-    """Initialize a new randomized session for this doctor."""
+def start_session(doctor_initials):
+    """Initialize a new randomized session."""
+    # Default initials if nothing entered
+    if not doctor_initials:
+        doctor_initials = "anonymous"
+
+    # No pairs found
     if not ALL_PAIRS:
-        status = "No matching image pairs found. Check your roentgen/ and pedisynth/ folders."
+        status = (
+            "No matching image pairs found. "
+            "Check your `roentgen/` and `pedisynth/` folders."
+        )
+        # Keep intro visible, keep session hidden
         return (
-            None,  # img_a
-            None,  # img_b
-            "",    # condition text
-            "",    # progress text
-            None,  # state
-            None,  # results file
-            status,
-            gr.update(value=None),  # clear choice
+            None,                      # img_a
+            None,                      # img_b
+            "",                        # condition_text
+            "",                        # progress_text
+            None,                      # state
+            status,                    # status_text
+            gr.update(value=None),     # clear choice
+            gr.update(visible=True),   # intro_group visible
+            gr.update(visible=False),  # session_group hidden
         )
 
-    if not doctor_id:
-        doctor_id = "anonymous"
-
     # Randomize order of pairs
     pairs = ALL_PAIRS.copy()
     random.shuffle(pairs)
@@ -87,7 +101,7 @@ def start_session(doctor_id):
         )
 
     state = {
-        "doctor_id": doctor_id,
+        "doctor_id": doctor_initials,
         "pairs": session_pairs,
         "idx": 0,
         "records": [],
@@ -96,40 +110,41 @@ def start_session(doctor_id):
     first = session_pairs[0]
     condition_text = f"**Condition:** {first['condition']}"
     progress_text = f"Pair 1 of {len(session_pairs)}"
-    status = f"Session started for doctor: {doctor_id}"
+    status = f"Session started for doctor: {doctor_initials}"
 
+    # Hide intro group, show session group
     return (
         first["left_path"],
         first["right_path"],
         condition_text,
         progress_text,
         state,
-        None,  # no CSV yet
         status,
-        gr.update(value=None),  # clear choice
+        gr.update(value=None),         # clear choice
+        gr.update(visible=False),      # intro_group hidden
+        gr.update(visible=True),       # session_group visible
     )
 
 
 def next_pair(choice, state):
     """Record the doctor's choice and move to the next pair."""
-    # If no active session
+    # No active session
     if state is None or not state.get("pairs"):
-        status = "No active session. Please start a session first."
+        status = "No active session. Please refresh and start a new session."
         return (
             None,
             None,
             "",
             "",
             state,
-            None,
             status,
-            gr.update(value=None),
+            gr.update(),  # leave choice alone
         )
 
     idx = state["idx"]
     pairs = state["pairs"]
 
-    # Ensure choice is made
+    # Ensure a choice is made
     if not choice:
         status = "Please select A or B before proceeding."
         cur_pair = pairs[idx]
@@ -141,7 +156,6 @@ def next_pair(choice, state):
             condition_text,
             progress_text,
             state,
-            None,
             status,
             gr.update(),  # keep current choice
         )
@@ -149,7 +163,9 @@ def next_pair(choice, state):
     cur_pair = pairs[idx]
 
     chosen_side = choice
-    chosen_model = cur_pair["left_model"] if choice == "A" else cur_pair["right_model"]
+    chosen_model = (
+        cur_pair["left_model"] if choice == "A" else cur_pair["right_model"]
+    )
 
     record = {
         "doctor_id": state["doctor_id"],
@@ -170,7 +186,7 @@ def next_pair(choice, state):
     state["idx"] += 1
     idx = state["idx"]
 
-    # If we've reached the end, save CSV
+    # If we've reached the end, save CSV (backend only)
     if idx >= len(pairs):
         df = pd.DataFrame(state["records"])
         ts = datetime.datetime.utcnow().strftime("%Y%m%d_%H%M%S")
@@ -183,19 +199,17 @@ def next_pair(choice, state):
 
         status = (
             f"Session complete. Saved {len(state['records'])} comparisons "
-            f"to `{csv_name}`."
+            f"to `{csv_name}` on the Space."
         )
 
-        # Optionally, reset session index or leave state as-is for audit
         return (
-            None,             # img_a
-            None,             # img_b
-            "Session finished.",  # condition text
-            "",               # progress
+            None,                   # img_a
+            None,                   # img_b
+            "Session finished.",    # condition_text
+            "",                     # progress_text
             state,
-            csv_path,         # file for download
             status,
-            gr.update(value=None),
+            gr.update(value=None),  # clear choice
         )
 
     # Otherwise, show next pair
@@ -210,7 +224,6 @@ def next_pair(choice, state):
         condition_text,
         progress_text,
         state,
-        None,
         status,
         gr.update(value=None),  # clear choice
     )
@@ -219,50 +232,50 @@ def next_pair(choice, state):
 with gr.Blocks() as demo:
     gr.Markdown("# Synthetic Chest X-ray A/B Evaluation")
 
-    with gr.Row():
-        doctor_id = gr.Textbox(
-            label="Doctor ID (or initials)",
-            placeholder="e.g., dr_smith",
+    # Intro screen: just initials + Start
+    with gr.Group() as intro_group:
+        with gr.Row():
+            doctor_initials = gr.Textbox(
+                label="Doctor initials",
+                placeholder="e.g., RG",
+            )
+            start_button = gr.Button("Start New Session", variant="primary")
+
+    # Main session UI (hidden initially)
+    with gr.Group(visible=False) as session_group:
+        state = gr.State()
+
+        with gr.Row():
+            condition_text = gr.Markdown("")
+            progress_text = gr.Markdown("")
+
+        with gr.Row():
+            img_a = gr.Image(label="Image A", interactive=False)
+            img_b = gr.Image(label="Image B", interactive=False)
+
+        choice = gr.Radio(
+            choices=["A", "B"],
+            label="Which image better matches the stated condition?",
+            interactive=True,
         )
-        start_button = gr.Button("Start New Session", variant="primary")
-
-    state = gr.State()
-
-    with gr.Row():
-        condition_text = gr.Markdown("")
-        progress_text = gr.Markdown("")
 
-    with gr.Row():
-        img_a = gr.Image(label="Image A", interactive=False)
-        img_b = gr.Image(label="Image B", interactive=False)
-
-    choice = gr.Radio(
-        choices=["A", "B"],
-        label="Which image better matches the stated condition?",
-        interactive=True,
-    )
-
-    next_button = gr.Button("Next")
-
-    results_file = gr.File(
-        label="Download session CSV",
-        interactive=False,
-    )
-    status_text = gr.Markdown("")
+        next_button = gr.Button("Next")
+        status_text = gr.Markdown("")
 
     # Wire callbacks
     start_button.click(
         fn=start_session,
-        inputs=doctor_id,
+        inputs=doctor_initials,
         outputs=[
             img_a,
             img_b,
             condition_text,
             progress_text,
             state,
-            results_file,
             status_text,
             choice,
+            intro_group,   # visibility toggle
+            session_group, # visibility toggle
         ],
     )
 
@@ -275,7 +288,6 @@ with gr.Blocks() as demo:
             condition_text,
             progress_text,
             state,
-            results_file,
             status_text,
             choice,
         ],