Update pages/ai_buddy.py

pages/ai_buddy.py

@@ -8,11 +8,14 @@ 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
 from playsound import playsound
 import pygame
-
+import sounddevice as sd
+from scipy.io import wavfile
 
 pygame.mixer.init()
 
@@ -240,6 +243,165 @@ 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 main():
     st.set_page_config(page_title="S.H.E.R.L.O.C.K. AI Buddy", page_icon="🕵️", layout="wide")
     
@@ -361,13 +523,15 @@ def main():
                 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()
 
-        # User input
-        if prompt := st.chat_input("What's on your mind?"):
+        if prompt:
             st.session_state.messages.append({"role": "user", "content": prompt})
             with st.chat_message("user"):
                 st.markdown(prompt)
@@ -387,12 +551,17 @@ def main():
                 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", "Recommendations"])
+        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("---")