Spaces:
Running
on
Zero
Running
on
Zero
File size: 3,277 Bytes
593f3bc |
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 92 93 94 95 |
# Copyright 2025 ByteDance and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import io
import os
import subprocess
import numpy as np
from scipy.io import wavfile
import pyloudnorm as pyln
from pydub import AudioSegment
def to_wav_bytes(wav, sr, norm=False):
wav = wav.astype(float)
if norm:
meter = pyln.Meter(sr) # create BS.1770 meter
loudness = meter.integrated_loudness(wav)
wav = pyln.normalize.loudness(wav, loudness, -18.0)
if np.abs(wav).max() >= 1:
wav = wav / np.abs(wav).max() * 0.95
wav = wav * 32767
bytes_io = io.BytesIO()
wavfile.write(bytes_io, sr, wav.astype(np.int16))
return bytes_io.getvalue()
def save_wav(wav_bytes, path):
with open(path[:-4] + '.wav', 'wb') as file:
file.write(wav_bytes)
if path[-4:] == '.mp3':
to_mp3(path[:-4])
def to_mp3(out_path):
if out_path[-4:] == '.wav':
out_path = out_path[:-4]
subprocess.check_call(
f'ffmpeg -threads 1 -loglevel error -i "{out_path}.wav" -vn -b:a 192k -y -hide_banner -async 1 "{out_path}.mp3"',
shell=True, stdin=subprocess.PIPE)
subprocess.check_call(f'rm -f "{out_path}.wav"', shell=True)
def convert_to_wav(wav_path):
# Check if the file exists
if not os.path.exists(wav_path):
print(f"The file '{wav_path}' does not exist.")
return
# Check if the file already has a .wav extension
if not wav_path.endswith(".wav"):
# Define the output path with a .wav extension
out_path = os.path.splitext(wav_path)[0] + ".wav"
# Load the audio file using pydub and convert it to WAV
audio = AudioSegment.from_file(wav_path)
audio.export(out_path, format="wav")
print(f"Converted '{wav_path}' to '{out_path}'")
def convert_to_wav_bytes(audio_binary):
# Load the audio binary using pydub and convert it to WAV
audio = AudioSegment.from_file(io.BytesIO(audio_binary))
wav_bytes = io.BytesIO()
audio.export(wav_bytes, format="wav")
wav_bytes.seek(0)
return wav_bytes
''' Smoothly combine audio segments using crossfade transitions." '''
def combine_audio_segments(segments, crossfade_duration=0.16, sr=24000):
window_length = int(sr * crossfade_duration)
hanning_window = np.hanning(2 * window_length)
# Combine
for i, segment in enumerate(segments):
if i == 0:
combined_audio = segment
else:
overlap = combined_audio[-window_length:] * hanning_window[window_length:] + segment[:window_length] * hanning_window[:window_length]
combined_audio = np.concatenate(
[combined_audio[:-window_length], overlap, segment[window_length:]]
)
return combined_audio |