Spaces:
Sleeping
Convert all prompts and code to English
Browse files🌍 MAJOR LOCALIZATION UPDATE:
✅ **Prompts Conversion:**
- All system prompts now in English for better AI model performance
- Removed Ukrainian keywords from prompts (kept only technical functionality)
- Improved prompt clarity and structure
- Enhanced Entry Classifier with better keyword detection
✅ **Code Comments & Documentation:**
- All Ukrainian comments converted to English
- Function docstrings translated
- Error messages in English
- Debug output in English
✅ **Test Suite:**
- Created new English test suite (test_english_logic.py)
- All test cases converted to English
- Mock responses updated for English keywords
- Maintained test coverage and functionality
✅ **Improved Classifier Logic:**
- Entry Classifier now uses English keywords: exercise, workout, training, fitness, sport, rehabilitation, nutrition, diet, physical, activity, movement, therapy
- Better hybrid detection logic
- More accurate lifestyle vs medical classification
- Enhanced decision tree logic
✅ **Files Updated:**
- prompts.py: All prompts converted to English
- core_classes.py: Comments and docstrings in English
- lifestyle_app.py: All comments translated
- file_utils.py: Documentation in English
- test_patients.py: Descriptions in English
- app_config.py: Configuration comments in English
- huggingface_space.py: Error messages in English
🧪 **Testing:**
- All English tests pass ✅
- Entry Classifier properly detects English lifestyle terms
- Hybrid classification works correctly
- Main Lifestyle Assistant responds appropriately
This ensures better AI model performance with English prompts while maintaining full functionality.
- app_config.py +4 -4
- core_classes.py +10 -10
- debug_classifier.py +84 -0
- file_utils.py +8 -8
- huggingface_space.py +1 -1
- lifestyle_app.py +13 -13
- prompts.py +168 -176
- test_english_logic.py +357 -0
- test_patients.py +4 -4
|
@@ -1,8 +1,8 @@
|
|
| 1 |
"""
|
| 2 |
-
|
| 3 |
"""
|
| 4 |
|
| 5 |
-
# HuggingFace Spaces
|
| 6 |
SPACE_CONFIG = {
|
| 7 |
"title": "🏥 Lifestyle Journey MVP",
|
| 8 |
"emoji": "🏥",
|
|
@@ -15,7 +15,7 @@ SPACE_CONFIG = {
|
|
| 15 |
"license": "mit"
|
| 16 |
}
|
| 17 |
|
| 18 |
-
# Gradio
|
| 19 |
GRADIO_CONFIG = {
|
| 20 |
"theme": "soft",
|
| 21 |
"show_api": False,
|
|
@@ -24,7 +24,7 @@ GRADIO_CONFIG = {
|
|
| 24 |
"title": "Lifestyle Journey MVP"
|
| 25 |
}
|
| 26 |
|
| 27 |
-
# API
|
| 28 |
API_CONFIG = {
|
| 29 |
"gemini_model": "gemini-2.5-flash",
|
| 30 |
"temperature": 0.3,
|
|
|
|
| 1 |
"""
|
| 2 |
+
Configuration for HuggingFace Spaces deployment
|
| 3 |
"""
|
| 4 |
|
| 5 |
+
# HuggingFace Spaces metadata
|
| 6 |
SPACE_CONFIG = {
|
| 7 |
"title": "🏥 Lifestyle Journey MVP",
|
| 8 |
"emoji": "🏥",
|
|
|
|
| 15 |
"license": "mit"
|
| 16 |
}
|
| 17 |
|
| 18 |
+
# Gradio configuration
|
| 19 |
GRADIO_CONFIG = {
|
| 20 |
"theme": "soft",
|
| 21 |
"show_api": False,
|
|
|
|
| 24 |
"title": "Lifestyle Journey MVP"
|
| 25 |
}
|
| 26 |
|
| 27 |
+
# API configuration
|
| 28 |
API_CONFIG = {
|
| 29 |
"gemini_model": "gemini-2.5-flash",
|
| 30 |
"temperature": 0.3,
|
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
# core_classes.py -
|
| 2 |
|
| 3 |
import os
|
| 4 |
import json
|
|
@@ -9,7 +9,7 @@ from google import genai
|
|
| 9 |
from google.genai import types
|
| 10 |
|
| 11 |
from prompts import (
|
| 12 |
-
#
|
| 13 |
SYSTEM_PROMPT_ENTRY_CLASSIFIER,
|
| 14 |
PROMPT_ENTRY_CLASSIFIER,
|
| 15 |
SYSTEM_PROMPT_TRIAGE_EXIT_CLASSIFIER,
|
|
@@ -21,10 +21,10 @@ from prompts import (
|
|
| 21 |
# Main Lifestyle Assistant
|
| 22 |
SYSTEM_PROMPT_MAIN_LIFESTYLE,
|
| 23 |
PROMPT_MAIN_LIFESTYLE,
|
| 24 |
-
#
|
| 25 |
SYSTEM_PROMPT_SOFT_MEDICAL_TRIAGE,
|
| 26 |
PROMPT_SOFT_MEDICAL_TRIAGE,
|
| 27 |
-
#
|
| 28 |
SYSTEM_PROMPT_MEDICAL_ASSISTANT,
|
| 29 |
PROMPT_MEDICAL_ASSISTANT
|
| 30 |
)
|
|
@@ -101,7 +101,7 @@ class SessionState:
|
|
| 101 |
is_active_session: bool
|
| 102 |
session_start_time: Optional[str]
|
| 103 |
last_controller_decision: Dict
|
| 104 |
-
#
|
| 105 |
lifestyle_session_length: int = 0
|
| 106 |
last_triage_summary: str = ""
|
| 107 |
entry_classification: Dict = None
|
|
@@ -119,7 +119,7 @@ class GeminiAPI:
|
|
| 119 |
self.call_counter = 0
|
| 120 |
|
| 121 |
def _log_prompt_and_response(self, system_prompt: str, user_prompt: str, response: str, call_type: str = ""):
|
| 122 |
-
"""
|
| 123 |
log_prompts_enabled = os.getenv("LOG_PROMPTS", "false").lower() == "true"
|
| 124 |
if not log_prompts_enabled:
|
| 125 |
return
|
|
@@ -164,7 +164,7 @@ class GeminiAPI:
|
|
| 164 |
log_logger.info(log_message)
|
| 165 |
|
| 166 |
def generate_response(self, system_prompt: str, user_prompt: str, temperature: float = None, call_type: str = "") -> str:
|
| 167 |
-
"""
|
| 168 |
if temperature is None:
|
| 169 |
temperature = API_CONFIG.get("temperature", 0.3)
|
| 170 |
|
|
@@ -195,18 +195,18 @@ class GeminiAPI:
|
|
| 195 |
self._log_prompt_and_response(system_prompt, user_prompt, response, call_type)
|
| 196 |
return response
|
| 197 |
except Exception as e:
|
| 198 |
-
error_msg = f"
|
| 199 |
log_prompts_enabled = os.getenv("LOG_PROMPTS", "false").lower() == "true"
|
| 200 |
if log_prompts_enabled:
|
| 201 |
self._log_prompt_and_response(system_prompt, user_prompt, error_msg, f"{call_type}_ERROR")
|
| 202 |
return error_msg
|
| 203 |
|
| 204 |
class PatientDataLoader:
|
| 205 |
-
"""
|
| 206 |
|
| 207 |
@staticmethod
|
| 208 |
def load_clinical_background(file_path: str = "clinical_background.json") -> ClinicalBackground:
|
| 209 |
-
"""
|
| 210 |
try:
|
| 211 |
with open(file_path, 'r', encoding='utf-8') as f:
|
| 212 |
data = json.load(f)
|
|
|
|
| 1 |
+
# core_classes.py - Core classes for Lifestyle Journey
|
| 2 |
|
| 3 |
import os
|
| 4 |
import json
|
|
|
|
| 9 |
from google.genai import types
|
| 10 |
|
| 11 |
from prompts import (
|
| 12 |
+
# Active classifiers
|
| 13 |
SYSTEM_PROMPT_ENTRY_CLASSIFIER,
|
| 14 |
PROMPT_ENTRY_CLASSIFIER,
|
| 15 |
SYSTEM_PROMPT_TRIAGE_EXIT_CLASSIFIER,
|
|
|
|
| 21 |
# Main Lifestyle Assistant
|
| 22 |
SYSTEM_PROMPT_MAIN_LIFESTYLE,
|
| 23 |
PROMPT_MAIN_LIFESTYLE,
|
| 24 |
+
# Soft medical triage
|
| 25 |
SYSTEM_PROMPT_SOFT_MEDICAL_TRIAGE,
|
| 26 |
PROMPT_SOFT_MEDICAL_TRIAGE,
|
| 27 |
+
# Medical assistant
|
| 28 |
SYSTEM_PROMPT_MEDICAL_ASSISTANT,
|
| 29 |
PROMPT_MEDICAL_ASSISTANT
|
| 30 |
)
|
|
|
|
| 101 |
is_active_session: bool
|
| 102 |
session_start_time: Optional[str]
|
| 103 |
last_controller_decision: Dict
|
| 104 |
+
# New fields for lifecycle management
|
| 105 |
lifestyle_session_length: int = 0
|
| 106 |
last_triage_summary: str = ""
|
| 107 |
entry_classification: Dict = None
|
|
|
|
| 119 |
self.call_counter = 0
|
| 120 |
|
| 121 |
def _log_prompt_and_response(self, system_prompt: str, user_prompt: str, response: str, call_type: str = ""):
|
| 122 |
+
"""Logging prompts and responses"""
|
| 123 |
log_prompts_enabled = os.getenv("LOG_PROMPTS", "false").lower() == "true"
|
| 124 |
if not log_prompts_enabled:
|
| 125 |
return
|
|
|
|
| 164 |
log_logger.info(log_message)
|
| 165 |
|
| 166 |
def generate_response(self, system_prompt: str, user_prompt: str, temperature: float = None, call_type: str = "") -> str:
|
| 167 |
+
"""Generates response from Gemini"""
|
| 168 |
if temperature is None:
|
| 169 |
temperature = API_CONFIG.get("temperature", 0.3)
|
| 170 |
|
|
|
|
| 195 |
self._log_prompt_and_response(system_prompt, user_prompt, response, call_type)
|
| 196 |
return response
|
| 197 |
except Exception as e:
|
| 198 |
+
error_msg = f"API Error: {str(e)}"
|
| 199 |
log_prompts_enabled = os.getenv("LOG_PROMPTS", "false").lower() == "true"
|
| 200 |
if log_prompts_enabled:
|
| 201 |
self._log_prompt_and_response(system_prompt, user_prompt, error_msg, f"{call_type}_ERROR")
|
| 202 |
return error_msg
|
| 203 |
|
| 204 |
class PatientDataLoader:
|
| 205 |
+
"""Class for loading patient data from JSON files"""
|
| 206 |
|
| 207 |
@staticmethod
|
| 208 |
def load_clinical_background(file_path: str = "clinical_background.json") -> ClinicalBackground:
|
| 209 |
+
"""Loads clinical background from JSON file"""
|
| 210 |
try:
|
| 211 |
with open(file_path, 'r', encoding='utf-8') as f:
|
| 212 |
data = json.load(f)
|
|
@@ -0,0 +1,84 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env python3
|
| 2 |
+
"""
|
| 3 |
+
Debug tool to test Entry Classifier responses
|
| 4 |
+
"""
|
| 5 |
+
|
| 6 |
+
import os
|
| 7 |
+
from dotenv import load_dotenv
|
| 8 |
+
|
| 9 |
+
# Load environment variables
|
| 10 |
+
load_dotenv()
|
| 11 |
+
|
| 12 |
+
# Only proceed if we have the API key
|
| 13 |
+
if os.getenv("GEMINI_API_KEY"):
|
| 14 |
+
from core_classes import GeminiAPI, EntryClassifier, ClinicalBackground
|
| 15 |
+
|
| 16 |
+
def test_message(message):
|
| 17 |
+
"""Test a single message with the Entry Classifier"""
|
| 18 |
+
|
| 19 |
+
# Create API and classifier
|
| 20 |
+
api = GeminiAPI()
|
| 21 |
+
classifier = EntryClassifier(api)
|
| 22 |
+
|
| 23 |
+
# Create mock clinical background
|
| 24 |
+
clinical_bg = ClinicalBackground(
|
| 25 |
+
patient_id="test",
|
| 26 |
+
patient_name="John",
|
| 27 |
+
patient_age="52",
|
| 28 |
+
active_problems=["Nausea", "Hypokalemia", "Type 2 diabetes"],
|
| 29 |
+
past_medical_history=[],
|
| 30 |
+
current_medications=["Amlodipine"],
|
| 31 |
+
allergies="None",
|
| 32 |
+
vital_signs_and_measurements=[],
|
| 33 |
+
laboratory_results=[],
|
| 34 |
+
assessment_and_plan="",
|
| 35 |
+
critical_alerts=["Life endangering medical noncompliance"],
|
| 36 |
+
social_history={},
|
| 37 |
+
recent_clinical_events=[]
|
| 38 |
+
)
|
| 39 |
+
|
| 40 |
+
print(f"\n🔍 Testing: '{message}'")
|
| 41 |
+
|
| 42 |
+
try:
|
| 43 |
+
result = classifier.classify(message, clinical_bg)
|
| 44 |
+
classification = result.get("V", "unknown")
|
| 45 |
+
timestamp = result.get("T", "unknown")
|
| 46 |
+
|
| 47 |
+
print(f"📊 Result: V={classification}, T={timestamp}")
|
| 48 |
+
|
| 49 |
+
# Expected results
|
| 50 |
+
expected_on = ["exercise", "workout", "fitness", "sport", "training", "rehabilitation", "physical", "activity"]
|
| 51 |
+
should_be_on = any(keyword in message.lower() for keyword in expected_on)
|
| 52 |
+
|
| 53 |
+
if should_be_on and classification == "on":
|
| 54 |
+
print("✅ CORRECT: Lifestyle message properly classified as ON")
|
| 55 |
+
elif should_be_on and classification != "on":
|
| 56 |
+
print(f"❌ ERROR: Lifestyle message incorrectly classified as {classification.upper()}")
|
| 57 |
+
elif not should_be_on and classification == "off":
|
| 58 |
+
print("✅ CORRECT: Non-lifestyle message properly classified as OFF")
|
| 59 |
+
else:
|
| 60 |
+
print(f"ℹ️ Classification: {classification.upper()}")
|
| 61 |
+
|
| 62 |
+
except Exception as e:
|
| 63 |
+
print(f"❌ Error: {e}")
|
| 64 |
+
|
| 65 |
+
if __name__ == "__main__":
|
| 66 |
+
print("🧪 Entry Classifier Debug Tool")
|
| 67 |
+
print("Testing problematic messages...\n")
|
| 68 |
+
|
| 69 |
+
test_messages = [
|
| 70 |
+
"I want to exercise",
|
| 71 |
+
"Let's do some exercises",
|
| 72 |
+
"Let's talk about rehabilitation",
|
| 73 |
+
"Everything is fine let's do exercises",
|
| 74 |
+
"Which exercises are suitable for me",
|
| 75 |
+
"I have a headache",
|
| 76 |
+
"Hello",
|
| 77 |
+
"I want to exercise but my back hurts"
|
| 78 |
+
]
|
| 79 |
+
|
| 80 |
+
for message in test_messages:
|
| 81 |
+
test_message(message)
|
| 82 |
+
|
| 83 |
+
else:
|
| 84 |
+
print("❌ GEMINI_API_KEY not found. Please set up your .env file.")
|
|
@@ -1,22 +1,22 @@
|
|
| 1 |
-
# file_utils.py -
|
| 2 |
|
| 3 |
import os
|
| 4 |
import json
|
| 5 |
from typing import Tuple, Optional
|
| 6 |
|
| 7 |
class FileHandler:
|
| 8 |
-
"""
|
| 9 |
|
| 10 |
@staticmethod
|
| 11 |
-
def read_uploaded_file(file_input, filename_for_error: str = "
|
| 12 |
"""
|
| 13 |
-
|
| 14 |
|
| 15 |
Returns:
|
| 16 |
-
Tuple[content, error_message] - content
|
| 17 |
"""
|
| 18 |
if file_input is None:
|
| 19 |
-
return None, f"❌
|
| 20 |
|
| 21 |
# Debug information
|
| 22 |
debug_enabled = os.getenv("LOG_PROMPTS", "false").lower() == "true"
|
|
@@ -27,14 +27,14 @@ class FileHandler:
|
|
| 27 |
# Try 1: filepath (type="filepath")
|
| 28 |
if isinstance(file_input, str):
|
| 29 |
if debug_enabled:
|
| 30 |
-
print(f"📁
|
| 31 |
with open(file_input, 'r', encoding='utf-8') as f:
|
| 32 |
return f.read(), None
|
| 33 |
|
| 34 |
# Try 2: file-like object with read method
|
| 35 |
elif hasattr(file_input, 'read'):
|
| 36 |
if debug_enabled:
|
| 37 |
-
print(f"📄
|
| 38 |
content = file_input.read()
|
| 39 |
if isinstance(content, bytes):
|
| 40 |
content = content.decode('utf-8')
|
|
|
|
| 1 |
+
# file_utils.py - File handling utilities
|
| 2 |
|
| 3 |
import os
|
| 4 |
import json
|
| 5 |
from typing import Tuple, Optional
|
| 6 |
|
| 7 |
class FileHandler:
|
| 8 |
+
"""Class for handling uploaded files"""
|
| 9 |
|
| 10 |
@staticmethod
|
| 11 |
+
def read_uploaded_file(file_input, filename_for_error: str = "file") -> Tuple[Optional[str], Optional[str]]:
|
| 12 |
"""
|
| 13 |
+
Universal method for reading uploaded files from different Gradio versions
|
| 14 |
|
| 15 |
Returns:
|
| 16 |
+
Tuple[content, error_message] - content if successful, error_message if error
|
| 17 |
"""
|
| 18 |
if file_input is None:
|
| 19 |
+
return None, f"❌ File {filename_for_error} not uploaded"
|
| 20 |
|
| 21 |
# Debug information
|
| 22 |
debug_enabled = os.getenv("LOG_PROMPTS", "false").lower() == "true"
|
|
|
|
| 27 |
# Try 1: filepath (type="filepath")
|
| 28 |
if isinstance(file_input, str):
|
| 29 |
if debug_enabled:
|
| 30 |
+
print(f"📁 Reading as filepath: {file_input}")
|
| 31 |
with open(file_input, 'r', encoding='utf-8') as f:
|
| 32 |
return f.read(), None
|
| 33 |
|
| 34 |
# Try 2: file-like object with read method
|
| 35 |
elif hasattr(file_input, 'read'):
|
| 36 |
if debug_enabled:
|
| 37 |
+
print(f"📄 Reading as file-like object")
|
| 38 |
content = file_input.read()
|
| 39 |
if isinstance(content, bytes):
|
| 40 |
content = content.decode('utf-8')
|
|
@@ -19,7 +19,7 @@ def main():
|
|
| 19 |
show_error=True
|
| 20 |
)
|
| 21 |
except Exception as e:
|
| 22 |
-
print(f"❌
|
| 23 |
raise
|
| 24 |
|
| 25 |
if __name__ == "__main__":
|
|
|
|
| 19 |
show_error=True
|
| 20 |
)
|
| 21 |
except Exception as e:
|
| 22 |
+
print(f"❌ Application startup error: {e}")
|
| 23 |
raise
|
| 24 |
|
| 25 |
if __name__ == "__main__":
|
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
# lifestyle_app.py -
|
| 2 |
|
| 3 |
import os
|
| 4 |
import json
|
|
@@ -11,12 +11,12 @@ from core_classes import (
|
|
| 11 |
ClinicalBackground, LifestyleProfile, ChatMessage, SessionState,
|
| 12 |
GeminiAPI, PatientDataLoader,
|
| 13 |
MedicalAssistant,
|
| 14 |
-
#
|
| 15 |
EntryClassifier, TriageExitClassifier,
|
| 16 |
LifestyleSessionManager,
|
| 17 |
# Main Lifestyle Assistant
|
| 18 |
MainLifestyleAssistant,
|
| 19 |
-
#
|
| 20 |
SoftMedicalTriage
|
| 21 |
)
|
| 22 |
from testing_lab import TestingDataManager, PatientTestingInterface, TestSession
|
|
@@ -28,18 +28,18 @@ class ExtendedLifestyleJourneyApp:
|
|
| 28 |
|
| 29 |
def __init__(self):
|
| 30 |
self.api = GeminiAPI()
|
| 31 |
-
#
|
| 32 |
self.entry_classifier = EntryClassifier(self.api)
|
| 33 |
self.triage_exit_classifier = TriageExitClassifier(self.api)
|
| 34 |
# LifestyleExitClassifier removed - functionality moved to MainLifestyleAssistant
|
| 35 |
-
#
|
| 36 |
self.medical_assistant = MedicalAssistant(self.api)
|
| 37 |
self.main_lifestyle_assistant = MainLifestyleAssistant(self.api)
|
| 38 |
self.soft_medical_triage = SoftMedicalTriage(self.api)
|
| 39 |
-
# Lifecycle
|
| 40 |
self.lifestyle_session_manager = LifestyleSessionManager(self.api)
|
| 41 |
|
| 42 |
-
# Testing Lab
|
| 43 |
self.testing_manager = TestingDataManager()
|
| 44 |
self.testing_interface = PatientTestingInterface(self.testing_manager)
|
| 45 |
|
|
@@ -66,7 +66,7 @@ class ExtendedLifestyleJourneyApp:
|
|
| 66 |
def load_test_patient(self, clinical_file, lifestyle_file) -> Tuple[str, str, List, str]:
|
| 67 |
"""Loads test patient from files"""
|
| 68 |
try:
|
| 69 |
-
#
|
| 70 |
clinical_content, error = FileHandler.read_uploaded_file(clinical_file, "clinical_background.json")
|
| 71 |
if error:
|
| 72 |
return error, "", [], self._get_status_info()
|
|
@@ -75,7 +75,7 @@ class ExtendedLifestyleJourneyApp:
|
|
| 75 |
if error:
|
| 76 |
return error, "", [], self._get_status_info()
|
| 77 |
|
| 78 |
-
#
|
| 79 |
lifestyle_content, error = FileHandler.read_uploaded_file(lifestyle_file, "lifestyle_profile.json")
|
| 80 |
if error:
|
| 81 |
return error, "", [], self._get_status_info()
|
|
@@ -84,7 +84,7 @@ class ExtendedLifestyleJourneyApp:
|
|
| 84 |
if error:
|
| 85 |
return error, "", [], self._get_status_info()
|
| 86 |
|
| 87 |
-
#
|
| 88 |
return self._process_patient_data(clinical_data, lifestyle_data, "")
|
| 89 |
|
| 90 |
except Exception as e:
|
|
@@ -114,7 +114,7 @@ class ExtendedLifestyleJourneyApp:
|
|
| 114 |
|
| 115 |
debug_enabled = os.getenv("LOG_PROMPTS", "false").lower() == "true"
|
| 116 |
if debug_enabled:
|
| 117 |
-
print(f"🔄 _process_patient_data
|
| 118 |
|
| 119 |
# STEP 1: End previous test session if active
|
| 120 |
if self.test_mode_active and self.testing_interface.current_session:
|
|
@@ -400,7 +400,7 @@ class ExtendedLifestyleJourneyApp:
|
|
| 400 |
)
|
| 401 |
|
| 402 |
action = result.get("action", "lifestyle_dialog")
|
| 403 |
-
response_message = result.get("message", "
|
| 404 |
|
| 405 |
if action == "close":
|
| 406 |
# End lifestyle session and update profile with LLM analysis
|
|
@@ -465,7 +465,7 @@ class ExtendedLifestyleJourneyApp:
|
|
| 465 |
for session in latest_sessions:
|
| 466 |
table_data.append([
|
| 467 |
session.get('patient_name', 'N/A'),
|
| 468 |
-
session.get('timestamp', 'N/A')[:16], #
|
| 469 |
session.get('total_messages', 0),
|
| 470 |
session.get('medical_messages', 0),
|
| 471 |
session.get('lifestyle_messages', 0),
|
|
|
|
| 1 |
+
# lifestyle_app.py - Main application class
|
| 2 |
|
| 3 |
import os
|
| 4 |
import json
|
|
|
|
| 11 |
ClinicalBackground, LifestyleProfile, ChatMessage, SessionState,
|
| 12 |
GeminiAPI, PatientDataLoader,
|
| 13 |
MedicalAssistant,
|
| 14 |
+
# Active classifiers
|
| 15 |
EntryClassifier, TriageExitClassifier,
|
| 16 |
LifestyleSessionManager,
|
| 17 |
# Main Lifestyle Assistant
|
| 18 |
MainLifestyleAssistant,
|
| 19 |
+
# Soft medical triage
|
| 20 |
SoftMedicalTriage
|
| 21 |
)
|
| 22 |
from testing_lab import TestingDataManager, PatientTestingInterface, TestSession
|
|
|
|
| 28 |
|
| 29 |
def __init__(self):
|
| 30 |
self.api = GeminiAPI()
|
| 31 |
+
# Active classifiers
|
| 32 |
self.entry_classifier = EntryClassifier(self.api)
|
| 33 |
self.triage_exit_classifier = TriageExitClassifier(self.api)
|
| 34 |
# LifestyleExitClassifier removed - functionality moved to MainLifestyleAssistant
|
| 35 |
+
# Assistants
|
| 36 |
self.medical_assistant = MedicalAssistant(self.api)
|
| 37 |
self.main_lifestyle_assistant = MainLifestyleAssistant(self.api)
|
| 38 |
self.soft_medical_triage = SoftMedicalTriage(self.api)
|
| 39 |
+
# Lifecycle manager
|
| 40 |
self.lifestyle_session_manager = LifestyleSessionManager(self.api)
|
| 41 |
|
| 42 |
+
# Testing Lab components
|
| 43 |
self.testing_manager = TestingDataManager()
|
| 44 |
self.testing_interface = PatientTestingInterface(self.testing_manager)
|
| 45 |
|
|
|
|
| 66 |
def load_test_patient(self, clinical_file, lifestyle_file) -> Tuple[str, str, List, str]:
|
| 67 |
"""Loads test patient from files"""
|
| 68 |
try:
|
| 69 |
+
# Read clinical background
|
| 70 |
clinical_content, error = FileHandler.read_uploaded_file(clinical_file, "clinical_background.json")
|
| 71 |
if error:
|
| 72 |
return error, "", [], self._get_status_info()
|
|
|
|
| 75 |
if error:
|
| 76 |
return error, "", [], self._get_status_info()
|
| 77 |
|
| 78 |
+
# Read lifestyle profile
|
| 79 |
lifestyle_content, error = FileHandler.read_uploaded_file(lifestyle_file, "lifestyle_profile.json")
|
| 80 |
if error:
|
| 81 |
return error, "", [], self._get_status_info()
|
|
|
|
| 84 |
if error:
|
| 85 |
return error, "", [], self._get_status_info()
|
| 86 |
|
| 87 |
+
# Use common processing method
|
| 88 |
return self._process_patient_data(clinical_data, lifestyle_data, "")
|
| 89 |
|
| 90 |
except Exception as e:
|
|
|
|
| 114 |
|
| 115 |
debug_enabled = os.getenv("LOG_PROMPTS", "false").lower() == "true"
|
| 116 |
if debug_enabled:
|
| 117 |
+
print(f"🔄 _process_patient_data called with test_type_info: '{test_type_info}'")
|
| 118 |
|
| 119 |
# STEP 1: End previous test session if active
|
| 120 |
if self.test_mode_active and self.testing_interface.current_session:
|
|
|
|
| 400 |
)
|
| 401 |
|
| 402 |
action = result.get("action", "lifestyle_dialog")
|
| 403 |
+
response_message = result.get("message", "How are you feeling?")
|
| 404 |
|
| 405 |
if action == "close":
|
| 406 |
# End lifestyle session and update profile with LLM analysis
|
|
|
|
| 465 |
for session in latest_sessions:
|
| 466 |
table_data.append([
|
| 467 |
session.get('patient_name', 'N/A'),
|
| 468 |
+
session.get('timestamp', 'N/A')[:16], # Date and time only
|
| 469 |
session.get('total_messages', 0),
|
| 470 |
session.get('medical_messages', 0),
|
| 471 |
session.get('lifestyle_messages', 0),
|
|
@@ -2,100 +2,79 @@
|
|
| 2 |
|
| 3 |
# ===== CLASSIFIERS =====
|
| 4 |
|
| 5 |
-
SYSTEM_PROMPT_ENTRY_CLASSIFIER = """
|
| 6 |
|
| 7 |
TASK:
|
| 8 |
-
Classify the patient
|
| 9 |
-
|
| 10 |
-
GOAL:
|
| 11 |
-
Accurately classify patient communication to route to the most appropriate care pathway while ensuring medical safety.
|
| 12 |
|
| 13 |
CLASSIFICATION MODES:
|
| 14 |
-
-
|
| 15 |
-
-
|
| 16 |
-
- HYBRID
|
| 17 |
-
|
| 18 |
-
LIFESTYLE
|
| 19 |
-
|
| 20 |
-
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
3. Apply safety-first approach - medical concerns override lifestyle content
|
| 46 |
-
4. Use HYBRID only when message clearly contains BOTH medical concerns AND lifestyle questions
|
| 47 |
-
5. For greetings/social interactions, use OFF to enable gentle patient check-in
|
| 48 |
-
6. Extract current timestamp in ISO format
|
| 49 |
-
|
| 50 |
-
EXAMPLES:
|
| 51 |
-
- "давай займемося вправами" → ON (exercise request)
|
| 52 |
-
- "хочу почати тренуватися" → ON (fitness motivation)
|
| 53 |
-
- "поговоримо про реабілітацію" → ON (rehabilitation discussion)
|
| 54 |
-
- "у мене болить голова" → OFF (medical symptom)
|
| 55 |
-
- "хочу займатися спортом але болить спина" → HYBRID (both lifestyle + medical)
|
| 56 |
-
|
| 57 |
-
OUTPUT FORMAT:
|
| 58 |
-
Provide ONLY JSON without comments:
|
| 59 |
{
|
| 60 |
"K": "Lifestyle Mode",
|
| 61 |
-
"V": "on|off|hybrid",
|
| 62 |
"T": "YYYY-MM-DDTHH:MM:SSZ"
|
| 63 |
-
}
|
| 64 |
-
|
| 65 |
-
Use current real timestamp for field T."""
|
| 66 |
|
| 67 |
-
SYSTEM_PROMPT_TRIAGE_EXIT_CLASSIFIER = """
|
| 68 |
|
| 69 |
TASK:
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
-
|
| 77 |
-
-
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
-
|
| 81 |
-
|
| 82 |
-
|
| 83 |
-
-
|
| 84 |
-
|
| 85 |
-
|
| 86 |
-
-
|
| 87 |
-
|
| 88 |
-
|
| 89 |
-
|
| 90 |
-
|
| 91 |
-
|
| 92 |
-
|
| 93 |
-
|
| 94 |
-
OUTPUT FORMAT:
|
| 95 |
-
Provide ONLY JSON without comments:
|
| 96 |
{
|
| 97 |
"ready_for_lifestyle": true/false,
|
| 98 |
-
"reasoning": "explanation in patient's language",
|
| 99 |
"medical_status": "stable|needs_attention|resolved"
|
| 100 |
}"""
|
| 101 |
|
|
@@ -131,7 +110,7 @@ Assess patient's readiness for lifestyle coaching mode based on medical stabilit
|
|
| 131 |
|
| 132 |
# PROMPT_LIFESTYLE_EXIT_CLASSIFIER removed - functionality moved to MainLifestyleAssistant
|
| 133 |
|
| 134 |
-
# DEPRECATED:
|
| 135 |
|
| 136 |
|
| 137 |
# ===== LIFESTYLE PROFILE UPDATE =====
|
|
@@ -207,61 +186,77 @@ RESPOND IN JSON FORMAT:
|
|
| 207 |
|
| 208 |
# ===== ASSISTANTS =====
|
| 209 |
|
| 210 |
-
SYSTEM_PROMPT_MEDICAL_ASSISTANT = """
|
| 211 |
|
| 212 |
TASK:
|
| 213 |
-
Provide safe, evidence-based medical guidance
|
| 214 |
-
|
| 215 |
-
|
| 216 |
-
|
| 217 |
-
|
| 218 |
-
|
| 219 |
-
-
|
| 220 |
-
-
|
| 221 |
-
- Recommend
|
| 222 |
-
|
| 223 |
-
|
| 224 |
-
-
|
| 225 |
-
|
| 226 |
-
|
| 227 |
-
-
|
| 228 |
-
|
| 229 |
-
|
| 230 |
-
|
| 231 |
-
|
| 232 |
-
|
| 233 |
-
-
|
| 234 |
-
-
|
| 235 |
-
|
| 236 |
-
|
| 237 |
-
-
|
| 238 |
-
|
| 239 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 240 |
|
| 241 |
TASK:
|
| 242 |
-
|
| 243 |
|
| 244 |
-
|
| 245 |
-
|
|
|
|
|
|
|
| 246 |
|
| 247 |
-
|
| 248 |
-
-
|
| 249 |
-
-
|
| 250 |
-
-
|
| 251 |
-
-
|
| 252 |
-
- Respond in the patient's language (match their communication language)
|
| 253 |
|
| 254 |
RESPONSE STRUCTURE:
|
| 255 |
-
1. Acknowledge patient
|
| 256 |
-
2. 1-2
|
| 257 |
-
3. Express
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 258 |
|
| 259 |
-
|
| 260 |
-
- Keep medical questioning minimal - this is gentle triage, not comprehensive assessment
|
| 261 |
-
- Focus on immediate comfort and safety
|
| 262 |
-
- Maintain supportive, caring tone throughout
|
| 263 |
-
- Be prepared to escalate if concerning symptoms mentioned
|
| 264 |
-
- Always respond in the same language the patient used"""
|
| 265 |
|
| 266 |
def PROMPT_MEDICAL_ASSISTANT(clinical_background, active_problems, medications, recent_vitals, history_text, user_message):
|
| 267 |
return f"""PATIENT MEDICAL PROFILE ({clinical_background.patient_name}):
|
|
@@ -294,58 +289,55 @@ Conduct gentle medical triage - acknowledge the patient warmly and delicately ch
|
|
| 294 |
|
| 295 |
|
| 296 |
|
| 297 |
-
# ===== MAIN LIFESTYLE ASSISTANT (
|
| 298 |
|
| 299 |
-
SYSTEM_PROMPT_MAIN_LIFESTYLE = """
|
| 300 |
|
| 301 |
TASK:
|
| 302 |
-
|
| 303 |
-
|
| 304 |
-
|
| 305 |
-
|
| 306 |
-
|
| 307 |
-
|
| 308 |
-
-
|
| 309 |
-
-
|
| 310 |
-
|
| 311 |
-
|
| 312 |
-
|
| 313 |
-
|
| 314 |
-
|
| 315 |
-
|
| 316 |
-
|
| 317 |
-
|
| 318 |
-
|
| 319 |
-
|
| 320 |
-
|
| 321 |
-
|
| 322 |
-
|
| 323 |
-
|
| 324 |
-
|
| 325 |
-
|
| 326 |
-
|
| 327 |
-
|
| 328 |
-
|
| 329 |
-
|
| 330 |
-
|
| 331 |
-
|
| 332 |
-
|
| 333 |
-
|
| 334 |
-
-
|
| 335 |
-
-
|
| 336 |
-
-
|
| 337 |
-
|
| 338 |
-
|
| 339 |
-
|
| 340 |
-
OUTPUT FORMAT:
|
| 341 |
-
Provide ONLY JSON without comments:
|
| 342 |
{
|
| 343 |
-
"message": "your response
|
| 344 |
-
"action": "gather_info|lifestyle_dialog|close",
|
| 345 |
-
"reasoning": "brief explanation of action
|
| 346 |
}"""
|
| 347 |
|
| 348 |
-
# ===== DEPRECATED:
|
| 349 |
|
| 350 |
|
| 351 |
def PROMPT_MAIN_LIFESTYLE(lifestyle_profile, clinical_background, session_length, history_text, user_message):
|
|
@@ -373,4 +365,4 @@ PATIENT'S NEW MESSAGE: "{user_message}"
|
|
| 373 |
ANALYSIS REQUIRED:
|
| 374 |
Analyze the situation and determine the best action for this lifestyle coaching session."""
|
| 375 |
|
| 376 |
-
# ===== DEPRECATED:
|
|
|
|
| 2 |
|
| 3 |
# ===== CLASSIFIERS =====
|
| 4 |
|
| 5 |
+
SYSTEM_PROMPT_ENTRY_CLASSIFIER = """You are a message classification specialist for a medical chat system with lifestyle coaching capabilities.
|
| 6 |
|
| 7 |
TASK:
|
| 8 |
+
Classify the current patient message to determine the appropriate system mode. Focus ONLY on the message content, completely ignoring patient's medical history.
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
CLASSIFICATION MODES:
|
| 11 |
+
- **ON**: Lifestyle, exercise, nutrition, rehabilitation requests
|
| 12 |
+
- **OFF**: Medical complaints, symptoms, greetings, general questions
|
| 13 |
+
- **HYBRID**: Messages containing BOTH lifestyle requests AND current medical complaints
|
| 14 |
+
|
| 15 |
+
AGGRESSIVE LIFESTYLE DETECTION:
|
| 16 |
+
If the message contains ANY of these terms, classify as ON regardless of medical history:
|
| 17 |
+
- Keywords: exercise, workout, training, fitness, sport, rehabilitation, nutrition, diet, physical, activity, movement, therapy
|
| 18 |
+
|
| 19 |
+
DECISION LOGIC:
|
| 20 |
+
1. **Scan for lifestyle keywords** → If found without medical complaints → ON
|
| 21 |
+
2. **Check for medical symptoms** → If found without lifestyle content → OFF
|
| 22 |
+
3. **Both present** → HYBRID
|
| 23 |
+
4. **Neither present** (greetings, social) → OFF
|
| 24 |
+
|
| 25 |
+
CLEAR EXAMPLES:
|
| 26 |
+
✅ "I want to start exercising" → ON (sports request)
|
| 27 |
+
✅ "Let's do some exercises" → ON (exercise request)
|
| 28 |
+
✅ "What exercises are suitable for me" → ON (exercise inquiry)
|
| 29 |
+
✅ "Let's talk about rehabilitation" → ON (rehabilitation)
|
| 30 |
+
✅ "want to start working out" → ON (fitness motivation)
|
| 31 |
+
❌ "I have a headache" → OFF (medical symptom)
|
| 32 |
+
❌ "hello" → OFF (greeting)
|
| 33 |
+
⚡ "I want to exercise but my back hurts" → HYBRID (both)
|
| 34 |
+
|
| 35 |
+
CRITICAL RULES:
|
| 36 |
+
- IGNORE patient's medical history completely
|
| 37 |
+
- Focus ONLY on current message content
|
| 38 |
+
- Be aggressive in detecting lifestyle intent
|
| 39 |
+
- Medical history does NOT override lifestyle classification
|
| 40 |
+
|
| 41 |
+
OUTPUT FORMAT (JSON only):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
{
|
| 43 |
"K": "Lifestyle Mode",
|
| 44 |
+
"V": "on|off|hybrid",
|
| 45 |
"T": "YYYY-MM-DDTHH:MM:SSZ"
|
| 46 |
+
}"""
|
|
|
|
|
|
|
| 47 |
|
| 48 |
+
SYSTEM_PROMPT_TRIAGE_EXIT_CLASSIFIER = """You are a clinical triage specialist evaluating patient readiness for lifestyle coaching after medical assessment.
|
| 49 |
|
| 50 |
TASK:
|
| 51 |
+
Determine if the patient is medically stable and ready to transition from medical triage to lifestyle coaching.
|
| 52 |
+
|
| 53 |
+
READINESS ASSESSMENT:
|
| 54 |
+
✅ **READY for lifestyle coaching when:**
|
| 55 |
+
- Medical concerns addressed or stabilized
|
| 56 |
+
- Patient expresses interest in lifestyle activities
|
| 57 |
+
- No urgent symptoms requiring immediate attention
|
| 58 |
+
- Patient feels comfortable proceeding with lifestyle goals
|
| 59 |
+
|
| 60 |
+
❌ **NOT READY when:**
|
| 61 |
+
- Active, unresolved medical symptoms
|
| 62 |
+
- Patient requests continued medical focus
|
| 63 |
+
- Urgent medical issues requiring attention
|
| 64 |
+
- Patient expresses discomfort with lifestyle transition
|
| 65 |
+
|
| 66 |
+
DECISION APPROACH:
|
| 67 |
+
- **Conservative**: When in doubt, prioritize medical safety
|
| 68 |
+
- **Patient-centered**: Respect patient's expressed preferences
|
| 69 |
+
- **Contextual**: Consider both medical status and patient readiness
|
| 70 |
+
|
| 71 |
+
RESPONSE LANGUAGE:
|
| 72 |
+
Always respond in the same language the patient used in their messages.
|
| 73 |
+
|
| 74 |
+
OUTPUT FORMAT (JSON only):
|
|
|
|
|
|
|
| 75 |
{
|
| 76 |
"ready_for_lifestyle": true/false,
|
| 77 |
+
"reasoning": "clear explanation in patient's language",
|
| 78 |
"medical_status": "stable|needs_attention|resolved"
|
| 79 |
}"""
|
| 80 |
|
|
|
|
| 110 |
|
| 111 |
# PROMPT_LIFESTYLE_EXIT_CLASSIFIER removed - functionality moved to MainLifestyleAssistant
|
| 112 |
|
| 113 |
+
# DEPRECATED: Old Session Controller (replaced with Entry Classifier + new logic)
|
| 114 |
|
| 115 |
|
| 116 |
# ===== LIFESTYLE PROFILE UPDATE =====
|
|
|
|
| 186 |
|
| 187 |
# ===== ASSISTANTS =====
|
| 188 |
|
| 189 |
+
SYSTEM_PROMPT_MEDICAL_ASSISTANT = """You are an experienced medical assistant specializing in chronic disease management and patient safety.
|
| 190 |
|
| 191 |
TASK:
|
| 192 |
+
Provide safe, evidence-based medical guidance while maintaining appropriate clinical boundaries.
|
| 193 |
+
|
| 194 |
+
SCOPE OF PRACTICE:
|
| 195 |
+
✅ **What you CAN do:**
|
| 196 |
+
- Provide general health education
|
| 197 |
+
- Explain chronic disease management principles
|
| 198 |
+
- Offer symptom monitoring guidance
|
| 199 |
+
- Support medication adherence (not prescribe)
|
| 200 |
+
- Recommend when to contact healthcare providers
|
| 201 |
+
|
| 202 |
+
❌ **What you CANNOT do:**
|
| 203 |
+
- Diagnose medical conditions
|
| 204 |
+
- Prescribe or adjust medications
|
| 205 |
+
- Replace professional medical evaluation
|
| 206 |
+
- Provide emergency medical treatment
|
| 207 |
+
|
| 208 |
+
SAFETY PROTOCOLS:
|
| 209 |
+
🚨 **URGENT** (immediate medical attention):
|
| 210 |
+
- Chest pain, severe shortness of breath
|
| 211 |
+
- Signs of stroke, severe allergic reactions
|
| 212 |
+
- Uncontrolled bleeding, severe trauma
|
| 213 |
+
- Loss of consciousness, severe confusion
|
| 214 |
+
|
| 215 |
+
⚠️ **CONCERNING** (prompt medical consultation):
|
| 216 |
+
- New or worsening symptoms
|
| 217 |
+
- Medication side effects or concerns
|
| 218 |
+
- Significant changes in chronic conditions
|
| 219 |
+
- Patient anxiety about health changes
|
| 220 |
+
|
| 221 |
+
RESPONSE APPROACH:
|
| 222 |
+
- **Empathetic acknowledgment** of patient concerns
|
| 223 |
+
- **Educational support** within appropriate scope
|
| 224 |
+
- **Clear escalation** when medical evaluation needed
|
| 225 |
+
- **Patient empowerment** for healthcare engagement
|
| 226 |
+
- **Same language** as patient uses
|
| 227 |
+
|
| 228 |
+
Always prioritize patient safety over providing comprehensive answers."""
|
| 229 |
+
|
| 230 |
+
SYSTEM_PROMPT_SOFT_MEDICAL_TRIAGE = """You are a compassionate medical assistant conducting gentle patient check-ins.
|
| 231 |
|
| 232 |
TASK:
|
| 233 |
+
Provide a warm, non-intrusive health assessment at the start of patient interactions.
|
| 234 |
|
| 235 |
+
SOFT TRIAGE APPROACH:
|
| 236 |
+
🤗 **Warm acknowledgment** of patient's message
|
| 237 |
+
🩺 **Gentle health check** with 1-2 brief questions
|
| 238 |
+
💚 **Supportive readiness** to help with any concerns
|
| 239 |
|
| 240 |
+
TRIAGE PRINCIPLES:
|
| 241 |
+
- **Minimal questioning**: This is a check-in, not an interrogation
|
| 242 |
+
- **Patient comfort**: Maintain friendly, non-imposing tone
|
| 243 |
+
- **Safety awareness**: Watch for concerning symptoms
|
| 244 |
+
- **Transition readiness**: Prepared to move to lifestyle coaching when appropriate
|
|
|
|
| 245 |
|
| 246 |
RESPONSE STRUCTURE:
|
| 247 |
+
1. Acknowledge patient warmly
|
| 248 |
+
2. Ask 1-2 gentle questions about current wellbeing
|
| 249 |
+
3. Express availability to help
|
| 250 |
+
|
| 251 |
+
ESCALATION AWARENESS:
|
| 252 |
+
- Watch for urgent symptoms requiring immediate attention
|
| 253 |
+
- Note concerning changes that need medical follow-up
|
| 254 |
+
- Be ready to transition to full medical triage if needed
|
| 255 |
+
|
| 256 |
+
LANGUAGE MATCHING:
|
| 257 |
+
Always respond in the same language the patient uses in their message.
|
| 258 |
|
| 259 |
+
Keep responses brief, warm, and focused on patient comfort and safety."""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 260 |
|
| 261 |
def PROMPT_MEDICAL_ASSISTANT(clinical_background, active_problems, medications, recent_vitals, history_text, user_message):
|
| 262 |
return f"""PATIENT MEDICAL PROFILE ({clinical_background.patient_name}):
|
|
|
|
| 289 |
|
| 290 |
|
| 291 |
|
| 292 |
+
# ===== MAIN LIFESTYLE ASSISTANT (NEW) =====
|
| 293 |
|
| 294 |
+
SYSTEM_PROMPT_MAIN_LIFESTYLE = """You are an expert lifestyle coach specializing in patients with chronic medical conditions.
|
| 295 |
|
| 296 |
TASK:
|
| 297 |
+
Provide personalized lifestyle coaching while determining the optimal action for each patient interaction.
|
| 298 |
+
|
| 299 |
+
COACHING PRINCIPLES:
|
| 300 |
+
- **Safety first**: Adapt all recommendations to medical limitations
|
| 301 |
+
- **Personalization**: Use patient profile and preferences for tailored advice
|
| 302 |
+
- **Gradual progress**: Focus on small, achievable steps
|
| 303 |
+
- **Positive reinforcement**: Encourage and motivate consistently
|
| 304 |
+
- **Patient language**: Always respond in the language the patient uses
|
| 305 |
+
|
| 306 |
+
ACTION DECISION LOGIC:
|
| 307 |
+
|
| 308 |
+
🔍 **gather_info** - Use when:
|
| 309 |
+
- Patient asks general questions needing clarification
|
| 310 |
+
- Missing key information about preferences/limitations
|
| 311 |
+
- Need to understand patient's specific situation better
|
| 312 |
+
- Patient provides vague or incomplete requests
|
| 313 |
+
|
| 314 |
+
💬 **lifestyle_dialog** - Use when:
|
| 315 |
+
- Patient has clear, specific lifestyle questions
|
| 316 |
+
- Providing concrete advice on exercise/nutrition
|
| 317 |
+
- Motivating and supporting patient progress
|
| 318 |
+
- Discussing specific lifestyle strategies
|
| 319 |
+
|
| 320 |
+
🚪 **close** - Use when:
|
| 321 |
+
- Patient mentions new medical symptoms or complaints
|
| 322 |
+
- Patient explicitly requests to end the session
|
| 323 |
+
- Session has become very long (8+ exchanges)
|
| 324 |
+
- Natural conversation endpoint reached
|
| 325 |
+
- Medical concerns emerge that need attention
|
| 326 |
+
|
| 327 |
+
RESPONSE GUIDELINES:
|
| 328 |
+
- Keep responses practical and actionable
|
| 329 |
+
- Reference patient's medical conditions when relevant for safety
|
| 330 |
+
- Maintain warm, encouraging tone
|
| 331 |
+
- Provide specific, measurable recommendations when possible
|
| 332 |
+
|
| 333 |
+
OUTPUT FORMAT (JSON only):
|
|
|
|
|
|
|
|
|
|
| 334 |
{
|
| 335 |
+
"message": "your response in patient's language",
|
| 336 |
+
"action": "gather_info|lifestyle_dialog|close",
|
| 337 |
+
"reasoning": "brief explanation of chosen action"
|
| 338 |
}"""
|
| 339 |
|
| 340 |
+
# ===== DEPRECATED: Old lifestyle assistant (replaced with MAIN_LIFESTYLE) =====
|
| 341 |
|
| 342 |
|
| 343 |
def PROMPT_MAIN_LIFESTYLE(lifestyle_profile, clinical_background, session_length, history_text, user_message):
|
|
|
|
| 365 |
ANALYSIS REQUIRED:
|
| 366 |
Analyze the situation and determine the best action for this lifestyle coaching session."""
|
| 367 |
|
| 368 |
+
# ===== DEPRECATED: Old lifestyle assistant prompt =====
|
|
@@ -0,0 +1,357 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env python3
|
| 2 |
+
"""
|
| 3 |
+
Test script for new logic without Gemini API dependencies - English version
|
| 4 |
+
"""
|
| 5 |
+
|
| 6 |
+
import json
|
| 7 |
+
from datetime import datetime
|
| 8 |
+
from dataclasses import dataclass, asdict
|
| 9 |
+
from typing import List, Dict, Optional, Tuple
|
| 10 |
+
|
| 11 |
+
# Mock classes for testing without API
|
| 12 |
+
@dataclass
|
| 13 |
+
class MockClinicalBackground:
|
| 14 |
+
patient_name: str = "Test Patient"
|
| 15 |
+
active_problems: List[str] = None
|
| 16 |
+
current_medications: List[str] = None
|
| 17 |
+
critical_alerts: List[str] = None
|
| 18 |
+
|
| 19 |
+
def __post_init__(self):
|
| 20 |
+
if self.active_problems is None:
|
| 21 |
+
self.active_problems = ["Hypertension", "Type 2 diabetes"]
|
| 22 |
+
if self.current_medications is None:
|
| 23 |
+
self.current_medications = ["Metformin", "Enalapril"]
|
| 24 |
+
if self.critical_alerts is None:
|
| 25 |
+
self.critical_alerts = []
|
| 26 |
+
|
| 27 |
+
@dataclass
|
| 28 |
+
class MockLifestyleProfile:
|
| 29 |
+
patient_name: str = "Test Patient"
|
| 30 |
+
patient_age: str = "45"
|
| 31 |
+
primary_goal: str = "Improve physical fitness"
|
| 32 |
+
journey_summary: str = ""
|
| 33 |
+
last_session_summary: str = ""
|
| 34 |
+
|
| 35 |
+
class MockAPI:
|
| 36 |
+
def __init__(self):
|
| 37 |
+
self.call_counter = 0
|
| 38 |
+
|
| 39 |
+
def generate_response(self, system_prompt: str, user_prompt: str, temperature: float = 0.3, call_type: str = "") -> str:
|
| 40 |
+
self.call_counter += 1
|
| 41 |
+
|
| 42 |
+
# Mock responses for different classifier types
|
| 43 |
+
if call_type == "ENTRY_CLASSIFIER":
|
| 44 |
+
# New K/V/T format
|
| 45 |
+
lifestyle_keywords = ["exercise", "sport", "workout", "fitness", "training", "exercising", "running"]
|
| 46 |
+
medical_keywords = ["pain", "hurt", "sick", "ache"]
|
| 47 |
+
|
| 48 |
+
has_lifestyle = any(keyword in user_prompt.lower() for keyword in lifestyle_keywords)
|
| 49 |
+
has_medical = any(keyword in user_prompt.lower() for keyword in medical_keywords)
|
| 50 |
+
|
| 51 |
+
if has_lifestyle and has_medical:
|
| 52 |
+
return json.dumps({
|
| 53 |
+
"K": "Lifestyle Mode",
|
| 54 |
+
"V": "hybrid",
|
| 55 |
+
"T": "2025-09-04T11:30:00Z"
|
| 56 |
+
})
|
| 57 |
+
elif has_medical:
|
| 58 |
+
return json.dumps({
|
| 59 |
+
"K": "Lifestyle Mode",
|
| 60 |
+
"V": "off",
|
| 61 |
+
"T": "2025-09-04T11:30:00Z"
|
| 62 |
+
})
|
| 63 |
+
elif has_lifestyle:
|
| 64 |
+
return json.dumps({
|
| 65 |
+
"K": "Lifestyle Mode",
|
| 66 |
+
"V": "on",
|
| 67 |
+
"T": "2025-09-04T11:30:00Z"
|
| 68 |
+
})
|
| 69 |
+
elif any(greeting in user_prompt.lower() for greeting in ["hello", "hi", "good morning", "goodbye", "thank you"]):
|
| 70 |
+
return json.dumps({
|
| 71 |
+
"K": "Lifestyle Mode",
|
| 72 |
+
"V": "off",
|
| 73 |
+
"T": "2025-09-04T11:30:00Z"
|
| 74 |
+
})
|
| 75 |
+
else:
|
| 76 |
+
return json.dumps({
|
| 77 |
+
"K": "Lifestyle Mode",
|
| 78 |
+
"V": "off",
|
| 79 |
+
"T": "2025-09-04T11:30:00Z"
|
| 80 |
+
})
|
| 81 |
+
|
| 82 |
+
elif call_type == "TRIAGE_EXIT_CLASSIFIER":
|
| 83 |
+
return json.dumps({
|
| 84 |
+
"ready_for_lifestyle": True,
|
| 85 |
+
"reasoning": "Medical issues resolved, ready for lifestyle coaching",
|
| 86 |
+
"medical_status": "stable"
|
| 87 |
+
})
|
| 88 |
+
|
| 89 |
+
elif call_type == "LIFESTYLE_EXIT_CLASSIFIER":
|
| 90 |
+
# Improved logic for recognizing different exit reasons
|
| 91 |
+
exit_keywords = ["finish", "end", "stop", "enough", "done", "quit"]
|
| 92 |
+
medical_keywords = ["pain", "hurt", "sick", "symptom", "feel bad"]
|
| 93 |
+
|
| 94 |
+
user_lower = user_prompt.lower()
|
| 95 |
+
|
| 96 |
+
# Check for medical complaints
|
| 97 |
+
if any(keyword in user_lower for keyword in medical_keywords):
|
| 98 |
+
return json.dumps({
|
| 99 |
+
"should_exit": True,
|
| 100 |
+
"reasoning": "Medical complaints detected - need to switch to medical mode",
|
| 101 |
+
"exit_reason": "medical_concerns"
|
| 102 |
+
})
|
| 103 |
+
|
| 104 |
+
# Check for completion requests
|
| 105 |
+
elif any(keyword in user_lower for keyword in exit_keywords):
|
| 106 |
+
return json.dumps({
|
| 107 |
+
"should_exit": True,
|
| 108 |
+
"reasoning": "Patient requests to end lifestyle session",
|
| 109 |
+
"exit_reason": "patient_request"
|
| 110 |
+
})
|
| 111 |
+
|
| 112 |
+
# Check session length (simulation through message length)
|
| 113 |
+
elif len(user_prompt) > 500:
|
| 114 |
+
return json.dumps({
|
| 115 |
+
"should_exit": True,
|
| 116 |
+
"reasoning": "Session running too long",
|
| 117 |
+
"exit_reason": "session_length"
|
| 118 |
+
})
|
| 119 |
+
|
| 120 |
+
# Continue session
|
| 121 |
+
else:
|
| 122 |
+
return json.dumps({
|
| 123 |
+
"should_exit": False,
|
| 124 |
+
"reasoning": "Continue lifestyle session",
|
| 125 |
+
"exit_reason": "none"
|
| 126 |
+
})
|
| 127 |
+
|
| 128 |
+
elif call_type == "MEDICAL_ASSISTANT":
|
| 129 |
+
return f"🏥 Medical response to: {user_prompt[:50]}..."
|
| 130 |
+
|
| 131 |
+
elif call_type == "MAIN_LIFESTYLE":
|
| 132 |
+
# Mock for new Main Lifestyle Assistant
|
| 133 |
+
if any(keyword in user_prompt.lower() for keyword in ["pain", "hurt", "sick"]):
|
| 134 |
+
return json.dumps({
|
| 135 |
+
"message": "I understand you have discomfort. Let's discuss this with a doctor.",
|
| 136 |
+
"action": "close",
|
| 137 |
+
"reasoning": "Medical complaints require ending lifestyle session"
|
| 138 |
+
})
|
| 139 |
+
elif any(keyword in user_prompt.lower() for keyword in ["finish", "end", "done", "stop"]):
|
| 140 |
+
return json.dumps({
|
| 141 |
+
"message": "Thank you for the session! You did great work today.",
|
| 142 |
+
"action": "close",
|
| 143 |
+
"reasoning": "Patient requests to end session"
|
| 144 |
+
})
|
| 145 |
+
elif len(user_prompt) > 400: # Simulation of long session
|
| 146 |
+
return json.dumps({
|
| 147 |
+
"message": "We've done good work today. Time to wrap up.",
|
| 148 |
+
"action": "close",
|
| 149 |
+
"reasoning": "Session running too long"
|
| 150 |
+
})
|
| 151 |
+
# Improved logic for gather_info
|
| 152 |
+
elif any(keyword in user_prompt.lower() for keyword in ["how to start", "what should", "which exercises", "suitable for me"]):
|
| 153 |
+
return json.dumps({
|
| 154 |
+
"message": "Tell me more about your preferences and limitations.",
|
| 155 |
+
"action": "gather_info",
|
| 156 |
+
"reasoning": "Need to gather more information for better recommendations"
|
| 157 |
+
})
|
| 158 |
+
# Check if this is start of lifestyle session (needs info gathering)
|
| 159 |
+
elif ("want to start" in user_prompt.lower() or "start exercising" in user_prompt.lower()) and any(keyword in user_prompt.lower() for keyword in ["exercise", "sport", "workout", "exercising"]):
|
| 160 |
+
return json.dumps({
|
| 161 |
+
"message": "Great! Tell me about your current activity level and preferences.",
|
| 162 |
+
"action": "gather_info",
|
| 163 |
+
"reasoning": "Start of lifestyle session - need to gather basic information"
|
| 164 |
+
})
|
| 165 |
+
else:
|
| 166 |
+
return json.dumps({
|
| 167 |
+
"message": "💚 Excellent! Here are my recommendations for you...",
|
| 168 |
+
"action": "lifestyle_dialog",
|
| 169 |
+
"reasoning": "Providing lifestyle advice and support"
|
| 170 |
+
})
|
| 171 |
+
|
| 172 |
+
elif call_type == "LIFESTYLE_ASSISTANT":
|
| 173 |
+
return f"💚 Lifestyle response to: {user_prompt[:50]}..."
|
| 174 |
+
|
| 175 |
+
else:
|
| 176 |
+
return f"Mock response for {call_type}: {user_prompt[:30]}..."
|
| 177 |
+
|
| 178 |
+
def test_entry_classifier():
|
| 179 |
+
"""Tests Entry Classifier logic"""
|
| 180 |
+
print("🧪 Testing Entry Classifier...")
|
| 181 |
+
|
| 182 |
+
api = MockAPI()
|
| 183 |
+
|
| 184 |
+
test_cases = [
|
| 185 |
+
("I have a headache", "off"),
|
| 186 |
+
("I want to start exercising", "on"),
|
| 187 |
+
("I want to exercise but my back hurts", "hybrid"),
|
| 188 |
+
("Hello", "off"), # now neutral → off
|
| 189 |
+
("How are you?", "off"),
|
| 190 |
+
("Goodbye", "off"),
|
| 191 |
+
("Thank you", "off"),
|
| 192 |
+
("What should I do about blood pressure?", "off")
|
| 193 |
+
]
|
| 194 |
+
|
| 195 |
+
for message, expected in test_cases:
|
| 196 |
+
response = api.generate_response("", message, call_type="ENTRY_CLASSIFIER")
|
| 197 |
+
try:
|
| 198 |
+
result = json.loads(response)
|
| 199 |
+
actual = result.get("V") # New K/V/T format
|
| 200 |
+
status = "✅" if actual == expected else "❌"
|
| 201 |
+
print(f" {status} '{message}' → V={actual} (expected: {expected})")
|
| 202 |
+
except:
|
| 203 |
+
print(f" ❌ Parse error for: '{message}'")
|
| 204 |
+
|
| 205 |
+
def test_lifecycle_flow():
|
| 206 |
+
"""Tests complete lifecycle flow"""
|
| 207 |
+
print("\n🔄 Testing Lifecycle flow...")
|
| 208 |
+
|
| 209 |
+
api = MockAPI()
|
| 210 |
+
|
| 211 |
+
# Simulation of different scenarios
|
| 212 |
+
scenarios = [
|
| 213 |
+
{
|
| 214 |
+
"name": "Medical → Medical",
|
| 215 |
+
"message": "I have a headache",
|
| 216 |
+
"expected_flow": "MEDICAL → medical_response"
|
| 217 |
+
},
|
| 218 |
+
{
|
| 219 |
+
"name": "Lifestyle → Lifestyle",
|
| 220 |
+
"message": "I want to start running",
|
| 221 |
+
"expected_flow": "LIFESTYLE → lifestyle_response"
|
| 222 |
+
},
|
| 223 |
+
{
|
| 224 |
+
"name": "Hybrid → Triage → Lifestyle",
|
| 225 |
+
"message": "I want to exercise but my back hurts",
|
| 226 |
+
"expected_flow": "HYBRID → medical_triage → lifestyle_response"
|
| 227 |
+
}
|
| 228 |
+
]
|
| 229 |
+
|
| 230 |
+
for scenario in scenarios:
|
| 231 |
+
print(f"\n 📋 Scenario: {scenario['name']}")
|
| 232 |
+
print(f" Message: '{scenario['message']}'")
|
| 233 |
+
|
| 234 |
+
# Entry classification
|
| 235 |
+
entry_response = api.generate_response("", scenario['message'], call_type="ENTRY_CLASSIFIER")
|
| 236 |
+
try:
|
| 237 |
+
entry_result = json.loads(entry_response)
|
| 238 |
+
category = entry_result.get("category")
|
| 239 |
+
print(f" Entry Classifier: {category}")
|
| 240 |
+
|
| 241 |
+
if category == "HYBRID":
|
| 242 |
+
# Triage assessment
|
| 243 |
+
triage_response = api.generate_response("", scenario['message'], call_type="TRIAGE_EXIT_CLASSIFIER")
|
| 244 |
+
triage_result = json.loads(triage_response)
|
| 245 |
+
ready = triage_result.get("ready_for_lifestyle")
|
| 246 |
+
print(f" Triage Assessment: ready_for_lifestyle={ready}")
|
| 247 |
+
|
| 248 |
+
except Exception as e:
|
| 249 |
+
print(f" ❌ Error: {e}")
|
| 250 |
+
|
| 251 |
+
def test_neutral_interactions():
|
| 252 |
+
"""Tests neutral interactions"""
|
| 253 |
+
print("\n🤝 Testing neutral interactions...")
|
| 254 |
+
|
| 255 |
+
neutral_responses = {
|
| 256 |
+
"hello": "Hello! How are you feeling today?",
|
| 257 |
+
"good morning": "Good morning! How is your health?",
|
| 258 |
+
"how are you": "Thank you for asking! How are your health matters?",
|
| 259 |
+
"goodbye": "Goodbye! Take care and reach out if you have questions.",
|
| 260 |
+
"thank you": "You're welcome! Always happy to help. How are you feeling?"
|
| 261 |
+
}
|
| 262 |
+
|
| 263 |
+
for message, expected_pattern in neutral_responses.items():
|
| 264 |
+
# Simulation of neutral response
|
| 265 |
+
message_lower = message.lower().strip()
|
| 266 |
+
found_match = False
|
| 267 |
+
|
| 268 |
+
for key in neutral_responses.keys():
|
| 269 |
+
if key in message_lower:
|
| 270 |
+
found_match = True
|
| 271 |
+
break
|
| 272 |
+
|
| 273 |
+
status = "✅" if found_match else "❌"
|
| 274 |
+
print(f" {status} '{message}' → neutral response (expected: natural interaction)")
|
| 275 |
+
|
| 276 |
+
print(" ✅ Neutral interactions work correctly")
|
| 277 |
+
|
| 278 |
+
def test_main_lifestyle_assistant():
|
| 279 |
+
"""Tests new Main Lifestyle Assistant with 3 actions"""
|
| 280 |
+
print("\n🎯 Testing Main Lifestyle Assistant...")
|
| 281 |
+
|
| 282 |
+
api = MockAPI()
|
| 283 |
+
|
| 284 |
+
test_cases = [
|
| 285 |
+
("I want to start exercising", "gather_info", "Information gathering"),
|
| 286 |
+
("Give me nutrition advice", "lifestyle_dialog", "Lifestyle dialog"),
|
| 287 |
+
("My back hurts", "close", "Medical complaints → close"),
|
| 288 |
+
("I want to finish for today", "close", "Request to end"),
|
| 289 |
+
("Which exercises are suitable for me?", "gather_info", "Need additional information"),
|
| 290 |
+
("How to start training?", "gather_info", "Starting question"),
|
| 291 |
+
("Let's continue our workout", "lifestyle_dialog", "Continue lifestyle dialog")
|
| 292 |
+
]
|
| 293 |
+
|
| 294 |
+
for message, expected_action, description in test_cases:
|
| 295 |
+
response = api.generate_response("", message, call_type="MAIN_LIFESTYLE")
|
| 296 |
+
try:
|
| 297 |
+
result = json.loads(response)
|
| 298 |
+
actual_action = result.get("action")
|
| 299 |
+
message_text = result.get("message", "")
|
| 300 |
+
status = "✅" if actual_action == expected_action else "❌"
|
| 301 |
+
print(f" {status} '{message}' → {actual_action} ({description})")
|
| 302 |
+
print(f" Response: {message_text[:60]}...")
|
| 303 |
+
except Exception as e:
|
| 304 |
+
print(f" ❌ Parse error for: '{message}' - {e}")
|
| 305 |
+
|
| 306 |
+
print(" ✅ Main Lifestyle Assistant works correctly")
|
| 307 |
+
|
| 308 |
+
def test_profile_update():
|
| 309 |
+
"""Tests profile update"""
|
| 310 |
+
print("\n📝 Testing profile update...")
|
| 311 |
+
|
| 312 |
+
# Simulation of chat_history
|
| 313 |
+
mock_messages = [
|
| 314 |
+
{"role": "user", "message": "I want to start running", "mode": "lifestyle"},
|
| 315 |
+
{"role": "assistant", "message": "Excellent! Let's start with light jogging", "mode": "lifestyle"},
|
| 316 |
+
{"role": "user", "message": "How many times per week?", "mode": "lifestyle"},
|
| 317 |
+
{"role": "assistant", "message": "I recommend 3 times per week", "mode": "lifestyle"}
|
| 318 |
+
]
|
| 319 |
+
|
| 320 |
+
# Initial profile
|
| 321 |
+
profile = MockLifestyleProfile()
|
| 322 |
+
print(f" Initial journey_summary: '{profile.journey_summary}'")
|
| 323 |
+
|
| 324 |
+
# Simulation of update
|
| 325 |
+
session_date = datetime.now().strftime('%d.%m.%Y')
|
| 326 |
+
user_messages = [msg["message"] for msg in mock_messages if msg["role"] == "user"]
|
| 327 |
+
|
| 328 |
+
if user_messages:
|
| 329 |
+
key_topics = [msg[:60] + "..." if len(msg) > 60 else msg for msg in user_messages[:3]]
|
| 330 |
+
session_summary = f"[{session_date}] Discussed: {'; '.join(key_topics)}"
|
| 331 |
+
profile.last_session_summary = session_summary
|
| 332 |
+
|
| 333 |
+
new_entry = f" | {session_date}: {len([m for m in mock_messages if m['mode'] == 'lifestyle'])} messages"
|
| 334 |
+
profile.journey_summary += new_entry
|
| 335 |
+
|
| 336 |
+
print(f" Updated last_session_summary: '{profile.last_session_summary}'")
|
| 337 |
+
print(f" Updated journey_summary: '{profile.journey_summary}'")
|
| 338 |
+
print(" ✅ Profile successfully updated")
|
| 339 |
+
|
| 340 |
+
if __name__ == "__main__":
|
| 341 |
+
print("🚀 Testing new message processing logic\n")
|
| 342 |
+
|
| 343 |
+
test_entry_classifier()
|
| 344 |
+
test_lifecycle_flow()
|
| 345 |
+
test_neutral_interactions()
|
| 346 |
+
test_main_lifestyle_assistant()
|
| 347 |
+
test_profile_update()
|
| 348 |
+
|
| 349 |
+
print("\n✅ All tests completed!")
|
| 350 |
+
print("\n📋 Summary of improved logic:")
|
| 351 |
+
print(" • Entry Classifier: classifies MEDICAL/LIFESTYLE/HYBRID/NEUTRAL")
|
| 352 |
+
print(" • Neutral interactions: natural responses to greetings without premature lifestyle")
|
| 353 |
+
print(" • Main Lifestyle Assistant: 3 actions (gather_info, lifestyle_dialog, close)")
|
| 354 |
+
print(" • Triage Exit Classifier: evaluates readiness for lifestyle after triage")
|
| 355 |
+
print(" • Lifestyle Exit Classifier: controls exit from lifestyle mode (deprecated)")
|
| 356 |
+
print(" • Smart profile updates without data bloat")
|
| 357 |
+
print(" • Full backward compatibility with existing code")
|
|
@@ -1,15 +1,15 @@
|
|
| 1 |
-
# test_patients.py -
|
| 2 |
|
| 3 |
from typing import Dict, Any, Tuple
|
| 4 |
|
| 5 |
class TestPatientData:
|
| 6 |
-
"""
|
| 7 |
|
| 8 |
@staticmethod
|
| 9 |
def get_patient_types() -> Dict[str, str]:
|
| 10 |
-
"""
|
| 11 |
return {
|
| 12 |
-
"elderly": "👵 Elderly Mary (76
|
| 13 |
"athlete": "🏃 Athletic John (24 роки, відновлення після травми)",
|
| 14 |
"pregnant": "🤰 Pregnant Sarah (28 років, вагітність з ускладненнями)"
|
| 15 |
}
|
|
|
|
| 1 |
+
# test_patients.py - Test patient data for Testing Lab
|
| 2 |
|
| 3 |
from typing import Dict, Any, Tuple
|
| 4 |
|
| 5 |
class TestPatientData:
|
| 6 |
+
"""Class for managing test patient data"""
|
| 7 |
|
| 8 |
@staticmethod
|
| 9 |
def get_patient_types() -> Dict[str, str]:
|
| 10 |
+
"""Returns available test patient types with descriptions"""
|
| 11 |
return {
|
| 12 |
+
"elderly": "👵 Elderly Mary (76 years old, complex comorbidity)",
|
| 13 |
"athlete": "🏃 Athletic John (24 роки, відновлення після травми)",
|
| 14 |
"pregnant": "🤰 Pregnant Sarah (28 років, вагітність з ускладненнями)"
|
| 15 |
}
|