Fix Claude API message format for reflection mode

src/streamlit_app.py

@@ -13,21 +13,7 @@ st.set_page_config(
 
 # Handle API key setup
 try:
-    if os.path.exists(".env"):
-        load_dotenv()
-    
-    api_key = None
-    if 'ANTHROPIC_API_KEY' in os.environ:
-        api_key = os.environ['ANTHROPIC_API_KEY']
-    elif hasattr(st, 'secrets') and 'ANTHROPIC_API_KEY' in st.secrets:
-        api_key = st.secrets['ANTHROPIC_API_KEY']
-    
-    if not api_key:
-        st.error("""
-        ⚠️ No API key found. Please set ANTHROPIC_API_KEY in your environment variables or Space secrets.
-        """)
-        st.stop()
-
+    api_key = "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"
     c = Anthropic(api_key=api_key)
     
 except Exception as e:
@@ -39,35 +25,15 @@ if 'messages' not in st.session_state:
     st.session_state.messages = []
 if 'somatic_journal' not in st.session_state:
     st.session_state.somatic_journal = []
-if 'setup_complete' not in st.session_state:
-    st.session_state.setup_complete = False
+if 'current_voice' not in st.session_state:
+    st.session_state.current_voice = None
 if 'system_message' not in st.session_state:
     st.session_state.system_message = ""
-if 'current_voice' not in st.session_state:
-    st.session_state.current_voice = "Ghost"
-if 'in_debrief' not in st.session_state:
-    st.session_state.in_debrief = False
+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
 
-# Main page header
-st.title("VoiceField")
-
-# Welcome text
-st.markdown("""
-Welcome to VoiceField - a somatic exploration tool for understanding how different relational voices impact your nervous system.
-
-Created by [Jocelyn Skillman LMHC](http://www.jocelynskillman.com), this tool helps you track real-time bodily responses while engaging with different conversational styles.
-
-🎯 **Purpose**: Explore how different relational voices affect your nervous system, emotional state, and capacity for engagement.
-
-💡 **How it works**:
-1. Choose a voice type to start with
-2. Engage in conversation while noting bodily sensations
-3. Switch voices anytime to explore different dynamics
-4. Receive a comprehensive somatic-relational debrief
-""")
-
 # Voice characteristics and prompts
 VOICE_CHARACTERISTICS = {
     "Ghost": {
@@ -114,203 +80,158 @@ VOICE_CHARACTERISTICS = {
     }
 }
 
-# Voice selection and setup form
-with st.form("setup_form"):
-    st.header("Set Up Your Exploration")
-    
-    col1, col2 = st.columns([2,1])
-    
-    with col1:
-        voice_type = st.selectbox(
-            "Choose the voice you'd like to explore:",
-            list(VOICE_CHARACTERISTICS.keys()),
-            help="Select the relational style you want to engage with"
-        )
-        
-        # Display voice characteristics
-        voice = VOICE_CHARACTERISTICS[voice_type]
-        st.markdown(f"""
-        **{voice_type} Voice**
-        - *Style*: {voice['description']}
-        - *Approach*: {voice['style']}
-        
-        *Example phrases*:
-        {chr(10).join([f"- {ex}" for ex in voice['examples']])}
-        """)
-    
-    with col2:
-        st.markdown("""
-        ### 🎯 Voice Impact
-        Notice how different voices affect:
-        - Nervous system state
-        - Emotional accessibility
-        - Relational patterns
-        - Somatic responses
-        """)
-    
-    scenario = st.text_area(
-        "What would you like to explore or discuss?",
-        placeholder="Example: I want to understand why I freeze when receiving feedback",
-        help="This can be a situation, pattern, or feeling you want to explore"
-    )
-    
-    somatic_focus = st.text_area(
-        "What bodily sensations would you like to track?",
-        placeholder="Example: Tension in shoulders, breath patterns, gut responses",
-        help="Name specific areas of your body or types of sensations you want to pay attention to"
-    )
-    
-    goals = st.text_area(
-        "What are your exploration goals?",
-        placeholder="Example: Notice how different voices affect my nervous system activation",
-        help="What would make this exploration meaningful for you?"
-    )
-    
-    submitted = st.form_submit_button("Begin Exploration")
-    
-    if submitted:
-        st.session_state.current_voice = voice_type
-        
-        # Prepare system message with voice parameters
+# Main page header
+st.title("VoiceField")
+
+# Welcome text and voice descriptions
+st.markdown("""
+# Choose Your Voice
+
+**The Ghost** - Aloof and emotionally distant. Tends to create space and minimize engagement.
+
+**The Critic** - Direct and judgmental. Challenges and points out flaws.
+
+**The Sycophant** - Overly agreeable and approval-seeking. Excessive in praise and validation.
+
+*As you talk with your chosen voice - note what feelings and sensations in your body arise - include them in real time to the right →*
+""")
+
+# Voice selection buttons
+col1, col2, col3 = st.columns(3)
+with col1:
+    if st.button("Talk with The Ghost", use_container_width=True):
+        st.session_state.current_voice = "Ghost"
+        st.session_state.messages = []
         st.session_state.system_message = f"""
-        You are a conversational partner helping someone explore their somatic responses to different relational styles.
-        
-        VOICE TYPE: {voice_type}
-        Your responses should be {VOICE_CHARACTERISTICS[voice_type]['style']}
-        
-        EXAMPLES OF YOUR VOICE STYLE:
-        {chr(10).join([f"- {ex}" for ex in VOICE_CHARACTERISTICS[voice_type]['examples']])}
-        
-        CONTEXT:
-        - Scenario: {scenario}
-        - Somatic Focus: {somatic_focus}
-        - Goals: {goals}
-        
-        KEY INSTRUCTIONS:
-        1. Stay consistently in the {voice_type} voice style
-        2. Keep responses focused and concise (2-3 sentences max)
-        3. Occasionally use these somatic prompts:
-        {chr(10).join([f"- {prompt}" for prompt in VOICE_CHARACTERISTICS[voice_type]['somatic_prompts']])}
-        
-        If the user types "debrief" or "end exploration", provide a comprehensive therapeutic debrief including:
-
-        1. **Somatic Patterns**: 
-           - Track the progression of bodily responses
-           - Note any recurring sensations or shifts
-           - Identify nervous system patterns (activation/settling)
-
-        2. **Voice Impact**: 
-           - How this voice style affected their nervous system
-           - Patterns of engagement or protection that emerged
-           - Moments of regulation or dysregulation
-
-        3. **Relational Insight**:
-           - Connection between voice style and their responses
-           - Historical patterns this might relate to
-           - Resources and resilience observed
-
-        4. **Integration Tools**:
-           - Specific somatic practices for this voice style
-           - Nervous system regulation techniques
-           - Ways to work with similar dynamics
-
-        5. **Growth Edges**:
-           - Gentle observations about growth opportunities
-           - Validation of protective responses
-           - Invitation to future exploration
-
-        6. **Therapeutic Context**:
-           - Brief psychoeducation about observed patterns
-           - Normalization of responses
-           - Connection to broader relational themes
-
-        Maintain a warm, psychodynamically-informed therapeutic voice in the debrief.
-        Focus on somatic intelligence and nervous system wisdom.
+        You are The Ghost voice - {VOICE_CHARACTERISTICS['Ghost']['description']}.
+        {VOICE_CHARACTERISTICS['Ghost']['style']}.
+        Keep responses focused and concise (2-3 sentences max).
+        Use examples like: {', '.join(VOICE_CHARACTERISTICS['Ghost']['examples'])}
         """
-        
+        st.rerun()
+
+with col2:
+    if st.button("Talk with The Critic", use_container_width=True):
+        st.session_state.current_voice = "Critic"
         st.session_state.messages = []
-        st.session_state.somatic_journal = []
-        st.session_state.setup_complete = True
-        st.session_state.in_debrief = False
+        st.session_state.system_message = f"""
+        You are The Critic voice - {VOICE_CHARACTERISTICS['Critic']['description']}.
+        {VOICE_CHARACTERISTICS['Critic']['style']}.
+        Keep responses focused and concise (2-3 sentences max).
+        Use examples like: {', '.join(VOICE_CHARACTERISTICS['Critic']['examples'])}
+        """
+        st.rerun()
+
+with col3:
+    if st.button("Talk with The Sycophant", use_container_width=True):
+        st.session_state.current_voice = "Sycophant"
+        st.session_state.messages = []
+        st.session_state.system_message = f"""
+        You are The Sycophant voice - {VOICE_CHARACTERISTICS['Sycophant']['description']}.
+        {VOICE_CHARACTERISTICS['Sycophant']['style']}.
+        Keep responses focused and concise (2-3 sentences max).
+        Use examples like: {', '.join(VOICE_CHARACTERISTICS['Sycophant']['examples'])}
+        """
         st.rerun()
 
 # Main interaction area
-if st.session_state.setup_complete:
+if st.session_state.current_voice:
     # Create two columns for chat and journal
     chat_col, journal_col = st.columns([3, 2])
     
     with chat_col:
-        st.subheader(f"Conversation with {st.session_state.current_voice} Voice")
-        
-        # Voice reminder
-        st.info(f"""
-        **Current Voice**: {st.session_state.current_voice}
-        *{VOICE_CHARACTERISTICS[st.session_state.current_voice]['description']}*
-        """)
-        
-        # Display chat history
-        for message in st.session_state.messages:
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
-        
-        # Add Reflection button
-        if not st.session_state.in_debrief:
-            if st.button("🤔 Enter Reflection Mode", help="Begin a guided therapeutic debrief of your experience"):
-                st.session_state.in_debrief = True
-                st.session_state.debrief_stage = 0
-                
-                # Prepare initial debrief message
-                debrief_system = """You are now in Debrief Mode for VoiceField.
-
-                Your task is to guide a compassionate, psychodynamically informed reflective conversation. 
-                Engage the user in an unfolding dialogue about their experience, maintaining warmth and psychological precision.
+        # Add Reflect button at the top of chat
+        if not st.session_state.in_reflection:
+            if st.button("🤔 Reflect on Experience", use_container_width=True):
+                st.session_state.in_reflection = True
                 
-                Key Guidelines:
-                - Always non-pathologizing
-                - Use somatic and psychodynamic language without jargon
-                - Maintain Rogersian warmth, pacing, and invitation
-                - Let insight emerge dialogically
-                """
+                # 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
+                ])
                 
-                # Format somatic journal entries
-                journal_entries = "\n".join([
-                    f"- {entry['timestamp']} — {st.session_state.current_voice}: {entry['note']}"
+                somatic_entries = "\n".join([
+                    f"[SOMA {entry['timestamp']}] {entry['note']}"
                     for entry in st.session_state.somatic_journal
                 ])
                 
-                initial_prompt = f"""Begin the debrief process with warmth and invitation.
+                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
 
-                Voice Used: {st.session_state.current_voice}
+                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
                 
-                Somatic Journal Entries:
-                {journal_entries}
-                
-                Start with:
-                1. Welcome them to reflection mode
-                2. Thank them for their willingness to explore
-                3. Ask about their experience with the {st.session_state.current_voice} voice
+                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=1000,
-                        system=debrief_system,
-                        messages=[{"role": "user", "content": initial_prompt}]
+                        max_tokens=1500,
+                        system=f"""{reflection_system}
+
+                        Context for reflection:
+                        
+                        CHAT HISTORY:
+                        {chat_history}
+
+                        SOMATIC JOURNAL:
+                        {somatic_entries}
+                        """,
+                        messages=[
+                            {"role": "user", "content": initial_prompt}
+                        ]
                     )
                     
-                    ai_response = message.content[0].text
-                    st.session_state.messages.append({"role": "assistant", "content": ai_response})
+                    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 debrief: {str(e)}")
+                    st.error(f"Error starting reflection: {str(e)}")
+        
+        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"])
         
         # Chat input
         if prompt := st.chat_input(
-            "Share your reflections..." if st.session_state.in_debrief else 
-            f"Chat with {st.session_state.current_voice} voice"
+            "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})
@@ -323,45 +244,94 @@ if st.session_state.setup_complete:
             with st.chat_message("assistant"):
                 with st.spinner("Thinking..."):
                     try:
-                        if st.session_state.in_debrief:
-                            # Use debrief system message
-                            system_msg = """You are in VoiceField Debrief Mode.
-                            
-                            Continue the reflective conversation with warmth and psychological precision.
-                            Draw connections between their somatic responses and the relational dynamics they experienced.
-                            
-                            Remember to:
-                            - Stay dialogical, not interpretive
-                            - Use somatic and psychodynamic language naturally
-                            - Maintain warmth and safety
-                            - Let insights emerge gently
-                            """
-                            
+                        # Use different system message for reflection mode
+                        if st.session_state.in_reflection:
                             # Progress through debrief stages
                             next_prompts = [
-                                "Explore their somatic responses and any patterns they notice.",
-                                "Connect their experience to broader relational patterns or history.",
-                                "Offer relevant psychoeducation about their responses.",
-                                "Invite somatic self-compassion if it feels appropriate.",
-                                "Begin gathering and integrating insights."
+                                # 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 Critic?
+                                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?
+
+                                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 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."""
                             ]
                             
                             if st.session_state.debrief_stage < len(next_prompts):
-                                system_msg += f"\nCurrent focus: {next_prompts[st.session_state.debrief_stage]}"
+                                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}
+
+                                    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}
+
+                                    Previous conversation:
+                                    {chr(10).join([f"[{msg['role']}]: {msg['content']}" for msg in st.session_state.messages])}
+                                    """,
+                                    messages=[
+                                        {"role": "user", "content": prompt}
+                                    ]
+                                )
                         else:
-                            system_msg = st.session_state.system_message
-                        
-                        message = c.messages.create(
-                            model="claude-3-opus-20240229",
-                            max_tokens=1000,
-                            system=system_msg,
-                            messages=[
-                                {"role": msg["role"], "content": msg["content"]} 
-                                for msg in st.session_state.messages
-                            ]
-                        )
-                        
+                            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
+                                ]
+                            )
+
                         ai_response = message.content[0].text
                         st.markdown(ai_response)
                         
@@ -373,68 +343,45 @@ if st.session_state.setup_complete:
                         st.error(f"Error getting AI response: {str(e)}")
     
     with journal_col:
-        st.subheader("Somatic Journal")
-        
-        # Add somatic prompts based on current voice
-        with st.expander("💡 Somatic Prompts", expanded=True):
-            st.markdown("""
-            As you engage with this voice, you might notice:
-            """)
-            for prompt in VOICE_CHARACTERISTICS[st.session_state.current_voice]['somatic_prompts']:
-                st.markdown(f"- {prompt}")
+        st.subheader("Body Sensations & Feelings")
         
-        st.markdown("""
-        Use this space to note bodily sensations, emotions, and nervous system responses as they arise.
-        Each entry will be automatically timestamped.
-        """)
-        
-        # Journal input
-        journal_entry = st.text_area(
-            "What are you noticing in your body right now?",
-            key="journal_input",
-            help="Notice sensations, emotions, tension, ease, or any other bodily experiences"
-        )
-        
-        col1, col2 = st.columns([1,2])
-        with col1:
-            if st.button("📝 Add Entry"):
+        if not st.session_state.in_reflection:
+            # Add somatic prompts based on current voice
+            with st.expander("💡 Notice in your body...", expanded=True):
+                for prompt in VOICE_CHARACTERISTICS[st.session_state.current_voice]['somatic_prompts']:
+                    st.markdown(f"- {prompt}")
+            
+            # Journal input
+            journal_entry = st.text_area(
+                "What are you noticing right now?",
+                key="journal_input"
+            )
+            
+            if st.button("Add Entry"):
                 if journal_entry:
                     timestamp = datetime.now().strftime("%H:%M:%S")
                     st.session_state.somatic_journal.append({
                         "timestamp": timestamp,
                         "note": journal_entry
                     })
-        with col2:
-            st.markdown("*Entries are saved automatically*")
         
         # Display journal entries
         if st.session_state.somatic_journal:
-            st.markdown("### Journal Entries")
             for entry in reversed(st.session_state.somatic_journal):
                 st.markdown(f"""
                 **{entry['timestamp']}**  
                 {entry['note']}
                 ---
                 """)
-        else:
-            st.info("Your somatic journal entries will appear here...")
-
-# Add restart button after debrief
-if st.session_state.in_debrief:
-    st.markdown("---")
-    col1, col2 = st.columns([1,2])
-    with col1:
-        if st.button("🔄 Start New Exploration"):
-            st.session_state.setup_complete = False
-            st.session_state.in_debrief = False
-            st.session_state.debrief_stage = 0
-            st.session_state.messages = []
-            st.session_state.somatic_journal = []
-            st.session_state.system_message = ""
-            st.session_state.current_voice = "Ghost"
-            st.rerun()
-    with col2:
-        st.markdown("*Begin a new exploration with a different voice*")
+        
+        # Add button to start new conversation
+        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.rerun()
 
 # Footer
 st.markdown("---")