File size: 4,280 Bytes
736c789 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
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 |