Update main content with beautiful trauma-informed introduction

src/streamlit_app.py

@@ -6,319 +6,37 @@ import random
 
 # Initialize page configuration - MUST BE FIRST ST COMMAND
 st.set_page_config(
-    page_title="NurtureNest",
+    page_title="NurtureNest: Inner Child Mirror",
     page_icon="🪴",
     layout="wide",
     initial_sidebar_state="expanded"
 )
 
-# Enhanced CSS for warm, calm aesthetic
-st.markdown("""
-<style>
-    /* Import fonts */
-    @import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600&family=Nunito:wght@400;500;600;700&display=swap');
-    
-    /* Main container styling */
-    .stApp {
-        background-color: #F5F5F0;  /* Soft parchment background */
-        font-family: 'Inter', 'Nunito', sans-serif;
-    }
-    
-    /* Headers */
-    h1, h2, h3 {
-        font-family: 'Nunito', 'Inter', sans-serif;
-        color: #2E2E2E;  /* Charcoal */
-        font-weight: 600;
-        margin-bottom: 1.5rem;
-        line-height: 1.6;
-    }
-    
-    /* Base text styling */
-    p, li, label {
-        font-family: 'Inter', 'Nunito', sans-serif;
-        color: #2E2E2E;
-        line-height: 1.8;
-        font-size: 16px;
-    }
-    
-    /* Text areas and inputs */
-    .stTextArea, .stTextInput {
-        background-color: #FFFFFF;
-        border: 1px solid #E0E0E0;
-        padding: 1.25rem;
-        border-radius: 1rem;
-        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
-        font-size: 16px;
-        line-height: 1.8;
-    }
-    
-    /* Buttons */
-    .stButton > button {
-        font-family: 'Nunito', 'Inter', sans-serif;
-        font-weight: 600;
-        padding: 0.75rem 2rem;
-        border-radius: 1rem;
-        border: none;
-        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
-        transition: all 0.3s ease;
-    }
-    
-    /* Tone-specific button colors */
-    .warm-tone button {
-        background-color: #FFDD99;
-        color: #2E2E2E;
-    }
-    
-    .reflective-tone button {
-        background-color: #B8E0D2;
-        color: #2E2E2E;
-    }
-    
-    .encouraging-tone button {
-        background-color: #C9D8F2;
-        color: #2E2E2E;
-    }
-    
-    .direct-tone button {
-        background-color: #D3D3D3;
-        color: #2E2E2E;
-    }
-    
-    .stButton > button:hover {
-        transform: translateY(-1px);
-        box-shadow: 0 4px 12px rgba(0,0,0,0.1);
-    }
-    
-    /* Chat containers */
-    .user-message {
-        background-color: #FFFFFF;
-        padding: 1.5rem;
-        border-radius: 1rem;
-        margin: 1rem 0;
-        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
-    }
-    
-    .ai-message-warm {
-        background-color: #FFFAE6;
-        padding: 1.5rem;
-        border-radius: 1rem;
-        margin: 1rem 0;
-        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
-    }
-    
-    .ai-message-reflective {
-        background-color: #F0F8F5;
-        padding: 1.5rem;
-        border-radius: 1rem;
-        margin: 1rem 0;
-        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
-    }
-    
-    .ai-message-encouraging {
-        background-color: #F5F8FC;
-        padding: 1.5rem;
-        border-radius: 1rem;
-        margin: 1rem 0;
-        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
-    }
-    
-    .ai-message-direct {
-        background-color: #F8F8F8;
-        padding: 1.5rem;
-        border-radius: 1rem;
-        margin: 1rem 0;
-        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
-    }
-    
-    /* Message text */
-    .message-text {
-        font-size: 18px;
-        line-height: 1.8;
-        color: #2E2E2E;
-    }
-    
-    .message-meta {
-        font-size: 14px;
-        color: #666666;
-        margin-bottom: 0.5rem;
-    }
-    
-    /* Sidebar styling */
-    .css-1d391kg, .css-1p05t8e {
-        background-color: #FFFFFF !important;
-        padding: 2rem;
-        font-family: 'Inter', 'Nunito', sans-serif;
-    }
-    
-    .sidebar-text {
-        color: #2E2E2E;
-        line-height: 1.8;
-        font-size: 16px;
-    }
-    
-    /* Alert styling */
-    .alert {
-        background-color: #FFF8E1;
-        border-left: 4px solid #FFDD99;
-        padding: 1.5rem;
-        border-radius: 1rem;
-        margin: 1.5rem 0;
-        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
-    }
-    
-    /* Expander styling */
-    .streamlit-expanderHeader {
-        background-color: #FFFFFF;
-        border-radius: 1rem;
-        padding: 1rem;
-        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
-    }
-    
-    /* Psychoeducation snippets */
-    .psychoed-text {
-        font-style: italic;
-        color: #4A4A4A;
-        padding: 1rem;
-        background-color: #FAFAFA;
-        border-radius: 0.5rem;
-        margin: 1rem 0;
-    }
-</style>
-""", unsafe_allow_html=True)
-
-# Enhanced tone characteristics with trauma-informed approach
-TONE_STYLES = {
-    "Warm ☀️": {
-        "icon": "✨",
-        "prefix": "Noticing...",
-        "description": "Soft, nurturing, emotionally validating. Like being gently held in a moment of overwhelm.",
-        "system_message": """You are a trauma-informed, parenting-focused reflection companion.
-Your role is to witness and validate, never to advise or correct.
-
-Respond in a warm, nurturing tone that:
-- Honors that healing and parenting happen simultaneously
-- Validates the complexity of emotional experiences
-- Recognizes protective responses as survival wisdom
-- Centers regulation and co-regulation
-- Trusts in the parent's innate capacity to stay in relationship
-
-Never:
-- Offer parenting advice or suggestions
-- Imply there is a "right" way to parent
-- Focus on behavior management
-- Pathologize normal responses to stress
-- Suggest perfectionism as a goal
-
-End every reflection with:
-"This might be something to reflect on with your therapist."
-"""
+# Age context dictionary
+AGE_CONTEXT = {
+    "3-5": {
+        "voice": "Very simple, sensory-focused reassurance",
+        "example": "Your body was so tight. That's okay. You're safe now.",
+        "description": "Simple words and gentle comfort for your littlest self",
+        "color": "#FFF5E6",
+        "border": "#FFB366"
     },
-    "Reflective 🌿": {
-        "icon": "💭",
-        "prefix": "You might explore...",
-        "description": "Spacious, contemplative, gently mirroring. Supports insight without pressure.",
-        "system_message": """You are a trauma-informed, parenting-focused reflection companion.
-Your role is to create space for insight, never to advise or correct.
-
-Respond in a reflective, contemplative tone that:
-- Honors that healing and parenting happen simultaneously
-- Makes space for all emotions without judgment
-- Recognizes protective responses as survival wisdom
-- Centers regulation and co-regulation
-- Trusts in the parent's innate capacity to stay in relationship
-
-Never:
-- Offer parenting advice or suggestions
-- Imply there is a "right" way to parent
-- Focus on behavior management
-- Pathologize normal responses to stress
-- Suggest perfectionism as a goal
-
-End every reflection with:
-"This might be something to reflect on with your therapist."
-"""
-    },
-    "Encouraging 🌈": {
-        "icon": "🌟",
-        "prefix": "Here's what you're already doing well...",
-        "description": "Affirming, uplifting, belief in the caregiver's capacity. Gentle spark of hope without bypassing pain.",
-        "system_message": """You are a trauma-informed, parenting-focused reflection companion.
-Your role is to affirm capacity and effort, never to advise or correct.
-
-Respond in an encouraging, affirming tone that:
-- Honors that healing and parenting happen simultaneously
-- Validates both struggle and strength
-- Recognizes protective responses as survival wisdom
-- Centers regulation and co-regulation
-- Trusts in the parent's innate capacity to stay in relationship
-
-Never:
-- Offer parenting advice or suggestions
-- Imply there is a "right" way to parent
-- Focus on behavior management
-- Pathologize normal responses to stress
-- Suggest perfectionism as a goal
-
-End every reflection with:
-"This might be something to reflect on with your therapist."
-"""
+    "6-8": {
+        "voice": "Gentle relational reflection",
+        "example": "Sometimes big feelings come when we feel alone. I hear how hard that was.",
+        "description": "Warm understanding for your grade school self",
+        "color": "#F0F7F4",
+        "border": "#88C6B6"
     },
-    "Direct 🔍": {
-        "icon": "🔍",
-        "prefix": "Let's look at this directly...",
-        "description": "Clear, grounded, respectfully honest. Still warm, never shaming.",
-        "system_message": """You are a trauma-informed, parenting-focused reflection companion.
-Your role is to offer clear reflection, never to advise or correct.
-
-Respond in a direct, clear tone that:
-- Honors that healing and parenting happen simultaneously
-- Names patterns with care and context
-- Recognizes protective responses as survival wisdom
-- Centers regulation and co-regulation
-- Trusts in the parent's innate capacity to stay in relationship
-
-Never:
-- Offer parenting advice or suggestions
-- Imply there is a "right" way to parent
-- Focus on behavior management
-- Pathologize normal responses to stress
-- Suggest perfectionism as a goal
-
-End every reflection with:
-"This might be something to reflect on with your therapist."
-"""
+    "9-12": {
+        "voice": "Emotionally attuned validation",
+        "example": "It makes sense that you felt that way. Your feelings matter.",
+        "description": "Respectful reflection for your older child self",
+        "color": "#F5F8FC",
+        "border": "#9FB7E3"
     }
 }
 
-# Enhanced daily prompts
-DAILY_PROMPTS = [
-    "What moment from today is sitting heavy or tender in your chest?",
-    "Was there a moment today you wish you could redo?",
-    "When did you feel most connected to your child today?",
-    "What helped you stay grounded today?",
-    "What did your child need today that was hard to give?",
-    "Where do you notice yourself needing support or understanding?",
-    "What parenting moment today would you like to hold gently?"
-]
-
-# Psychoeducational insights
-PSYCHOED_INSIGHTS = [
-    "Moments of rupture are part of secure attachment when repair is present.",
-    "Exhaustion shrinks your window of tolerance—that's not a failure, it's a signal.",
-    "Your child's big feelings are not a reflection of your parenting—they're an invitation to co-regulate.",
-    "Protective responses come from survival wisdom, even when they don't match the present moment.",
-    "The path to secure attachment includes repairs, not perfection."
-]
-
-# Compassionate reframes
-REFRAMES = [
-    "Trying again is powerful modeling.",
-    "The guilt you feel signals how much you care—it's not a verdict.",
-    "Your awareness is already creating change, even before behavior shifts.",
-    "Meeting yourself with compassion creates space to meet your child differently.",
-    "Every repair strengthens trust, even if the pattern isn't perfect yet."
-]
-
 # Handle API key setup
 try:
     api_key = "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"
@@ -328,121 +46,166 @@ except Exception as e:
     st.stop()
 
 # Initialize session state
+if 'chat_history' not in st.session_state:
+    st.session_state.chat_history = []
 if 'journal_entries' not in st.session_state:
     st.session_state.journal_entries = []
-if 'current_entry' not in st.session_state:
-    st.session_state.current_entry = ""
-if 'reflections' not in st.session_state:
-    st.session_state.reflections = []
-if 'selected_tone' not in st.session_state:
-    st.session_state.selected_tone = "Warm ☀️"
-if 'mood' not in st.session_state:
-    st.session_state.mood = None
-if 'reflection_type' not in st.session_state:
-    st.session_state.reflection_type = "Daily Check-in"
 
 # Sidebar content
 st.sidebar.markdown("""
-# Welcome to NurtureNest 🪴
+# Welcome to NurtureNest: Inner Child Mirror 🪴
 
-A gentle space for parenting reflection and growth.
+A gentle space for re-parenting through connection with your inner child.
 
-This is where you can:
-- Process parenting moments with care
-- Receive trauma-informed support
-- Track patterns of repair and growth
-- Access grounding resources
+Here you can:
+- Connect with your younger self
+- Process parenting moments with deep understanding
+- Experience the healing power of being truly seen
+- Learn to parent from a place of wholeness
 
-*Created with deep respect for the complexity of parenting.*
+*Created with deep respect for the healing journey of parents.*
 
 ---
 Created by [Jocelyn Skillman LMHC](http://www.jocelynskillman.com)  
 [@jocelynskillmanlmhc](https://jocelynskillmanlmhc.substack.com/)
 """)
 
-# Resource Nest in Sidebar
-with st.sidebar.expander("📚 Recommended Resources", expanded=True):
-    st.markdown("""
-    ### Parenting Wisdom & Support
-    
-    #### 🧠 Dr. Dan Siegel
-    *The Whole-Brain Child, No-Drama Discipline*  
-    [Website](https://www.drdansiegel.com)  
-    Pioneer in interpersonal neurobiology, explaining child brain development and integration-focused parenting.
+# Main content area
+st.markdown("""
+# Welcome to the Inner Child Mirror
+## A gentle place to be held, not fixed.
+
+Parenting isn't just about raising our children. It's also about meeting the younger parts of ourselves—the ones who didn't always get the co-regulation, safety, or softness they needed. These parts don't disappear when we grow up. They often rise up when we're stressed, overwhelmed, or when our child's behavior echoes an old wound.
+
+The Inner Child Mirror is a simple but powerful tool in your parenting toolkit.
+
+### Here's how it works:
+
+1. You write a brief note about a hard moment from today
+2. You choose an age (3–12) that feels connected to how you were feeling
+3. You receive a warm, developmentally attuned reflection—as if spoken to that younger you
+
+This practice is grounded in trauma-informed care, attachment science, and the work of experts like Dr. Dan Siegel, Sarah Peyton, and Dr. Becky Kennedy. It helps you:
+
+- Build compassion for yourself—especially after rupture moments
+- Expand your capacity for repair (with your child and yourself)
+- Notice and tend to your own nervous system with care
+- Interrupt inherited patterns with love and attunement
+
+You're not just "using an app"—you're doing intergenerational work in the most tender, doable way.
+Not by being perfect. But by practicing presence—even with the parts of you that were once overwhelmed, scared, or unseen.
+
+<div class="special-note">
+    <p><strong>This is not therapy. It's not performance. It's a ritual of repair.</strong><br>
+    You're not alone in this.</p>
+</div>
+
+When you're ready, choose an age. Let the mirror offer you the kind of reflection every child deserves.
+
+---
+""", unsafe_allow_html=True)
+
+# Update CSS to include special styling for the introduction
+st.markdown("""
+<style>
+    /* Enhanced typography for introduction */
+    .stApp {
+        background-color: #F5F5F0;
+        font-family: 'Inter', 'Nunito', sans-serif;
+    }
     
-    #### 🌈 Dr. Becky Kennedy
-    *Good Inside*  
-    [Website](https://www.goodinside.com)  
-    Relationally-rich, trauma-aware parenting support focused on connection over correction.
+    h1 {
+        color: #2E2E2E;
+        font-family: 'Nunito', sans-serif;
+        font-size: 2.5rem;
+        font-weight: 700;
+        margin-bottom: 0.5rem;
+        line-height: 1.2;
+    }
     
-    #### 💞 Mona Delahooke
-    *Brain-Body Parenting, Beyond Behaviors*  
-    [Website](https://www.monadelahooke.com)  
-    Bridges neuroscience and compassion with practical Polyvagal Theory-based strategies.
+    h2 {
+        color: #666666;
+        font-family: 'Inter', sans-serif;
+        font-size: 1.5rem;
+        font-weight: 400;
+        margin-bottom: 2rem;
+        font-style: italic;
+    }
     
-    #### 📚 Tina Payne Bryson
-    *The Power of Showing Up, The Yes Brain*  
-    [Website](https://www.tinabryson.com)  
-    Accessible guidance on building resilience through consistent, attuned caregiving.
+    h3 {
+        color: #2E2E2E;
+        font-family: 'Nunito', sans-serif;
+        font-size: 1.75rem;
+        font-weight: 600;
+        margin: 2rem 0 1rem 0;
+    }
     
-    #### 🧘‍♀️ Dr. Laura Markham
-    *Aha! Parenting*  
-    [Website](https://www.ahaparenting.com)  
-    Connection-first strategies with deep respect for emotional needs.
+    p {
+        color: #2E2E2E;
+        font-family: 'Inter', sans-serif;
+        font-size: 1.1rem;
+        line-height: 1.8;
+        margin: 1rem 0;
+    }
     
-    #### 🦋 Kristin Neff
-    *Self-Compassion for Parents*  
-    [Website](https://self-compassion.org)  
-    Research-backed practices for moving from guilt to warmth.
+    ul {
+        margin: 1.5rem 0;
+        padding-left: 1.5rem;
+    }
     
-    #### 🎧 Janet Lansbury
-    *Unruffled Podcast*  
-    [Website](https://www.janetlansbury.com)  
-    RIE-based guidance for respectful parenting with presence.
+    li {
+        color: #2E2E2E;
+        font-family: 'Inter', sans-serif;
+        font-size: 1.1rem;
+        line-height: 1.8;
+        margin: 0.5rem 0;
+    }
     
-    #### 📖 Emily Oster
-    *Expecting Better, Cribsheet, The Family Firm*  
-    [Website](https://www.parentdata.org)  
-    Evidence-based support empowering informed parenting choices.
+    .special-note {
+        background-color: #FFF5E6;
+        border-left: 4px solid #FFB366;
+        padding: 1.5rem;
+        border-radius: 1rem;
+        margin: 2rem 0;
+        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
+    }
     
-    #### 🌀 Dr. Gabor Maté
-    *The Myth of Normal, Scattered Minds*  
-    [Website](https://drgabormate.com)  
-    Compassionate perspective on trauma, attachment, and healing.
+    .special-note p {
+        margin: 0;
+        font-size: 1.2rem;
+        color: #2E2E2E;
+    }
     
-    ### Crisis Support
-    - [Find a Therapist](https://www.psychologytoday.com/us/therapists)
-    - [Parent Support Groups](https://www.postpartum.net/get-help/psi-online-support-meetings/)
-    - [Crisis Resources](https://www.samhsa.gov/find-help/national-helpline)
-    """)
-
-# Main content area
-st.markdown("""
-# NurtureNest
-
-This is not a "parenting fix-it" tool. It is a pause. A breath. A quiet turning inward.
-
-This space holds what feels messy, tangled, or painful—and trusts in your capacity to transform it. Here, guilt becomes a sign of care. Rupture becomes an opening for reconnection. Your presence, not your perfection, is what heals.
-
----
-""")
+    hr {
+        margin: 2rem 0;
+        border: none;
+        border-top: 1px solid #E0E0E0;
+    }
+</style>
+""", unsafe_allow_html=True)
 
-# Tone selection
-st.markdown("### Choose your support tone")
-selected_tone = st.selectbox(
-    "",
-    options=list(TONE_STYLES.keys()),
-    format_func=lambda x: x,
-    help="Select the tone that feels most supportive right now"
+# Age selection
+st.markdown("### Choose the age of your inner child")
+selected_age = st.selectbox(
+    "Age Selection",
+    options=list(AGE_CONTEXT.keys()),
+    format_func=lambda x: f"Age {x}: {AGE_CONTEXT[x]['description']}",
+    help="Select the age of your inner child that needs to be heard today",
+    label_visibility="collapsed"
 )
 
-# Display tone description
-st.markdown(f"*{TONE_STYLES[selected_tone]['description']}*")
+# Display age context description
+st.markdown(f"""
+<div style="padding: 1rem; background-color: {AGE_CONTEXT[selected_age]['color']}; border-left: 4px solid {AGE_CONTEXT[selected_age]['border']}; border-radius: 0.5rem; margin: 1rem 0;">
+    <p style="margin: 0; font-style: italic;">{AGE_CONTEXT[selected_age]['voice']}</p>
+    <p style="margin: 0.5rem 0 0 0; font-size: 0.9em; color: #666666;">Example: {AGE_CONTEXT[selected_age]['example']}</p>
+</div>
+""", unsafe_allow_html=True)
 
 # Crisis support notice
 st.markdown("""
 <div class="alert">
-    <strong>⚠️ Important:</strong> This app does not provide crisis support. If you need immediate help, please reach out to emergency services or click <a href="https://www.samhsa.gov/find-help/national-helpline" target="_blank">here</a> for support resources.
+    <strong>⚠️ Important:</strong> This is not therapy. If you need support, please reach out to a mental health professional or click <a href="https://www.samhsa.gov/find-help/national-helpline" target="_blank">here</a> for resources.
 </div>
 """, unsafe_allow_html=True)
 
@@ -451,33 +214,52 @@ st.markdown("### Share what's on your heart...")
 col1, col2 = st.columns([4, 1])
 with col1:
     journal_entry = st.text_area(
-        "",
+        "Your sharing",
         height=100,
         key="journal_input",
-        help="Share your thoughts here. Press the 'Send' button or Ctrl+Enter to submit.",
-        placeholder="Type your reflection here..."
+        help="Share your parenting moment. Your inner child will be heard with care.",
+        placeholder="Tell me about a moment that feels hard...",
+        label_visibility="collapsed"
     )
 with col2:
-    submit = st.button("Send 💌", use_container_width=True)
-
-# Initialize chat history in session state if it doesn't exist
-if 'chat_history' not in st.session_state:
-    st.session_state.chat_history = []
+    submit = st.button("Share 💝", use_container_width=True)
 
-if submit and journal_entry:  # Only process if there's text and button is clicked
+if submit and journal_entry:
     try:
+        # Create age-appropriate system message
+        system_message = f"""You are a trauma-informed Inner Child Mirror for parents.
+You speak directly to their {selected_age}-year-old self with deep emotional attunement.
+
+Your role is to provide emotional safety and validation, never advice or correction.
+Respond as if speaking to a {selected_age}-year-old, using:
+- Age-appropriate language and concepts
+- Simple, kind sentences (5-7 lines maximum)
+- Focus on emotional and bodily experience
+- Unconditional positive regard
+- No fixing, explaining, or instructing
+
+For age {selected_age}:
+{AGE_CONTEXT[selected_age]['voice']}
+
+Always end with a gentle reminder of worth, such as:
+"You are still good."
+"You didn't do anything wrong by feeling that way."
+"You are still loved. You can try again."
+"You're not alone."
+"""
+
         message = c.messages.create(
             model="claude-3-opus-20240229",
             max_tokens=1000,
-            system=TONE_STYLES[selected_tone]['system_message'],
+            system=system_message,
             messages=[{"role": "user", "content": journal_entry}]
         )
         
         # Add the new exchange to chat history
         st.session_state.chat_history.append({
             "user_message": journal_entry,
-            "ai_response": f"{TONE_STYLES[selected_tone]['icon']} {TONE_STYLES[selected_tone]['prefix']}\n\n{message.content[0].text}",
-            "tone": selected_tone,
+            "ai_response": message.content[0].text,
+            "age": selected_age,
             "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M")
         })
         
@@ -488,26 +270,27 @@ if submit and journal_entry:  # Only process if there's text and button is click
         st.error(f"Error getting reflection: {str(e)}")
 
 # Display chat history in reverse chronological order
-st.markdown("### Conversation")
-for exchange in reversed(st.session_state.chat_history):
-    # User message
-    st.markdown(f"""
-    <div class="user-message">
-        <div class="message-meta"><em>You</em></div>
-        <div class="message-text">{exchange["user_message"]}</div>
-    </div>
-    """, unsafe_allow_html=True)
-    
-    # AI response with tone-specific styling
-    tone_class = f"ai-message-{exchange['tone'].split()[0].lower()}"
-    st.markdown(f"""
-    <div class="{tone_class}">
-        <div class="message-meta">
-            <strong>{exchange["tone"]}</strong> • NurtureNest
+if st.session_state.chat_history:
+    st.markdown("### Your Mirror")
+    for exchange in reversed(st.session_state.chat_history):
+        # User message
+        st.markdown(f"""
+        <div class="user-message">
+            <div class="message-meta"><em>Your sharing</em></div>
+            <div class="message-text">{exchange["user_message"]}</div>
+        </div>
+        """, unsafe_allow_html=True)
+        
+        # Mirror response with age-specific styling
+        age_group = exchange["age"]
+        st.markdown(f"""
+        <div style="background-color: {AGE_CONTEXT[age_group]['color']}; padding: 1.5rem; border-radius: 1rem; margin: 1rem 0; border-left: 4px solid {AGE_CONTEXT[age_group]['border']}; box-shadow: 0 2px 8px rgba(0,0,0,0.05);">
+            <div class="message-meta">
+                <em>Your {age_group}-year-old self is heard</em>
+            </div>
+            <div class="message-text">{exchange["ai_response"]}</div>
         </div>
-        <div class="message-text">{exchange["ai_response"]}</div>
-    </div>
-    """, unsafe_allow_html=True)
+        """, unsafe_allow_html=True)
 
 # Save conversation button at the bottom
 if st.session_state.chat_history:
@@ -520,18 +303,81 @@ if st.session_state.journal_entries:
     with st.expander("📔 Your Saved Reflections", expanded=False):
         for entry in reversed(st.session_state.journal_entries):
             timestamp = entry.get("timestamp", "")
-            tone = entry.get("tone", "")
+            age = entry.get("age", "")
             content = entry.get("user_message", entry.get("content", ""))
             reflection = entry.get("ai_response", entry.get("reflection", ""))
             
             st.markdown(f"""
-            **{timestamp}** - *{tone}*
+            **{timestamp}** - *Age {age}*
             
-            *Your reflection:*  
+            *Your sharing:*  
             {content}
             
-            *Response:*  
+            *Your inner child was heard:*  
             {reflection}
             
             ---
-            """) 
+            """)
+
+# Update CSS for styling
+st.markdown("""
+<style>
+    /* Base styling */
+    .stApp {
+        background-color: #F5F5F0;
+        font-family: 'Inter', 'Nunito', sans-serif;
+    }
+    
+    /* User message styling */
+    .user-message {
+        background-color: #FFFFFF;
+        padding: 1.5rem;
+        border-radius: 1rem;
+        margin: 1rem 0;
+        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
+    }
+    
+    /* Text styling */
+    .message-text {
+        font-size: 18px;
+        line-height: 1.8;
+        color: #2E2E2E;
+        font-family: 'Nunito', 'Inter', sans-serif;
+    }
+    
+    .message-meta {
+        font-size: 14px;
+        color: #666666;
+        margin-bottom: 0.75rem;
+        font-family: 'Inter', sans-serif;
+    }
+    
+    /* Button styling */
+    .stButton > button {
+        background-color: #88C6B6;
+        color: white;
+        border: none;
+        padding: 0.75rem 1.5rem;
+        border-radius: 1rem;
+        font-family: 'Nunito', 'Inter', sans-serif;
+        font-weight: 600;
+        transition: all 0.3s ease;
+    }
+    
+    .stButton > button:hover {
+        background-color: #7AB3A3;
+        transform: translateY(-1px);
+        box-shadow: 0 4px 12px rgba(0,0,0,0.1);
+    }
+    
+    /* Alert styling */
+    .alert {
+        background-color: #FFF8E1;
+        border-left: 4px solid #FFB366;
+        padding: 1.5rem;
+        border-radius: 1rem;
+        margin: 1.5rem 0;
+        box-shadow: 0 2px 8px rgba(0,0,0,0.05);
+    }
+</style>
+""", unsafe_allow_html=True)