|
|
import requests
|
|
|
import pyaudio
|
|
|
import wave
|
|
|
import json
|
|
|
import os
|
|
|
import serial
|
|
|
from serial.tools import list_ports
|
|
|
def send_data_to_com(com, data_to_send):
|
|
|
# 创建Serial对象，设置串口参数
|
|
|
ser = serial.Serial(
|
|
|
port=com, # 串口名称
|
|
|
baudrate=115200, # 波特率，根据实际情况设置
|
|
|
parity=serial.PARITY_NONE, # 奇偶校验，默认无
|
|
|
stopbits=serial.STOPBITS_ONE, # 停止位，默认1
|
|
|
bytesize=serial.EIGHTBITS, # 数据位，默认8
|
|
|
timeout=None # 没有超时，阻塞读取
|
|
|
)
|
|
|
|
|
|
# 检查串口是否成功打开
|
|
|
if not ser.isOpen():
|
|
|
print("无法打开串口!")
|
|
|
return
|
|
|
|
|
|
print("串口 COM3 已打开...")
|
|
|
|
|
|
# 写入数据
|
|
|
ser.write(data_to_send.encode()) # 必须将数据转换为字节，例如字符串需用encode()方法
|
|
|
|
|
|
# 为了让示例更完整，添加一个关闭串口的步骤
|
|
|
ser.close()
|
|
|
print("串口已关闭")
|
|
|
def find_serial_ports():
|
|
|
"""Find and return a list of available serial ports."""
|
|
|
available_ports = list_ports.comports()
|
|
|
for port in available_ports:
|
|
|
print(f"Found: {port.device} - {port.description}")
|
|
|
return available_ports
|
|
|
def get_audio(filepath, aa):
|
|
|
if aa == str("是") :
|
|
|
CHUNK = 256
|
|
|
FORMAT = pyaudio.paInt16
|
|
|
CHANNELS = 1 # 声道数
|
|
|
RATE = 11025 # 采样率
|
|
|
RECORD_SECONDS = 5
|
|
|
WAVE_OUTPUT_FILENAME = filepath
|
|
|
p = pyaudio.PyAudio()
|
|
|
|
|
|
stream = p.open(format=FORMAT,
|
|
|
channels=CHANNELS,
|
|
|
rate=RATE,
|
|
|
input=True,
|
|
|
frames_per_buffer=CHUNK)
|
|
|
|
|
|
print("*"*10, "开始录音：请在5秒内输入语音")
|
|
|
frames = []
|
|
|
for i in range(0, int(RATE / CHUNK * RECORD_SECONDS)):
|
|
|
data = stream.read(CHUNK)
|
|
|
frames.append(data)
|
|
|
print("*"*10, "录音结束\n")
|
|
|
|
|
|
stream.stop_stream()
|
|
|
stream.close()
|
|
|
p.terminate()
|
|
|
|
|
|
wf = wave.open(WAVE_OUTPUT_FILENAME, 'wb')
|
|
|
wf.setnchannels(CHANNELS)
|
|
|
wf.setsampwidth(p.get_sample_size(FORMAT))
|
|
|
wf.setframerate(RATE)
|
|
|
wf.writeframes(b''.join(frames))
|
|
|
wf.close()
|
|
|
else:
|
|
|
print("无效输入，请重新选择")
|
|
|
exit()
|
|
|
if __name__ == "__main__":
|
|
|
com_string = find_serial_ports()
|
|
|
com = str(com_string[0]).split()[0]
|
|
|
# 请求地址
|
|
|
url = "http://127.0.0.1:9977/api"
|
|
|
input_filename = "input.wav" # 麦克风采集的语音输入
|
|
|
input_filepath = "audio" # 输入文件的path
|
|
|
in_path = os.path.join(input_filepath, input_filename)
|
|
|
while True:
|
|
|
aa = str(input("是否开始录音?（是/否）"))
|
|
|
if aa == "否":
|
|
|
break
|
|
|
get_audio(in_path, aa)
|
|
|
# 请求参数 file:音视频文件，language：语言代码，model：模型，response_format:text|json|srt
|
|
|
# 返回 code==0 成功，其他失败，msg==成功为ok，其他失败原因，data=识别后返回文字
|
|
|
files = {"file": open(in_path, "rb")}
|
|
|
data={"language":"zh","model":"small","response_format":"json"}
|
|
|
response = requests.request("POST", url, timeout=600, data=data,files=files)
|
|
|
try:
|
|
|
data = json.loads(response.json()['data']['answer'])['instruction']
|
|
|
send_data_to_com(com, data)
|
|
|
print(data)
|
|
|
except:
|
|
|
print("描述信息不充分，请修改明确描述问题。")
|