pages/ai_buddy.py

import streamlit as st
import random
from langchain.chat_models import ChatOpenAI
from langchain.schema import HumanMessage, SystemMessage
import os
from dotenv import load_dotenv
import pandas as pd
from datetime import datetime
import plotly.express as px
import json
import tempfile
import time
import numpy as np
import threading
#import sounddevice as sd
#from playsound import playsound
#import pygame
#from scipy.io import wavfile

#pygame.mixer.init()

# Load environment variables
load_dotenv()

AI71_BASE_URL = "https://api.ai71.ai/v1/"
AI71_API_KEY = os.getenv('AI71_API_KEY')

# Initialize the Falcon model
chat = ChatOpenAI(
    model="tiiuae/falcon-180B-chat",
    api_key=AI71_API_KEY,
    base_url=AI71_BASE_URL,
    streaming=True,
)

# Expanded Therapy techniques
THERAPY_TECHNIQUES = {
    "CBT": "Use Cognitive Behavioral Therapy techniques to help the user identify and change negative thought patterns.",
    "Mindfulness": "Guide the user through mindfulness exercises to promote present-moment awareness and reduce stress.",
    "Solution-Focused": "Focus on the user's strengths and resources to help them find solutions to their problems.",
    "Emotion-Focused": "Help the user identify, experience, and regulate their emotions more effectively.",
    "Psychodynamic": "Explore the user's past experiences and unconscious patterns to gain insight into current issues.",
    "ACT": "Use Acceptance and Commitment Therapy to help the user accept their thoughts and feelings while committing to positive changes.",
    "DBT": "Apply Dialectical Behavior Therapy techniques to help the user manage intense emotions and improve relationships.",
    "Gestalt": "Use Gestalt therapy techniques to focus on the present moment and increase self-awareness.",
    "Existential": "Explore existential themes such as meaning, freedom, and responsibility to help the user find purpose.",
    "Narrative": "Use storytelling and narrative techniques to help the user reframe their life experiences and create new meaning.",
}

def get_ai_response(user_input, buddy_config, therapy_technique=None):
    system_message = f"You are {buddy_config['name']}, an AI companion with the following personality: {buddy_config['personality']}. "
    system_message += f"Additional details about you: {buddy_config['details']}. "
    
    if therapy_technique:
        system_message += f"In this conversation, {THERAPY_TECHNIQUES[therapy_technique]}"
    
    messages = [
        SystemMessage(content=system_message),
        HumanMessage(content=user_input)
    ]
    response = chat.invoke(messages).content
    return response

#def play_sound_loop(sound_file, stop_event):
#    while not stop_event.is_set():
#        playsound(sound_file)
#
#def play_sound_for_duration(sound_file, duration):
#    start_time = time.time()
#    while time.time() - start_time < duration:
#        playsound(sound_file, block=False)
#        time.sleep(0.1)  # Short sleep to prevent excessive CPU usage
#    # Ensure the sound stops after the duration
#    pygame.mixer.quit()
#
#def get_sound_files(directory):
#    return [f for f in os.listdir(directory) if f.endswith('.mp3')]
#
#def get_sound_file_path(sound_name, sound_dir):
#    # Convert the sound name to a filename
#    filename = f"{sound_name.lower().replace(' ', '_')}.mp3"
#    return os.path.join(sound_dir, filename)
#
#SOUND_OPTIONS = [
#    "Gentle Rain", "Ocean Waves", "Forest Ambience", "Soft Wind Chimes",
#    "Tibetan Singing Bowls", "Humming Song", "Crackling Fireplace",
#    "Birdsong", "White Noise", "Zen River", "Heartbeat", "Deep Space",
#    "Whale Songs", "Bamboo Flute", "Thunderstorm", "Cat Purring",
#    "Campfire", "Windchimes", "Waterfall", "Beach Waves", "Cicadas",
#    "Coffee Shop Ambience", "Grandfather Clock", "Rainstorm on Tent",
#    "Tropical Birds", "Subway Train", "Washing Machine", "Fan White Noise",
#    "Tibetan Bells", "Wind in Trees", "Meditation Bowl", "Meditation Bowl2", "Birds Singing Rainy Day"
#]
#
#def show_meditation_timer():
#    st.subheader("🧘‍♀️ Enhanced Meditation Timer")
#    
#    sound_dir = os.path.join(os.path.dirname(__file__), "..", "sounds")
#    
#    col1, col2 = st.columns(2)
#    
#    with col1:
#        duration = st.slider("Select duration (minutes)", 1, 60, 5)
#        background_sound = st.selectbox("Background Sound", SOUND_OPTIONS)
#        
#    with col2:
#        interval_options = ["None", "Every 5 minutes", "Every 10 minutes"]
#        interval_reminder = st.selectbox("Interval Reminders", interval_options)
#        end_sound = st.selectbox("End of Session Sound", SOUND_OPTIONS)
#    
#    if st.button("Start Meditation", key="start_meditation"):
#        progress_bar = st.progress(0)
#        status_text = st.empty()
#        
#        # Initialize pygame mixer
#        pygame.mixer.init()
#        
#        # Load background sound
#        background_sound_file = get_sound_file_path(background_sound, sound_dir)
#        if not os.path.exists(background_sound_file):
#            st.error(f"Background sound file not found: {background_sound_file}")
#            return
#        
#        # Load end of session sound
#        end_sound_file = get_sound_file_path(end_sound, sound_dir)
#        if not os.path.exists(end_sound_file):
#            st.error(f"End sound file not found: {end_sound_file}")
#            return
#        
#        # Play background sound on loop
#        pygame.mixer.music.load(background_sound_file)
#        pygame.mixer.music.play(-1)  # -1 means loop indefinitely
#        
#        start_time = time.time()
#        end_time = start_time + (duration * 60)
#        
#        try:
#            while time.time() < end_time:
#                elapsed_time = time.time() - start_time
#                progress = elapsed_time / (duration * 60)
#                progress_bar.progress(progress)
#                
#                remaining_time = end_time - time.time()
#                mins, secs = divmod(int(remaining_time), 60)
#                status_text.text(f"Time remaining: {mins:02d}:{secs:02d}")
#                
#                if interval_reminder != "None":
#                    interval = 5 if interval_reminder == "Every 5 minutes" else 10
#                    if int(elapsed_time) > 0 and int(elapsed_time) % (interval * 60) == 0:
#                        st.toast(f"{interval} minutes passed", icon="⏰")
#                
#                # Check if 10 seconds remaining
#                if remaining_time <= 10 and remaining_time > 9:
#                    pygame.mixer.music.stop()  # Stop background sound
#                    pygame.mixer.Sound(end_sound_file).play()  # Play end sound
#                
#                if remaining_time <= 0:
#                    break
#                
#                time.sleep(0.1)  # Update more frequently for smoother countdown
#        finally:
#            # Stop all sounds
#            pygame.mixer.quit()
#        
#        # Ensure the progress bar is full and time remaining shows 00:00
#        progress_bar.progress(1.0)
#        status_text.text("Time remaining: 00:00")
#        
#        st.success("Meditation complete!")
#        st.balloons()
#        
#        if 'achievements' not in st.session_state:
#            st.session_state.achievements = set()
#        st.session_state.achievements.add("Zen Master")
#        st.success("Achievement Unlocked: Zen Master 🧘‍♀️")

def show_personalized_recommendations():
    st.subheader("🎯 Personalized Recommendations")
    
    recommendation_categories = [
        "Mental Health",
        "Physical Health",
        "Personal Development",
        "Relationships",
        "Career",
        "Hobbies",
    ]
    
    selected_category = st.selectbox("Choose a category", recommendation_categories)
    
    recommendations = {
        "Mental Health": [
            "Practice daily gratitude journaling",
            "Try a guided meditation for stress relief",
            "Explore cognitive behavioral therapy techniques",
            "Start a mood tracking journal",
            "Learn about mindfulness practices",
        ],
        "Physical Health": [
            "Start a 30-day yoga challenge",
            "Try intermittent fasting",
            "Begin a couch to 5K running program",
            "Experiment with new healthy recipes",
            "Create a sleep hygiene routine",
        ],
        "Personal Development": [
            "Start learning a new language",
            "Read personal development books",
            "Take an online course in a subject you're interested in",
            "Practice public speaking",
            "Start a daily writing habit",
        ],
        "Relationships": [
            "Practice active listening techniques",
            "Plan regular date nights or friend meetups",
            "Learn about love languages",
            "Practice expressing gratitude to loved ones",
            "Join a local community or interest group",
        ],
        "Career": [
            "Update your resume and LinkedIn profile",
            "Network with professionals in your industry",
            "Set SMART career goals",
            "Learn a new skill relevant to your field",
            "Start a side project or freelance work",
        ],
        "Hobbies": [
            "Start a garden or learn about plant care",
            "Try a new art form like painting or sculpting",
            "Learn to play a musical instrument",
            "Start a DIY home improvement project",
            "Explore photography or videography",
        ],
    }
    
    st.write("Here are some personalized recommendations for you:")
    for recommendation in recommendations[selected_category]:
        st.markdown(f"- {recommendation}")
    
    if st.button("Get More Recommendations"):
        st.write("More tailored recommendations:")
        additional_recs = random.sample(recommendations[selected_category], 3)
        for rec in additional_recs:
            st.markdown(f"- {rec}")

#def generate_binaural_beat(freq1, freq2, duration_seconds, sample_rate=44100):
#    t = np.linspace(0, duration_seconds, int(sample_rate * duration_seconds), False)
#    left_channel = np.sin(2 * np.pi * freq1 * t)
#    right_channel = np.sin(2 * np.pi * freq2 * t)
#    stereo_audio = np.vstack((left_channel, right_channel)).T
#    return (stereo_audio * 32767).astype(np.int16)
#
#def get_binary_file_downloader_html(bin_file, file_label='File'):
#    b64 = base64.b64encode(bin_file).decode()
#    return f'<a href="data:application/octet-stream;base64,{b64}" download="{file_label}.wav" class="download-link">Download {file_label}</a>'
#
#def show_binaural_beats():
#    st.subheader("🎵 Binaural Beats Generator")
#    
#    st.markdown("""
#    <style>
#    .stButton>button {
#        background-color: #4CAF50;
#        color: white;
#        font-weight: bold;
#    }
#    .download-link {
#        background-color: #008CBA;
#        color: white;
#        padding: 10px 15px;
#        text-align: center;
#        text-decoration: none;
#        display: inline-block;
#        font-size: 16px;
#        margin: 4px 2px;
#        cursor: pointer;
#        border-radius: 4px;
#    }
#    .stop-button {
#        background-color: #f44336;
#        color: white;
#        font-weight: bold;
#    }
#    </style>
#    """, unsafe_allow_html=True)
#    
#    st.write("Binaural beats are created when two slightly different frequencies are played in each ear, potentially influencing brainwave activity.")
#    
#    preset_beats = {
#        "Deep Relaxation (Delta)": {"base": 100, "beat": 2},
#        "Meditation (Theta)": {"base": 150, "beat": 6},
#        "Relaxation (Alpha)": {"base": 200, "beat": 10},
#        "Light Focus (Low Beta)": {"base": 250, "beat": 14},
#        "High Focus (Mid Beta)": {"base": 300, "beat": 20},
#        "Alertness (High Beta)": {"base": 350, "beat": 30},
#        "Gamma Consciousness": {"base": 400, "beat": 40},
#        "Lucid Dreaming": {"base": 180, "beat": 3},
#        "Memory Enhancement": {"base": 270, "beat": 12},
#        "Creativity Boost": {"base": 220, "beat": 8},
#        "Pain Relief": {"base": 130, "beat": 4},
#        "Mood Elevation": {"base": 315, "beat": 18}
#    }
#    
#    col1, col2 = st.columns(2)
#    
#    with col1:
#        beat_type = st.selectbox("Choose a preset or custom:", ["Custom"] + list(preset_beats.keys()))
#    
#    with col2:
#        duration = st.slider("Duration (minutes):", 1, 60, 15)
#    
#    if beat_type == "Custom":
#        col3, col4 = st.columns(2)
#        with col3:
#            base_freq = st.slider("Base Frequency (Hz):", 100, 500, 200)
#        with col4:
#            beat_freq = st.slider("Desired Beat Frequency (Hz):", 1, 40, 10)
#    else:
#        base_freq = preset_beats[beat_type]["base"]
#        beat_freq = preset_beats[beat_type]["beat"]
#        st.info(f"Base Frequency: {base_freq} Hz, Beat Frequency: {beat_freq} Hz")
#    
#    if 'audio_playing' not in st.session_state:
#        st.session_state.audio_playing = False
#    
#    if 'start_time' not in st.session_state:
#        st.session_state.start_time = None
#    
#    if 'end_time' not in st.session_state:
#        st.session_state.end_time = None
#    
#    # Create persistent placeholders for UI elements
#    progress_bar = st.empty()
#    status_text = st.empty()
#    stop_button = st.empty()
#    
#    generate_button = st.button("Generate and Play Binaural Beat")
#    
#    if generate_button:
#        try:
#            # Stop any currently playing audio
#            if st.session_state.audio_playing:
#                pygame.mixer.music.stop()
#                st.session_state.audio_playing = False
#            
#            audio_data = generate_binaural_beat(base_freq, base_freq + beat_freq, duration * 60)
#            
#            # Save the generated audio to a temporary file
#            with tempfile.NamedTemporaryFile(delete=False, suffix='.wav') as temp_file:
#                temp_filename = temp_file.name
#                wavfile.write(temp_filename, 44100, audio_data)
#            
#            # Initialize pygame mixer
#            pygame.mixer.init(frequency=44100, size=-16, channels=2)
#            
#            # Load and play the audio
#            pygame.mixer.music.load(temp_filename)
#            pygame.mixer.music.play()
#            st.session_state.audio_playing = True
#            
#            st.session_state.start_time = time.time()
#            st.session_state.end_time = st.session_state.start_time + (duration * 60)
#        
#        except Exception as e:
#            st.error(f"An error occurred: {str(e)}")
#            st.info("Please try again or contact support if the issue persists.")
#    
#    if st.session_state.audio_playing:
#        stop_button_active = stop_button.button("Stop Binaural Beat", key="stop_binaural", type="primary")
#        current_time = time.time()
#        
#        if stop_button_active:
#            pygame.mixer.music.stop()
#            st.session_state.audio_playing = False
#            st.session_state.start_time = None
#            st.session_state.end_time = None
#        
#        elif current_time < st.session_state.end_time:
#            elapsed_time = current_time - st.session_state.start_time
#            progress = elapsed_time / (st.session_state.end_time - st.session_state.start_time)
#            progress_bar.progress(progress)
#            
#            remaining_time = st.session_state.end_time - current_time
#            mins, secs = divmod(int(remaining_time), 60)
#            status_text.text(f"Time remaining: {mins:02d}:{secs:02d}")
#        else:
#            pygame.mixer.music.stop()
#            st.session_state.audio_playing = False
#            st.session_state.start_time = None
#            st.session_state.end_time = None
#            progress_bar.empty()
#            status_text.text("Binaural beat session complete!")
#    
#    # Offer download of the generated audio
#    if not st.session_state.audio_playing and 'audio_data' in locals():
#        with io.BytesIO() as buffer:
#            wavfile.write(buffer, 44100, audio_data)
#            st.markdown(get_binary_file_downloader_html(buffer.getvalue(), f"binaural_beat_{base_freq}_{beat_freq}Hz"), unsafe_allow_html=True)
#
#    # Ensure the app updates every second
#    if st.session_state.audio_playing:
#        time.sleep(1)
#        st.experimental_rerun()

def main():
    st.set_page_config(page_title="S.H.E.R.L.O.C.K. AI Buddy", page_icon="🕵️", layout="wide")
    
    # Custom CSS for improved styling
    st.markdown("""
    <style>
    .stApp {
        background-color: #f0f2f6;
    }
    .stButton>button {
        background-color: #4CAF50;
        color: white;
        font-weight: bold;
    }
    .stTextInput>div>div>input {
        background-color: #ffffff;
    }
    </style>
    """, unsafe_allow_html=True)
    
    st.title("🕵️ S.H.E.R.L.O.C.K. AI Buddy")
    st.markdown("Your personalized AI companion for conversation, therapy, and personal growth.")

    # Initialize session state
    if 'buddy_name' not in st.session_state:
        st.session_state.buddy_name = "Sherlock"
    if 'buddy_personality' not in st.session_state:
        st.session_state.buddy_personality = "Friendly, empathetic, and insightful"
    if 'buddy_details' not in st.session_state:
        st.session_state.buddy_details = "Knowledgeable about various therapy techniques and always ready to listen"
    if 'messages' not in st.session_state:
        st.session_state.messages = []

    # Sidebar for AI Buddy configuration and additional features
    with st.sidebar:
        st.header("🤖 Configure Your AI Buddy")
        st.session_state.buddy_name = st.text_input("Name your AI Buddy", value=st.session_state.buddy_name)
        st.session_state.buddy_personality = st.text_area("Describe your buddy's personality", value=st.session_state.buddy_personality)
        st.session_state.buddy_details = st.text_area("Additional details about your buddy", value=st.session_state.buddy_details)
        
        st.header("🧘 Therapy Session")
        therapy_mode = st.checkbox("Enable Therapy Mode")
        if therapy_mode:
            therapy_technique = st.selectbox("Select Therapy Technique", list(THERAPY_TECHNIQUES.keys()))
        else:
            therapy_technique = None
        
        st.markdown("---")
        
        st.subheader("📋 Todo List")
        if 'todos' not in st.session_state:
            st.session_state.todos = []
        
        new_todo = st.text_input("Add a new todo:")
        if st.button("Add Todo", key="add_todo"):
            if new_todo:
                st.session_state.todos.append({"task": new_todo, "completed": False})
                st.success("Todo added successfully!")
            else:
                st.warning("Please enter a todo item.")
        
        for i, todo in enumerate(st.session_state.todos):
            col1, col2, col3 = st.columns([0.05, 0.8, 0.15])
            with col1:
                todo['completed'] = st.checkbox("", todo['completed'], key=f"todo_{i}")
            with col2:
                st.write(todo['task'], key=f"todo_text_{i}")
            with col3:
                if st.button("🗑️", key=f"delete_{i}", help="Delete todo"):
                    st.session_state.todos.pop(i)
                    st.experimental_rerun()

        st.subheader("⏱️ Pomodoro Timer")
        pomodoro_duration = st.slider("Pomodoro Duration (minutes)", 1, 60, 25)
        if st.button("Start Pomodoro"):
            progress_bar = st.progress(0)
            for i in range(pomodoro_duration * 60):
                time.sleep(1)
                progress_bar.progress((i + 1) / (pomodoro_duration * 60))
            st.success("Pomodoro completed!")
            if 'achievements' not in st.session_state:
                st.session_state.achievements = set()
            st.session_state.achievements.add("Consistent Learner")
        
        st.markdown("---")
        st.markdown("Powered by Falcon-180B and Streamlit")

        st.markdown("---")
        st.header("📔 Daily Journal")
        journal_entry = st.text_area("Write your thoughts for today")
        if st.button("Save Journal Entry"):
            if 'journal_entries' not in st.session_state:
                st.session_state.journal_entries = []
            st.session_state.journal_entries.append({
                'date': datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
                'entry': journal_entry
            })
            st.success("Journal entry saved!")
            st.toast("Journal entry saved successfully!", icon="✅")
        
        if 'journal_entries' in st.session_state and st.session_state.journal_entries:
            st.subheader("Previous Entries")
            for entry in st.session_state.journal_entries[-5:]:  # Show last 5 entries
                st.text(entry['date'])
                st.write(entry['entry'])
                st.markdown("---")

    # Main content area
    tab1, tab2 = st.tabs(["Chat", "Tools"])
    
    with tab1:
        # Chat interface
        st.header("🗨️ Chat with Your AI Buddy")
        
        # Display chat history
        chat_container = st.container()
        with chat_container:
            for message in st.session_state.messages:
                with st.chat_message(message["role"]):
                    st.markdown(message["content"])
        
        # User input
        prompt = st.chat_input("What's on your mind?")
        
        # Clear chat history button
        if st.button("Clear Chat History"):
            st.session_state.messages = []
            st.experimental_rerun()

        if prompt:
            st.session_state.messages.append({"role": "user", "content": prompt})
            with st.chat_message("user"):
                st.markdown(prompt)

            buddy_config = {
                "name": st.session_state.buddy_name,
                "personality": st.session_state.buddy_personality,
                "details": st.session_state.buddy_details
            }

            with st.chat_message("assistant"):
                message_placeholder = st.empty()
                full_response = ""
                for chunk in chat.stream(get_ai_response(prompt, buddy_config, therapy_technique)):
                    full_response += chunk.content
                    message_placeholder.markdown(full_response + "▌")
                message_placeholder.markdown(full_response)
            st.session_state.messages.append({"role": "assistant", "content": full_response})

            # Force a rerun to update the chat history immediately
            st.experimental_rerun()

    with tab2:
        tool_choice = st.selectbox("Select a tool", ["Meditation Timer", "Binaural Beats", "Recommendations"])
        if tool_choice == "Meditation Timer":
            show_meditation_timer()
        elif tool_choice == "Recommendations":
            show_personalized_recommendations()
        elif tool_choice == "Binaural Beats":
            show_binaural_beats()

    # Mood tracker
    st.sidebar.markdown("---")
    st.sidebar.header("😊 Mood Tracker")
    mood = st.sidebar.slider("How are you feeling today?", 1, 10, 5)
    if st.sidebar.button("Log Mood"):
        st.sidebar.success(f"Mood logged: {mood}/10")
        st.balloons()

    # Resources and Emergency Contact
    st.sidebar.markdown("---")
    st.sidebar.header("🆘 Resources")
    st.sidebar.info("If you're in crisis, please reach out for help:")
    st.sidebar.markdown("- [Mental Health Resources](https://www.mentalhealth.gov/get-help/immediate-help)")
    st.sidebar.markdown("- Emergency Contact: 911 or your local emergency number")

    # Inspiration Quote
    st.sidebar.markdown("---")
    st.sidebar.header("💡 Daily Inspiration")
    if st.sidebar.button("Get Inspirational Quote"):
        quotes = [
            "The only way to do great work is to love what you do. - Steve Jobs",
            "Believe you can and you're halfway there. - Theodore Roosevelt",
            "The future belongs to those who believe in the beauty of their dreams. - Eleanor Roosevelt",
            "Strive not to be a success, but rather to be of value. - Albert Einstein",
            "The only limit to our realization of tomorrow will be our doubts of today. - Franklin D. Roosevelt",
            "Do not wait to strike till the iron is hot; but make it hot by striking. - William Butler Yeats",
            "What lies behind us and what lies before us are tiny matters compared to what lies within us. - Ralph Waldo Emerson",
            "Success is not final, failure is not fatal: It is the courage to continue that counts. - Winston Churchill",
            "Life is what happens when you're busy making other plans. - John Lennon",
            "You miss 100% of the shots you don't take. - Wayne Gretzky",
            "The best way to predict the future is to create it. - Peter Drucker",
            "It is not the strongest of the species that survive, nor the most intelligent, but the one most responsive to change. - Charles Darwin",
            "Whether you think you can or you think you can't, you're right. - Henry Ford",
            "The only place where success comes before work is in the dictionary. - Vidal Sassoon",
            "Do what you can, with what you have, where you are. - Theodore Roosevelt",
            "The purpose of our lives is to be happy. - Dalai Lama",
            "Success usually comes to those who are too busy to be looking for it. - Henry David Thoreau",
            "Your time is limited, so don't waste it living someone else's life. - Steve Jobs",
            "Don't be afraid to give up the good to go for the great. - John D. Rockefeller",
            "I find that the harder I work, the more luck I seem to have. - Thomas Jefferson",
            "Success is not the key to happiness. Happiness is the key to success. - Albert Schweitzer",
            "It does not matter how slowly you go, as long as you do not stop. - Confucius",
            "If you set your goals ridiculously high and it's a failure, you will fail above everyone else's success. - James Cameron",
            "Don't watch the clock; do what it does. Keep going. - Sam Levenson",
            "Hardships often prepare ordinary people for an extraordinary destiny. - C.S. Lewis",
            "Don't count the days, make the days count. - Muhammad Ali",
            "The best revenge is massive success. - Frank Sinatra",
            "The only impossible journey is the one you never begin. - Tony Robbins",
            "Act as if what you do makes a difference. It does. - William James",
            "You are never too old to set another goal or to dream a new dream. - C.S. Lewis",
            "If you're going through hell, keep going. - Winston Churchill",
            "Dream big and dare to fail. - Norman Vaughan",
            "In the middle of every difficulty lies opportunity. - Albert Einstein",
            "What we achieve inwardly will change outer reality. - Plutarch",
            "I have not failed. I've just found 10,000 ways that won't work. - Thomas Edison",
            "It always seems impossible until it's done. - Nelson Mandela",
            "The future depends on what you do today. - Mahatma Gandhi",
            "Don't wait. The time will never be just right. - Napoleon Hill",
            "Quality is not an act, it is a habit. - Aristotle",
            "Your life does not get better by chance, it gets better by change. - Jim Rohn",
            "The only thing standing between you and your goal is the story you keep telling yourself as to why you can't achieve it. - Jordan Belfort",
            "Challenges are what make life interesting; overcoming them is what makes life meaningful. - Joshua J. Marine",
            "Opportunities don't happen, you create them. - Chris Grosser",
            "I can't change the direction of the wind, but I can adjust my sails to always reach my destination. - Jimmy Dean",
            "Start where you are. Use what you have. Do what you can. - Arthur Ashe",
            "The secret of getting ahead is getting started. - Mark Twain",
            "You don’t have to be great to start, but you have to start to be great. - Zig Ziglar",
            "Keep your eyes on the stars, and your feet on the ground. - Theodore Roosevelt",
            "The only way to achieve the impossible is to believe it is possible. - Charles Kingsleigh"
        ]

        random_quote = random.choice(quotes)
        st.sidebar.success(random_quote)
        
    # Chat Export
    st.sidebar.markdown("---")
    if st.sidebar.button("Export Chat History"):
        chat_history = "\n".join([f"{msg['role']}: {msg['content']}" for msg in st.session_state.messages])
        st.sidebar.download_button(
            label="Download Chat History",
            data=chat_history,
            file_name="ai_buddy_chat_history.txt",
            mime="text/plain"
        )
        
        st.sidebar.success("Chat history ready for download!")

    # Display achievements
    if 'achievements' in st.session_state and st.session_state.achievements:
        st.sidebar.markdown("---")
        st.sidebar.header("🏆 Achievements")
        for achievement in st.session_state.achievements:
            st.sidebar.success(f"Unlocked: {achievement}")

if __name__ == "__main__":
    main()