Update pages/ai_buddy.py

pages/ai_buddy.py

@@ -1,368 +1,424 @@
-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 speech_recognition as sr
-from gtts import gTTS
-import pygame
-from io import BytesIO
-
-# Load environment variables
-load_dotenv()
-
-AI71_BASE_URL = "https://api.ai71.ai/v1/"
-AI71_API_KEY = "api71-api-92fc2ef9-9f3c-47e5-a019-18e257b04af2"
-
-# 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 show_progress_dashboard():
-    st.subheader("Your Progress Dashboard")
-    
-    # Placeholder data for demonstration
-    mood_logs = [{"date": "2023-08-01", "mood": 7}, {"date": "2023-08-02", "mood": 8}, {"date": "2023-08-03", "mood": 6}]
-    
-    # Mood trend
-    if mood_logs:
-        df = pd.DataFrame(mood_logs)
-        fig = px.line(df, x="date", y="mood", title="Mood Trend")
-        st.plotly_chart(fig)
-    else:
-        st.info("No mood data available yet. Start logging your mood!")
-    
-    # Journal entries
-    st.subheader("Recent Journal Entries")
-    journal_entries = [
-        {"date": "2023-08-03", "entry": "Had a productive day at work..."},
-        {"date": "2023-08-02", "entry": "Feeling a bit stressed about..."},
-        {"date": "2023-08-01", "entry": "Started a new book today..."}
-    ]
-    for entry in journal_entries[-5:]:
-        st.text(f"{entry['date']}: {entry['entry'][:50]}...")
-    
-    # Goals progress
-    st.subheader("Goals Progress")
-    goals = [
-        {"description": "Exercise 3 times a week", "progress": 60},
-        {"description": "Read 2 books this month", "progress": 75},
-        {"description": "Learn Python programming", "progress": 40}
-    ]
-    for goal in goals:
-        st.progress(goal['progress'])
-        st.text(f"{goal['description']}: {goal['progress']}% complete")
-
-def show_goal_setting_interface():
-    st.subheader("Set Your Goals")
-    goal_description = st.text_input("Describe your goal")
-    goal_progress = st.slider("Initial progress", 0, 100, 0)
-    
-    if st.button("Add Goal"):
-        st.success("Goal added successfully!")
-
-def show_meditation_timer():
-    st.subheader("Meditation Timer")
-    duration = st.slider("Select duration (minutes)", 1, 60, 5)
-    if st.button("Start Meditation"):
-        progress_bar = st.progress(0)
-        for i in range(duration * 60):
-            progress_bar.progress((i + 1) / (duration * 60))
-            st.empty().text(f"Time remaining: {duration - (i // 60)}:{59 - (i % 60):02d}")
-            time.sleep(1)
-        st.success("Meditation complete!")
-
-def show_community_forum():
-    st.subheader("Community Forum")
-    st.warning("This feature is not yet implemented. Check back soon!")
-
-def show_personalized_recommendations():
-    st.subheader("Personalized Recommendations")
-    
-    # Placeholder for demonstration
-    avg_mood = 6
-    
-    if avg_mood < 4:
-        st.markdown("Based on your recent mood, we recommend:")
-        st.markdown("- Practice daily gratitude journaling")
-        st.markdown("- Try a guided meditation for stress relief")
-        st.markdown("- Read 'The Happiness Trap' by Russ Harris")
-    elif avg_mood < 7:
-        st.markdown("To maintain and improve your mood, consider:")
-        st.markdown("- Start a new hobby or learn a new skill")
-        st.markdown("- Practice mindfulness meditation")
-        st.markdown("- Read 'Atomic Habits' by James Clear")
-    else:
-        st.markdown("Great job maintaining a positive mood! To keep it up:")
-        st.markdown("- Share your positivity with others")
-        st.markdown("- Set ambitious goals for personal growth")
-        st.markdown("- Read 'Flow' by Mihaly Csikszentmihalyi")
-
-def start_voice_interaction():
-    st.subheader("Voice Interaction")
-    if st.button("Start Voice Recognition"):
-        r = sr.Recognizer()
-        with sr.Microphone() as source:
-            st.info("Listening... Speak now!")
-            audio = r.listen(source)
-            try:
-                text = r.recognize_google(audio)
-                st.success(f"You said: {text}")
-                
-                # Get AI response
-                buddy_config = {
-                    "name": st.session_state.buddy_name,
-                    "personality": st.session_state.buddy_personality,
-                    "details": st.session_state.buddy_details
-                }
-                response = get_ai_response(text, buddy_config)
-                
-                # Convert response to speech
-                tts = gTTS(text=response, lang='en')
-                fp = BytesIO()
-                tts.write_to_fp(fp)
-                fp.seek(0)
-                
-                # Play the response
-                pygame.mixer.init()
-                pygame.mixer.music.load(fp)
-                pygame.mixer.music.play()
-                while pygame.mixer.music.get_busy():
-                    pygame.time.Clock().tick(10)
-                
-                st.success(f"AI response: {response}")
-            except sr.UnknownValueError:
-                st.error("Sorry, I couldn't understand what you said.")
-            except sr.RequestError as e:
-                st.error(f"Could not request results; {e}")
-
-def show_daily_challenge():
-    st.subheader("Daily Personal Growth Challenge")
-    challenges = [
-        "Write down three things you're grateful for today.",
-        "Reach out to a friend or family member you haven't spoken to in a while.",
-        "Try a new healthy recipe for dinner tonight.",
-        "Take a 15-minute walk in nature and practice mindfulness.",
-        "Learn five new words in a language you're interested in.",
-        "Declutter one area of your living space.",
-        "Practice active listening in your next conversation.",
-        "Try a new form of exercise or physical activity.",
-        "Write a short story or poem expressing your current emotions.",
-        "Perform a random act of kindness for someone."
-    ]
-    
-    if 'daily_challenge' not in st.session_state:
-        st.session_state.daily_challenge = random.choice(challenges)
-    
-    st.info(f"Your challenge for today: {st.session_state.daily_challenge}")
-    
-    if st.button("Complete Challenge"):
-        st.success("Great job completing today's challenge! Keep up the good work.")
-        st.session_state.daily_challenge = random.choice(challenges)
-
-def show_therapist_scheduling_interface():
-    st.subheader("Schedule a Video Therapy Session")
-    st.warning("This feature would integrate with a real therapist booking system. For demonstration purposes, we'll use a simplified version.")
-    
-    therapists = ["Dr. Smith", "Dr. Johnson", "Dr. Williams", "Dr. Brown"]
-    selected_therapist = st.selectbox("Choose a therapist", therapists)
-    
-    date = st.date_input("Select a date")
-    time = st.time_input("Select a time")
-    
-    if st.button("Schedule Session"):
-        st.success(f"Session scheduled with {selected_therapist} on {date} at {time}. You will receive a confirmation email with further details.")
-
-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
-    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.markdown("Powered by Falcon-180B and Streamlit")
-
-    # Main content area
-    tab1, tab2, tab3, tab4, tab5 = st.tabs(["Chat", "Progress", "Goals", "Community", "Tools"])
-    
-    with tab1:
-        # Chat interface
-        st.header("🗨️ Chat with Your AI Buddy")
-        chat_container = st.container()
-        with chat_container:
-            for message in st.session_state.messages[-5:]:  # Display last 5 messages
-                with st.chat_message(message["role"]):
-                    st.markdown(message["content"])
-
-        # User input
-        if prompt := st.chat_input("What's on your mind?"):
-            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})
-
-    with tab2:
-        show_progress_dashboard()
-
-    with tab3:
-        show_goal_setting_interface()
-
-    with tab4:
-        show_community_forum()
-
-    with tab5:
-        tool_choice = st.selectbox("Select a tool", ["Meditation Timer", "Recommendations", "Voice Interaction", "Daily Challenge", "Schedule Therapy"])
-        if tool_choice == "Meditation Timer":
-            show_meditation_timer()
-        elif tool_choice == "Recommendations":
-            show_personalized_recommendations()
-        elif tool_choice == "Voice Interaction":
-            start_voice_interaction()
-        elif tool_choice == "Daily Challenge":
-            show_daily_challenge()
-        elif tool_choice == "Schedule Therapy":
-            show_therapist_scheduling_interface()
-
-    # 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()
-
-    # Journaling feature
-    st.sidebar.markdown("---")
-    st.sidebar.header("📔 Daily Journal")
-    journal_entry = st.sidebar.text_area("Write your thoughts for today")
-    if st.sidebar.button("Save Journal Entry"):
-        st.sidebar.success("Journal entry saved!")
-        st.toast("Journal entry saved successfully!", icon="✅")
-
-    # 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"
-        ]
-        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!")
-
-if __name__ == "__main__":
+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 time
+from playsound import playsound
+
+# Load environment variables
+load_dotenv()
+
+AI71_BASE_URL = "https://api.ai71.ai/v1/"
+AI71_API_KEY = "api71-api-92fc2ef9-9f3c-47e5-a019-18e257b04af2"
+
+# 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 show_meditation_timer():
+    st.subheader("🧘‍♀️ Enhanced Meditation Timer")
+    col1, col2 = st.columns(2)
+    
+    with col1:
+        duration = st.slider("Select duration (minutes)", 1, 60, 5)
+        background_options = ["Forest", "Beach", "Rain", "White Noise"]
+        background_sound = st.selectbox("Background Sound", background_options)
+        
+    with col2:
+        interval_options = ["None", "Every 5 minutes", "Every 10 minutes"]
+        interval_reminder = st.selectbox("Interval Reminders", interval_options)
+        end_sound_options = ["Gentle Chime", "Tibetan Singing Bowl", "Ocean Wave"]
+        end_sound = st.selectbox("End of Session Sound", end_sound_options)
+    
+    if st.button("Start Meditation", key="start_meditation"):
+        progress_bar = st.progress(0)
+        status_text = st.empty()
+        
+        # Play background sound
+        background_sound_file = f"sounds/{background_sound.lower().replace(' ', '_')}.mp3"
+        playsound(background_sound_file, block=False)
+        
+        for i in range(duration * 60):
+            progress_bar.progress((i + 1) / (duration * 60))
+            mins, secs = divmod(duration * 60 - i - 1, 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 i > 0 and i % (interval * 60) == 0:
+                    st.toast(f"{interval} minutes passed", icon="⏰")
+            
+            time.sleep(1)
+        
+        # Stop background sound (you may need to implement this depending on how playsound works)
+        # stop_sound(background_sound_file)
+        
+        st.success("Meditation complete!")
+        st.balloons()
+        
+        # Play end of session sound
+        end_sound_file = f"sounds/{end_sound.lower().replace(' ', '_')}.mp3"
+        playsound(end_sound_file)
+        
+        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 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"])
+        
+        # Clear chat history button
+        if st.button("Clear Chat History"):
+            st.session_state.messages = []
+            st.experimental_rerun()
+
+        # User input
+        if prompt := st.chat_input("What's on your mind?"):
+            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})
+
+    with tab2:
+        tool_choice = st.selectbox("Select a tool", ["Meditation Timer", "Recommendations"])
+        if tool_choice == "Meditation Timer":
+            show_meditation_timer()
+        elif tool_choice == "Recommendations":
+            show_personalized_recommendations()
+
+    # 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()