Update pages/ai_buddy.py

pages/ai_buddy.py

@@ -61,119 +61,119 @@ 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_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 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")
     
@@ -243,164 +243,164 @@ def show_personalized_recommendations():
         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 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")