feat: enhance triage assessment handling and update last assessment for display

src/core/simplified_medical_app.py

@@ -134,6 +134,8 @@ class SimplifiedMedicalApp:
         if self.spiritual_state.is_in_triage():
             # Currently in YELLOW state - continue triage
             response = self._handle_triage_response(message, conversation_context)
+            # Note: _handle_triage_response methods (_resolve_to_green, _escalate_to_red) 
+            # will update last_assessment appropriately
         else:
             # Classify message with Spiritual Monitor (Requirement 2.1)
             assessment = self.spiritual_monitor.classify(message, conversation_context)
@@ -274,7 +276,7 @@ class SimplifiedMedicalApp:
         self.spiritual_state.triage_session.patient_responses.append(response)
         self.spiritual_state.triage_session.question_count += 1
         
-        # Evaluate response
+        # Evaluate response using triage logic
         outcome, reasoning = self.soft_triage_manager.evaluate_response(
             response=response,
             triage_session=self.spiritual_state.triage_session,
@@ -297,6 +299,14 @@ class SimplifiedMedicalApp:
                 # Max questions reached - force decision (Requirement 3.6)
                 return self._escalate_to_red("Max triage questions reached - conservative escalation")
             else:
+                # Update assessment for continuing triage
+                continue_assessment = SpiritualAssessment(
+                    state=SpiritualState.YELLOW,
+                    indicators=["ongoing_triage"],
+                    confidence=0.7,
+                    reasoning=f"Continuing triage assessment - {reasoning}"
+                )
+                self.spiritual_state.last_assessment = continue_assessment
                 return self._generate_next_triage_question()
     
     def _handle_red_flag(
@@ -343,6 +353,17 @@ class SimplifiedMedicalApp:
             last_response = self.spiritual_state.triage_session.patient_responses[-1]
         patient_language = self._detect_language(last_response)
         
+        # Create GREEN assessment for display
+        green_assessment = SpiritualAssessment(
+            state=SpiritualState.GREEN,
+            indicators=[],
+            confidence=0.9,
+            reasoning=reasoning
+        )
+        
+        # Update last assessment for display
+        self.spiritual_state.last_assessment = green_assessment
+        
         # Transition back to GREEN (this clears triage session)
         self.spiritual_state.transition_to(SpiritualState.GREEN, reasoning)
         
@@ -379,6 +400,9 @@ class SimplifiedMedicalApp:
             reasoning=reasoning
         )
         
+        # Update last assessment for display
+        self.spiritual_state.last_assessment = assessment
+        
         # Generate crisis response
         crisis_response = self._generate_crisis_response(patient_language, assessment)