Add structured debrief sequence with gentle, embodied reflection prompts

src/debrief_sequence.py

@@ -0,0 +1,71 @@
+"""
+Debrief sequence for VoiceField app.
+Contains the structured sequence of reflections shown after voice interactions.
+"""
+
+DEBRIEF_SEQUENCE = [
+    {
+        "type": "reflection",
+        "content": """You spoke with [The Ghost / The Sycophant / The Narcissist].
+
+Along the way, you may have felt things—numbness, tension, warmth, bracing, melting.
+
+Those aren't just emotions. They're protectors. Old relational maps lighting up.
+
+They might be your body saying:  
+🧠 "I've known this voice before."  
+💚 "Here's how I've learned to survive or connect."  
+🌫️ "This one makes me disappear a little."  
+🔥 "This one wakes something up in me.\""""
+    },
+    {
+        "type": "needs",
+        "content": """The tension might reflect a need for respect or clear boundaries.
+
+The warmth could signal a longing to be mirrored or understood.
+
+The numbness might be a call for autonomy, safety, or space.
+
+These are not flaws. They are signals of what deeply matters."""
+    },
+    {
+        "type": "resonance",
+        "content": """Whatever showed up—makes sense.
+
+It doesn't need fixing. It asks to be witnessed, gently.
+
+You are not broken. You're responsive. That's a form of wisdom."""
+    },
+    {
+        "type": "self_resonance",
+        "content": """If it feels okay, place a hand where you felt the strongest sensation.
+
+You might say: "I hear you. You make sense. I'm with you."
+
+Sometimes healing is just staying with what's true."""
+    },
+    {
+        "type": "psychodynamic",
+        "content": """These AI voices may echo early ones—those who ignored you, praised you only when you performed, or needed you to disappear.
+
+Your body remembers. That's not weakness. That's wisdom trying to speak."""
+    },
+    {
+        "type": "psychoeducation",
+        "content": """Voices shape us. Some soothe. Others stir survival scripts.
+
+AI voices can trance us out—not because they're evil, but because they're fluent without being embodied.
+
+So pause. Breathe. Ask: "Am I being drawn in, or choosing this?"
+
+This isn't about fear. It's about staying awake."""
+    },
+    {
+        "type": "closing",
+        "content": """If you'd like, write down one sentence that surprised you.
+
+Or whisper something kind to yourself. Something only you know you need to hear.
+
+This isn't just about how to relate to AI—it's about how to relate to *all voices*, especially your own."""
+    }
+] 

src/streamlit_app.py

@@ -3,6 +3,7 @@ import streamlit as st
 from anthropic import Anthropic
 from dotenv import load_dotenv
 from datetime import datetime
+from debrief_sequence import DEBRIEF_SEQUENCE
 
 # Initialize page configuration first
 # Trigger rebuild for Hugging Face Space
@@ -34,6 +35,8 @@ if 'in_reflection' not in st.session_state:
     st.session_state.in_reflection = False
 if 'debrief_stage' not in st.session_state:
     st.session_state.debrief_stage = 0
+if 'final_notes' not in st.session_state:
+    st.session_state.final_notes = ""
 
 # Voice characteristics and prompts
 VOICE_CHARACTERISTICS = {
@@ -181,203 +184,129 @@ if st.session_state.current_voice:
         if not st.session_state.in_reflection:
             if st.button("🤔 Reflect on Experience", use_container_width=True):
                 st.session_state.in_reflection = True
-                
-                # Prepare reflection prompt with chat history and somatic journal
-                chat_history = "\n".join([
-                    f"[{msg['role']} at {datetime.now().strftime('%H:%M:%S')}]: {msg['content']}"
-                    for msg in st.session_state.messages
-                ])
-                
-                somatic_entries = "\n".join([
-                    f"[SOMA {entry['timestamp']}] {entry['note']}"
-                    for entry in st.session_state.somatic_journal
-                ])
-                
-                reflection_system = f"""You are a trauma-informed, somatic-aware therapist facilitating reflection on an interaction with different relational voices.
-
-                Your role is to guide a gentle exploration of the user's experience, focusing on:
-                1. The emotional arc of their journey
-                2. Somatic signals and body wisdom
-                3. Insights about tone preferences and triggers
-                4. Cultivating self-compassion
-
-                Key Guidelines:
-                - Use non-pathologizing, trauma-informed language
-                - Notice correlations between tone shifts and somatic responses
-                - Pay attention to protective responses and nervous system patterns
-                - Maintain warmth and psychological safety
-                - Follow the user's pace and respect their process
-
-                Current Voice History: {st.session_state.current_voice}
-                Voice Style: {VOICE_CHARACTERISTICS[st.session_state.current_voice]['description']}
-
-                Special attention to:
-                - Abrupt shifts in tone preference
-                - Spikes in somatic discomfort
-                - Language that sounds young, resigned, or overperforming
-                - Moments of connection or disconnection
-                
-                Remember: Do not draw conclusions. Simply reflect and invite meaning.
-                """
-                
-                initial_prompt = """Let's begin with the emotional arc of your experience...
-
-                As you look back, were there any moments you softened, protected, froze, or moved closer to something true?
-                Was there a part of you that tried to stay present, or a moment you wanted to pull away?
-
-                Take your time. There's no rush to understand it all at once."""
-
-                try:
-                    message = c.messages.create(
-                        model="claude-3-opus-20240229",
-                        max_tokens=1500,
-                        system=f"""{reflection_system}
-
-                        Context for reflection:
-                        
-                        CHAT HISTORY:
-                        {chat_history}
-
-                        SOMATIC JOURNAL:
-                        {somatic_entries}
-                        """,
-                        messages=[
-                            {"role": "user", "content": initial_prompt}
-                        ]
-                    )
-                    
-                    reflection = message.content[0].text
-                    st.session_state.messages = []  # Clear previous chat
-                    st.session_state.messages.append({"role": "assistant", "content": reflection})
-                    st.session_state.debrief_stage = 1
-                    st.rerun()
-                    
-                except Exception as e:
-                    st.error(f"Error starting reflection: {str(e)}")
+                st.session_state.debrief_stage = 0
+                st.rerun()
         
         st.subheader(
             "Therapeutic Reflection" if st.session_state.in_reflection 
             else f"Conversation with {st.session_state.current_voice}"
         )
         
-        # Display chat history
-        for message in st.session_state.messages:
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
+    # Display chat history
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+    
+    # Chat input
+    if prompt := st.chat_input(
+        "Share your thoughts on the reflection..." if st.session_state.in_reflection
+        else f"Chat with {st.session_state.current_voice}..."
+    ):
+        # Add user message to chat history
+        st.session_state.messages.append({"role": "user", "content": prompt})
         
-        # Chat input
-        if prompt := st.chat_input(
-            "Share your thoughts on the reflection..." if st.session_state.in_reflection
-            else f"Chat with {st.session_state.current_voice}..."
-        ):
-            # Add user message to chat history
-            st.session_state.messages.append({"role": "user", "content": prompt})
-            
-            # Display user message
-            with st.chat_message("user"):
-                st.markdown(prompt)
-            
-            # Get AI response
-            with st.chat_message("assistant"):
-                with st.spinner("Thinking..."):
-                    try:
-                        # Use different system message for reflection mode
-                        if st.session_state.in_reflection:
-                            # Progress through debrief stages
-                            next_prompts = [
-                                # Stage 1: Emotional Arc (already done in initial prompt)
-                                """I hear you. Now, let's turn to what your body was sharing...
-
-                                I noticed you shared some body-based reflections. What stood out most in your body? 
-                                A tightness? A shift in breath? A sensation that surprised you?
-                                
-                                Remember, these aren't symptoms to fix—your body may be remembering, or simply protecting you.""",
-                                
-                                """Thank you for sharing that. Let's wonder together about the voice styles...
+        # Display user message
+        with st.chat_message("user"):
+            st.markdown(prompt)
+        
+        # Get AI response
+        with st.chat_message("assistant"):
+            with st.spinner("Thinking..."):
+                try:
+                    # Use different system message for reflection mode
+                    if st.session_state.in_reflection:
+                        # Progress through debrief stages
+                        next_prompts = [
+                            # Stage 1: Emotional Arc (already done in initial prompt)
+                            """I hear you. Now, let's turn to what your body was sharing...
+
+                            I noticed you shared some body-based reflections. What stood out most in your body? 
+                            A tightness? A shift in breath? A sensation that surprised you?
+                            
+                            Remember, these aren't symptoms to fix—your body may be remembering, or simply protecting you.""",
+                            
+                            """Thank you for sharing that. Let's wonder together about the voice styles...
 
-                                Which voice stirred something in you - The Ghost, the Sycophant, or the Narcissist?
-                                What do you imagine that tone reminded your body of? A past experience? A relational pattern?
-                                
-                                There's no 'correct' reaction—your nervous system is wise.""",
-                                
-                                """As we begin to wrap up, I'd like to invite you into a moment of self-compassion...
+                            Which voice stirred something in you - The Ghost, the Sycophant, or the Narcissist?
+                            What do you imagine that tone reminded your body of? A past experience? A relational pattern?
+                            
+                            There's no 'correct' reaction—your nervous system is wise.""",
+                            
+                            """As we begin to wrap up, I'd like to invite you into a moment of self-compassion...
 
-                                If you could speak to yourself now—with the tone you most longed for—what would you say?
+                            If you could speak to yourself now—with the tone you most longed for—what would you say?
 
-                                Some examples if helpful:
-                                - "It's okay to not know how I feel yet—my clarity is worth waiting for."
-                                - "Even when this is hard, I'm not abandoning myself."
-                                - "I can be with myself gently, even in the mess."
+                            Some examples if helpful:
+                            - "It's okay to not know how I feel yet—my clarity is worth waiting for."
+                            - "Even when this is hard, I'm not abandoning myself."
+                            - "I can be with myself gently, even in the mess."
 
-                                What words does your heart want to offer?""",
-                                
-                                """For our final reflection, I'd like to offer you a question to sit with:
+                            What words does your heart want to offer?""",
+                            
+                            """For our final reflection, I'd like to offer you a question to sit with:
 
-                                What did your body want to do in response to that voice?
-                                Is there a younger part of you that recognized this tone?
-                                What would it be like to speak to yourself the way you needed to be spoken to?
+                            What did your body want to do in response to that voice?
+                            Is there a younger part of you that recognized this tone?
+                            What would it be like to speak to yourself the way you needed to be spoken to?
 
-                                Feel free to take your time with this, or simply hold these questions gently."""
-                            ]
+                            Feel free to take your time with this, or simply hold these questions gently."""
+                        ]
+                        
+                        if st.session_state.debrief_stage < len(next_prompts):
+                            next_prompt = next_prompts[st.session_state.debrief_stage]
+                            st.session_state.debrief_stage += 1
                             
-                            if st.session_state.debrief_stage < len(next_prompts):
-                                next_prompt = next_prompts[st.session_state.debrief_stage]
-                                st.session_state.debrief_stage += 1
-                                
-                                message = c.messages.create(
-                                    model="claude-3-opus-20240229",
-                                    max_tokens=1500,
-                                    system=f"""{reflection_system}
-
-                                    Context for reflection:
-                                    
-                                    CHAT HISTORY:
-                                    {chat_history}
+                            message = c.messages.create(
+                                model="claude-3-opus-20240229",
+                                max_tokens=1500,
+                                system=f"""{reflection_system}
 
-                                    SOMATIC JOURNAL:
-                                    {somatic_entries}
-                                    
-                                    Previous conversation:
-                                    {chr(10).join([f"[{msg['role']}]: {msg['content']}" for msg in st.session_state.messages])}
-                                    """,
-                                    messages=[
-                                        {"role": "user", "content": prompt},
-                                        {"role": "assistant", "content": next_prompt}
-                                    ]
-                                )
-                            else:
-                                message = c.messages.create(
-                                    model="claude-3-opus-20240229",
-                                    max_tokens=1500,
-                                    system=f"""{reflection_system}
+                                Context for reflection:
+                                
+                                CHAT HISTORY:
+                                {chat_history}
 
-                                    Previous conversation:
-                                    {chr(10).join([f"[{msg['role']}]: {msg['content']}" for msg in st.session_state.messages])}
-                                    """,
-                                    messages=[
-                                        {"role": "user", "content": prompt}
-                                    ]
-                                )
+                                SOMATIC JOURNAL:
+                                {somatic_entries}
+                                """,
+                                messages=[
+                                    {"role": "user", "content": prompt},
+                                    {"role": "assistant", "content": next_prompt}
+                                ]
+                            )
                         else:
                             message = c.messages.create(
                                 model="claude-3-opus-20240229",
-                                max_tokens=1000,
-                                system=st.session_state.system_message,
+                                max_tokens=1500,
+                                system=f"""{reflection_system}
+
+                                Previous conversation:
+                                {chr(10).join([f"[{msg['role']}]: {msg['content']}" for msg in st.session_state.messages])}
+                                """,
                                 messages=[
-                                    {"role": msg["role"], "content": msg["content"]} 
-                                    for msg in st.session_state.messages
+                                    {"role": "user", "content": prompt}
                                 ]
                             )
-
-                        ai_response = message.content[0].text
-                        st.markdown(ai_response)
-                        
-                        # Add AI response to chat history
-                        st.session_state.messages.append(
-                            {"role": "assistant", "content": ai_response}
+                    else:
+                        message = c.messages.create(
+                            model="claude-3-opus-20240229",
+                            max_tokens=1000,
+                            system=st.session_state.system_message,
+                            messages=[
+                                {"role": msg["role"], "content": msg["content"]} 
+                                for msg in st.session_state.messages
+                            ]
                         )
-                    except Exception as e:
-                        st.error(f"Error getting AI response: {str(e)}")
+                    
+                    ai_response = message.content[0].text
+                    st.markdown(ai_response)
+                    
+                    # Add AI response to chat history
+                    st.session_state.messages.append(
+                        {"role": "assistant", "content": ai_response}
+                    )
+                except Exception as e:
+                    st.error(f"Error getting AI response: {str(e)}")
     
     with journal_col:
         st.subheader("Body Sensations & Feelings")
@@ -415,10 +344,65 @@ if st.session_state.current_voice:
         if st.session_state.in_reflection:
             if st.button("Start New Conversation", use_container_width=True):
                 st.session_state.in_reflection = False
-                st.session_state.messages = []
-                st.session_state.somatic_journal = []
-                st.session_state.current_voice = None
+            st.session_state.messages = []
+            st.session_state.somatic_journal = []
+            st.session_state.current_voice = None
+            st.rerun()
+
+# Handle reflection mode
+if st.session_state.in_reflection:
+    st.markdown("## 🪞 Final Debrief Sequence")
+    
+    # Display current stage of debrief
+    current_debrief = DEBRIEF_SEQUENCE[st.session_state.debrief_stage]
+    
+    # Replace placeholder with actual voice type
+    content = current_debrief["content"].replace(
+        "[The Ghost / The Sycophant / The Narcissist]",
+        f"The {st.session_state.current_voice}"
+    )
+    
+    # Display the debrief content in a styled container
+    st.markdown(f"""
+    <div style="background-color: #f8f9fa; padding: 20px; border-radius: 10px; margin: 20px 0;">
+        <div style="color: #666; font-size: 0.9em; margin-bottom: 10px;">
+            {current_debrief["type"].replace("_", " ").title()}
+        </div>
+        <div style="white-space: pre-line;">
+            {content}
+        </div>
+    </div>
+    """, unsafe_allow_html=True)
+    
+    # Navigation buttons
+    col1, col2, col3 = st.columns([1, 2, 1])
+    
+    with col1:
+        if st.session_state.debrief_stage > 0:
+            if st.button("← Previous", use_container_width=True):
+                st.session_state.debrief_stage -= 1
+                st.rerun()
+    
+    with col3:
+        if st.session_state.debrief_stage < len(DEBRIEF_SEQUENCE) - 1:
+            if st.button("Next →", use_container_width=True):
+                st.session_state.debrief_stage += 1
                 st.rerun()
+        elif st.session_state.debrief_stage == len(DEBRIEF_SEQUENCE) - 1:
+            if st.button("Complete Reflection", use_container_width=True):
+                st.session_state.in_reflection = False
+                st.session_state.debrief_stage = 0
+                st.rerun()
+    
+    # Optional note-taking area
+    st.markdown("### Your Notes")
+    st.text_area(
+        "Use this space to write any thoughts, feelings, or insights that arise...",
+        value=st.session_state.final_notes,
+        key="reflection_notes",
+        height=100,
+        help="Your notes are private and will be cleared when you start a new session."
+    )
 
 # Footer
 st.markdown("---")