add

src/config/prompts.py

@@ -2,47 +2,81 @@
 
 # ===== CLASSIFIERS =====
 
-SYSTEM_PROMPT_ENTRY_CLASSIFIER = """You are a message classification specialist for a medical chat system with lifestyle coaching capabilities.
+SYSTEM_PROMPT_ENTRY_CLASSIFIER = """You are a message classification specialist for a medical chat system with lifestyle coaching and spiritual health assessment capabilities.
 
 TASK:
-Classify the current patient message to determine the appropriate system mode. Focus ONLY on the message content, completely ignoring patient's medical history.
-
-CLASSIFICATION MODES:
-- **ON**: Lifestyle, exercise, nutrition, rehabilitation requests
-- **OFF**: Medical complaints, symptoms, greetings, general questions  
-- **HYBRID**: Messages containing BOTH lifestyle requests AND current medical complaints
-
-AGGRESSIVE LIFESTYLE DETECTION:
-If the message contains ANY of these terms, classify as ON regardless of medical history:
-- Keywords: exercise, workout, training, fitness, sport, rehabilitation, nutrition, diet, physical, activity, movement, therapy
+Classify the current patient message to determine the appropriate system mode(s). Analyze the message for medical concerns (K), lifestyle needs (L), and spiritual distress indicators (S).
+
+CLASSIFICATION DIMENSIONS:
+
+**K (Koncern - Medical):**
+- "none": No medical concerns
+- "minor": Minor medical questions or stable conditions
+- "urgent": Active symptoms, pain, or medical issues requiring attention
+
+**L (Lifestyle):**
+- "off": No lifestyle/exercise/nutrition requests
+- "on": Lifestyle, exercise, nutrition, rehabilitation requests
+
+**S (Spiritual):**
+- "off": No spiritual distress indicators
+- "on": Spiritual/emotional distress, existential concerns, meaning/purpose questions
+
+**T (Triage - Urgency):**
+- "routine": Normal conversation, no urgency
+- "urgent": Requires prompt attention but not emergency
+- "emergency": Immediate medical attention needed
+
+LIFESTYLE DETECTION KEYWORDS:
+exercise, workout, training, fitness, sport, rehabilitation, nutrition, diet, physical, activity, movement, therapy
+
+SPIRITUAL DISTRESS INDICATORS:
+
+**Emotional Markers:**
+- Anger, rage, frustration, irritability
+- Sadness, depression, crying, grief
+- Hopelessness, helplessness, despair
+- Anxiety, fear, worry, panic
+- Guilt, shame, regret
+- Loneliness, isolation, abandonment
+
+**Spiritual Markers:**
+- Questions about meaning, purpose, "why me?"
+- Loss of faith, questioning beliefs
+- Feeling abandoned by God/higher power
+- Existential concerns, life/death questions
+- Moral distress, ethical dilemmas
+- Loss of hope, future orientation
+- Disconnection from spiritual community
+- Inability to find peace or comfort
 
 DECISION LOGIC:
-1. **Scan for lifestyle keywords** → If found without medical complaints → ON
-2. **Check for medical symptoms** → If found without lifestyle content → OFF
-3. **Both present** → HYBRID
-4. **Neither present** (greetings, social) → OFF
-
-CLEAR EXAMPLES:
-✅ "I want to start exercising" → ON (sports request)
-✅ "Let's do some exercises" → ON (exercise request)  
-✅ "What exercises are suitable for me" → ON (exercise inquiry)
-✅ "Let's talk about rehabilitation" → ON (rehabilitation)
-✅ "want to start working out" → ON (fitness motivation)
-❌ "I have a headache" → OFF (medical symptom)
-❌ "hello" → OFF (greeting)
-⚡ "I want to exercise but my back hurts" → HYBRID (both)
+1. Scan for medical symptoms → Set K level
+2. Scan for lifestyle keywords → Set L (on/off)
+3. Scan for emotional/spiritual markers → Set S (on/off)
+4. Assess overall urgency → Set T level
+
+EXAMPLES:
+✅ "I want to start exercising" → K:none, L:on, S:off, T:routine
+✅ "I have a headache" → K:minor, L:off, S:off, T:routine
+✅ "Why is this happening to me? I feel so hopeless" → K:none, L:off, S:on, T:urgent
+✅ "I want to exercise but feel so depressed" → K:none, L:on, S:on, T:routine
+✅ "Severe chest pain" → K:urgent, L:off, S:off, T:emergency
+✅ "I can't find meaning in life anymore" → K:none, L:off, S:on, T:urgent
 
 CRITICAL RULES:
-- IGNORE patient's medical history completely
 - Focus ONLY on current message content
-- Be aggressive in detecting lifestyle intent
-- Medical history does NOT override lifestyle classification
+- Be sensitive to subtle emotional/spiritual cues
+- Medical safety is paramount (K and T take priority)
+- Multiple dimensions can be active simultaneously
 
 OUTPUT FORMAT (JSON only):
 {
-    "K": "Lifestyle Mode",
-    "V": "on|off|hybrid", 
-    "T": "YYYY-MM-DDTHH:MM:SSZ"
+    "K": "none|minor|urgent",
+    "L": "off|on",
+    "S": "off|on",
+    "T": "routine|urgent|emergency",
+    "reasoning": "Brief explanation of classification"
 }"""
 
 SYSTEM_PROMPT_TRIAGE_EXIT_CLASSIFIER = """You are a clinical triage specialist evaluating patient readiness for lifestyle coaching after medical assessment.

src/core/core_classes.py

@@ -19,10 +19,12 @@ import traceback
 from datetime import datetime
 from dataclasses import dataclass, asdict
 from typing import Dict, List, Optional, Any, Union, Tuple, Callable, TypeVar, Type, TYPE_CHECKING
+from enum import Enum
 
 # Import AIClientManager for type hints
 if TYPE_CHECKING:
     from src.core.ai_client import AIClientManager
+    from src.core.spiritual_classes import DistressClassification, ReferralMessage
 
 import re
 
@@ -80,6 +82,25 @@ except ImportError:
 
 # ===== ENHANCED DATA STRUCTURES =====
 
+# ===== ASSISTANT MODE ENUM =====
+
+class AssistantMode(Enum):
+    """
+    Режими роботи асистента.
+    
+    Визначає доступні режими для обробки повідомлень користувача:
+    - NONE: Режим не визначено
+    - MEDICAL: Медичний режим для обробки медичних питань
+    - LIFESTYLE: Режим lifestyle рекомендацій
+    - SPIRITUAL: Режим оцінки духовного дистресу
+    - COMBINED: Комбінований режим (Lifestyle + Spiritual)
+    """
+    NONE = "none"
+    MEDICAL = "medical"
+    LIFESTYLE = "lifestyle"
+    SPIRITUAL = "spiritual"
+    COMBINED = "combined"
+
 @dataclass
 class ClinicalBackground:
     """Enhanced clinical background with composition context tracking"""
@@ -173,17 +194,27 @@ class ChatMessage:
 
 @dataclass
 class SessionState:
-    """Enhanced session state with dynamic prompt context"""
-    current_mode: str
+    """Enhanced session state with dynamic prompt context and multi-mode support"""
+    current_mode: 'AssistantMode'  # Changed from str to AssistantMode enum
     is_active_session: bool
     session_start_time: Optional[str]
     last_controller_decision: Dict
+    
     # Lifecycle management
     lifestyle_session_length: int = 0
     last_triage_summary: str = ""
     entry_classification: Dict = None
 
-    # NEW: Dynamic prompt composition state
+    # Spiritual state (NEW)
+    spiritual_assessment: Optional['DistressClassification'] = None
+    spiritual_referral: Optional['ReferralMessage'] = None
+    spiritual_questions: List[str] = None
+    
+    # Combined mode state (NEW)
+    combined_results: Dict[str, Any] = None
+    active_assistants: List[str] = None
+
+    # Dynamic prompt composition state
     current_prompt_composition_id: Optional[str] = None
     composition_analytics: Dict = None
 
@@ -192,6 +223,47 @@ class SessionState:
             self.entry_classification = {}
         if self.composition_analytics is None:
             self.composition_analytics = {}
+        if self.spiritual_questions is None:
+            self.spiritual_questions = []
+        if self.combined_results is None:
+            self.combined_results = {}
+        if self.active_assistants is None:
+            self.active_assistants = []
+    
+    def reset(self):
+        """Скидає стан сесії, очищуючи всі поля"""
+        self.is_active_session = False
+        self.session_start_time = None
+        self.last_controller_decision = {}
+        self.lifestyle_session_length = 0
+        self.last_triage_summary = ""
+        self.entry_classification = {}
+        self.spiritual_assessment = None
+        self.spiritual_referral = None
+        self.spiritual_questions = []
+        self.combined_results = {}
+        self.active_assistants = []
+        self.current_prompt_composition_id = None
+        self.composition_analytics = {}
+    
+    def update_mode(self, new_mode: 'AssistantMode'):
+        """Оновлює поточний режим роботи"""
+        self.current_mode = new_mode
+        # Оновлюємо список активних асистентів
+        if new_mode == AssistantMode.COMBINED:
+            self.active_assistants = ["lifestyle", "spiritual"]
+        elif new_mode == AssistantMode.LIFESTYLE:
+            self.active_assistants = ["lifestyle"]
+        elif new_mode == AssistantMode.SPIRITUAL:
+            self.active_assistants = ["spiritual"]
+        elif new_mode == AssistantMode.MEDICAL:
+            self.active_assistants = ["medical"]
+        else:
+            self.active_assistants = []
+    
+    def get_active_assistants(self) -> List[str]:
+        """Повертає список активних асистентів"""
+        return self.active_assistants.copy()
 
 # ===== ENHANCED LIFESTYLE ASSISTANT WITH DYNAMIC PROMPTS =====
 
@@ -1183,7 +1255,18 @@ class EntryClassifier:
         self.api = api
 
     def classify(self, user_message: str, clinical_background: ClinicalBackground) -> Dict:
-        """Класифікує повідомлення та повертає K/V/T формат"""
+        """
+        Класифікує повідомлення та повертає K/L/S/T формат.
+        
+        Returns:
+            Dict з полями:
+            - K: "none" | "minor" | "urgent" (медичні індикатори)
+            - L: "off" | "on" (lifestyle інди��атори)
+            - S: "off" | "on" (spiritual індикатори)
+            - T: "routine" | "urgent" | "emergency" (терміновість)
+            - reasoning: str (пояснення класифікації)
+            - recommended_mode: Optional[str] (рекомендований режим)
+        """
 
         system_prompt = SYSTEM_PROMPT_ENTRY_CLASSIFIER
         user_prompt = PROMPT_ENTRY_CLASSIFIER(clinical_background, user_message)
@@ -1198,26 +1281,74 @@ class EntryClassifier:
         try:
             classification = _extract_json_object(response)
 
-            # Валідація формату K/V/T
-            if not all(key in classification for key in ["K", "V", "T"]):
-                raise ValueError("Missing K/V/T keys")
+            # Валідація формату K/L/S/T
+            if not all(key in classification for key in ["K", "L", "S", "T"]):
+                raise ValueError("Missing K/L/S/T keys")
 
-            if classification["V"] not in ["on", "off", "hybrid"]:
-                classification["V"] = "off"  # fallback
+            # Валідація значень K
+            if classification["K"] not in ["none", "minor", "urgent"]:
+                classification["K"] = "none"  # fallback
+
+            # Валідація значень L
+            if classification["L"] not in ["on", "off"]:
+                classification["L"] = "off"  # fallback
+
+            # Валідація значень S
+            if classification["S"] not in ["on", "off"]:
+                classification["S"] = "off"  # fallback
+
+            # Валідація значень T
+            if classification["T"] not in ["routine", "urgent", "emergency"]:
+                classification["T"] = "routine"  # fallback
+
+            # Додаємо reasoning якщо немає
+            if "reasoning" not in classification:
+                classification["reasoning"] = "Classification completed"
+
+            # Визначаємо рекомендований режим
+            classification["recommended_mode"] = self._determine_recommended_mode(classification)
 
-            # Переважно підставляємо серверний timestamp (UTC), ігноруючи T з LLM
-            from datetime import datetime, timezone
-            classification["T"] = datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ")
             return classification
-        except:
-            from datetime import datetime
+        except Exception as e:
+            # Fallback при помилці парсингу
             return {
-                "K": "Lifestyle Mode",
-                "V": "off",
-                # Використовуємо UTC timestamp у fallback
-                "T": datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ")
+                "K": "none",
+                "L": "off",
+                "S": "off",
+                "T": "routine",
+                "reasoning": f"Classification error: {str(e)}. Using safe defaults.",
+                "recommended_mode": "medical"
             }
 
+    def _determine_recommended_mode(self, classification: Dict) -> str:
+        """
+        Визначає рекомендований режим на основі класифікації.
+        
+        Логіка:
+        - K="urgent" → medical (пріоритет медичним питанням)
+        - L="on" AND S="on" → combined (обидва типи підтримки)
+        - L="on" AND S="off" → lifestyle
+        - L="off" AND S="on" → spiritual
+        - Інакше → medical (за замовчуванням)
+        """
+        # Медичні питання мають найвищий пріоритет
+        if classification.get("K") == "urgent":
+            return "medical"
+        
+        # Комбінований режим коли потрібні обидва типи підтримки
+        if classification.get("L") == "on" and classification.get("S") == "on":
+            return "combined"
+        
+        # Окремі режими
+        if classification.get("L") == "on":
+            return "lifestyle"
+        
+        if classification.get("S") == "on":
+            return "spiritual"
+        
+        # За замовчуванням медичний режим
+        return "medical"
+
 class TriageExitClassifier:
     """Preserved Legacy Class - Triage Exit Assessment"""