Spaces:
Running
Running
| import os | |
| import librosa | |
| import numpy as np | |
| import gradio as gr | |
| import soundfile as sf | |
| from pedalboard import ( | |
| Pedalboard, Reverb, Compressor, HighpassFilter, | |
| LowShelfFilter, HighShelfFilter, NoiseGate, Chorus | |
| ) | |
| from pedalboard.io import AudioFile | |
| from pydub import AudioSegment | |
| BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) | |
| def display_progress(percent, message, progress=gr.Progress()): | |
| progress(percent, desc=message) | |
| def combine_audio(vocal_path, instrumental_path, output_path, vocal_gain, instrumental_gain, output_format): | |
| vocal_format = vocal_path.split('.')[-1] | |
| instrumental_format = instrumental_path.split('.')[-1] | |
| vocal = AudioSegment.from_file(vocal_path, format=vocal_format) | |
| instrumental = AudioSegment.from_file(instrumental_path, format=instrumental_format) | |
| vocal += vocal_gain | |
| instrumental += instrumental_gain | |
| combined = vocal.overlay(instrumental) | |
| combined.export(output_path, format=output_format) | |
| def add_audio_effects(vocal_audio_path, instrumental_audio_path, reverb_rm_size, reverb_wet, reverb_dry, reverb_damping, reverb_width, | |
| low_shelf_gain, high_shelf_gain, compressor_ratio, compressor_threshold, noise_gate_threshold, noise_gate_ratio, | |
| noise_gate_attack, noise_gate_release, chorus_rate_hz, chorus_depth, chorus_centre_delay_ms, chorus_feedback, | |
| chorus_mix, output_format, vocal_gain, instrumental_gain, progress=gr.Progress()): | |
| if not vocal_audio_path or not instrumental_audio_path: | |
| raise ValueError("Оба пути к аудиофайлам должны быть заполнены.") | |
| display_progress(0.2, "Применение аудиоэффектов к вокалу...", progress) | |
| board = Pedalboard( | |
| [ | |
| HighpassFilter(), | |
| Compressor(ratio=compressor_ratio, threshold_db=compressor_threshold), | |
| NoiseGate(threshold_db=noise_gate_threshold, ratio=noise_gate_ratio, attack_ms=noise_gate_attack, release_ms=noise_gate_release), | |
| Reverb(room_size=reverb_rm_size, dry_level=reverb_dry, wet_level=reverb_wet, damping=reverb_damping, width=reverb_width), | |
| LowShelfFilter(gain_db=low_shelf_gain), | |
| HighShelfFilter(gain_db=high_shelf_gain), | |
| Chorus(rate_hz=chorus_rate_hz, depth=chorus_depth, centre_delay_ms=chorus_centre_delay_ms, feedback=chorus_feedback, mix=chorus_mix), | |
| ] | |
| ) | |
| vocal_output_path = f'Vocal_Effects.wav' | |
| with AudioFile(vocal_audio_path) as f: | |
| with AudioFile(vocal_output_path, 'w', f.samplerate, 2) as o: | |
| while f.tell() < f.frames: | |
| chunk = f.read(int(f.samplerate)) | |
| chunk = np.tile(chunk, (2, 1)).T | |
| effected = board(chunk, f.samplerate, reset=False) | |
| o.write(effected) | |
| display_progress(0.5, "Объединение вокала и инструментальной части...", progress) | |
| output_dir = os.path.join(BASE_DIR, 'processed_output') | |
| if not os.path.exists(output_dir): | |
| os.makedirs(output_dir) | |
| combined_output_path = os.path.join(output_dir, f'AiCover_combined.{output_format}') | |
| if os.path.exists(combined_output_path): | |
| os.remove(combined_output_path) | |
| combine_audio(vocal_output_path, instrumental_audio_path, combined_output_path, vocal_gain, instrumental_gain, output_format) | |
| display_progress(1.0, "Готово!", progress) | |
| return combined_output_path | |