mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-02-19 00:37:23 +08:00
71 lines
1.8 KiB
Python
71 lines
1.8 KiB
Python
import wave
|
|
import pysilk
|
|
from pydub import AudioSegment
|
|
|
|
|
|
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 mp3_to_wav(mp3_path, wav_path):
|
|
"""
|
|
把mp3格式转成pcm文件
|
|
"""
|
|
audio = AudioSegment.from_mp3(mp3_path)
|
|
audio.export(wav_path, format="wav")
|
|
|
|
def any_to_wav(any_path, wav_path):
|
|
"""
|
|
把任意格式转成wav文件
|
|
"""
|
|
if any_path.endswith('.wav'):
|
|
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 pcm_to_silk(pcm_path, silk_path):
|
|
"""
|
|
wav 文件转成 silk
|
|
return 声音长度，毫秒
|
|
"""
|
|
audio = AudioSegment.from_wav(pcm_path)
|
|
wav_data = audio.raw_data
|
|
silk_data = pysilk.encode(
|
|
wav_data, data_rate=audio.frame_rate, sample_rate=audio.frame_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)
|
|
wav_data = audio.raw_data
|
|
silk_data = pysilk.encode(
|
|
wav_data, data_rate=audio.frame_rate, sample_rate=audio.frame_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)
|