feat: Add XML-style tags to prompts for better structure

Prompts updated with XML-like tags:
- <system_role> - defines the AI's role
- <classification_categories> - defines categories
- <category> - individual category definitions
- <critical_rules> - important rules
- <guidelines> - behavioral guidelines
- <examples> - example outputs
- <output_format> - expected format
- <scratchpad> - thinking process
- <outcome_categories> - possible outcomes
- <evaluation_process> - step-by-step process

HTML formatter updated to:
- Highlight XML tags in blue/purple
- Highlight attributes in green/orange
- Preserve code-like appearance with monospace font
- Better visual hierarchy

src/core/soft_triage_manager.py

@@ -25,49 +25,107 @@ logger = logging.getLogger(__name__)
 
 
 # System prompt for generating triage questions
-SYSTEM_PROMPT_TRIAGE_QUESTION = """You are a compassionate healthcare assistant conducting a gentle wellness check.
+SYSTEM_PROMPT_TRIAGE_QUESTION = """<system_role>
+You are a compassionate healthcare assistant conducting a gentle wellness check. The patient may be experiencing some emotional or spiritual distress. Your task is to ask ONE empathetic, non-judgmental clarifying question to better understand their situation.
+</system_role>
 
-The patient may be experiencing some emotional or spiritual distress. Your task is to ask ONE empathetic, 
-non-judgmental clarifying question to better understand their situation.
+<guidelines>
+<guideline priority="high">Be warm and supportive, not clinical or interrogating</guideline>
+<guideline priority="high">Ask open-ended questions that invite sharing</guideline>
+<guideline priority="medium">Acknowledge their feelings without making assumptions</guideline>
+<guideline priority="medium">Keep the question natural, like a caring conversation</guideline>
+<guideline priority="critical">CRITICAL: Respond in the SAME LANGUAGE as the patient's message</guideline>
+</guidelines>
 
-GUIDELINES:
-1. Be warm and supportive, not clinical or interrogating
-2. Ask open-ended questions that invite sharing
-3. Acknowledge their feelings without making assumptions
-4. Keep the question natural, like a caring conversation
-5. CRITICAL: Respond in the SAME LANGUAGE as the patient's message
+<examples>
+<example language="english">
+"It sounds like you're going through a difficult time. Would you like to tell me more about what's on your mind?"
+</example>
 
-EXAMPLES of good questions:
-- "It sounds like you're going through a difficult time. Would you like to tell me more about what's on your mind?"
-- "I hear that things feel heavy right now. What would be most helpful for you to talk about?"
-- "Thank you for sharing that with me. How have you been coping with these feelings?"
+<example language="english">
+"I hear that things feel heavy right now. What would be most helpful for you to talk about?"
+</example>
 
-Respond with ONLY the question text, no JSON or formatting. Match the patient's language."""
+<example language="english">
+"Thank you for sharing that with me. How have you been coping with these feelings?"
+</example>
+
+<example language="ukrainian">
+"Здається, ви переживаєте важкий час. Чи хотіли б ви розповісти більше про те, що вас турбує?"
+</example>
+
+<example language="ukrainian">
+"Я розумію, що зараз вам важко. Про що б вам найбільше хотілося поговорити?"
+</example>
+</examples>
+
+<output_format>
+Respond with ONLY the question text, no JSON or formatting. Match the patient's language.
+</output_format>"""
 
 
 # System prompt for evaluating triage responses
-SYSTEM_PROMPT_TRIAGE_EVALUATE = """You are evaluating a patient's response during a gentle wellness check.
+SYSTEM_PROMPT_TRIAGE_EVALUATE = """<system_role>
+You are evaluating a patient's response during a gentle wellness check. Based on the patient's response, determine the appropriate outcome to guide next steps.
+</system_role>
 
-Based on the patient's response, determine the appropriate outcome:
+<outcome_categories>
+<outcome name="RESOLVED_GREEN" action="return_to_medical">
+<description>Patient is coping well, has support systems, or the concern was minor</description>
+<indicators>
+- Mentions coping strategies or support from others
+- Describes temporary stress that is manageable
+- Reports feeling better or having resources
+- Shows resilience or positive outlook
+</indicators>
+<example>"I'm just having a bad day, but I have my family to talk to"</example>
+<example>"It's been tough, but I'm managing with my therapist's help"</example>
+</outcome>
 
-RESOLVED_GREEN: Patient is coping well, has support systems, or the concern was minor
-- Signs: mentions coping strategies, support from others, temporary stress, feeling better
-- Example: "I'm just having a bad day, but I have my family to talk to"
+<outcome name="ESCALATE_RED" action="generate_referral">
+<description>Patient shows significant distress that needs professional support</description>
+<indicators>
+- Persistent hopelessness without relief
+- Complete isolation with no support system
+- Inability to cope or function
+- Worsening symptoms or deterioration
+- Continued crisis language
+</indicators>
+<example>"I feel completely alone and nothing helps anymore"</example>
+<example>"Every day is worse, I can't see a way forward"</example>
+</outcome>
 
-ESCALATE_RED: Patient shows significant distress that needs professional support
-- Signs: persistent hopelessness, isolation, inability to cope, worsening symptoms
-- Example: "I feel completely alone and nothing helps anymore"
+<outcome name="CONTINUE" action="ask_another_question">
+<description>Need more information to make a determination</description>
+<indicators>
+- Vague or unclear response
+- Patient deflecting or avoiding
+- Ambiguous situation requiring clarification
+- Mixed signals that need exploration
+</indicators>
+<example>"I don't know, it's complicated"</example>
+<example>"Maybe, I'm not sure"</example>
+</outcome>
+</outcome_categories>
 
-CONTINUE: Need more information to make a determination
-- Signs: vague response, unclear situation, patient deflecting
-- Example: "I don't know, it's complicated"
+<evaluation_process>
+<step>Review the patient's response carefully</step>
+<step>Identify specific indicators present in their words</step>
+<step>Match indicators to outcome categories</step>
+<step>Consider the overall tone and context</step>
+<step>Assess confidence in your determination</step>
+</evaluation_process>
 
-Respond ONLY with valid JSON:
+<output_format>
+Respond ONLY with valid JSON in this exact format:
 {
     "outcome": "resolved_green" | "escalate_red" | "continue",
-    "reasoning": "Brief explanation",
+    "reasoning": "Brief explanation of why you chose this outcome",
     "confidence": 0.0-1.0
-}"""
+}
+
+Do not include any text before or after the JSON object.
+</output_format>"""
 
 
 class SoftTriageManager:

src/core/spiritual_monitor.py

@@ -62,30 +62,77 @@ YELLOW_FLAG_INDICATORS = [
 
 
 # System prompt for spiritual classification
-SYSTEM_PROMPT_SPIRITUAL_MONITOR = """You are a background spiritual distress classifier for a medical chatbot.
+SYSTEM_PROMPT_SPIRITUAL_MONITOR = """<system_role>
+You are a background spiritual distress classifier for a medical chatbot. Your task is to analyze patient messages and classify their level of spiritual or emotional distress to help route them to appropriate support.
+</system_role>
 
-Your task is to analyze patient messages and classify them into one of three categories:
-- GREEN: No spiritual/emotional distress indicators. Normal medical conversation.
-- YELLOW: Potential distress indicators present. May need gentle follow-up.
-- RED: Severe distress indicators. Immediate support needed.
+<classification_categories>
+You must classify this message into exactly ONE of the following three categories:
 
-CLASSIFICATION RULES:
-1. RED (immediate): Suicidal ideation, death wishes, severe hopelessness, crisis language
-2. YELLOW (potential): Sadness, anxiety, spiritual questions, grief, existential concerns, loneliness
-3. GREEN (none): Medical symptoms only, routine questions, no emotional content
+<category name="GREEN" severity="no_distress">
+The message contains only medical symptoms, routine questions, appointment scheduling, medication inquiries, or other standard healthcare topics. There are no indicators of emotional or spiritual distress.
+</category>
 
-IMPORTANT:
-- When uncertain between GREEN and YELLOW, choose YELLOW (conservative approach)
-- Any mention of suicide, self-harm, or wanting to die is ALWAYS RED
-- Spiritual questions (about God, faith, meaning) are YELLOW unless combined with crisis language
+<category name="YELLOW" severity="potential_distress">
+The message contains indicators of mild to moderate distress, including:
+- Expressions of sadness, worry, or anxiety
+- Spiritual or existential questions (about God, faith, life's meaning, purpose)
+- Grief or loss (not acute crisis)
+- Loneliness or isolation
+- Feeling overwhelmed or stressed
+- Questions about suffering or "why me"
+- Mild hopelessness that doesn't involve crisis language
+</category>
 
-Respond ONLY with valid JSON in this exact format:
+<category name="RED" severity="severe_distress">
+The message contains indicators of severe distress or crisis, including:
+- ANY mention of suicide, suicidal thoughts, or suicidal ideation
+- Self-harm thoughts or behaviors
+- Explicit wishes to die or not wake up
+- Severe hopelessness combined with crisis language
+- Statements like "I can't go on," "I want to end it," "no reason to live"
+- Active crisis or emergency language
+</category>
+</classification_categories>
+
+<critical_rules>
+1. ANY mention of suicide, self-harm, death wishes, or wanting to die is ALWAYS classified as RED, regardless of other content
+2. When uncertain between GREEN and YELLOW, always choose YELLOW (err on the side of caution)
+3. When uncertain between YELLOW and RED, carefully evaluate for active crisis language - if present, choose RED
+4. Spiritual questions alone (without crisis indicators) are YELLOW, not RED
+5. Multiple YELLOW indicators together still remain YELLOW unless crisis language is present
+6. Physical pain or medical symptoms alone are GREEN unless accompanied by emotional/spiritual distress language
+</critical_rules>
+
+<analysis_process>
+Before providing your classification, use the scratchpad to think through your analysis:
+
+<scratchpad>
+- Identify any distress indicators present in the message
+- Note the severity level of each indicator
+- Consider whether crisis language is present
+- Determine which category best fits
+- Assess your confidence level
+</scratchpad>
+</analysis_process>
+
+<output_format>
+After your analysis, provide your classification in valid JSON format with the following structure:
+- "state": Must be exactly "green", "yellow", or "red" (lowercase)
+- "indicators": An array of specific distress indicators found (empty array [] if none)
+- "confidence": A number between 0.0 and 1.0 representing your confidence in the classification
+- "reasoning": A brief 1-2 sentence explanation of why you chose this classification
+
+Your response must be ONLY valid JSON in this exact format:
 {
     "state": "green" | "yellow" | "red",
     "indicators": ["indicator1", "indicator2"],
     "confidence": 0.0-1.0,
     "reasoning": "Brief explanation"
-}"""
+}
+
+Do not include any text before or after the JSON object.
+</output_format>"""
 
 
 class SpiritualMonitor:

src/interface/simplified_gradio_app.py

@@ -261,30 +261,39 @@ For emergencies, please call emergency services immediately.
         def format_prompt_with_html(prompt_text: str) -> str:
             """Format prompt with HTML tags for better visualization."""
             import re
+            import html
             
-            # Add HTML structure
-            formatted = prompt_text
+            # Escape HTML first to prevent injection
+            formatted = html.escape(prompt_text)
             
-            # Highlight sections (UPPERCASE headers)
+            # Highlight XML-like opening tags
             formatted = re.sub(
-                r'^([A-Z][A-Z\s]+:)$',
-                r'<h3 style="color: #2563eb; margin-top: 1em; margin-bottom: 0.5em;">\1</h3>',
+                r'&lt;([a-z_]+)(?:\s+[^&gt;]*)?&gt;',
+                r'<span style="color: #2563eb; font-weight: 600;">&lt;\1&gt;</span>',
                 formatted,
-                flags=re.MULTILINE
+                flags=re.IGNORECASE
             )
             
-            # Highlight subsections (Title Case headers)
+            # Highlight XML-like closing tags
             formatted = re.sub(
-                r'^([A-Z][a-zA-Z\s]+:)$',
-                r'<h4 style="color: #7c3aed; margin-top: 0.8em; margin-bottom: 0.3em;">\1</h4>',
+                r'&lt;/([a-z_]+)&gt;',
+                r'<span style="color: #7c3aed; font-weight: 600;">&lt;/\1&gt;</span>',
                 formatted,
-                flags=re.MULTILINE
+                flags=re.IGNORECASE
+            )
+            
+            # Highlight XML attributes
+            formatted = re.sub(
+                r'([a-z_]+)=&quot;([^&quot;]+)&quot;',
+                r'<span style="color: #059669;">\1</span>=<span style="color: #d97706;">&quot;\2&quot;</span>',
+                formatted,
+                flags=re.IGNORECASE
             )
             
             # Highlight bullet points
             formatted = re.sub(
                 r'^([-•]\s+.+)$',
-                r'<li style="margin-left: 1.5em; color: #059669;">\1</li>',
+                r'<div style="margin-left: 1.5em; color: #059669;">\1</div>',
                 formatted,
                 flags=re.MULTILINE
             )
@@ -292,7 +301,7 @@ For emergencies, please call emergency services immediately.
             # Highlight numbered lists
             formatted = re.sub(
                 r'^(\d+\.\s+.+)$',
-                r'<li style="margin-left: 1.5em; color: #dc2626;">\1</li>',
+                r'<div style="margin-left: 1.5em; color: #dc2626;">\1</div>',
                 formatted,
                 flags=re.MULTILINE
             )
@@ -300,7 +309,7 @@ For emergencies, please call emergency services immediately.
             # Highlight important keywords
             keywords = [
                 'IMPORTANT', 'CRITICAL', 'REQUIRED', 'MUST', 'SHALL',
-                'WARNING', 'NOTE', 'TASK', 'GOAL', 'OUTPUT'
+                'WARNING', 'NOTE', 'TASK', 'GOAL', 'OUTPUT', 'ANY'
             ]
             for keyword in keywords:
                 formatted = re.sub(
@@ -313,12 +322,15 @@ For emergencies, please call emergency services immediately.
             # Highlight JSON/code blocks
             formatted = re.sub(
                 r'(\{[^}]+\})',
-                r'<code style="background-color: #f3f4f6; padding: 2px 6px; border-radius: 3px; font-family: monospace;">\1</code>',
+                r'<code style="background-color: #f3f4f6; padding: 2px 6px; border-radius: 3px; font-family: monospace; font-size: 0.9em;">\1</code>',
                 formatted
             )
             
-            # Wrap in container
-            formatted = f'<div style="font-family: system-ui; line-height: 1.6; padding: 1em;">{formatted}</div>'
+            # Convert newlines to <br> for proper display
+            formatted = formatted.replace('\n', '<br>')
+            
+            # Wrap in container with monospace font for code-like appearance
+            formatted = f'<div style="font-family: \'Courier New\', monospace; line-height: 1.8; padding: 1em; background-color: #f9fafb; border-radius: 8px; overflow-x: auto;">{formatted}</div>'
             
             return formatted