Update app.py

app.py

@@ -1,10 +1,9 @@
 import streamlit as st
 import numpy as np
+from pydub import AudioSegment
+from pydub.generators import Sine
+import simpleaudio as sa
 import time
-import pygame
-
-# Initialize Pygame mixer
-pygame.mixer.init()
 
 # Constants
 KEYS = ['C', 'C#', 'D', 'D#', 'E', 'F', 'F#', 'G', 'G#', 'A', 'A#', 'B']
@@ -35,15 +34,26 @@ def get_chord_notes(root, chord_type):
         intervals = [0, 4, 7]
     return [KEYS[(root_index + interval) % len(KEYS)] for interval in intervals]
 
+def create_tone(freq, duration_ms):
+    return Sine(freq).to_audio_segment(duration=duration_ms)
+
 def play_arpeggio(root, chord_type, tempo):
     chord_notes = get_chord_notes(root, chord_type)
-    duration = 60 / tempo / 4  # 16th notes
+    duration_ms = int(60 / tempo * 250)  # 16th notes
+    
+    arpeggio = AudioSegment.silent(duration=1)
     for note in chord_notes:
         freq = note_to_freq(f"{note}4")
-        pygame.mixer.Sound(pygame.sndarray.make_sound(
-            (np.sin(2 * np.pi * np.arange(44100) * freq / 44100) * 32767).astype(np.int16)
-        )).play()
-        time.sleep(duration)
+        tone = create_tone(freq, duration_ms)
+        arpeggio = arpeggio.overlay(tone)
+    
+    play_obj = sa.play_buffer(
+        arpeggio.raw_data,
+        num_channels=arpeggio.channels,
+        bytes_per_sample=arpeggio.sample_width,
+        sample_rate=arpeggio.frame_rate
+    )
+    play_obj.wait_done()
 
 def main():
     st.title("12-Bar Blues Arpeggio Game")
@@ -70,8 +80,6 @@ def main():
     # Play/Stop button
     if st.button("Play/Stop 12-Bar Blues"):
         for i, chord in enumerate(st.session_state.blues_progression):
-            if st.stop_button:
-                break
             st.session_state.current_bar = i
             play_arpeggio(root_key, chord, tempo)
             st.session_state.chord_history.append(f"{root_key}{chord}")