Add enhanced features: voice switching, quick reactions, and detailed somatic tracking

src/streamlit_app.py

@@ -2,283 +2,347 @@ import os
 import streamlit as st
 from anthropic import Anthropic
 from dotenv import load_dotenv
+from datetime import datetime
 
 # Initialize page configuration first
 st.set_page_config(
-    page_title="Attachment Style Roleplay Simulator",
+    page_title="VoiceField",
     page_icon="🗣️",
     layout="wide"
 )
 
-# Handle API key
+# Handle API key setup
 try:
-    # Load from .env file for local development
     if os.path.exists(".env"):
         load_dotenv()
     
-    # Try multiple ways to get the API key
     api_key = None
-    
-    # Try direct environment variable
     if 'ANTHROPIC_API_KEY' in os.environ:
         api_key = os.environ['ANTHROPIC_API_KEY']
-    
-    # Try getenv
-    if not api_key:
-        api_key = os.getenv('ANTHROPIC_API_KEY')
-    
-    # Try Streamlit secrets
-    if not api_key and hasattr(st, 'secrets') and 'ANTHROPIC_API_KEY' in st.secrets:
+    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 make sure:
-        1. You've added the secret in Hugging Face Space settings
-        2. The secret is named exactly 'ANTHROPIC_API_KEY'
-        3. The Space has been rebuilt after adding the secret
+        ⚠️ No API key found. Please set ANTHROPIC_API_KEY in your environment variables or Space secrets.
         """)
         st.stop()
 
-    # Initialize Anthropic client
     c = Anthropic(api_key=api_key)
     
 except Exception as e:
-    st.error(f"""
-    ⚠️ Error initializing Anthropic client: {str(e)}
-    
-    Please check:
-    1. Your ANTHROPIC_API_KEY is set correctly
-    2. You're using a valid API key from Anthropic
-    """)
+    st.error(f"Error initializing Anthropic client: {str(e)}")
     st.stop()
 
-# Initialize session state variables
+# Initialize session state
 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 'system_message' not in st.session_state:
     st.session_state.system_message = ""
-if 'simulation_params' not in st.session_state:
-    st.session_state.simulation_params = {}
+if 'current_voice' not in st.session_state:
+    st.session_state.current_voice = "Nurturing"
 if 'in_debrief' not in st.session_state:
     st.session_state.in_debrief = False
+if 'voice_transitions' not in st.session_state:
+    st.session_state.voice_transitions = []
 
 # Main page header
-st.title("Attachment Style Roleplay Simulator")
+st.title("VoiceField")
 
-# Welcome text in main content area
+# Welcome text
 st.markdown("""
-Welcome!
-
-Created by [Jocelyn Skillman LMHC](http://www.jocelynskillman.com), this tool is designed to help you practice and understand relational dynamics through the lens of attachment theory.
-
-To learn more about attachment theory and therapeutic approaches, check out my newsletter: [@jocelynskillmanlmhc](https://jocelynskillmanlmhc.substack.com/)
-""")
-
-st.markdown("""
-This specific project — the Attachment Roleplay Simulator — invites users to practice emotionally difficult conversations, like setting boundaries, through roleplay tailored to their attachment style. It's designed to help people feel into relational patterns, develop language for self-advocacy, and build nervous system capacity for connection and repair.
-
-💡 Not sure about your attachment style?
-You can take the free quiz by Sarah Peyton — or simply choose the one that resonates when you read it:
-
-Anxious – "I often worry I've upset people or need to explain myself."
-
-Avoidant – "I'd rather deal with things alone and not depend on anyone."
+Welcome to VoiceField - a somatic exploration tool for understanding how different relational voices impact your nervous system.
 
-Disorganized – "I want closeness, but also feel unsafe or mistrusting."
+Created by [Jocelyn Skillman LMHC](http://www.jocelynskillman.com), this tool helps you track real-time bodily responses while engaging with different conversational styles.
 
-Secure – "I can express needs and handle conflict without losing connection."
+🎯 **Purpose**: Explore how different relational voices affect your nervous system, emotional state, and capacity for engagement.
 
-Choose what vibes — this is a practice space, not a test.
+💡 **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
 """)
 
-# Sidebar welcome and introduction
-with st.sidebar:
-    st.markdown("""
-Hi, I'm Jocelyn Skillman, LMHC — a clinical therapist, relational design ethicist, and creator of experimental tools that explore how AI can support (not replace) human care.
-
-Each tool in this collection is thoughtfully designed to:
-
-Extend therapeutic support between sessions
-
-Model emotional safety and relational depth
-
-Help clients and clinicians rehearse courage, regulation, and repair
-
-Stay grounded in trauma-informed, developmentally sensitive frameworks
-    """)
-    
-    st.divider()
-    
-    # Add the moved content to the bottom of sidebar
-    st.markdown("""
-I use Claude (by Anthropic) as the primary language model for these tools, chosen for its relational tone, contextual nuance, and responsiveness to emotionally complex prompts.
-
-As a practicing therapist, I imagine these resources being especially helpful to clinicians like myself — companions in the work of tending to others with insight, warmth, and care.
-
-To learn more about my work, visit:
-🌐 [jocelynskillman.com](http://www.jocelynskillman.com)
-📬 [Substack: Relational Code](https://jocelynskillmanlmhc.substack.com/)
-    """)
-
-# Simulation setup form
+# Voice selection and setup form
 with st.form("setup_form"):
-    st.header("Set Up Your Simulation")
+    st.header("Set Up Your Exploration")
     
-   tone_archetype = st.selectbox(
-    "Select the voice you'd like to engage with:",
-    ["Ghost", "Sycophant", "Critic"]
-)
+    voice_type = st.selectbox(
+        "Choose the voice you'd like to explore:",
+        ["Nurturing", "Challenging", "Neutral"],
+        help="""
+        🫂 Nurturing: Warm, empathetic, and supportive. Creates safety and attunement.
+        🔍 Challenging: Direct, provocative, growth-oriented. Pushes edges with respect.
+        ⚖️ Neutral: Objective, clear, and grounded. Holds space without strong emotional coloring.
+        """
+    )
     
     scenario = st.text_area(
-        "Describe the scenario you'd like to practice:",
-        placeholder="Example: Having a difficult conversation with my partner about feeling disconnected"
+        "What would you like to explore or discuss?",
+        placeholder="Example: I want to understand why I freeze when receiving feedback"
     )
     
-    tone = st.text_input(
-        "How would you like the AI to respond?",
-        placeholder="Example: Direct but understanding"
+    somatic_focus = st.text_area(
+        "What bodily sensations would you like to track?",
+        placeholder="Example: Tension in shoulders, breath patterns, gut responses"
     )
     
     goals = st.text_area(
-        "What are your practice goals?",
-        placeholder="Example: Practice expressing needs without becoming defensive"
+        "What are your exploration goals?",
+        placeholder="Example: Notice how different voices affect my nervous system activation"
     )
     
-    submitted = st.form_submit_button("Start Simulation")
+    submitted = st.form_submit_button("Begin Exploration")
     
     if submitted:
-        # Store simulation parameters for debrief
-        st.session_state.simulation_params = {
-            "tone_archetype": tone_archetype,
-            "scenario": scenario,
-            "tone": tone,
-            "goals": goals
+        st.session_state.current_voice = voice_type
+        st.session_state.voice_transitions = [(datetime.now().strftime("%H:%M:%S"), voice_type)]
+        
+        # Prepare system message with voice parameters
+        voice_characteristics = {
+            "Nurturing": {
+                "style": "warm, empathetic, and supportive",
+                "language": "gentle, validating, and attuned",
+                "pacing": "slow and patient",
+                "focus": "creating safety and connection"
+            },
+            "Challenging": {
+                "style": "direct, provocative, and growth-oriented",
+                "language": "clear, bold, and respectfully confrontational",
+                "pacing": "dynamic and engaging",
+                "focus": "expanding awareness and capacity"
+            },
+            "Neutral": {
+                "style": "objective, clear, and grounded",
+                "language": "balanced, factual, and measured",
+                "pacing": "steady and consistent",
+                "focus": "observing patterns and processes"
+            }
         }
         
-        # Prepare system message with simulation parameters
+        voice = voice_characteristics[voice_type]
         st.session_state.system_message = f"""
-        You are a conversational partner helping someone practice difficult conversations. 
-        The user has a {tone_archetype} attachment style and wants to practice: {scenario}
+        You are a conversational partner helping someone explore their somatic responses to different relational styles.
         
-        Respond in a {tone} manner. Help them achieve their goals: {goals}
+        VOICE TYPE: {voice_type}
+        Style: {voice['style']}
+        Language: {voice['language']}
+        Pacing: {voice['pacing']}
+        Focus: {voice['focus']}
         
-        Maintain a realistic conversation while providing gentle guidance when needed.
+        CONTEXT:
+        - Scenario: {scenario}
+        - Somatic Focus: {somatic_focus}
+        - Goals: {goals}
         
-        IMPORTANT: If the user types "debrief" or "end roleplay", switch to debrief mode using this format:
-
-        📝 DEBRIEF SUMMARY
-
-        **Emotional Arc**: [Analyze how the user shifted emotionally during the interaction, noting moments of courage, freeze, protest, backtracking, or people-pleasing]
-
-        **Goal Alignment**: [Reflect on how well they stayed aligned with their stated goal. Note small wins and struggles, affirm effort and awareness]
-
-        **Attachment Insight**: [Offer one insight based on their {tone_archetype} attachment style. Normalize the response as protective. Gently suggest a next growth step without shaming]
-
-        **Practical Tool**: [Offer a brief NVC or DBT tool they could try next time (e.g., needs check-in, opposite action, self-validation, distress tolerance)]
-
-        **Bold Reframe**: [Offer one bold, loving sentence they might say if they trusted their relational worth]
-
-        **Journaling Prompt**: [Suggest one reflective question or embodied inquiry for integration]
-
-        Speak warmly and with psychological precision, like an emotionally attuned therapist.
+        KEY INSTRUCTIONS:
+        1. Stay consistently in the {voice_type} voice style
+        2. Keep responses focused and concise (2-3 sentences max)
+        3. Occasionally invite somatic awareness with questions like:
+           - "What sensations are you noticing right now?"
+           - "How is your breath responding to this?"
+           - "Where do you feel this in your body?"
+        
+        If the user types "debrief" or "end exploration", provide a comprehensive therapeutic debrief including:
+
+        1. **Somatic Patterns**: 
+           - Track patterns in bodily responses across different voice types
+           - Note shifts in nervous system state (activation/settling)
+           - Identify somatic resources and challenges
+
+        2. **Nervous System Insights**:
+           - Connect responses to polyvagal theory (fight/flight/freeze/fawn)
+           - Explore window of tolerance and regulation capacity
+           - Highlight moments of co-regulation or dysregulation
+
+        3. **Relational Dynamics**:
+           - Analyze engagement patterns with each voice type
+           - Note attachment themes or protective patterns
+           - Identify relational resources and growth edges
+
+        4. **Integration Tools**:
+           - Offer specific somatic practices for regulation
+           - Suggest relational experiments for expanding capacity
+           - Provide nervous system education relevant to their experience
+
+        5. **Growth Edges**:
+           - Frame challenges within a developmental context
+           - Suggest gentle next steps for expanding capacity
+           - Normalize and validate protective responses
+
+        6. **Therapeutic Frame**:
+           - Provide relevant psychoeducation about their patterns
+           - Connect individual experience to broader human themes
+           - Offer compassionate reframes of challenges
+
+        Maintain a warm, psychodynamically-informed therapeutic voice in the debrief.
         """
         
-        # Reset message history and state
         st.session_state.messages = []
+        st.session_state.somatic_journal = []
         st.session_state.setup_complete = True
         st.session_state.in_debrief = False
         st.rerun()
 
-# Display status or chat interface
-if not st.session_state.setup_complete:
-    st.info("Please complete the simulation setup to begin.")
-else:
-    # Display chat history
-    for message in st.session_state.messages:
-        with st.chat_message(message["role"]):
-            st.markdown(message["content"])
-
-    # Optional somatic input logging
-st.subheader("🧘 Somatic Reflection (Optional)")
-somatic_input = st.text_input("What’s happening in your body right now?", key="soma_input")
-
-if st.button("Log Somatic Entry"):
-    if 'soma_log' not in st.session_state:
-        st.session_state.soma_log = []
-    st.session_state.soma_log.append({
-        "timestamp": datetime.now().isoformat(),
-        "tone": st.session_state.simulation_params.get('tone_archetype', st.session_state.simulation_params.get('tone_archetype', 'Unknown')),
-        "somatic_input": somatic_input
-    })
-    st.success("Somatic reflection logged.")
-
+# Main interaction area
+if st.session_state.setup_complete:
+    # Create two columns for chat and journal
+    chat_col, journal_col = st.columns([3, 2])
     
-    # Chat input
-    if prompt := st.chat_input("Type your message here (type 'debrief' or 'end roleplay' when ready to reflect)"):
-        # Check for debrief trigger
-        if prompt.lower() in ["debrief", "end roleplay"] and not st.session_state.in_debrief:
-            st.session_state.in_debrief = True
-            # Construct conversation summary
-            conversation_summary = "\n".join([
-                f"{msg['role']}: {msg['content']}" 
-                for msg in st.session_state.messages
-            ])
+    with chat_col:
+        # Voice switching interface
+        if not st.session_state.in_debrief:
+            new_voice = st.selectbox(
+                "Switch voice type:",
+                ["Nurturing", "Challenging", "Neutral"],
+                index=["Nurturing", "Challenging", "Neutral"].index(st.session_state.current_voice)
+            )
             
-            # Add debrief request to messages
-            prompt = f"""Please provide a therapeutic debrief for this roleplay session using:
-
-            Attachment Style: {st.session_state.simulation_params['tone_archetype']}
-            Scenario: {st.session_state.simulation_params['scenario']}
-            Goals: {st.session_state.simulation_params['goals']}
-            AI Tone: {st.session_state.simulation_params['tone']}
+            if new_voice != st.session_state.current_voice:
+                st.session_state.current_voice = new_voice
+                timestamp = datetime.now().strftime("%H:%M:%S")
+                st.session_state.voice_transitions.append((timestamp, new_voice))
+                st.rerun()
+        
+        st.subheader(f"Conversation with {st.session_state.current_voice} 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(f"Chat with {st.session_state.current_voice} voice (type 'debrief' when ready to reflect)"):
+            # Check for debrief trigger
+            if prompt.lower() in ["debrief", "end exploration"] and not st.session_state.in_debrief:
+                st.session_state.in_debrief = True
+                # Prepare debrief request
+                voice_history = "\n".join([
+                    f"[{time}] Switched to {voice} voice"
+                    for time, voice in st.session_state.voice_transitions
+                ])
+                
+                journal_summary = "\n".join([
+                    f"[{entry['timestamp']}] {entry['note']}"
+                    for entry in st.session_state.somatic_journal
+                ])
+                
+                prompt = f"""Please provide a therapeutic debrief for this somatic exploration:
+
+                Voice Transitions:
+                {voice_history}
+                
+                Somatic Journal Entries:
+                {journal_summary}
+                
+                Conversation History:
+                {chr(10).join([f"{msg['role']}: {msg['content']}" for msg in st.session_state.messages])}
+                """
             
-            Conversation Summary:
-            {conversation_summary}
-            """
+            # 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:
+                        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)
+                        
+                        # 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("Somatic Journal")
         
-        # Add user message to chat history
-        st.session_state.messages.append({"role": "user", "content": prompt})
+        # Quick reaction buttons
+        st.write("Quick Reactions:")
+        reaction_cols = st.columns(3)
+        
+        reactions = {
+            "Nervous System": ["🔴 Activated", "🟡 Alert", "🟢 Settled"],
+            "Emotional": ["😌 Safe", "😰 Tense", "😶 Numb"],
+            "Engagement": ["👥 Connected", "🛡️ Protected", "❌ Disconnected"]
+        }
+        
+        for i, (category, buttons) in enumerate(reactions.items()):
+            with reaction_cols[i]:
+                st.write(f"**{category}**")
+                for button in buttons:
+                    if st.button(button):
+                        timestamp = datetime.now().strftime("%H:%M:%S")
+                        st.session_state.somatic_journal.append({
+                            "timestamp": timestamp,
+                            "note": f"Quick reaction: {button}"
+                        })
+                        st.rerun()
+        
+        st.markdown("""
+        Use this space to note bodily sensations, emotions, and nervous system responses as they arise.
+        Each entry will be automatically timestamped.
+        
+        💡 Consider tracking:
+        - Physical sensations
+        - Breath patterns
+        - Muscle tension/release
+        - Energy levels
+        - Emotional shifts
+        - Relational impulses
+        """)
         
-        # Display user message
-        with st.chat_message("user"):
-            st.markdown(prompt)
+        # Journal input
+        journal_entry = st.text_area("What are you noticing in your body right now?", key="journal_input")
+        if st.button("Add Journal Entry"):
+            if journal_entry:
+                timestamp = datetime.now().strftime("%H:%M:%S")
+                st.session_state.somatic_journal.append({
+                    "timestamp": timestamp,
+                    "note": journal_entry
+                })
+                st.rerun()
         
-        # Get AI response
-        with st.chat_message("assistant"):
-            with st.spinner("Thinking..."):
-                try:
-                    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)
-                    
-                    # 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)}")
-                    st.error("Please try again or contact support if the issue persists.")
+        # Display journal entries
+        st.markdown("### Journal Timeline")
+        for entry in reversed(st.session_state.somatic_journal):
+            st.markdown(f"""
+            **{entry['timestamp']}**  
+            {entry['note']}
+            ---
+            """)
 
 # Add restart button after debrief
 if st.session_state.in_debrief:
-    if st.button("Start New Roleplay"):
+    if st.button("Start New Exploration"):
         st.session_state.setup_complete = False
         st.session_state.in_debrief = False
         st.session_state.messages = []
+        st.session_state.somatic_journal = []
+        st.session_state.voice_transitions = []
         st.session_state.system_message = ""
-        st.session_state.simulation_params = {}
+        st.session_state.current_voice = "Nurturing"
         st.rerun()
 
 # Footer