mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-06-16 22:17:28 +08:00
lanvent
105 lines
3.1 KiB
Python
105 lines
3.1 KiB
Python
import shutil
|
|
import wave
|
|
import pysilk
|
|
from pydub import AudioSegment
|
|
|
|
sil_supports=[8000, 12000, 16000, 24000, 32000, 44100, 48000] # slk转wav时，支持的采样率
|
|
def find_closest_sil_supports(sample_rate):
|
|
"""
|
|
找到最接近的支持的采样率
|
|
"""
|
|
if sample_rate in sil_supports:
|
|
return sample_rate
|
|
closest = 0
|
|
mindiff = 9999999
|
|
for rate in sil_supports:
|
|
diff = abs(rate - sample_rate)
|
|
if diff < mindiff:
|
|
closest = rate
|
|
mindiff = diff
|
|
return closest
|
|
|
|
def get_pcm_from_wav(wav_path):
|
|
"""
|
|
从 wav 文件中读取 pcm
|
|
|
|
:param wav_path: wav 文件路径
|
|
:returns: pcm 数据
|
|
"""
|
|
wav = wave.open(wav_path, "rb")
|
|
return wav.readframes(wav.getnframes())
|
|
|
|
def any_to_wav(any_path, wav_path):
|
|
"""
|
|
把任意格式转成wav文件
|
|
"""
|
|
if any_path.endswith('.wav'):
|
|
shutil.copy2(any_path, wav_path)
|
|
return
|
|
if any_path.endswith('.sil') or any_path.endswith('.silk') or any_path.endswith('.slk'):
|
|
return sil_to_wav(any_path, wav_path)
|
|
audio = AudioSegment.from_file(any_path)
|
|
audio.export(wav_path, format="wav")
|
|
|
|
def any_to_sil(any_path, sil_path):
|
|
"""
|
|
把任意格式转成sil文件
|
|
"""
|
|
if any_path.endswith('.sil') or any_path.endswith('.silk') or any_path.endswith('.slk'):
|
|
shutil.copy2(any_path, sil_path)
|
|
return 10000
|
|
if any_path.endswith('.wav'):
|
|
return pcm_to_sil(any_path, sil_path)
|
|
if any_path.endswith('.mp3'):
|
|
return mp3_to_sil(any_path, sil_path)
|
|
raise NotImplementedError("Not support file type: {}".format(any_path))
|
|
|
|
def mp3_to_wav(mp3_path, wav_path):
|
|
"""
|
|
把mp3格式转成pcm文件
|
|
"""
|
|
audio = AudioSegment.from_mp3(mp3_path)
|
|
audio.export(wav_path, format="wav")
|
|
|
|
def pcm_to_sil(pcm_path, silk_path):
|
|
"""
|
|
wav 文件转成 silk
|
|
return 声音长度，毫秒
|
|
"""
|
|
audio = AudioSegment.from_wav(pcm_path)
|
|
rate = find_closest_sil_supports(audio.frame_rate)
|
|
# Convert to PCM_s16
|
|
pcm_s16 = audio.set_sample_width(2)
|
|
pcm_s16 = pcm_s16.set_frame_rate(rate)
|
|
wav_data = pcm_s16.raw_data
|
|
silk_data = pysilk.encode(
|
|
wav_data, data_rate=rate, sample_rate=rate)
|
|
with open(silk_path, "wb") as f:
|
|
f.write(silk_data)
|
|
return audio.duration_seconds * 1000
|
|
|
|
def mp3_to_sil(mp3_path, silk_path):
|
|
"""
|
|
mp3 文件转成 silk
|
|
return 声音长度，毫秒
|
|
"""
|
|
audio = AudioSegment.from_mp3(mp3_path)
|
|
rate = find_closest_sil_supports(audio.frame_rate)
|
|
# Convert to PCM_s16
|
|
pcm_s16 = audio.set_sample_width(2)
|
|
pcm_s16 = pcm_s16.set_frame_rate(rate)
|
|
wav_data = pcm_s16.raw_data
|
|
silk_data = pysilk.encode(wav_data, data_rate=rate, sample_rate=rate)
|
|
# Save the silk file
|
|
with open(silk_path, "wb") as f:
|
|
f.write(silk_data)
|
|
return audio.duration_seconds * 1000
|
|
|
|
def sil_to_wav(silk_path, wav_path, rate: int = 24000):
|
|
"""
|
|
silk 文件转 wav
|
|
"""
|
|
wav_data = pysilk.decode_file(silk_path, to_wav=True, sample_rate=rate)
|
|
with open(wav_path, "wb") as f:
|
|
f.write(wav_data)
|