TalkHeal_Simple.py

"""
TalkHeal - Simplified Version for Streamlit Cloud
This version removes OAuth to ensure successful deployment
"""

import streamlit as st
from auth.auth_utils import init_db
from components.login_page import show_login_page
from core.utils import save_conversations, load_conversations, set_authenticated_user
from components.mood_dashboard import MoodTracker, render_mood_dashboard
import plotly.express as px

st.set_page_config(page_title="TalkHeal", page_icon="💬", layout="wide")

# --- DB Initialization ---
if "db_initialized" not in st.session_state:
    init_db()
    st.session_state["db_initialized"] = True
    
# --- LOGIN PAGE ---
if not st.session_state.get("authenticated", False):
    show_login_page()
    st.stop()

# Add responsive navigation CSS
st.markdown("""
<style>
@media (max-width: 768px) {
    .nav-button-container {
        display: flex;
        flex-wrap: wrap;
        gap: 0.5rem;
        justify-content: center;
        margin-bottom: 1rem;
    }
    .nav-button-container .stButton {
        flex: 1 1 auto;
        min-width: 100px;
        max-width: 140px;
    }
    .nav-button-container .stButton > button {
        font-size: 0.85rem !important;
        padding: 0.4rem 0.6rem !important;
    }
}
@media (max-width: 480px) {
    .nav-button-container {
        flex-direction: column;
        align-items: stretch;
    }
    .nav-button-container .stButton {
        max-width: none;
        margin-bottom: 0.25rem;
    }
}
</style>
""", unsafe_allow_html=True)

# Responsive navigation layout
col_spacer, col_buttons = st.columns([1, 4])
with col_spacer:
    pass
with col_buttons:
    st.markdown('<div class="nav-button-container">', unsafe_allow_html=True)
    
    nav_cols = st.columns([1, 1.5, 1, 1])
    
    with nav_cols[0]:
        is_dark = st.session_state.get('dark_mode', False)
        if st.button("🌙" if is_dark else "☀", key="top_theme_toggle", help="Toggle Light/Dark Mode", use_container_width=True):
                st.session_state.dark_mode = not is_dark
                st.session_state.theme_changed = True
                st.rerun()
    with nav_cols[1]:
        if st.button("🚨 Emergency Help", key="emergency_main_btn", help="Open crisis resources", use_container_width=True, type="secondary"):
            st.session_state.show_emergency_page = True
            st.rerun()
    with nav_cols[2]:
        if st.button("ℹ About", key="about_btn", help="About TalkHeal", use_container_width=True):
            st.switch_page("pages/About.py")
    with nav_cols[3]:
        if st.button("Logout", key="logout_btn", help="Sign out", use_container_width=True):
            for key in ["authenticated", "user_profile"]:
                if key in st.session_state:
                    del st.session_state[key]
                    st.rerun()
    
    st.markdown('</div>', unsafe_allow_html=True)

from core.config import configure_gemini, PAGE_CONFIG
from core.utils import get_current_time, create_new_conversation
from css.styles import apply_custom_css
from components.header import render_header
from components.sidebar import render_sidebar
from components.chat_interface import render_chat_interface, handle_chat_input, render_session_controls
from components.mood_dashboard import MoodTracker
from components.emergency_page import render_emergency_page
from components.focus_session import render_focus_session
from components.profile import apply_global_font_size
from components.games import show_games_page

# --- 1. INITIALIZE SESSION STATE ---
if "chat_history" not in st.session_state:
    st.session_state.chat_history = []
if "conversations" not in st.session_state:
    st.session_state.conversations = load_conversations()
if "active_conversation" not in st.session_state:
    st.session_state.active_conversation = -1
if "show_emergency_page" not in st.session_state:
    st.session_state.show_emergency_page = False
if "show_focus_session" not in st.session_state:
    st.session_state.show_focus_session = False
if "show_mood_dashboard" not in st.session_state:
    st.session_state.show_mood_dashboard = False
if "sidebar_state" not in st.session_state:
    st.session_state.sidebar_state = "expanded"
if "mental_disorders" not in st.session_state:
    st.session_state.mental_disorders = [
        "Depression & Mood Disorders", "Anxiety & Panic Disorders", "Bipolar Disorder",
        "PTSD & Trauma", "OCD & Related Disorders", "Eating Disorders",
        "Substance Use Disorders", "ADHD & Neurodevelopmental", "Personality Disorders",
        "Sleep Disorders"
    ]
if "selected_tone" not in st.session_state:
    st.session_state.selected_tone = "Compassionate Listener"
if "pinned_messages" not in st.session_state:
    st.session_state.pinned_messages = []

if "active_page" not in st.session_state:
    st.session_state.active_page = "TalkHeal"

# --- Footer Navigation State ---
if "show_privacy_policy" not in st.session_state:
    st.session_state.show_privacy_policy = False

if st.session_state.show_privacy_policy:
    from pages.PrivacyPolicy import show as show_privacy
    show_privacy()
    from components.footer import show_footer
    show_footer()
    st.stop()

# --- 2. SET PAGE CONFIG ---
apply_global_font_size()

# --- 3. APPLY STYLES & CONFIGURATIONS ---
apply_custom_css()
model = configure_gemini()

# --- 4. TONE SELECTION DROPDOWN IN SIDEBAR ---
TONE_OPTIONS = {
    "Compassionate Listener": "You are a compassionate listener — soft, empathetic, patient — like a therapist who listens without judgment.",
    "Motivating Coach": "You are a motivating coach — energetic, encouraging, and action-focused — helping the user push through rough days.",
    "Wise Friend": "You are a wise friend — thoughtful, poetic, and reflective — giving soulful responses and timeless advice.",
    "Neutral Therapist": "You are a neutral therapist — balanced, logical, and non-intrusive — asking guiding questions using CBT techniques.",
    "Mindfulness Guide": "You are a mindfulness guide — calm, slow, and grounding — focused on breathing, presence, and awareness."
}

def get_tone_prompt():
    return TONE_OPTIONS.get(st.session_state.get("selected_tone", "Compassionate Listener"), TONE_OPTIONS["Compassionate Listener"])

# --- 6. RENDER SIDEBAR ---
render_sidebar()

# --- 7. PAGE ROUTING ---
main_area = st.container()

if not st.session_state.conversations:
    saved_conversations = load_conversations()
    if saved_conversations:
        st.session_state.conversations = saved_conversations
        if st.session_state.active_conversation == -1:
            st.session_state.active_conversation = 0
    else:
        create_new_conversation()
        st.session_state.active_conversation = 0
    st.rerun()

# --- 8. FEATURE CARDS FUNCTION ---
def render_feature_cards():
    """Render beautiful feature cards showcasing app capabilities"""

    # Hero Welcome Section
    st.markdown(f"""
        <div class="hero-welcome-section">
            <div class="hero-content">
                <h1 class="hero-title">Welcome to TalkHeal, {st.session_state.user_profile.get("name", "User")}! 💬</h1>
                <p class="hero-subtitle">Your Mental Health Companion 💙</p>
            </div>
        </div>
    """, unsafe_allow_html=True)

    # Define all feature cards data
    cards_data = [
        {
            "icon": "🧘‍♀",
            "title": "Yoga & Meditation",
            "action": lambda: st.switch_page("pages/Yoga.py"),
            "key": "yoga_btn",
            "button_text": "🧘‍♀ Start Yoga",
            "css_class": "yoga-card"
        },
        {
            "icon": "🌬",
            "title": "Breathing Exercises",
            "action": lambda: st.switch_page("pages/Breathing_Exercise.py"),
            "key": "breathing_btn",
            "button_text": "🌬 Start Breathing",
            "css_class": "breathing-card"
        },
        {
            "icon": "📝",
            "title": "Personal Journaling",
            "action": lambda: st.switch_page("pages/Journaling.py"),
            "key": "journal_btn",
            "button_text": "📝 Open Journal",
            "css_class": "journal-card"
        },
        {
            "icon": "👨‍⚕",
            "title": "Doctor Specialist",
            "action": lambda: st.switch_page("pages/doctor_spec.py"),
            "key": "doctor_btn",
            "button_text": "👨‍⚕ Find Specialists",
            "css_class": "doctor-card"
        },
        {
            "icon": "🎮",
            "title": "Mental Wellness Games",
            "action": lambda: setattr(st.session_state, 'active_page', 'Games') or st.rerun(),
            "key": "games_btn",
            "button_text": "🎮 Play Games",
            "css_class": "mood-card"
        },
        {
            "icon": "🛠",
            "title": "Self-Help Tools",
            "action": lambda: st.switch_page("pages/selfHelpTools.py"),
            "key": "tools_btn",
            "button_text": "🛠 Explore Tools",
            "css_class": "tools-card"
        },
        {
            "icon": "🌿",
            "title": "Habit Builder",
            "action": lambda: st.switch_page("pages/Habit_Builder.py"),
            "key": "habits_btn",
            "button_text": "🌿 Build Habits",
            "css_class": "habits-card"
        },
        {
            "icon": "🌟",
            "title": "Wellness Resource Hub",
            "action": lambda: st.switch_page("pages/WellnessResourceHub.py"),
            "key": "wellness_btn",
            "button_text": "🌟 Wellness Hub",
            "css_class": "wellness-card"
        },
        {
            "icon": "💬",
            "title": "Community Forum",
            "action": lambda: st.switch_page("pages/CommunityForum.py"),
            "key": "forum_btn",
            "button_text": "💬 Join Community",
            "css_class": "community-card"
        },
        {
            "icon": "❓",
            "title": "Q&A Support",
            "action": lambda: st.switch_page("pages/QnA.py"),
            "key": "qna_btn",
            "button_text": "❓ Ask Questions",
            "css_class": "qna-card"
        }
    ]

    # Use Streamlit's native columns to create the grid layout
    num_columns = 5
    
    if len(cards_data) % num_columns != 0:
        st.warning(f"Please use a number of cards that is a multiple of {num_columns} for a perfect grid.")

    cols = st.columns(num_columns)
    
    for i, card in enumerate(cards_data):
        with cols[i % num_columns]:
            st.markdown(f"""
            <div class="feature-card primary-card {card['css_class']}">
                <div class="card-icon" style="font-size: 3rem; margin-bottom: 1rem;">{card['icon']}</div>
                <h3 style="margin-bottom: 1rem; color: white; font-size: 1.1rem;">{card['title']}</h3>
            </div>
            """, unsafe_allow_html=True)
            
            if st.button(card['button_text'], key=card['key'], use_container_width=True):
                try:
                    card['action']()
                except Exception as e:
                    st.error(f"Error navigating to {card['title']}: {str(e)}")
                                        
# --- 9. RENDER PAGE ---
if st.session_state.get("show_emergency_page"):
    with main_area:
        render_emergency_page()
elif st.session_state.get("show_focus_session"):
    with main_area:
        render_focus_session()
elif st.session_state.get("show_mood_dashboard"):
    with main_area:
        render_mood_dashboard()

# Handles rendering the "Pinned Messages" page.
elif st.session_state.active_page == "PinnedMessages":
    with main_area:
        from pages.Pinned_msg import render_pinned_messages_page

        # Back to Home Button
        if st.button("⬅ Back to Home", key="back_to_home_btn"):
            st.session_state.active_page = "TalkHeal"
            st.rerun()

        render_pinned_messages_page()

elif st.session_state.active_page == "CommunityForum":
    with main_area:
        import pages.CommunityForum as community_forum

elif st.session_state.active_page == "Games":
    with main_area:
        # Back to Home Button
        if st.button("⬅ Back to Home", key="back_to_home_from_games"):
            st.session_state.active_page = "TalkHeal"
            st.rerun()
        
        # Show Games Page
        show_games_page()
else:
    with main_area:
        # Render the beautiful feature cards layout
        render_feature_cards()
        
        # AI Tone Selection in main area
        with st.expander("🧠 Customize Your AI Companion", expanded=False):
            st.markdown("*Choose how your AI companion should respond to you:*")
            selected_tone = st.selectbox(
                "Select AI personality:",
                options=list(TONE_OPTIONS.keys()),
                index=list(TONE_OPTIONS.keys()).index(st.session_state.selected_tone),
                help="Different tones provide different therapeutic approaches"
            )
            if selected_tone != st.session_state.selected_tone:
                st.session_state.selected_tone = selected_tone
                st.rerun()
            
            st.info(f"*Current Style*: {TONE_OPTIONS[selected_tone]}")
            
        # Current AI Tone Display
        st.markdown(f"""
        <div class="current-tone-display">
            <div class="tone-content">
                <span class="tone-label">🧠 Current AI Personality:</span>
                <span class="tone-value">{st.session_state['selected_tone']}</span>
            </div>
        </div>
        """, unsafe_allow_html=True)
    
        # Mood Tracking Section
        st.markdown("""
        <div class="mood-tracking-section">
            <h3>😊 How are you feeling today?</h3>
            <p>Track your mood to help your AI companion provide better support</p>
        </div>
        """, unsafe_allow_html=True)
        
        # Initialize mood tracker if not already done
        if "mood_tracker" not in st.session_state:
            st.session_state.mood_tracker = MoodTracker()
        
        tracker = st.session_state.mood_tracker
        
        # Mood Entry Form
        with st.form("mood_entry_form"):
            st.markdown("###   Record Your Mood")
            
            # Mood Level Selection
            mood_options = {
                "very_low": "  Very Low",
                "low": "😔 Low", 
                "okay": "  Okay",
                "good": "😊 Good",
                "great": " 😄 Great"
            }
            
            selected_mood = st.selectbox(
                "How are you feeling right now?",
                options=list(mood_options.keys()),
                format_func=lambda x: mood_options[x],
                help="Select your current emotional state"
            )
            
            # Context/Reason
            context_options = [
                "Work/School related",
                "Family matters",
                "Health concerns",
                "Social interactions",
                "Financial stress",
                "Weather/environment",
                "Sleep quality",
                "Physical activity",
                "Food/Nutrition",
                "Personal achievement",
                "Relationship issues",
                "Future worries",
                "Other"
            ]
            
            context_reason = st.selectbox(
                "What's influencing your mood today?",
                options=context_options,
                help="Understanding context helps provide better support"
            )
            
            # Activities
            activity_options = [
                "Exercise/Physical activity",
                "Meditation/Mindfulness",
                "Reading",
                "Writing/Journaling",
                "Socializing",
                "Hobbies/Creative work",
                "Watching TV/Movies",
                "Gaming",
                "Cooking/Eating",
                "Shopping",
                "Housework/Chores",
                "Learning/Education",
                "Music/Audio",
                "Nature/Outdoors",
                "Resting/Sleeping",
                "Other"
            ]
            
            selected_activities = st.multiselect(
                "What activities have you done today?",
                options=activity_options,
                help="Select all that apply"
            )
            
            # Notes
            mood_notes = st.text_area(
                "Additional notes (optional)",
                height=100,
                placeholder="Share any thoughts, feelings, or details about your day...",
                help="This helps your AI companion understand you better"
            )
            
            # Submit button
            submitted = st.form_submit_button("💾 Save Mood Entry")
            
            if submitted:
                try:
                    # Save the mood entry
                    tracker.add_mood_entry(
                        mood_level=selected_mood,
                        notes=mood_notes,
                        context_reason=context_reason,
                        activities=selected_activities
                    )
                    
                    st.success("✅ Your mood has been recorded successfully!")
                    
                    # Show personalized response based on mood
                    mood_responses = {
                        "very_low": "🤗 I'm here for you. Consider reaching out to a trusted friend or professional if you need support.",
                        "low": "📝 Journaling your thoughts might help process your feelings. Would you like to talk about what's bothering you?",
                        "okay": "🚶‍♀ A short walk or some light stretching might help you feel more balanced.",
                        "good": "✨ Great to hear you're feeling good! What positive things happened today?",
                        "great": "🌟 You're shining today! Keep spreading that positivity with a kind act."
                    }
                    
                    st.info(mood_responses.get(selected_mood, "Thanks for sharing how you're feeling!"))
                    
                except Exception as e:
                    st.error(f"❌ Error saving mood entry: {str(e)}")
        
        # Quick Mood Stats
        st.markdown("---")
        st.markdown("### 📊 Your Recent Mood Summary")
        
        try:
            # Get recent mood data
            recent_df = tracker.get_mood_dataframe(days=7)
            
            if not recent_df.empty:
                # Add responsive CSS for mood metrics
                st.markdown("""
                <style>
                @media (max-width: 768px) {
                    .mood-metrics-container {
                        display: flex;
                        flex-direction: column;
                        gap: 0.5rem;
                    }
                    .mood-metrics-row {
                        display: flex;
                        gap: 0.5rem;
                    }
                    .mood-metrics-row > div {
                        flex: 1;
                    }
                }
                </style>
                """, unsafe_allow_html=True)
                
                # Responsive layout - 2 rows of 2 columns on mobile, 3 columns on desktop
                col1, col2 = st.columns(2)
                
                with col1:
                    avg_mood = recent_df['mood_level'].apply(tracker.get_mood_numeric).mean()
                    st.metric("Average Mood (7 days)", f"{avg_mood:.1f}/5")
                
                with col2:
                    total_entries = len(recent_df)
                    st.metric("Entries This Week", total_entries)
                
                # Third metric in a centered column
                col3_container = st.columns([1, 2, 1])
                with col3_container[1]:
                    most_common = recent_df['mood_level'].mode().iloc[0] if not recent_df.empty else "N/A"
                    st.metric("Most Common Mood", tracker.get_mood_label(most_common))
                
                # Quick chart
                st.markdown("#### Mood Trend (Last 7 Days)")
                fig = px.line(recent_df, x='date', y=recent_df['mood_level'].apply(tracker.get_mood_numeric), 
                             markers=True, line_shape='linear')
                fig.update_layout(
                    xaxis_title="Date",
                    yaxis_title="Mood Level",
                    yaxis=dict(tickmode='array', tickvals=[1,2,3,4,5], 
                              ticktext=['Very Low', 'Low', 'Okay', 'Good', 'Great']),
                    height=200
                )
                st.plotly_chart(fig, use_container_width=True)
            else:
                st.info("📝 Start tracking your mood to see insights here!")
                
        except Exception as e:
            st.warning("Unable to load mood statistics. This is normal if you haven't tracked your mood yet.")
        
        st.markdown("---")
        
        # Mood Dashboard Access
        col1, col2, col3 = st.columns([1, 2, 1])
        with col2:
            if st.button("📊 View Mood Dashboard", use_container_width=True, type="primary"):
                st.session_state.show_mood_dashboard = True
                st.rerun()
        
        st.markdown("---")
        
        # Chat Interface
        render_chat_interface()
        handle_chat_input(model, system_prompt=get_tone_prompt())
        render_session_controls()

        # --- Footer ---
        from components.footer import show_footer
        show_footer()

# --- 10. SCROLL SCRIPT ---
st.markdown("""
<script>
    function scrollToBottom() {
        var chatContainer = document.querySelector('.chat-container');
        if (chatContainer) {
            chatContainer.scrollTop = chatContainer.scrollHeight;
        }
    }
    setTimeout(scrollToBottom, 100);
</script>
""", unsafe_allow_html=True)