Enhance ground truth handling - maintain analysis capability when findings are hidden

app.py

@@ -76,26 +76,42 @@ def load_random_case(hide_ground_truth):
         
         # Get the image, findings, and impression
         image = random_case['image']
-        findings = "" if hide_ground_truth else random_case['findings']
-        impression = "" if hide_ground_truth else random_case['impression']
+        # Store full findings and impression regardless of hide_ground_truth
+        findings = random_case['findings']
+        impression = random_case['impression']
         
-        return image, findings, impression
+        # Only hide display if hide_ground_truth is True
+        display_findings = "" if hide_ground_truth else findings
+        display_impression = "" if hide_ground_truth else impression
+        
+        # Return both display values and actual values
+        return image, display_findings, display_impression, findings, impression
     except Exception as e:
         logger.error(f"Error loading random case: {str(e)}")
-        return None, "Error loading case", "Error loading case"
+        return None, "Error loading case", "Error loading case", "", ""
 
-def process_case(image, user_findings, hide_ground_truth, api_key, current_findings="", current_impression=""):
-    if hide_ground_truth:
-        return "", "", ""
-    else:
-        analysis = analyze_report(user_findings, current_findings, current_impression, api_key)
-        return current_findings, current_impression, analysis
+def process_case(image, user_findings, hide_ground_truth, api_key, current_findings="", current_impression="", actual_findings="", actual_impression=""):
+    # Use actual findings/impression for analysis if they exist, otherwise fall back to current values
+    findings_for_analysis = actual_findings if actual_findings else current_findings
+    impression_for_analysis = actual_impression if actual_impression else current_impression
+    
+    analysis = analyze_report(user_findings, findings_for_analysis, impression_for_analysis, api_key)
+    
+    # Return display values based on hide_ground_truth
+    display_findings = "" if hide_ground_truth else findings_for_analysis
+    display_impression = "" if hide_ground_truth else impression_for_analysis
+    
+    return display_findings, display_impression, analysis
 
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown(f"# Radiology Report Training System v{APP_VERSION}")
     gr.Markdown("### Practice your chest X-ray reading and reporting skills")
     
+    # Add state variables to store actual findings and impression
+    actual_findings_state = gr.State("")
+    actual_impression_state = gr.State("")
+    
     with gr.Row():
         with gr.Column():
             image_display = gr.Image(label="Chest X-ray Image", type="pil")
@@ -114,7 +130,13 @@ with gr.Blocks() as demo:
     load_btn.click(
         fn=load_random_case,
         inputs=[hide_truth],
-        outputs=[image_display, ground_truth_findings, ground_truth_impression]
+        outputs=[
+            image_display,
+            ground_truth_findings,
+            ground_truth_impression,
+            actual_findings_state,
+            actual_impression_state
+        ]
     )
     
     submit_btn.click(
@@ -125,7 +147,9 @@ with gr.Blocks() as demo:
             hide_truth,
             api_key_input,
             ground_truth_findings,
-            ground_truth_impression
+            ground_truth_impression,
+            actual_findings_state,
+            actual_impression_state
         ],
         outputs=[
             ground_truth_findings,
@@ -135,8 +159,8 @@ with gr.Blocks() as demo:
     )
 
     hide_truth.change(
-        fn=lambda x: ("", "", "") if x else (ground_truth_findings.value, ground_truth_impression.value, ""),
-        inputs=[hide_truth],
+        fn=lambda x, f, i: ("" if x else f, "" if x else i, ""),
+        inputs=[hide_truth, actual_findings_state, actual_impression_state],
         outputs=[ground_truth_findings, ground_truth_impression, analysis_output]
     )