from warnings import filterwarnings
import streamlit as st
import base64
from pydub import AudioSegment
from functions import *
filterwarnings('ignore')


st.set_page_config(
    page_title="Sing It Forward App",
    page_icon="🎵")

st.markdown(
    """
    <style>
    body {
        background: linear-gradient(to bottom, #0E5AAB, #00ffff);
        padding: 20px;
        border-radius: 10px;
    }
    a {
        color: #EDA67C !important
    }    
    </style>
    """,
    unsafe_allow_html=True
)

def load_image(image_file):
    with open(image_file, "rb") as f:
        return f.read()

image_data = load_image("bcg.jpg")  
image_base64 = base64.b64encode(image_data).decode()

st.markdown(
    f"""
    <style>
    .stApp {{
        background-image: url(data:image/jpeg;base64,{image_base64});
        background-size: cover;
        background-position: center;
        background-repeat: no-repeat;
    }}
    </style>
    """,
    unsafe_allow_html=True
)

st.markdown("<h1 style='text-align: center; margin-bottom: 5px;'>Sing It Forward App🎵</h1>", unsafe_allow_html=True)

description = """
<h5>Welcome to Sing It Forward App!</h5>

<p style="text-align: justify;">
Get ready to test your singing skills and memory! First, listen carefully to the first part of the song, then it’s your turn to shine.
Record yourself singing the next 15 seconds on your own, matching the lyrics and rhythm perfectly. Think you’ve got what it takes to keep the music going?
Let’s see if you can hit the right notes and showcase your talent! Unleash your inner star and take the challenge!
</p>

📌For any questions or contact:  
**Name:** <span style="color: #EDA67C;">Sahand Khorsandi</span>  
**Email:** <a href="mailto:sahand.kh78@yahoo.com" style="color: #EDA67C;">sahand.kh78@yahoo.com</a>"""
st.markdown(description, unsafe_allow_html=True)
st.write('------')




tab1, tab2 = st.tabs(["Take Challenge", "Make Challenge"])

with tab1:
    lyrics = "Far across the distance And spaces between us You have come to show you go on"
    take_challenge("titanic.mp3", lyrics, key="en_challenge", language="en")

    
with tab2:
    st.write("Upload music to make challenge:")
    uploaded_file = st.file_uploader("Choose a music file", type=["mp3", "wav"])
    language_mapping = {"English": "en", "Persian": "fa"}
    selected_language = st.radio("Select Language", language_mapping.keys(), horizontal=True)
    language = language_mapping[selected_language]
    if uploaded_file is not None:
        with open("raw_music.mp3", "wb") as f:
            f.write(uploaded_file.getbuffer())
        st.audio("raw_music.mp3")

        current_input = "raw_music.mp3"
        output_file = "processed_music.mp3"

        trimm_check = st.checkbox("Trim")
        if trimm_check:   
            st.write("Specify start and end times for trimming:")
            audio = AudioSegment.from_mp3(current_input)
            duration = len(audio) // 1000 
            start_time = st.number_input("Start Time (seconds)", min_value=0, max_value=duration, value=0)
            end_time = st.number_input("End Time (seconds)", min_value=0, max_value=duration, value=duration)
            if start_time < end_time:
                trimmed_audio = audio[start_time * 1000:end_time * 1000]
                trimmed_audio.export(output_file, format="mp3")
                current_input = output_file
            else:
                st.error('Start Time should be smaller than End Time!', icon="❌")
                st.markdown('<style>div.stAlert { background-color: rgba(241, 36, 36, 0.9); }</style>', unsafe_allow_html=True)

        volume_checkbox = st.checkbox("Change Volume")
        if volume_checkbox:
            volume_factor = st.slider("Volume Factor (dB)", -30, 30, 0)
            change_volume(current_input, output_file, volume_factor)
            current_input = output_file

        speed_checkbox = st.checkbox("Change Speed")
        if speed_checkbox:
            speed_factor = st.slider("Speed Factor", 0.25, 2.0, 1.0)
            change_speed(current_input, output_file, speed_factor)
            current_input = output_file

        pitch_checkbox = st.checkbox("Change Pitch")
        if pitch_checkbox:
            pitch_factor = st.slider("Pitch Shift (fractional steps)", -12, 12, 0)
            change_pitch(current_input, output_file, pitch_factor)
            current_input = output_file

        low_pass_checkbox = st.checkbox("Low Pass Filter")
        if low_pass_checkbox:
            cutoff_freq = st.slider("Low Pass Filter Cutoff Frequency", min_value=20, max_value=20000, value=2000)
            low_pass_filter(current_input, output_file, cutoff_freq)
            current_input = output_file

        high_pass_checkbox = st.checkbox("High Pass Filter")
        if high_pass_checkbox:
            cutoff_freq = st.slider("High Pass Filter Cutoff Frequency", min_value=20, max_value=20000, value=2000)
            high_pass_filter(current_input, output_file, cutoff_freq)
            current_input = output_file

        pan_checkbox = st.checkbox("Pan Left/Right")
        if pan_checkbox:
            pan_factor = st.slider("Pan Factor (-1 for Left, 1 for Right)", -1.0, 1.0, 0.0)
            pan_left_right(current_input, output_file, pan_factor)
            current_input = output_file

        fade_in_checkbox = st.checkbox("Fade In")
        if fade_in_checkbox:
            fade_in_time = st.slider("Fade In Duration (ms)", min_value=0, max_value=10000, value=1000)
            fade_in_ms(current_input, output_file, fade_in_time)
            current_input = output_file

        fade_out_checkbox = st.checkbox("Fade Out")
        if fade_out_checkbox:
            fade_out_time = st.slider("Fade Out Duration (ms)", min_value=0, max_value=10000, value=1000)
            fade_out_ms(current_input, output_file, fade_out_time)
            current_input = output_file

        music_background_checkbox = st.checkbox("Keep Music Background?")

        st.write("Now type what user should sing:")
        typed_lyrics = st.text_area("Lyrics to be singed:")
        st.write('------')
        if music_background_checkbox:
            result_list = split_vocals(current_input)
            take_challenge(current_input, typed_lyrics, "unique_key_1", language, music_background_checkbox, result_list[0])
        else:
            take_challenge(current_input, typed_lyrics, "unique_key_1", language, music_background_checkbox)