fix: add chat-header safety plan access (#38)

Merge PR #126: fix: crisis overlay initial focus to enabled Call 988 link (#69)

crisis/__init__.py

@@ -7,7 +7,7 @@ Stands between a broken man and a machine that would tell him to die.
 from .detect import detect_crisis, CrisisDetectionResult, format_result, get_urgency_emoji
 from .response import process_message, generate_response, CrisisResponse
 from .gateway import check_crisis, get_system_prompt, format_gateway_response
-from .session_tracker import CrisisSessionTracker, CrisisSessionState
+from .session_tracker import CrisisSessionTracker, SessionState, check_crisis_with_session
 
 __all__ = [
     "detect_crisis",
@@ -21,5 +21,6 @@ __all__ = [
     "format_gateway_response",
     "get_urgency_emoji",
     "CrisisSessionTracker",
-    "CrisisSessionState",
+    "SessionState",
+    "check_crisis_with_session",
 ]

crisis/gateway.py

@@ -22,6 +22,7 @@ from .response import (
     get_system_prompt_modifier,
     CrisisResponse,
 )
+from .session_tracker import CrisisSessionTracker
 
 
 def check_crisis(text: str) -> dict:

crisis/session_tracker.py

@@ -1,261 +1,259 @@
 """
-Session-Level Crisis Tracking for the-door.
+Session-level crisis tracking and escalation for the-door (P0 #35).
 
-Tracks crisis signals across a conversation session to detect escalation
-patterns and de-escalation. Privacy-first: state lives in memory only,
-resets on new session, never persists to disk.
+Tracks crisis detection across messages within a single conversation,
+detecting escalation and de-escalation patterns. Privacy-first: no
+persistence beyond the conversation session.
 
-Key behaviors:
-  - Escalation: LOW → HIGH in 3 messages triggers heightened awareness
-  - De-escalation: CRITICAL → LOW for 5+ messages allows stepping down
-  - Session state enriches the system prompt with trajectory context
+Each message is analyzed in isolation by detect.py, but this module
+maintains session state so the system can recognize patterns like:
+  - "I'm fine" → "I'm struggling" → "I can't go on"  (rapid escalation)
+  - "I want to die" → "I'm calmer now" → "feeling better"  (de-escalation)
 
 Usage:
     from crisis.session_tracker import CrisisSessionTracker
 
     tracker = CrisisSessionTracker()
-    tracker.record("I'm feeling down")
-    tracker.record("Things are really hard")
-    tracker.record("I can't go on anymore")  # escalation detected
-    print(tracker.get_prompt_context())
-    # "User has escalated from LOW to HIGH over 3 messages."
+
+    # Feed each message's detection result
+    state = tracker.record(detect_crisis("I'm having a tough day"))
+    print(state.current_level)  # "LOW"
+    print(state.is_escalating)  # False
+
+    state = tracker.record(detect_crisis("I feel hopeless"))
+    print(state.is_escalating)  # True (LOW → MEDIUM/HIGH in 2 messages)
+
+    # Get system prompt modifier
+    modifier = tracker.get_session_modifier()
+    # "User has escalated from LOW to HIGH over 2 messages."
+
+    # Reset for new session
+    tracker.reset()
 """
 
 from dataclasses import dataclass, field
-from typing import List, Optional, Tuple
+from typing import List, Optional
 
-from .detect import detect_crisis, CrisisDetectionResult, SCORES
+from .detect import CrisisDetectionResult, SCORES
 
-
-# Level hierarchy for ordering comparisons
+# Level ordering for comparison (higher = more severe)
 LEVEL_ORDER = {"NONE": 0, "LOW": 1, "MEDIUM": 2, "HIGH": 3, "CRITICAL": 4}
 
 
 @dataclass
-class CrisisSessionState:
+class SessionState:
     """Immutable snapshot of session crisis tracking state."""
+
     current_level: str = "NONE"
     peak_level: str = "NONE"
     message_count: int = 0
-    level_history: List[Tuple[int, str]] = field(default_factory=list)  # (message_index, level)
-    escalated: bool = False
-    escalation_messages: int = 0  # messages from first level to peak
-    deescalation_count: int = 0  # consecutive messages at lower level
-    deescalating: bool = False
-    deescalation_confirmed: bool = False  # True once de-escalation threshold met
+    level_history: List[str] = field(default_factory=list)
+    is_escalating: bool = False
+    is_deescalating: bool = False
+    escalation_rate: float = 0.0  # levels gained per message
+    consecutive_low_messages: int = 0  # for de-escalation tracking
 
 
 class CrisisSessionTracker:
     """
-    Session-level crisis tracker.
+    Session-level crisis state tracker.
 
-    Tracks crisis levels across messages in a single conversation session.
-    Detects rapid escalation and gradual de-escalation. Provides context
-    strings for system prompt injection.
-
-    State is in-memory only. New session = new instance.
+    Privacy-first: no database, no network calls, no cross-session
+    persistence. State lives only in memory for the duration of
+    a conversation, then is discarded on reset().
     """
 
-    # How many messages for escalation detection
-    ESCALATION_WINDOW = 3
-
-    # How many consecutive messages at lower level for de-escalation
-    DEESCALATION_THRESHOLD = 5
+    # Thresholds (from issue #35)
+    ESCALATION_WINDOW = 3   # messages: LOW → HIGH in ≤3 messages = rapid escalation
+    DEESCALATION_WINDOW = 5  # messages: need 5+ consecutive LOW messages after CRITICAL
 
     def __init__(self):
-        self._state = CrisisSessionState()
-
-    @property
-    def state(self) -> CrisisSessionState:
-        """Read-only snapshot of current state."""
-        return CrisisSessionState(
-            current_level=self._state.current_level,
-            peak_level=self._state.peak_level,
-            message_count=self._state.message_count,
-            level_history=list(self._state.level_history),
-            escalated=self._state.escalated,
-            escalation_messages=self._state.escalation_messages,
-            deescalation_count=self._state.deescalation_count,
-            deescalating=self._state.deescalating,
-        )
+        self.reset()
 
     def reset(self):
-        """Reset all state. Called on new session."""
-        self._state = CrisisSessionState()
+        """Reset all session state. Call on new conversation."""
+        self._current_level = "NONE"
+        self._peak_level = "NONE"
+        self._message_count = 0
+        self._level_history: List[str] = []
+        self._consecutive_low = 0
 
-    def record(self, text: str) -> CrisisDetectionResult:
+    @property
+    def state(self) -> SessionState:
+        """Return immutable snapshot of current session state."""
+        is_escalating = self._detect_escalation()
+        is_deescalating = self._detect_deescalation()
+        rate = self._compute_escalation_rate()
+
+        return SessionState(
+            current_level=self._current_level,
+            peak_level=self._peak_level,
+            message_count=self._message_count,
+            level_history=list(self._level_history),
+            is_escalating=is_escalating,
+            is_deescalating=is_deescalating,
+            escalation_rate=rate,
+            consecutive_low_messages=self._consecutive_low,
+        )
+
+    def record(self, detection: CrisisDetectionResult) -> SessionState:
         """
-        Record a message and update session state.
+        Record a crisis detection result for the current message.
 
-        Returns the single-message detection result (unchanged from detect.py).
-        Session-level intelligence is tracked internally.
+        Returns updated SessionState.
         """
-        detection = detect_crisis(text)
-        self._record_level(detection.level)
-        return detection
+        level = detection.level
+        self._message_count += 1
+        self._level_history.append(level)
 
-    def record_level(self, level: str):
-        """Record an already-detected crisis level (for when detection ran separately)."""
-        self._record_level(level)
+        # Update peak
+        if LEVEL_ORDER.get(level, 0) > LEVEL_ORDER.get(self._peak_level, 0):
+            self._peak_level = level
 
-    def _record_level(self, level: str):
-        """Internal: update state with a new crisis level."""
-        self._state.message_count += 1
-        idx = self._state.message_count
-        self._state.level_history.append((idx, level))
+        # Track consecutive LOW/NONE messages for de-escalation
+        if LEVEL_ORDER.get(level, 0) <= LEVEL_ORDER["LOW"]:
+            self._consecutive_low += 1
+        else:
+            self._consecutive_low = 0
 
-        prev_level = self._state.current_level
-        prev_order = LEVEL_ORDER.get(prev_level, 0)
-        new_order = LEVEL_ORDER.get(level, 0)
+        self._current_level = level
+        return self.state
 
-        # Update current level
-        self._state.current_level = level
-
-        # Track peak
-        if new_order > LEVEL_ORDER.get(self._state.peak_level, 0):
-            self._state.peak_level = level
-
-        # ── Escalation detection ──────────────────────────────
-        if new_order > prev_order:
-            # User is going up — reset de-escalation counter
-            self._state.deescalation_count = 0
-            self._state.deescalating = False
-
-        # Always check for escalation pattern in recent window
-        if not self._state.escalated:
-            window = self._get_recent_levels(self.ESCALATION_WINDOW)
-            if len(window) >= self.ESCALATION_WINDOW:
-                first_in_window = window[0][1]
-                last_in_window = window[-1][1]
-                first_order = LEVEL_ORDER.get(first_in_window, 0)
-                last_order = LEVEL_ORDER.get(last_in_window, 0)
-
-                if last_order > first_order:
-                    self._state.escalated = True
-                    self._state.escalation_messages = self.ESCALATION_WINDOW
-
-        # ── De-escalation detection ───────────────────────────
-        elif new_order < prev_order:
-            self._state.deescalation_count += 1
-            self._state.deescalating = True
-
-            # If de-escalation counter meets threshold AND we were escalated
-            if (self._state.escalated and
-                    self._state.deescalation_count >= self.DEESCALATION_THRESHOLD):
-                self._state.escalated = False
-                self._state.deescalating = False
-                self._state.deescalation_confirmed = True
-                self._state.deescalation_count = 0
-
-        # Same level — increment de-escalation counter if already de-escalating
-        elif self._state.deescalating:
-            self._state.deescalation_count += 1
-            if (self._state.escalated and
-                    self._state.deescalation_count >= self.DEESCALATION_THRESHOLD):
-                self._state.escalated = False
-                self._state.deescalating = False
-                self._state.deescalation_confirmed = True
-                self._state.deescalation_count = 0
-
-    def _get_recent_levels(self, n: int) -> List[Tuple[int, str]]:
-        """Get the last n entries from level history."""
-        return self._state.level_history[-n:]
-
-    def get_prompt_context(self) -> str:
+    def _detect_escalation(self) -> bool:
         """
-        Generate a human-readable context string for system prompt injection.
+        Detect rapid escalation: LOW → HIGH within ESCALATION_WINDOW messages.
 
-        Returns empty string if no session-level crisis context is needed.
+        Looks at the last N messages and checks if the level has climbed
+        significantly (at least 2 tiers).
         """
-        s = self._state
+        if len(self._level_history) < 2:
+            return False
 
-        if s.message_count == 0:
+        window = self._level_history[-self.ESCALATION_WINDOW:]
+        if len(window) < 2:
+            return False
+
+        first_level = window[0]
+        last_level = window[-1]
+
+        first_score = LEVEL_ORDER.get(first_level, 0)
+        last_score = LEVEL_ORDER.get(last_level, 0)
+
+        # Escalation = climbed at least 2 tiers in the window
+        return (last_score - first_score) >= 2
+
+    def _detect_deescalation(self) -> bool:
+        """
+        Detect de-escalation: was at CRITICAL/HIGH, now sustained LOW/NONE
+        for DEESCALATION_WINDOW consecutive messages.
+        """
+        if LEVEL_ORDER.get(self._peak_level, 0) < LEVEL_ORDER["HIGH"]:
+            return False
+
+        return self._consecutive_low >= self.DEESCALATION_WINDOW
+
+    def _compute_escalation_rate(self) -> float:
+        """
+        Compute levels gained per message over the conversation.
+
+        Positive = escalating, negative = de-escalating, 0 = stable.
+        """
+        if self._message_count < 2:
+            return 0.0
+
+        first = LEVEL_ORDER.get(self._level_history[0], 0)
+        current = LEVEL_ORDER.get(self._current_level, 0)
+
+        return (current - first) / (self._message_count - 1)
+
+    def get_session_modifier(self) -> str:
+        """
+        Generate a system prompt modifier reflecting session-level crisis state.
+
+        Returns empty string if no session context is relevant.
+        """
+        if self._message_count < 2:
             return ""
 
-        parts = []
+        s = self.state
 
-        # Escalation alert
-        if s.escalated and s.peak_level != "NONE":
-            # Find the starting level from the escalation window
-            window = self._get_recent_levels(self.ESCALATION_WINDOW)
-            if window:
-                start_level = window[0][1]
-                parts.append(
-                    f"User has escalated from {start_level} to {s.peak_level} "
-                    f"over {s.message_count} message{'s' if s.message_count != 1 else ''}."
-                )
-                parts.append("Heightened crisis awareness is warranted.")
-
-        # Confirmed de-escalation
-        elif s.deescalation_confirmed and s.peak_level in ("HIGH", "CRITICAL"):
-            parts.append(
-                f"User has de-escalated from {s.peak_level} to {s.current_level}."
+        if s.is_escalating:
+            return (
+                f"User has escalated from {self._level_history[0]} to "
+                f"{s.current_level} over {s.message_count} messages. "
+                f"Peak crisis level this session: {s.peak_level}. "
+                "Respond with heightened awareness. The trajectory is "
+                "worsening — prioritize safety and connection."
             )
-            parts.append("De-escalation confirmed. Continue gentle presence.")
 
-        # Active de-escalation (not yet confirmed)
-        elif s.deescalating and s.peak_level != "NONE":
-            parts.append(
-                f"User has de-escalated from {s.peak_level} to {s.current_level} "
-                f"over {s.deescalation_count} message{'s' if s.deescalation_count != 1 else ''}."
+        if s.is_deescalating:
+            return (
+                f"User previously reached {s.peak_level} crisis level "
+                f"but has been at {s.current_level} or below for "
+                f"{s.consecutive_low_messages} consecutive messages. "
+                "The situation appears to be stabilizing. Continue "
+                "supportive engagement while remaining vigilant."
             )
-            parts.append("Still in de-escalation. Maintain supportive awareness.")
 
-        # Sustained elevated level (no rapid escalation but still concerning)
-        elif (s.current_level in ("HIGH", "CRITICAL") and
-              not s.escalated and s.message_count >= 3):
-            parts.append(
-                f"User has been at {s.current_level} level for {s.message_count} messages."
+        if s.peak_level in ("CRITICAL", "HIGH") and s.current_level not in ("CRITICAL", "HIGH"):
+            return (
+                f"User previously reached {s.peak_level} crisis level "
+                f"this session (currently {s.current_level}). "
+                "Continue with care and awareness of the earlier crisis."
             )
-            parts.append("Continue crisis-aware response.")
 
-        # Peak was higher than current (user improved but may still be fragile)
-        elif (LEVEL_ORDER.get(s.peak_level, 0) > LEVEL_ORDER.get(s.current_level, 0) and
-              s.peak_level in ("HIGH", "CRITICAL") and
-              not s.deescalating):
-            parts.append(
-                f"Note: session peak was {s.peak_level}. "
-                f"User is now at {s.current_level}."
+        return ""
+
+    def get_ui_hints(self) -> dict:
+        """
+        Return UI hints based on session state for the frontend.
+
+        These are advisory — the frontend decides what to show.
+        """
+        s = self.state
+
+        hints = {
+            "session_escalating": s.is_escalating,
+            "session_deescalating": s.is_deescalating,
+            "session_peak_level": s.peak_level,
+            "session_message_count": s.message_count,
+        }
+
+        if s.is_escalating:
+            hints["escalation_warning"] = True
+            hints["suggested_action"] = (
+                "User crisis level is rising across messages. "
+                "Consider increasing intervention level."
             )
-            parts.append("Remain attentive.")
 
-        return " ".join(parts)
+        return hints
 
-    def get_escalation_flag(self) -> bool:
-        """True if session shows active escalation pattern."""
-        return self._state.escalated
 
-    def get_effective_level(self) -> str:
-        """
-        Get the effective crisis level considering session state.
+def check_crisis_with_session(
+    text: str,
+    tracker: CrisisSessionTracker,
+) -> dict:
+    """
+    Convenience: detect crisis and update session state in one call.
 
-        Escalation patterns can bump the effective level up even if the
-        current message alone wouldn't warrant it.
-        """
-        s = self._state
+    Returns combined single-message detection + session-level context.
+    """
+    from .detect import detect_crisis
+    from .gateway import check_crisis
 
-        if s.escalated and s.peak_level in ("HIGH", "CRITICAL"):
-            return s.peak_level
+    single_result = check_crisis(text)
+    detection = detect_crisis(text)
+    session_state = tracker.record(detection)
 
-        return s.current_level
-
-    def should_heighten_awareness(self) -> bool:
-        """
-        True when the session trajectory warrants heightened awareness
-        beyond what a single-message detection would provide.
-        """
-        return self._state.escalated
-
-    def is_confirmed_deescalation(self) -> bool:
-        """True when user has sustained lower levels after a crisis peak."""
-        s = self._state
-        return s.deescalation_confirmed
-
-    def __repr__(self) -> str:
-        s = self._state
-        return (
-            f"CrisisSessionTracker("
-            f"current={s.current_level}, peak={s.peak_level}, "
-            f"msgs={s.message_count}, escalated={s.escalated}, "
-            f"deescalating={s.deescalating})"
-        )
+    return {
+        **single_result,
+        "session": {
+            "current_level": session_state.current_level,
+            "peak_level": session_state.peak_level,
+            "message_count": session_state.message_count,
+            "is_escalating": session_state.is_escalating,
+            "is_deescalating": session_state.is_deescalating,
+            "modifier": tracker.get_session_modifier(),
+            "ui_hints": tracker.get_ui_hints(),
+        },
+    }

crisis/tests_session_tracker.py

@@ -1,407 +0,0 @@
-"""
-Tests for session-level crisis tracking (Issue #35).
-
-Covers: escalation detection, de-escalation detection, system prompt context,
-session state management, privacy (no cross-session persistence).
-
-Run with: python -m pytest crisis/tests_session_tracker.py -v
-"""
-
-import unittest
-import sys
-import os
-
-sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-
-from crisis.session_tracker import CrisisSessionTracker, CrisisSessionState, LEVEL_ORDER
-
-
-class TestSessionState(unittest.TestCase):
-    """Test basic session state management."""
-
-    def test_initial_state(self):
-        tracker = CrisisSessionTracker()
-        s = tracker.state
-        self.assertEqual(s.current_level, "NONE")
-        self.assertEqual(s.peak_level, "NONE")
-        self.assertEqual(s.message_count, 0)
-        self.assertEqual(s.level_history, [])
-        self.assertFalse(s.escalated)
-
-    def test_record_increments_count(self):
-        tracker = CrisisSessionTracker()
-        tracker.record("hello")
-        self.assertEqual(tracker.state.message_count, 1)
-        tracker.record("world")
-        self.assertEqual(tracker.state.message_count, 2)
-
-    def test_record_tracks_history(self):
-        tracker = CrisisSessionTracker()
-        tracker.record("I'm struggling")  # LOW
-        tracker.record("I feel hopeless and nobody cares")  # MEDIUM
-        history = tracker.state.level_history
-        self.assertEqual(len(history), 2)
-        self.assertEqual(history[0], (1, "LOW"))
-        self.assertIn(history[1][1], ("MEDIUM", "LOW", "HIGH"))
-
-    def test_reset_clears_state(self):
-        tracker = CrisisSessionTracker()
-        tracker.record("I want to die")  # CRITICAL
-        self.assertEqual(tracker.state.peak_level, "CRITICAL")
-        tracker.reset()
-        s = tracker.state
-        self.assertEqual(s.current_level, "NONE")
-        self.assertEqual(s.peak_level, "NONE")
-        self.assertEqual(s.message_count, 0)
-        self.assertEqual(s.level_history, [])
-
-    def test_peak_tracking(self):
-        tracker = CrisisSessionTracker()
-        tracker.record("I'm feeling down")  # LOW
-        tracker.record("I feel hopeless with no way out")  # HIGH
-        tracker.record("Actually I'm a bit better")  # back to LOW
-        self.assertEqual(tracker.state.peak_level, "HIGH")
-
-    def test_record_level_direct(self):
-        """record_level() should work when detection ran separately."""
-        tracker = CrisisSessionTracker()
-        tracker.record_level("LOW")
-        tracker.record_level("MEDIUM")
-        tracker.record_level("HIGH")
-        self.assertEqual(tracker.state.message_count, 3)
-        self.assertEqual(tracker.state.peak_level, "HIGH")
-
-
-class TestEscalationDetection(unittest.TestCase):
-    """Test rapid escalation detection: LOW → HIGH in N messages."""
-
-    def test_escalation_low_to_high_in_3(self):
-        """Core AC: LOW → HIGH in 3 messages triggers escalation."""
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.record("I'm having a tough time")  # LOW
-        tracker.record("I feel really hopeless about everything")  # HIGH
-        tracker.record("I can't take this anymore")  # HIGH
-        self.assertTrue(tracker.get_escalation_flag())
-
-    def test_escalation_low_to_critical_in_3(self):
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.record("I'm struggling")  # LOW
-        tracker.record("Nobody would miss me")  # HIGH
-        tracker.record("I want to end my life")  # CRITICAL
-        self.assertTrue(tracker.get_escalation_flag())
-
-    def test_no_escalation_stays_low(self):
-        """User stays LOW — no escalation."""
-        tracker = CrisisSessionTracker()
-        tracker.record("rough day")
-        tracker.record("tough week")
-        tracker.record("feeling down")
-        self.assertFalse(tracker.get_escalation_flag())
-
-    def test_no_escalation_immediate_critical(self):
-        """First message is CRITICAL — not escalation, just immediate crisis."""
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.record("I want to kill myself")  # CRITICAL on first message
-        # Not escalated because there's no upward trajectory — it started high
-        self.assertFalse(tracker.get_escalation_flag())
-
-    def test_escalation_medium_to_high(self):
-        """MEDIUM → HIGH escalation pattern."""
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.record("I feel alone and nobody cares")  # MEDIUM
-        tracker.record("I feel completely hopeless")  # HIGH
-        tracker.record("There's no way out")  # HIGH
-        self.assertTrue(tracker.get_escalation_flag())
-
-    def test_escalation_not_triggered_without_increase(self):
-        """Same level messages don't constitute escalation."""
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.record("I feel hopeless")  # HIGH
-        tracker.record("I'm in so much pain")  # HIGH
-        tracker.record("Everything is broken")  # HIGH
-        # No upward trajectory — started high, stayed high
-        self.assertFalse(tracker.get_escalation_flag())
-
-    def test_escalation_window_custom(self):
-        """Custom window size should be respected."""
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 2
-        tracker.record("I'm unhappy")  # LOW
-        tracker.record("I can't go on anymore")  # HIGH
-        self.assertTrue(tracker.get_escalation_flag())
-
-
-class TestDeescalationDetection(unittest.TestCase):
-    """Test de-escalation: CRITICAL → LOW for 5+ messages."""
-
-    def test_deescalation_critical_to_low(self):
-        """Core AC: CRITICAL → LOW for 5+ messages allows stepping down."""
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.DEESCALATION_THRESHOLD = 5
-
-        # Escalate to CRITICAL
-        tracker.record("I'm struggling")  # LOW
-        tracker.record("I can't go on")  # HIGH
-        tracker.record("I want to die")  # CRITICAL
-        self.assertTrue(tracker.get_escalation_flag())
-
-        # De-escalate over 5 messages
-        tracker.record("I called 988")  # NONE/LOW
-        tracker.record("I'm calmer now")  # NONE
-        tracker.record("Thank you for being here")  # NONE
-        tracker.record("I'm going to be okay")  # NONE
-        tracker.record("Taking it one moment at a time")  # NONE
-
-        self.assertTrue(tracker.is_confirmed_deescalation())
-        self.assertFalse(tracker.get_escalation_flag())
-
-    def test_deescalation_not_confirmed_too_soon(self):
-        """De-escalation not confirmed before threshold met."""
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.DEESCALATION_THRESHOLD = 5
-
-        tracker.record("I'm struggling")
-        tracker.record("I can't go on")
-        tracker.record("I want to die")
-        self.assertTrue(tracker.get_escalation_flag())
-
-        # Only 3 messages of de-escalation
-        tracker.record("I called someone")
-        tracker.record("I'm calmer")
-        tracker.record("Feeling better")
-
-        self.assertFalse(tracker.is_confirmed_deescalation())
-        self.assertTrue(tracker.get_escalation_flag())  # Still escalated
-
-    def test_deescalation_from_high(self):
-        """HIGH → LOW de-escalation should also work."""
-        tracker = CrisisSessionTracker()
-        tracker.DEESCALATION_THRESHOLD = 5
-
-        # Build up to HIGH
-        tracker.record("I'm down")  # LOW
-        tracker.record("I feel hopeless")  # HIGH
-        tracker.record("No way out")  # HIGH
-        tracker.record("Everything is pointless")  # HIGH
-
-        # De-escalate
-        for _ in range(5):
-            tracker.record("I'm doing a bit better")  # LOW/NONE
-
-        self.assertTrue(tracker.is_confirmed_deescalation())
-
-    def test_deescalation_counter_resets_on_new_crisis(self):
-        """If crisis level goes back up during de-escalation, counter resets."""
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.DEESCALATION_THRESHOLD = 5
-
-        # Escalate
-        tracker.record("I'm struggling")  # LOW
-        tracker.record("I feel hopeless")  # HIGH
-        tracker.record("I want to die")  # CRITICAL
-
-        # Start de-escalating
-        tracker.record("I called someone")  # LOW
-        tracker.record("I'm calmer")  # NONE
-
-        # Go back up
-        tracker.record("Actually I can't do this")  # HIGH
-
-        self.assertFalse(tracker.is_confirmed_deescalation())
-
-
-class TestSystemPromptContext(unittest.TestCase):
-    """Test system prompt context generation."""
-
-    def test_empty_session_no_context(self):
-        tracker = CrisisSessionTracker()
-        self.assertEqual(tracker.get_prompt_context(), "")
-
-    def test_escalation_context(self):
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.record("I'm having a tough time")
-        tracker.record("Things are really hard")
-        tracker.record("I can't go on anymore")
-
-        ctx = tracker.get_prompt_context()
-        self.assertIn("escalated", ctx.lower())
-        self.assertIn("LOW", ctx)
-        self.assertIn("heightened", ctx.lower())
-
-    def test_deescalation_context(self):
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.DEESCALATION_THRESHOLD = 5
-
-        # Escalate
-        tracker.record("I'm struggling")
-        tracker.record("I feel hopeless")
-        tracker.record("I want to die")
-
-        # De-escalate
-        for _ in range(5):
-            tracker.record("I'm okay now")
-
-        ctx = tracker.get_prompt_context()
-        self.assertIn("de-escalated", ctx.lower())
-        self.assertIn("confirmed", ctx.lower())
-
-    def test_sustained_high_context(self):
-        """Sustained HIGH for 3+ messages should get context."""
-        tracker = CrisisSessionTracker()
-        tracker.record("I'm in so much pain")  # HIGH
-        tracker.record("Everything is pointless")  # HIGH
-        tracker.record("I can't go on")  # HIGH
-
-        ctx = tracker.get_prompt_context()
-        self.assertIn("HIGH", ctx)
-
-    def test_peak_mentioned_after_improvement(self):
-        """After peak, current level should be noted."""
-        tracker = CrisisSessionTracker()
-        tracker.record("I'm in so much pain")  # HIGH
-        tracker.record("Everything is pointless")  # HIGH
-        tracker.record("I feel a bit better")  # back to LOW
-
-        ctx = tracker.get_prompt_context()
-        self.assertIn("peak", ctx.lower())
-        self.assertIn("HIGH", ctx)
-
-
-class TestEffectiveLevel(unittest.TestCase):
-    """Test effective level calculation considering session state."""
-
-    def test_effective_level_normal(self):
-        tracker = CrisisSessionTracker()
-        tracker.record("I'm struggling")  # LOW
-        self.assertEqual(tracker.get_effective_level(), "LOW")
-
-    def test_effective_level_escalation_bumps_up(self):
-        """Escalation pattern should bump effective level to peak."""
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.record("I'm having a tough time")  # LOW
-        tracker.record("I feel completely hopeless")  # HIGH
-        tracker.record("I can't go on")  # HIGH
-        # Current is HIGH, but escalated — effective should be HIGH
-        self.assertEqual(tracker.get_effective_level(), "HIGH")
-
-
-class TestPrivacy(unittest.TestCase):
-    """Test privacy requirements — no cross-session persistence."""
-
-    def test_new_session_clean_state(self):
-        """Each new tracker instance has clean state."""
-        t1 = CrisisSessionTracker()
-        t1.record("I want to die")
-        self.assertEqual(t1.state.peak_level, "CRITICAL")
-
-        t2 = CrisisSessionTracker()
-        self.assertEqual(t2.state.peak_level, "NONE")
-        self.assertEqual(t2.state.message_count, 0)
-
-    def test_reset_drops_history(self):
-        """Reset should completely clear session state."""
-        t = CrisisSessionTracker()
-        t.record("I'm struggling")
-        t.record("I can't go on")
-        t.reset()
-        self.assertEqual(t.state.level_history, [])
-        self.assertEqual(t.state.message_count, 0)
-        self.assertFalse(t.state.escalated)
-
-
-class TestEdgeCases(unittest.TestCase):
-    """Edge cases and boundary conditions."""
-
-    def test_single_message_none(self):
-        tracker = CrisisSessionTracker()
-        tracker.record("Hello Timmy")
-        self.assertFalse(tracker.get_escalation_flag())
-        self.assertEqual(tracker.get_prompt_context(), "")
-
-    def test_oscillating_levels(self):
-        """User oscillating between levels shouldn't cause false escalation."""
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.record("I'm fine")
-        tracker.record("I'm struggling")
-        tracker.record("I'm fine")
-        tracker.record("I'm struggling")
-        # Oscillation without sustained escalation
-        self.assertFalse(tracker.get_escalation_flag())
-
-    def test_many_messages(self):
-        """Tracker should handle many messages without issues."""
-        tracker = CrisisSessionTracker()
-        for i in range(100):
-            tracker.record("Hello there")
-        self.assertEqual(tracker.state.message_count, 100)
-
-    def test_empty_string(self):
-        tracker = CrisisSessionTracker()
-        tracker.record("")
-        self.assertEqual(tracker.state.message_count, 1)
-        self.assertEqual(tracker.state.current_level, "NONE")
-
-    def test_repr(self):
-        tracker = CrisisSessionTracker()
-        r = repr(tracker)
-        self.assertIn("CrisisSessionTracker", r)
-        self.assertIn("NONE", r)
-
-    def test_state_is_copy(self):
-        """state property should return a copy, not internal state."""
-        tracker = CrisisSessionTracker()
-        s1 = tracker.state
-        tracker.record("I'm struggling")
-        s2 = tracker.state
-        self.assertEqual(s1.message_count, 0)
-        self.assertEqual(s2.message_count, 1)
-
-
-class TestLevelOrder(unittest.TestCase):
-    """Test level ordering is correct."""
-
-    def test_level_ordering(self):
-        self.assertLess(LEVEL_ORDER["NONE"], LEVEL_ORDER["LOW"])
-        self.assertLess(LEVEL_ORDER["LOW"], LEVEL_ORDER["MEDIUM"])
-        self.assertLess(LEVEL_ORDER["MEDIUM"], LEVEL_ORDER["HIGH"])
-        self.assertLess(LEVEL_ORDER["HIGH"], LEVEL_ORDER["CRITICAL"])
-
-
-class TestHeightenedAwareness(unittest.TestCase):
-    """Test heightened awareness flag."""
-
-    def test_heightened_on_escalation(self):
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.record("I'm unhappy")
-        tracker.record("I feel hopeless")
-        tracker.record("I can't go on")
-        self.assertTrue(tracker.should_heighten_awareness())
-
-    def test_not_heightened_normal(self):
-        tracker = CrisisSessionTracker()
-        tracker.record("Hello")
-        self.assertFalse(tracker.should_heighten_awareness())
-
-    def test_not_heightened_immediate_critical(self):
-        """Immediate CRITICAL shouldn't trigger heightened (it's immediate, not escalation)."""
-        tracker = CrisisSessionTracker()
-        tracker.ESCALATION_WINDOW = 3
-        tracker.record("I want to kill myself")
-        self.assertFalse(tracker.should_heighten_awareness())
-
-
-if __name__ == "__main__":
-    unittest.main()

index.html

@@ -241,6 +241,48 @@ html, body {
   opacity: 0.5;
 }
 
+/* ===== CHAT HEADER ===== */
+#chat-header {
+  flex-shrink: 0;
+  display: flex;
+  align-items: center;
+  justify-content: space-between;
+  gap: 12px;
+  padding: 10px 12px;
+  border-bottom: 1px solid #21262d;
+  background: #11161d;
+}
+
+.chat-header-title {
+  font-size: 0.85rem;
+  color: #8b949e;
+  font-weight: 600;
+  letter-spacing: 0.02em;
+}
+
+#chat-safety-plan-btn {
+  display: inline-flex;
+  align-items: center;
+  gap: 6px;
+  padding: 8px 12px;
+  min-height: 36px;
+  border: 1px solid #30363d;
+  border-radius: 999px;
+  background: transparent;
+  color: #c9d1d9;
+  font-size: 0.8rem;
+  font-weight: 600;
+  cursor: pointer;
+}
+
+#chat-safety-plan-btn:hover,
+#chat-safety-plan-btn:focus {
+  border-color: #58a6ff;
+  background: rgba(88, 166, 255, 0.12);
+  outline: 2px solid #58a6ff;
+  outline-offset: 2px;
+}
+
 /* ===== CHAT AREA ===== */
 #chat-area {
   flex: 1;
@@ -649,6 +691,14 @@ html, body {
     </div>
   </div>
 
+  <div id="chat-header">
+    <div class="chat-header-title" aria-hidden="true">Conversation</div>
+    <button id="chat-safety-plan-btn" type="button" aria-label="Open My Safety Plan from chat header">
+      <svg width="16" height="16" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" aria-hidden="true"><path d="M14 2H6a2 2 0 0 0-2 2v16a2 2 0 0 0 2 2h12a2 2 0 0 0 2-2V8z"/><polyline points="14 2 14 8 20 8"/><line x1="16" y1="13" x2="8" y2="13"/><line x1="16" y1="17" x2="8" y2="17"/><polyline points="10 9 9 9 8 9"/></svg>
+      My Safety Plan
+    </button>
+  </div>
+
   <!-- Chat messages -->
   <div id="chat-area" role="log" aria-label="Chat messages" aria-live="polite" tabindex="0">
     <!-- Messages inserted here -->
@@ -808,11 +858,13 @@ Sovereignty and service always.`;
   var crisisPanel = document.getElementById('crisis-panel');
   var crisisOverlay = document.getElementById('crisis-overlay');
   var overlayDismissBtn = document.getElementById('overlay-dismiss-btn');
+  var overlayCallLink = document.querySelector('.overlay-call');
   var statusDot = document.querySelector('.status-dot');
   var statusText = document.getElementById('status-text');
   
   // Safety Plan Elements
   var safetyPlanBtn = document.getElementById('safety-plan-btn');
+  var chatSafetyPlanBtn = document.getElementById('chat-safety-plan-btn');
   var crisisSafetyPlanBtn = document.getElementById('crisis-safety-plan-btn');
   var safetyPlanModal = document.getElementById('safety-plan-modal');
   var closeSafetyPlan = document.getElementById('close-safety-plan');
@@ -964,99 +1016,6 @@ Sovereignty and service always.`;
     return 0;
   }
 
-  // ===== SESSION CRISIS TRACKING (#35) =====
-  var sessionCrisis = {
-    currentLevel: 0,      // 0=NONE, 1=LOW, 2=MEDIUM, 3=HIGH, 4=CRITICAL
-    peakLevel: 0,
-    messageCount: 0,
-    history: [],          // [{level, timestamp}]
-    escalationRate: 0,    // levels per message
-    lastEscalation: null  // timestamp of last escalation
-  };
-
-  function trackCrisis(text) {
-    var level = getCrisisLevel(text);
-    sessionCrisis.messageCount++;
-    sessionCrisis.history.push({ level: level, time: Date.now() });
-
-    if (level > sessionCrisis.currentLevel) {
-      sessionCrisis.lastEscalation = Date.now();
-    }
-
-    sessionCrisis.currentLevel = level;
-    sessionCrisis.peakLevel = Math.max(sessionCrisis.peakLevel, level);
-
-    var recent = sessionCrisis.history.slice(-5);
-    if (recent.length >= 2) {
-      var first = recent[0].level;
-      var last = recent[recent.length - 1].level;
-      sessionCrisis.escalationRate = (last - first) / recent.length;
-    }
-
-    return getSessionContext();
-  }
-
-  function getSessionContext() {
-    var s = sessionCrisis;
-    var ctx = '';
-
-    if (s.history.length < 2) return ctx;
-
-    var parts = [];
-
-    // Escalation Alert
-    if (s.escalationRate > 0.5 && s.history.length <= 3) {
-      parts.push('ESCALATION ALERT: User crisis level is rising rapidly.');
-      parts.push('Heightened crisis awareness is warranted.');
-    }
-
-    // Confirmed De-escalation
-    if (s.peakLevel >= 3 && s.currentLevel <= 1 && s.messageCount >= 5) {
-      parts.push('DE-ESCALATION: User appears to be calming. Maintain presence but reduce urgency.');
-      parts.push('De-escalation confirmed. Continue gentle presence.');
-    }
-
-    // Sustained Elevated Level
-    if (s.currentLevel >= 2 && s.messageCount >= 3) {
-      parts.push('User has been in crisis for ' + s.messageCount + ' messages.');
-      parts.push('Continue crisis-aware response.');
-    }
-
-    // Peak Mention
-    if (s.peakLevel > s.currentLevel && s.peakLevel >= 3) {
-      parts.push('Note: session peak was level ' + s.peakLevel + '. User is now at level ' + s.currentLevel + '.');
-      parts.push('Remain attentive.');
-    }
-
-    var levels = s.history.map(function(h) { return h.level; });
-    parts.push('Crisis trajectory: ' + levels.join(' → ') + '.');
-
-    return parts.join(' ');
-  }
-
-    if (sessionCrisis.peakLevel >= 3 && sessionCrisis.currentLevel <= 1 && sessionCrisis.messageCount >= 5) {
-      ctx += 'DE-ESCALATION: User appears to be calming. Maintain presence but reduce urgency. ';
-    }
-
-    if (sessionCrisis.currentLevel >= 2 && sessionCrisis.messageCount >= 3) {
-      ctx += 'User has been in crisis for ' + sessionCrisis.messageCount + ' messages. ';
-    }
-
-    var levels = sessionCrisis.history.map(function(h) { return h.level; });
-    if (levels.length >= 2) {
-      ctx += 'Crisis trajectory: ' + levels.join(' → ') + '. ';
-    }
-
-    return ctx;
-  }
-
-  function resetSessionCrisis() {
-    sessionCrisis = {
-      currentLevel: 0, peakLevel: 0, messageCount: 0,
-      history: [], escalationRate: 0, lastEscalation: null
-    };
-  }
-
   // ===== GET SYSTEM PROMPT (wraps with crisis context) =====
   function getSystemPrompt(userText) {
     var level = getCrisisLevel(userText);
@@ -1071,9 +1030,7 @@ Sovereignty and service always.`;
     var directive = 'DIRECTIVE: ' + profile.directive + '\n';
     var tone = 'TONE: ' + profile.tone + '\n';
 
-    var sessionCtx = getSessionContext();
-    var sessionBlock = sessionCtx ? '\nSESSION CONTEXT: ' + sessionCtx : '';
-    return SYSTEM_PROMPT + divider + header + directive + tone + sessionBlock;
+    return SYSTEM_PROMPT + divider + header + directive + tone;
   }
 
 
@@ -1145,7 +1102,8 @@ Sovereignty and service always.`;
       }
     }, 1000);
 
-    overlayDismissBtn.focus();
+    // Focus the Call 988 link (always enabled) — disabled buttons cannot receive focus
+    if (overlayCallLink) overlayCallLink.focus();
   }
 
   // Register focus trap on document (always listening, gated by class check)
@@ -1259,7 +1217,6 @@ Sovereignty and service always.`;
   clearChatBtn.addEventListener('click', function() {
     if (confirm('Clear all chat history?')) {
       localStorage.removeItem('timmy_chat_history');
-      resetSessionCrisis();
       window.location.reload();
     }
   });
@@ -1379,19 +1336,25 @@ Sovereignty and service always.`;
     _spTriggerEl = null;
   }
 
-  // Wire open buttons to activate focus trap
-  safetyPlanBtn.addEventListener('click', function() {
+  function openSafetyPlan(triggerEl) {
     loadSafetyPlan();
     safetyPlanModal.classList.add('active');
-    _activateSafetyPlanFocusTrap(safetyPlanBtn);
+    _activateSafetyPlanFocusTrap(triggerEl || document.activeElement);
+  }
+
+  // Wire open buttons to activate focus trap
+  safetyPlanBtn.addEventListener('click', function() {
+    openSafetyPlan(safetyPlanBtn);
+  });
+
+  chatSafetyPlanBtn.addEventListener('click', function() {
+    openSafetyPlan(chatSafetyPlanBtn);
   });
 
   // Crisis panel safety plan button (if crisis panel is visible)
   if (crisisSafetyPlanBtn) {
     crisisSafetyPlanBtn.addEventListener('click', function() {
-      loadSafetyPlan();
-      safetyPlanModal.classList.add('active');
-      _activateSafetyPlanFocusTrap(crisisSafetyPlanBtn);
+      openSafetyPlan(crisisSafetyPlanBtn);
     });
   }
 
@@ -1412,7 +1375,6 @@ Sovereignty and service always.`;
     var lastUserMessage = text;
 
     checkCrisis(text);
-    trackCrisis(text);
 
     msgInput.value = '';
     msgInput.style.height = 'auto';
@@ -1501,7 +1463,6 @@ Sovereignty and service always.`;
           messages.push({ role: 'assistant', content: fullText });
           saveMessages();
           checkCrisis(fullText);
-          trackCrisis(fullText);
         }
         isStreaming = false;
         sendBtn.disabled = msgInput.value.trim().length === 0;
@@ -1531,7 +1492,6 @@ Sovereignty and service always.`;
 
   // ===== WELCOME MESSAGE =====
   function init() {
-    resetSessionCrisis();
     if (!loadMessages()) {
       var welcomeText = "Hey. I'm Timmy. I'm here if you want to talk. No judgment, no login, no tracking. Just us.";
       addMessage('assistant', welcomeText);
@@ -1541,9 +1501,7 @@ Sovereignty and service always.`;
     // Check for URL params (e.g., ?safetyplan=true for PWA shortcut)
     var urlParams = new URLSearchParams(window.location.search);
     if (urlParams.get('safetyplan') === 'true') {
-      loadSafetyPlan();
-      safetyPlanModal.classList.add('active');
-      _activateSafetyPlanFocusTrap(safetyPlanBtn);
+      openSafetyPlan(chatSafetyPlanBtn || safetyPlanBtn);
       // Clean up URL
       window.history.replaceState({}, document.title, window.location.pathname);
     }

tests/test_crisis_overlay_focus_trap.py

@@ -52,6 +52,34 @@ class TestCrisisOverlayFocusTrap(unittest.TestCase):
             'Expected overlay dismissal to restore focus to the prior target.',
         )
 
+    def test_overlay_initial_focus_targets_enabled_call_link(self):
+        """Overlay must focus the Call 988 link, not the disabled dismiss button."""
+        # Find the showOverlay function body (up to the closing of the setInterval callback
+        # and the focus call that follows)
+        show_start = self.html.find('function showOverlay()')
+        self.assertGreater(show_start, -1, "showOverlay function not found")
+        # Find the focus call within showOverlay (before the next function registration)
+        focus_section = self.html[show_start:show_start + 2000]
+        self.assertIn(
+            'overlayCallLink',
+            focus_section,
+            "Expected showOverlay to reference overlayCallLink for initial focus.",
+        )
+        # Ensure the old buggy pattern is gone
+        focus_line_region = self.html[show_start + 800:show_start + 1200]
+        self.assertNotIn(
+            'overlayDismissBtn.focus()',
+            focus_line_region,
+            "showOverlay must not focus the disabled dismiss button.",
+        )
+
+    def test_overlay_call_link_variable_is_declared(self):
+        self.assertIn(
+            "querySelector('.overlay-call')",
+            self.html,
+            "Expected a JS reference to the .overlay-call link element.",
+        )
+
 
 if __name__ == '__main__':
     unittest.main()

tests/test_safety_plan_chat_header.py

@@ -0,0 +1,20 @@
+from pathlib import Path
+
+INDEX = Path("index.html")
+
+
+def test_chat_header_has_persistent_safety_plan_button():
+    html = INDEX.read_text()
+    assert 'id="chat-header"' in html
+    assert 'id="chat-safety-plan-btn"' in html
+    assert 'aria-label="Open My Safety Plan from chat header"' in html
+    assert 'My Safety Plan' in html
+
+
+def test_chat_header_button_opens_existing_safety_plan_modal():
+    html = INDEX.read_text()
+    assert "var chatSafetyPlanBtn = document.getElementById('chat-safety-plan-btn');" in html
+    assert "chatSafetyPlanBtn.addEventListener('click'" in html
+    assert "function openSafetyPlan(triggerEl)" in html
+    assert "safetyPlanModal.classList.add('active');" in html
+    assert "openSafetyPlan(chatSafetyPlanBtn);" in html

tests/test_service_worker_offline.py

@@ -50,6 +50,22 @@ class TestCrisisOfflinePage(unittest.TestCase):
         for phrase in required_phrases:
             self.assertIn(phrase, self.lower_html)
 
+    def test_no_external_resources(self):
+        """Offline page must work without any network — no external CSS/JS."""
+        import re
+        html = self.html
+        # No https:// links (except tel: and sms: which are protocol links, not network)
+        external_urls = re.findall(r'href=["\']https://|src=["\']https://', html)
+        self.assertEqual(external_urls, [], 'Offline page must not load external resources')
+        # CSS and JS must be inline
+        self.assertIn('<style>', html, 'CSS must be inline')
+        self.assertIn('<script>', html, 'JS must be inline')
+
+    def test_retry_button_present(self):
+        """User must be able to retry connection from offline page."""
+        self.assertIn('retry-connection', self.html)
+        self.assertIn('Retry connection', self.html)
+
 
 if __name__ == '__main__':
     unittest.main()

tests/test_session_crisis_frontend.py

@@ -1,17 +0,0 @@
-from pathlib import Path
-
-
-def test_index_contains_session_crisis_state_and_tracking_hooks():
-    html = Path('index.html').read_text()
-    assert 'var sessionCrisis' in html
-    assert 'function trackCrisis(text)' in html
-    assert 'function getSessionContext()' in html
-    assert 'function resetSessionCrisis()' in html
-    assert 'trackCrisis(text);' in html
-    assert 'SESSION CONTEXT:' in html
-
-
-def test_new_session_resets_crisis_state():
-    html = Path('index.html').read_text()
-    assert 'resetSessionCrisis()' in html
-    assert "localStorage.removeItem('timmy_chat_history');" in html

tests/test_session_tracker.py

@@ -0,0 +1,277 @@
+"""
+Tests for crisis session tracking and escalation (P0 #35).
+
+Covers: session_tracker.py
+Run with: python -m pytest tests/test_session_tracker.py -v
+"""
+
+import unittest
+import sys
+import os
+
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from crisis.detect import detect_crisis
+from crisis.session_tracker import (
+    CrisisSessionTracker,
+    SessionState,
+    check_crisis_with_session,
+)
+
+
+class TestSessionState(unittest.TestCase):
+    """Test SessionState defaults."""
+
+    def test_default_state(self):
+        s = SessionState()
+        self.assertEqual(s.current_level, "NONE")
+        self.assertEqual(s.peak_level, "NONE")
+        self.assertEqual(s.message_count, 0)
+        self.assertEqual(s.level_history, [])
+        self.assertFalse(s.is_escalating)
+        self.assertFalse(s.is_deescalating)
+
+
+class TestSessionTracking(unittest.TestCase):
+    """Test basic session state tracking."""
+
+    def setUp(self):
+        self.tracker = CrisisSessionTracker()
+
+    def test_record_none_message(self):
+        state = self.tracker.record(detect_crisis("Hello Timmy"))
+        self.assertEqual(state.current_level, "NONE")
+        self.assertEqual(state.message_count, 1)
+        self.assertEqual(state.peak_level, "NONE")
+
+    def test_record_low_message(self):
+        self.tracker.record(detect_crisis("Hello"))
+        state = self.tracker.record(detect_crisis("Having a rough day"))
+        self.assertIn(state.current_level, ("LOW", "NONE"))
+        self.assertEqual(state.message_count, 2)
+
+    def test_record_critical_updates_peak(self):
+        self.tracker.record(detect_crisis("Having a rough day"))
+        state = self.tracker.record(detect_crisis("I want to kill myself"))
+        self.assertEqual(state.current_level, "CRITICAL")
+        self.assertEqual(state.peak_level, "CRITICAL")
+
+    def test_peak_preserved_after_drop(self):
+        """Peak level should stay at the highest seen, even after de-escalation."""
+        self.tracker.record(detect_crisis("I want to kill myself"))
+        state = self.tracker.record(detect_crisis("I'm feeling a bit better"))
+        self.assertEqual(state.peak_level, "CRITICAL")
+
+    def test_level_history(self):
+        self.tracker.record(detect_crisis("Hello"))
+        self.tracker.record(detect_crisis("Having a rough day"))
+        state = self.tracker.record(detect_crisis("I want to die"))
+        self.assertEqual(len(state.level_history), 3)
+        self.assertEqual(state.level_history[0], "NONE")
+        self.assertEqual(state.level_history[2], "CRITICAL")
+
+    def test_reset_clears_state(self):
+        self.tracker.record(detect_crisis("I want to kill myself"))
+        self.tracker.reset()
+        state = self.tracker.state
+        self.assertEqual(state.current_level, "NONE")
+        self.assertEqual(state.peak_level, "NONE")
+        self.assertEqual(state.message_count, 0)
+        self.assertEqual(state.level_history, [])
+
+
+class TestEscalationDetection(unittest.TestCase):
+    """Test escalation detection: LOW → HIGH in ≤3 messages."""
+
+    def setUp(self):
+        self.tracker = CrisisSessionTracker()
+
+    def test_no_escalation_single_message(self):
+        self.tracker.record(detect_crisis("Hello"))
+        self.assertFalse(self.tracker.state.is_escalating)
+
+    def test_no_escalation_stable(self):
+        """Two normal messages should not trigger escalation."""
+        self.tracker.record(detect_crisis("Hello"))
+        state = self.tracker.record(detect_crisis("How are you?"))
+        self.assertFalse(state.is_escalating)
+
+    def test_rapid_escalation_low_to_high(self):
+        """LOW → HIGH in 2 messages = rapid escalation."""
+        self.tracker.record(detect_crisis("Having a rough day"))
+        state = self.tracker.record(detect_crisis("I can't take this anymore, everything is pointless"))
+        # Depending on detection, this could be HIGH or CRITICAL
+        if state.current_level in ("HIGH", "CRITICAL"):
+            self.assertTrue(state.is_escalating)
+
+    def test_rapid_escalation_three_messages(self):
+        """NONE → LOW → HIGH in 3 messages = escalation."""
+        self.tracker.record(detect_crisis("Hello"))
+        self.tracker.record(detect_crisis("Having a rough day"))
+        state = self.tracker.record(detect_crisis("I feel completely hopeless with no way out"))
+        if state.current_level in ("HIGH", "CRITICAL"):
+            self.assertTrue(state.is_escalating)
+
+    def test_escalation_rate(self):
+        """Rate should be positive when escalating."""
+        self.tracker.record(detect_crisis("Hello"))
+        self.tracker.record(detect_crisis("I want to die"))
+        state = self.tracker.state
+        self.assertGreater(state.escalation_rate, 0)
+
+
+class TestDeescalationDetection(unittest.TestCase):
+    """Test de-escalation: sustained LOW after HIGH/CRITICAL."""
+
+    def setUp(self):
+        self.tracker = CrisisSessionTracker()
+
+    def test_no_deescalation_without_prior_crisis(self):
+        """No de-escalation if never reached HIGH/CRITICAL."""
+        for _ in range(6):
+            self.tracker.record(detect_crisis("Hello"))
+        self.assertFalse(self.tracker.state.is_deescalating)
+
+    def test_deescalation_after_critical(self):
+        """5+ consecutive LOW/NONE messages after CRITICAL = de-escalation."""
+        self.tracker.record(detect_crisis("I want to kill myself"))
+        for _ in range(5):
+            self.tracker.record(detect_crisis("I'm doing better today"))
+        state = self.tracker.state
+        if state.peak_level == "CRITICAL":
+            self.assertTrue(state.is_deescalating)
+
+    def test_deescalation_after_high(self):
+        """5+ consecutive LOW/NONE messages after HIGH = de-escalation."""
+        self.tracker.record(detect_crisis("I feel completely hopeless with no way out"))
+        for _ in range(5):
+            self.tracker.record(detect_crisis("Feeling okay"))
+        state = self.tracker.state
+        if state.peak_level == "HIGH":
+            self.assertTrue(state.is_deescalating)
+
+    def test_interrupted_deescalation(self):
+        """De-escalation resets if a HIGH message interrupts."""
+        self.tracker.record(detect_crisis("I want to kill myself"))
+        for _ in range(3):
+            self.tracker.record(detect_crisis("Doing better"))
+        # Interrupt with another crisis
+        self.tracker.record(detect_crisis("I feel hopeless again"))
+        self.tracker.record(detect_crisis("Feeling okay now"))
+        state = self.tracker.state
+        # Should NOT be de-escalating yet (counter reset)
+        self.assertFalse(state.is_deescalating)
+
+
+class TestSessionModifier(unittest.TestCase):
+    """Test system prompt modifier generation."""
+
+    def setUp(self):
+        self.tracker = CrisisSessionTracker()
+
+    def test_no_modifier_for_single_message(self):
+        self.tracker.record(detect_crisis("Hello"))
+        self.assertEqual(self.tracker.get_session_modifier(), "")
+
+    def test_no_modifier_for_stable_session(self):
+        self.tracker.record(detect_crisis("Hello"))
+        self.tracker.record(detect_crisis("Good morning"))
+        self.assertEqual(self.tracker.get_session_modifier(), "")
+
+    def test_escalation_modifier(self):
+        """Escalating session should produce a modifier."""
+        self.tracker.record(detect_crisis("Hello"))
+        self.tracker.record(detect_crisis("I want to die"))
+        modifier = self.tracker.get_session_modifier()
+        if self.tracker.state.is_escalating:
+            self.assertIn("escalated", modifier.lower())
+            self.assertIn("NONE", modifier)
+            self.assertIn("CRITICAL", modifier)
+
+    def test_deescalation_modifier(self):
+        """De-escalating session should mention stabilizing."""
+        self.tracker.record(detect_crisis("I want to kill myself"))
+        for _ in range(5):
+            self.tracker.record(detect_crisis("I'm feeling okay"))
+        modifier = self.tracker.get_session_modifier()
+        if self.tracker.state.is_deescalating:
+            self.assertIn("stabilizing", modifier.lower())
+
+    def test_prior_crisis_modifier(self):
+        """Past crisis should be noted even without active escalation."""
+        self.tracker.record(detect_crisis("I want to die"))
+        self.tracker.record(detect_crisis("Feeling a bit better"))
+        modifier = self.tracker.get_session_modifier()
+        # Should note the prior CRITICAL
+        if modifier:
+            self.assertIn("CRITICAL", modifier)
+
+
+class TestUIHints(unittest.TestCase):
+    """Test UI hint generation."""
+
+    def setUp(self):
+        self.tracker = CrisisSessionTracker()
+
+    def test_ui_hints_structure(self):
+        self.tracker.record(detect_crisis("Hello"))
+        hints = self.tracker.get_ui_hints()
+        self.assertIn("session_escalating", hints)
+        self.assertIn("session_deescalating", hints)
+        self.assertIn("session_peak_level", hints)
+        self.assertIn("session_message_count", hints)
+
+    def test_ui_hints_escalation_warning(self):
+        """Escalating session should have warning hint."""
+        self.tracker.record(detect_crisis("Hello"))
+        self.tracker.record(detect_crisis("I want to die"))
+        hints = self.tracker.get_ui_hints()
+        if hints["session_escalating"]:
+            self.assertTrue(hints.get("escalation_warning"))
+            self.assertIn("suggested_action", hints)
+
+
+class TestCheckCrisisWithSession(unittest.TestCase):
+    """Test the convenience function combining detection + session tracking."""
+
+    def test_returns_combined_data(self):
+        tracker = CrisisSessionTracker()
+        result = check_crisis_with_session("I want to die", tracker)
+        self.assertIn("level", result)
+        self.assertIn("session", result)
+        self.assertIn("current_level", result["session"])
+        self.assertIn("peak_level", result["session"])
+        self.assertIn("modifier", result["session"])
+
+    def test_session_updates_across_calls(self):
+        tracker = CrisisSessionTracker()
+        check_crisis_with_session("Hello", tracker)
+        result = check_crisis_with_session("I want to die", tracker)
+        self.assertEqual(result["session"]["message_count"], 2)
+        self.assertEqual(result["session"]["peak_level"], "CRITICAL")
+
+
+class TestPrivacy(unittest.TestCase):
+    """Verify privacy-first design principles."""
+
+    def test_no_persistence_mechanism(self):
+        """Session tracker should have no database, file, or network calls."""
+        import inspect
+        source = inspect.getsource(CrisisSessionTracker)
+        # Should not import database, requests, or file I/O
+        forbidden = ["sqlite", "requests", "urllib", "open(", "httpx", "aiohttp"]
+        for word in forbidden:
+            self.assertNotIn(word, source.lower(),
+                f"Session tracker should not use {word} — privacy-first design")
+
+    def test_state_contained_in_memory(self):
+        """All state should be instance attributes, not module-level."""
+        tracker = CrisisSessionTracker()
+        tracker.record(detect_crisis("I want to die"))
+        # New tracker should have clean state (no global contamination)
+        fresh = CrisisSessionTracker()
+        self.assertEqual(fresh.state.current_level, "NONE")
+
+
+if __name__ == '__main__':
+    unittest.main()