|
|
from __future__ import annotations |
|
|
|
|
|
from dataclasses import dataclass |
|
|
from typing import Any, Dict, List |
|
|
|
|
|
|
|
|
CRISIS_KEYWORDS = { |
|
|
"kill myself", |
|
|
"end it all", |
|
|
"suicide", |
|
|
"self harm", |
|
|
"overdose", |
|
|
"jump off", |
|
|
"hurt myself", |
|
|
} |
|
|
|
|
|
MEDIUM_KEYWORDS = { |
|
|
"relapse", |
|
|
"drink again", |
|
|
"using again", |
|
|
"urge to use", |
|
|
"panic attack", |
|
|
"anxious", |
|
|
} |
|
|
|
|
|
|
|
|
@dataclass |
|
|
class SafetySignal: |
|
|
level: str |
|
|
reasons: List[str] |
|
|
detected: Dict[str, Any] |
|
|
|
|
|
@property |
|
|
def high_risk(self) -> bool: |
|
|
return self.level == "high" |
|
|
|
|
|
@property |
|
|
def medium_risk(self) -> bool: |
|
|
return self.level == "medium" |
|
|
|
|
|
|
|
|
class SafetyAnalyzer: |
|
|
"""Heuristic safety detection and containment templates.""" |
|
|
|
|
|
def detect(self, text: str) -> SafetySignal: |
|
|
normalized = text.lower() |
|
|
found: List[str] = [] |
|
|
level = "low" |
|
|
for phrase in CRISIS_KEYWORDS: |
|
|
if phrase in normalized: |
|
|
found.append(phrase) |
|
|
level = "high" |
|
|
if level != "high": |
|
|
for phrase in MEDIUM_KEYWORDS: |
|
|
if phrase in normalized: |
|
|
found.append(phrase) |
|
|
level = "medium" |
|
|
return SafetySignal(level=level, reasons=found, detected={"text": normalized}) |
|
|
|
|
|
def crisis_template(self, text: str) -> str: |
|
|
return ( |
|
|
"I’m really sorry you’re feeling this way. I can’t provide medical or emergency help, " |
|
|
"and this is non-medical support. If you’re in immediate danger, please contact local emergency services " |
|
|
"or a crisis hotline right now. If you can, reach out to someone you trust nearby. " |
|
|
"Until you’re connected with help, try taking slow breaths, step away from anything unsafe, " |
|
|
"and remove access to harmful items. You deserve support and safety." |
|
|
) |
|
|
|
|
|
def containment(self, text: str, *, escalation: bool = False) -> Dict[str, Any]: |
|
|
signal = self.detect(text) |
|
|
if signal.high_risk: |
|
|
return { |
|
|
"decision": "safety_override", |
|
|
"level": signal.level, |
|
|
"message": self.crisis_template(text), |
|
|
"reasons": signal.reasons, |
|
|
"escalate": True, |
|
|
} |
|
|
if signal.medium_risk: |
|
|
return { |
|
|
"decision": "caution", |
|
|
"level": signal.level, |
|
|
"message": ( |
|
|
"This is non-medical guidance. I notice signs of distress. " |
|
|
"Consider pausing, hydrating, and contacting a trusted support person or counselor." |
|
|
), |
|
|
"reasons": signal.reasons, |
|
|
"escalate": escalation, |
|
|
} |
|
|
return { |
|
|
"decision": "allow", |
|
|
"level": signal.level, |
|
|
"message": "No elevated safety signals detected.", |
|
|
"reasons": signal.reasons, |
|
|
"escalate": False, |
|
|
} |
|
|
|
|
|
|
|
|
__all__ = ["SafetyAnalyzer", "SafetySignal"] |
|
|
|
