mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-05 08:06:15 +00:00
153 lines
4.5 KiB
Python
153 lines
4.5 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
macOS 麦克风录音脚本
|
||
需要安装: pip install pyaudio wave
|
||
"""
|
||
|
||
import pyaudio
|
||
import wave
|
||
import sys
|
||
import os
|
||
from datetime import datetime
|
||
|
||
|
||
class AudioRecorder:
|
||
"""macOS 音频录制器"""
|
||
|
||
def __init__(self, output_dir="recordings"):
|
||
"""
|
||
初始化录音器
|
||
|
||
Args:
|
||
output_dir: 录音文件保存目录
|
||
"""
|
||
self.output_dir = output_dir
|
||
self.chunk = 1024 # 每次读取的音频块大小
|
||
self.format = pyaudio.paInt16 # 16位深度
|
||
self.channels = 1 # 单声道
|
||
self.rate = 44100 # 采样率 44.1kHz
|
||
|
||
# 创建输出目录
|
||
if not os.path.exists(output_dir):
|
||
os.makedirs(output_dir)
|
||
|
||
def record(self, duration=5, filename=None):
|
||
"""
|
||
录制音频
|
||
|
||
Args:
|
||
duration: 录制时长（秒）
|
||
filename: 输出文件名，如果为None则自动生成
|
||
|
||
Returns:
|
||
str: 保存的文件路径
|
||
"""
|
||
# 生成文件名
|
||
if filename is None:
|
||
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||
filename = f"recording_{timestamp}.wav"
|
||
|
||
filepath = os.path.join(self.output_dir, filename)
|
||
|
||
# 初始化PyAudio
|
||
audio = pyaudio.PyAudio()
|
||
|
||
try:
|
||
# 打开音频流（这会触发macOS的麦克风权限请求）
|
||
print("正在请求麦克风权限...")
|
||
stream = audio.open(
|
||
format=self.format,
|
||
channels=self.channels,
|
||
rate=self.rate,
|
||
input=True,
|
||
frames_per_buffer=self.chunk
|
||
)
|
||
|
||
print(f"开始录音，时长: {duration} 秒")
|
||
print("录音中...")
|
||
|
||
frames = []
|
||
|
||
# 录制音频
|
||
for i in range(0, int(self.rate / self.chunk * duration)):
|
||
data = stream.read(self.chunk)
|
||
frames.append(data)
|
||
|
||
# 显示进度
|
||
progress = (i + 1) / (self.rate / self.chunk * duration) * 100
|
||
sys.stdout.write(f"\r进度: {progress:.1f}%")
|
||
sys.stdout.flush()
|
||
|
||
print("\n录音完成！")
|
||
|
||
# 停止并关闭流
|
||
stream.stop_stream()
|
||
stream.close()
|
||
|
||
# 保存为WAV文件
|
||
print(f"正在保存到: {filepath}")
|
||
wf = wave.open(filepath, 'wb')
|
||
wf.setnchannels(self.channels)
|
||
wf.setsampwidth(audio.get_sample_size(self.format))
|
||
wf.setframerate(self.rate)
|
||
wf.writeframes(b''.join(frames))
|
||
wf.close()
|
||
|
||
print(f"✓ 文件已保存: {filepath}")
|
||
return filepath
|
||
|
||
except Exception as e:
|
||
print(f"\n错误: {e}")
|
||
print("\n提示:")
|
||
print("1. 请确保已安装 pyaudio: pip install pyaudio")
|
||
print("2. 在macOS上，首次运行会弹出权限请求对话框")
|
||
print("3. 如果权限被拒绝，请前往 系统偏好设置 > 安全性与隐私 > 隐私 > 麦克风")
|
||
return None
|
||
|
||
finally:
|
||
audio.terminate()
|
||
|
||
def record_interactive(self):
|
||
"""交互式录音"""
|
||
print("=" * 50)
|
||
print("macOS 麦克风录音工具")
|
||
print("=" * 50)
|
||
|
||
try:
|
||
duration = input("\n请输入录音时长（秒，默认5秒）: ").strip()
|
||
duration = int(duration) if duration else 5
|
||
|
||
filename = input("请输入文件名（留空自动生成）: ").strip()
|
||
filename = filename if filename else None
|
||
if filename and not filename.endswith('.wav'):
|
||
filename += '.wav'
|
||
|
||
print()
|
||
self.record(duration=duration, filename=filename)
|
||
|
||
except KeyboardInterrupt:
|
||
print("\n\n录音已取消")
|
||
except ValueError:
|
||
print("输入无效，请输入数字")
|
||
|
||
|
||
def main():
|
||
"""主函数"""
|
||
recorder = AudioRecorder()
|
||
|
||
if len(sys.argv) > 1:
|
||
# 命令行模式
|
||
try:
|
||
duration = int(sys.argv[1])
|
||
filename = sys.argv[2] if len(sys.argv) > 2 else None
|
||
recorder.record(duration=duration, filename=filename)
|
||
except ValueError:
|
||
print("用法: python record_audio.py [时长(秒)] [文件名(可选)]")
|
||
print("示例: python record_audio.py 10 my_recording.wav")
|
||
else:
|
||
# 交互式模式
|
||
recorder.record_interactive()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|