import streamlit as st
import noisereduce as nr
import soundfile as sf
import io
from pydub import AudioSegment

# Define a Streamlit app
st.title("Audio Processing App")

# Upload the input audio file
uploaded_audio = st.file_uploader("Upload an audio file", type=["mp3", "wav", "ogg", "flac", "wma", "m4a"])

# Speed factor input
speed_factor = st.slider("Playback Speed", min_value=0.1, max_value=2.0, step=0.1, value=1.0)

if uploaded_audio is not None:
    audio_bytes = uploaded_audio.read()

    # Convert audio file to numpy array
    audio, sample_rate = sf.read(io.BytesIO(audio_bytes))

    # Apply noise reduction
    st.write("Applying noise reduction...")
    reduced_audio_data = nr.reduce_noise(y=audio, sr=sample_rate)

    # Create an AudioSegment from the reduced audio data
    reduced_audio = AudioSegment(
        reduced_audio_data.tobytes(),
        frame_rate=sample_rate,
        sample_width=reduced_audio_data.dtype.itemsize,
        channels=1
    )

    # Slow down the audio based on the user's input speed factor
    st.write(f"Slowing down audio to {speed_factor}x speed...")
    slowed_audio = reduced_audio.speedup(playback_speed=1/speed_factor)

    # Provide a link to download the processed audio
    st.audio(slowed_audio.export(format="wav").read(), format="audio/wav")

# Run the Streamlit app
if __name__ == "__main__":
    st.write("Upload an audio file, set the playback speed, and apply noise reduction.")