mosibi/RVC_HFv2

1from pedalboard import Pedalboard, Compressor, Reverb, NoiseGate2from pedalboard.io import AudioFile3import sys4import os5now_dir = os.getcwd()6sys.path.append(now_dir)7from i18n import I18nAuto8i18n = I18nAuto()9from pydub import AudioSegment10import numpy as np11import soundfile as sf12from pydub.playback import play13 14def process_audio(input_path, output_path, reverb_enabled, compressor_enabled, noise_gate_enabled, ):15    print(reverb_enabled)16    print(compressor_enabled)17    print(noise_gate_enabled)18    effects = []19    if reverb_enabled:20        effects.append(Reverb(room_size=0.01))21    if compressor_enabled:22        effects.append(Compressor(threshold_db=-10, ratio=25))23    if noise_gate_enabled:24        effects.append(NoiseGate(threshold_db=-16, ratio=1.5, release_ms=250))25 26    board = Pedalboard(effects)27 28    with AudioFile(input_path) as f:29        with AudioFile(output_path, 'w', f.samplerate, f.num_channels) as o:30            while f.tell() < f.frames:31                chunk = f.read(f.samplerate)32                effected = board(chunk, f.samplerate, reset=False)33                o.write(effected)34 35    result = i18n("Processed audio saved at: ") +  output_path36    print(result)37    return output_path