Spaces:
Running
Running
| 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 | |
| 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 |