| import os |
| from pydub import AudioSegment |
| from pydub.silence import detect_silence, detect_nonsilent |
|
|
| SEPERATE_DIR = os.path.join(os.getcwd(), "seperate") |
| TEMP_DIR = os.path.join(SEPERATE_DIR, "temp") |
| cache = {} |
|
|
| os.makedirs(SEPERATE_DIR, exist_ok=True) |
| os.makedirs(TEMP_DIR, exist_ok=True) |
|
|
| def cache_result(func): |
| def wrapper(*args, **kwargs): |
| key = (args, frozenset(kwargs.items())) |
| if key in cache: |
| return cache[key] |
| else: |
| result = func(*args, **kwargs) |
| cache[key] = result |
| return result |
| return wrapper |
|
|
| def get_non_silent(audio_name, audio, min_silence, silence_thresh, seek_step, keep_silence): |
| """ |
| Function to get non-silent parts of the audio. |
| """ |
| nonsilent_ranges = detect_nonsilent(audio, min_silence_len=min_silence, silence_thresh=silence_thresh, seek_step=seek_step) |
| nonsilent_files = [] |
| for index, range in enumerate(nonsilent_ranges): |
| nonsilent_name = os.path.join(SEPERATE_DIR, f"{audio_name}_min{min_silence}_t{silence_thresh}_ss{seek_step}_ks{keep_silence}", f"nonsilent{index}-{audio_name}.wav") |
| start, end = range[0] - keep_silence, range[1] + keep_silence |
| audio[start:end].export(nonsilent_name, format="wav") |
| nonsilent_files.append(nonsilent_name) |
| return nonsilent_files |
|
|
| def get_silence(audio_name, audio, min_silence, silence_thresh, seek_step, keep_silence): |
| """ |
| Function to get silent parts of the audio. |
| """ |
| silence_ranges = detect_silence(audio, min_silence_len=min_silence, silence_thresh=silence_thresh, seek_step=seek_step) |
| silence_files = [] |
| for index, range in enumerate(silence_ranges): |
| silence_name = os.path.join(SEPERATE_DIR, f"{audio_name}_min{min_silence}_t{silence_thresh}_ss{seek_step}_ks{keep_silence}", f"silence{index}-{audio_name}.wav") |
| start, end = range[0] + keep_silence, range[1] - keep_silence |
| audio[start:end].export(silence_name, format="wav") |
| silence_files.append(silence_name) |
| return silence_files |
|
|
| @cache_result |
| def split_silence_nonsilent(input_path, min_silence=500, silence_thresh=-40, seek_step=1, keep_silence=100): |
| """ |
| Function to split the audio into silent and non-silent parts. |
| """ |
| audio_name = os.path.splitext(os.path.basename(input_path))[0] |
| os.makedirs(os.path.join(SEPERATE_DIR, f"{audio_name}_min{min_silence}_t{silence_thresh}_ss{seek_step}_ks{keep_silence}"), exist_ok=True) |
| audio = AudioSegment.silent(duration=1000) + AudioSegment.from_file(input_path) + AudioSegment.silent(duration=1000) |
| silence_files = get_silence(audio_name, audio, min_silence, silence_thresh, seek_step, keep_silence) |
| nonsilent_files = get_non_silent(audio_name, audio, min_silence, silence_thresh, seek_step, keep_silence) |
| return silence_files, nonsilent_files |
|
|
| def adjust_audio_lengths(original_audios, inferred_audios): |
| """ |
| Function to adjust the lengths of the inferred audio files list to match the original audio files length. |
| """ |
| adjusted_audios = [] |
| for original_audio, inferred_audio in zip(original_audios, inferred_audios): |
| audio_1 = AudioSegment.from_file(original_audio) |
| audio_2 = AudioSegment.from_file(inferred_audio) |
| |
| if len(audio_1) > len(audio_2): |
| audio_2 += AudioSegment.silent(duration=len(audio_1) - len(audio_2)) |
| else: |
| audio_2 = audio_2[:len(audio_1)] |
| |
| adjusted_file = os.path.join(TEMP_DIR, f"adjusted-{os.path.basename(inferred_audio)}") |
| audio_2.export(adjusted_file, format="wav") |
| adjusted_audios.append(adjusted_file) |
| |
| return adjusted_audios |
|
|
| def combine_silence_nonsilent(silence_files, nonsilent_files, keep_silence, output): |
| """ |
| Function to combine the silent and non-silent parts of the audio. |
| """ |
| combined = AudioSegment.empty() |
| for silence, nonsilent in zip(silence_files, nonsilent_files): |
| combined += AudioSegment.from_wav(silence) + AudioSegment.from_wav(nonsilent) |
| combined += AudioSegment.from_wav(silence_files[-1]) |
| combined = AudioSegment.silent(duration=keep_silence) + combined[1000:-1000] + AudioSegment.silent(duration=keep_silence) |
| combined.export(output, format="wav") |
| return output |
