Update app.py

app.py

@@ -84,6 +84,44 @@ def save_evaluations(evaluations, doctor_name):
     df.to_csv(filepath, index=False)
     return filepath, filename
 
+def record_preference(choice):
+    """Record the doctor's preference and move to next image."""
+    if not state["session_active"]:
+        return None, None, "Session not active.", ""
+    
+    current_pair = state["image_pairs"][state["current_index"]]
+    
+    if choice == "A":
+        preferred_source = state["current_left_source"]
+    else:  # choice == "B"
+        preferred_source = state["current_right_source"]
+    
+    state["evaluations"].append({
+        'timestamp': datetime.now().isoformat(),
+        'condition': current_pair['condition'],
+        'base_name': current_pair['base_name'],
+        'preference': preferred_source.capitalize(),
+        'doctor': state["doctor_name"]
+    })
+    
+    state["current_index"] += 1
+    
+    left_img, right_img, condition, progress = get_current_display()
+    
+    if left_img is None:
+        state["session_active"] = False
+        return None, None, f"{progress} | Condition: {condition}", ""
+    
+    return left_img, right_img, f"{progress} | Condition: {condition}", ""
+
+def get_preference_a():
+    """Wrapper for preference A."""
+    return record_preference("A")
+
+def get_preference_b():
+    """Wrapper for preference B."""
+    return record_preference("B")
+
 def start_session(doctor_name):
     """Start a new evaluation session."""
     if not doctor_name.strip():
@@ -238,13 +276,18 @@ with gr.Blocks(title="X-Ray Quality Evaluation") as demo:
     )
     
     # Event handlers
+    def start_and_display(doctor_name):
+        result = start_session(doctor_name)
+        if result[0]["visible"] == False:
+            return result[0], result[1], result[2], None, None, ""
+        else:
+            left_img, right_img, condition, progress = get_current_display()
+            return result[0], result[1], result[2], left_img, right_img, progress
+    
     start_btn.click(
-        start_session,
+        start_and_display,
         inputs=doctor_name_input,
-        outputs=[eval_section, error_section, result_message]
-    ).then(
-        update_display,
-        outputs=[eval_section, image_a, image_b, progress_text]
+        outputs=[eval_section, error_section, result_message, image_a, image_b, progress_text]
     )
     
     pref_a.click(