from gtts import gTTS

from pydub import AudioSegment

from pydub.generators import Sine

# Lyrics (Burmese)

lyrics = """

မနက်အလင်းရောင်တွေ ချွတ်ချွတ်လက်နဲ့လာ

ကျောင်းမဂေါက်ဝတံတားထဲ ဝင်လာကြမယ်

စာအုပ်နဲ့ အိပ်မက်တွေကို ထမ်းဆောင်ပြီး

ကျွန်တော်နဲ့ ကျောင်း နေ့တိုင်းအတူရှိနေမယ်

"""

# Step 1: Generate voice (ballad style Burmese TTS)

tts = gTTS(lyrics, lang="my")

tts.save("school_song.mp3")

# Step 2: Add Jersey style beat (simple kick-snare pattern)

voice = AudioSegment.from_mp3("school_song.mp3")

kick = Sine(60).to_audio_segment(duration=150).apply_gain(-5)

snare = Sine(180).to_audio_segment(duration=80).apply_gain(-10)

beat = (kick + AudioSegment.silent(100) + snare + AudioSegment.silent(100)) * 20

beat = beat - 12 # lower volume

# Step 3: Mix voice + beat

final = voice.overlay(beat)

final.export("school_song_with_beat.mp3", format="mp3")

print("✅ mp3 ဖိုင် ထွက်ပြီးပါပြီ: school_song_with_beat.mp3")