Add comprehensive debrief functionality

src/streamlit_app.py

@@ -60,6 +60,10 @@ 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 'in_debrief' not in st.session_state:
+    st.session_state.in_debrief = False
 
 # Main page header
 st.title("Attachment Style Roleplay Simulator")
@@ -146,6 +150,14 @@ with st.form("setup_form"):
     submitted = st.form_submit_button("Start Simulation")
     
     if submitted:
+        # Store simulation parameters for debrief
+        st.session_state.simulation_params = {
+            "attachment_style": attachment_style,
+            "scenario": scenario,
+            "tone": tone,
+            "goals": goals
+        }
+        
         # Prepare system message with simulation parameters
         st.session_state.system_message = f"""
         You are a conversational partner helping someone practice difficult conversations. 
@@ -154,11 +166,30 @@ with st.form("setup_form"):
         Respond in a {tone} manner. Help them achieve their goals: {goals}
         
         Maintain a realistic conversation while providing gentle guidance when needed.
+        
+        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 {attachment_style} 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.
         """
         
-        # Reset message history
+        # Reset message history and state
         st.session_state.messages = []
         st.session_state.setup_complete = True
+        st.session_state.in_debrief = False
         st.rerun()
 
 # Display status or chat interface
@@ -171,7 +202,28 @@ else:
             st.markdown(message["content"])
     
     # Chat input
-    if prompt := st.chat_input("Type your message here..."):
+    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
+            ])
+            
+            # Add debrief request to messages
+            prompt = f"""Please provide a therapeutic debrief for this roleplay session using:
+
+            Attachment Style: {st.session_state.simulation_params['attachment_style']}
+            Scenario: {st.session_state.simulation_params['scenario']}
+            Goals: {st.session_state.simulation_params['goals']}
+            AI Tone: {st.session_state.simulation_params['tone']}
+            
+            Conversation Summary:
+            {conversation_summary}
+            """
+        
         # Add user message to chat history
         st.session_state.messages.append({"role": "user", "content": prompt})
         
@@ -182,14 +234,6 @@ else:
         # Get AI response
         with st.chat_message("assistant"):
             with st.spinner("Thinking..."):
-                # Construct conversation history
-                messages = []
-                if st.session_state.system_message:
-                    messages.append({"role": "system", "content": st.session_state.system_message})
-                for msg in st.session_state.messages:
-                    messages.append({"role": msg["role"], "content": msg["content"]})
-                
-                # Get completion
                 try:
                     message = c.messages.create(
                         model="claude-3-opus-20240229",
@@ -212,6 +256,16 @@ else:
                     st.error(f"Error getting AI response: {str(e)}")
                     st.error("Please try again or contact support if the issue persists.")
 
+# Add restart button after debrief
+if st.session_state.in_debrief:
+    if st.button("Start New Roleplay"):
+        st.session_state.setup_complete = False
+        st.session_state.in_debrief = False
+        st.session_state.messages = []
+        st.session_state.system_message = ""
+        st.session_state.simulation_params = {}
+        st.rerun()
+
 # Footer
 st.markdown("---")
 st.markdown(