mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-10-07 23:48:48 +08:00
78 lines
2.8 KiB
Python
78 lines
2.8 KiB
Python
import logging
|
|
import os
|
|
import platform
|
|
import traceback
|
|
import sys
|
|
import ffmpeg
|
|
import numpy as np
|
|
|
|
|
|
def load_audio(file, sr):
|
|
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 = clean_path(file) # 防止小白拷路径头尾带了空格和"和回车
|
|
if os.path.exists(file) == False:
|
|
raise RuntimeError(
|
|
"You input a wrong audio path that does not exists, please fix it!"
|
|
)
|
|
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:
|
|
traceback.print_exc()
|
|
raise RuntimeError(f"Failed to load audio: {e}")
|
|
|
|
return np.frombuffer(out, np.float32).flatten()
|
|
|
|
|
|
def clean_path(path_str):
|
|
if platform.system() == 'Windows':
|
|
path_str = path_str.replace('/', '\\')
|
|
return path_str.strip(" ").strip('"').strip("\n").strip('"').strip(" ")
|
|
|
|
COLORS = {
|
|
'WARNING' : '\033[33m', # Yellow
|
|
'ERROR' : '\033[31m', # Red
|
|
'CRITICAL': '\033[35m',
|
|
'RESET' : '\033[0m', # Reset color
|
|
}
|
|
|
|
class ColoredConsoleHandler(logging.StreamHandler):
|
|
def emit(self, record):
|
|
# 获取日志级别对应的颜色
|
|
color = COLORS.get(record.levelname, '')
|
|
# 重置颜色
|
|
reset = COLORS['RESET']
|
|
# 设置日志消息的颜色
|
|
self.setFormatter(logging.Formatter(color + '%(asctime)s - %(name)s - %(levelname)s - %(message)s' + reset))
|
|
# 输出日志消息
|
|
super().emit(record)
|
|
|
|
class Tools_Logger():
|
|
def __init__(self, logger_name, log_level='info', log_path=None):
|
|
assert type(log_level) == str and log_level.upper() in ['CRITICAL', 'ERROR', 'WARNING', 'INFO', 'DEBUG']
|
|
log_level = log_level.upper()
|
|
self.logger = logging.getLogger(logger_name)
|
|
self.logger.setLevel(log_level)
|
|
|
|
if not self.logger.hasHandlers():
|
|
ch = ColoredConsoleHandler()
|
|
ch.setLevel(log_level)
|
|
# formatter = logging.Formatter()
|
|
# ch.setFormatter(formatter)
|
|
self.logger.addHandler(ch)
|
|
|
|
# if log_path is not None:
|
|
# fh = logging.FileHandler(log_path)
|
|
# fh.setLevel(log_level)
|
|
# fh.setFormatter(formatter)
|
|
# self.logger.addHandler(fh)
|
|
|
|
def getLogger(self):
|
|
return self.logger
|
|
|
|
ASR_Logger = Tools_Logger('ASR').getLogger() |