Update pages/ai_buddy.py

pages/ai_buddy.py

@@ -9,7 +9,12 @@ from datetime import datetime
 import plotly.express as px
 import json
 import time
+import threading
 from playsound import playsound
+import pygame
+
+
+pygame.mixer.init()
 
 # Load environment variables
 load_dotenv()
@@ -53,51 +58,114 @@ def get_ai_response(user_input, buddy_config, therapy_technique=None):
     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", "Crackling Fireplace", "Flowing Stream",
+    "Birdsong", "White Noise", "Zen Garden", "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"
+]
+
 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_options = ["Forest", "Beach", "Rain", "White Noise"]
-        background_sound = st.selectbox("Background Sound", background_options)
+        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_options = ["Gentle Chime", "Tibetan Singing Bowl", "Ocean Wave"]
-        end_sound = st.selectbox("End of Session Sound", end_sound_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()
         
-        # Play background sound
-        background_sound_file = f"sounds/{background_sound.lower().replace(' ', '_')}.mp3"
-        playsound(background_sound_file, block=False)
+        # 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
         
-        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)
+        # Play background sound on loop
+        pygame.mixer.music.load(background_sound_file)
+        pygame.mixer.music.play(-1)  # -1 means loop indefinitely
         
-        # Stop background sound (you may need to implement this depending on how playsound works)
-        # stop_sound(background_sound_file)
+        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()
         
-        # 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")