Update app.py

app.py

@@ -33,17 +33,40 @@ def generate_animation(script, output_video_file="animated_script.mp4"):
 
     return output_video_file
 
-# Function to generate simple music (for kids content)
+import numpy as np
+from scipy.io.wavfile import write
+from pydub import AudioSegment
+
+def generate_sine_wave(frequency, duration, sample_rate=44100, amplitude=0.5):
+    """Generate a sine wave."""
+    t = np.linspace(0, duration, int(sample_rate * duration), False)
+    wave = amplitude * np.sin(2 * np.pi * frequency * t)
+    return wave
+
 def generate_kids_music(theme, output_music_file="kids_music.wav"):
-    # Simple sine wave generation using pydub for kids' music
-    music = AudioSegment.sine(frequency=261.63)  # C4 note
-    music += AudioSegment.sine(frequency=293.66)  # D4 note
-    music += AudioSegment.sine(frequency=329.63)  # E4 note
+    # Generate a sine wave for C4, D4, E4 notes
+    sample_rate = 44100
+    duration = 2  # 2 seconds per note
+    
+    c4_wave = generate_sine_wave(261.63, duration)  # C4
+    d4_wave = generate_sine_wave(293.66, duration)  # D4
+    e4_wave = generate_sine_wave(329.63, duration)  # E4
+    
+    # Combine the waves
+    wave = np.concatenate([c4_wave, d4_wave, e4_wave])
+    
+    # Normalize to 16-bit PCM format
+    audio_wave = np.int16(wave * 32767)
+    
+    # Save as a wav file
+    write(output_music_file, sample_rate, audio_wave)
+    
+    # Convert to AudioSegment (optional if you need to modify)
+    audio_segment = AudioSegment.from_wav(output_music_file)
     
-    # Extend for more complexity, then export
-    music.export(output_music_file, format="wav")
     return output_music_file
 
+
 # Function to combine kids music with a simple animation
 def generate_kids_animation_with_music(theme, output_video_file="kids_animation.mp4"):
     # Step 1: Generate kids music