preprocess/fai/noisereduce.py

import noisereduce as nr
from pedalboard import Compressor, Gain, LowShelfFilter, NoiseGate, Pedalboard
from preprocess.fai.base import AudioPreprocessor


class RNAudioPreprocessor(AudioPreprocessor):
    def __init__(self, audio, sr):
        super().__init__(audio, sr)

    def preprocess(self):
        reduced_noise = nr.reduce_noise(
            y=self.audio, sr=self.sr, stationary=True, prop_decrease=0.75
        )
        return reduced_noise

    def enhance(self):
        board = Pedalboard(
            [
                NoiseGate(threshold_db=-30, ratio=1.5, release_ms=250),
                Compressor(threshold_db=-16, ratio=4),
                LowShelfFilter(cutoff_frequency_hz=400, gain_db=10, q=1),
                Gain(gain_db=2),
            ]
        )
        enhanced = board(self.reduced_noise, self.sr)
        return enhanced