Spaces:
Runtime error
Runtime error
| import librosa | |
| import numpy as np | |
| import av | |
| from io import BytesIO | |
| import ffmpeg | |
| import os | |
| import sys | |
| import random | |
| from infer.lib.csvutil import CSVutil | |
| #import csv | |
| platform_stft_mapping = { | |
| 'linux': 'stftpitchshift', | |
| 'darwin': 'stftpitchshift', | |
| 'win32': 'stftpitchshift.exe', | |
| } | |
| stft = platform_stft_mapping.get(sys.platform) | |
| def wav2(i, o, format): | |
| inp = av.open(i, 'rb') | |
| if format == "m4a": format = "mp4" | |
| out = av.open(o, 'wb', format=format) | |
| if format == "ogg": format = "libvorbis" | |
| if format == "mp4": format = "aac" | |
| ostream = out.add_stream(format) | |
| for frame in inp.decode(audio=0): | |
| for p in ostream.encode(frame): out.mux(p) | |
| for p in ostream.encode(None): out.mux(p) | |
| out.close() | |
| inp.close() | |
| def audio2(i, o, format, sr): | |
| inp = av.open(i, 'rb') | |
| out = av.open(o, 'wb', format=format) | |
| if format == "ogg": format = "libvorbis" | |
| if format == "f32le": format = "pcm_f32le" | |
| ostream = out.add_stream(format, channels=1) | |
| ostream.sample_rate = sr | |
| for frame in inp.decode(audio=0): | |
| for p in ostream.encode(frame): out.mux(p) | |
| out.close() | |
| inp.close() | |
| def load_audion(file, sr): | |
| try: | |
| file = ( | |
| file.strip(" ").strip('"').strip("\n").strip('"').strip(" ") | |
| ) # 防止小白拷路径头尾带了空格和"和回车 | |
| with open(file, "rb") as f: | |
| with BytesIO() as out: | |
| audio2(f, out, "f32le", sr) | |
| return np.frombuffer(out.getvalue(), np.float32).flatten() | |
| except AttributeError: | |
| audio = file[1] / 32768.0 | |
| if len(audio.shape) == 2: | |
| audio = np.mean(audio, -1) | |
| return librosa.resample(audio, orig_sr=file[0], target_sr=16000) | |
| except Exception as e: | |
| raise RuntimeError(f"Failed to load audio: {e}") | |
| def load_audio(file, sr, DoFormant=False, Quefrency=1.0, Timbre=1.0): | |
| converted = False | |
| DoFormant, Quefrency, Timbre = CSVutil("csvdb/formanting.csv", "r", "formanting") | |
| try: | |
| # https://github.com/openai/whisper/blob/main/whisper/audio.py#L26 | |
| # This launches a subprocess to decode audio while down-mixing and resampling as necessary. | |
| # Requires the ffmpeg CLI and `ffmpeg-python` package to be installed. | |
| file = ( | |
| file.strip(" ").strip('"').strip("\n").strip('"').strip(" ") | |
| ) # 防止小白拷路径头尾带了空格和"和回车 | |
| file_formanted = file.strip(" ").strip('"').strip("\n").strip('"').strip(" ") | |
| # print(f"dofor={bool(DoFormant)} timbr={Timbre} quef={Quefrency}\n") | |
| if ( | |
| lambda DoFormant: True | |
| if DoFormant.lower() == "true" | |
| else (False if DoFormant.lower() == "false" else DoFormant) | |
| )(DoFormant): | |
| numerator = round(random.uniform(1, 4), 4) | |
| # os.system(f"stftpitchshift -i {file} -q {Quefrency} -t {Timbre} -o {file_formanted}") | |
| # print('stftpitchshift -i "%s" -p 1.0 --rms -w 128 -v 8 -q %s -t %s -o "%s"' % (file, Quefrency, Timbre, file_formanted)) | |
| if not file.endswith(".wav"): | |
| if not os.path.isfile(f"{file_formanted}.wav"): | |
| converted = True | |
| # print(f"\nfile = {file}\n") | |
| # print(f"\nfile_formanted = {file_formanted}\n") | |
| converting = ( | |
| ffmpeg.input(file_formanted, threads=0) | |
| .output(f"{file_formanted}.wav") | |
| .run( | |
| cmd=["ffmpeg", "-nostdin"], | |
| capture_stdout=True, | |
| capture_stderr=True, | |
| ) | |
| ) | |
| else: | |
| pass | |
| file_formanted = ( | |
| f"{file_formanted}.wav" | |
| if not file_formanted.endswith(".wav") | |
| else file_formanted | |
| ) | |
| print(f" · Formanting {file_formanted}...\n") | |
| os.system( | |
| '%s -i "%s" -q "%s" -t "%s" -o "%sFORMANTED_%s.wav"' | |
| % ( | |
| stft, | |
| file_formanted, | |
| Quefrency, | |
| Timbre, | |
| file_formanted, | |
| str(numerator), | |
| ) | |
| ) | |
| print(f" · Formanted {file_formanted}!\n") | |
| # filepraat = (os.path.abspath(os.getcwd()) + '\\' + file).replace('/','\\') | |
| # file_formantedpraat = ('"' + os.path.abspath(os.getcwd()) + '/' + 'formanted'.join(file_formanted) + '"').replace('/','\\') | |
| # print("%sFORMANTED_%s.wav" % (file_formanted, str(numerator))) | |
| out, _ = ( | |
| ffmpeg.input( | |
| "%sFORMANTED_%s.wav" % (file_formanted, str(numerator)), threads=0 | |
| ) | |
| .output("-", format="f32le", acodec="pcm_f32le", ac=1, ar=sr) | |
| .run( | |
| cmd=["ffmpeg", "-nostdin"], capture_stdout=True, capture_stderr=True | |
| ) | |
| ) | |
| try: | |
| os.remove("%sFORMANTED_%s.wav" % (file_formanted, str(numerator))) | |
| except Exception: | |
| pass | |
| print("couldn't remove formanted type of file") | |
| else: | |
| out, _ = ( | |
| ffmpeg.input(file, threads=0) | |
| .output("-", format="f32le", acodec="pcm_f32le", ac=1, ar=sr) | |
| .run( | |
| cmd=["ffmpeg", "-nostdin"], capture_stdout=True, capture_stderr=True | |
| ) | |
| ) | |
| except Exception as e: | |
| raise RuntimeError(f"Failed to load audio: {e}") | |
| if converted: | |
| try: | |
| os.remove(file_formanted) | |
| except Exception: | |
| pass | |
| print("couldn't remove converted type of file") | |
| converted = False | |
| return np.frombuffer(out, np.float32).flatten() | |
| def check_audio_duration(file): | |
| try: | |
| file = file.strip(" ").strip('"').strip("\n").strip('"').strip(" ") | |
| probe = ffmpeg.probe(file) | |
| duration = float(probe['streams'][0]['duration']) | |
| if duration < 0.76: | |
| print( | |
| f"\n------------\n" | |
| f"Audio file, {file.split('/')[-1]}, under ~0.76s detected - file is too short. Target at least 1-2s for best results." | |
| f"\n------------\n\n" | |
| ) | |
| return False | |
| return True | |
| except Exception as e: | |
| raise RuntimeError(f"Failed to check audio duration: {e}") |