import librosa
from src.config import N_MELS, SR

# chosen to be able to use for modeling downstream 

def load_audio(audio_fp, sr=None, res_type='soxr_hq'):
    wave, sr = librosa.load(audio_fp, sr=sr, res_type=res_type)
    return wave, sr


def get_melspec(y, sr=None, plot=False):
    if not sr: sr = SR  # default
    mel_power = librosa.feature.melspectrogram(y=y, sr=sr, n_mels=N_MELS, hop_length=1000)
    mel_dB = librosa.power_to_db(mel_power)
    if plot:
        pass
    return mel_dB