Spaces:
Running
Running
File size: 2,825 Bytes
5690e11 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import http.client
import json
from typing import List
from config.config import my_config
from services.alinls.token import getToken
from tools.utils import must_have_value
class AliRecognitionResult:
def __init__(self, text, begin_time, end_time):
self.text = text
self.begin_time = begin_time
self.end_time = end_time
class AliRecognitionService:
def __init__(self):
self.ALI_ACCESS_AKID = my_config['audio']['Ali']['access_key_id']
self.ALI_ACCESS_AKKEY = my_config['audio']['Ali']['access_key_secret']
self.ALI_APP_KEY = my_config['audio']['Ali']['app_key']
must_have_value(self.ALI_ACCESS_AKID, "请设置Ali access key id")
must_have_value(self.ALI_ACCESS_AKKEY, "请设置Ali access key secret")
must_have_value(self.ALI_APP_KEY, "请设置Ali app key")
self.token = getToken(self.ALI_ACCESS_AKID, self.ALI_ACCESS_AKKEY)
self.format = "wav"
self.sampleRate = 16000
self.url = 'https://nls-gateway-cn-shanghai.aliyuncs.com/stream/v1/FlashRecognizer'
def process(self, audioFile) -> List[AliRecognitionResult]:
result_list = []
# 设置RESTful请求参数
request = self.url + '?appkey=' + self.ALI_APP_KEY
request = request + '&token=' + self.token
request = request + '&format=' + self.format
request = request + '&sample_rate=' + str(self.sampleRate)
# 读取音频文件
with open(audioFile, mode='rb') as f:
audioContent = f.read()
host = 'nls-gateway-cn-shanghai.aliyuncs.com'
# 设置HTTP请求头部
httpHeaders = {
'Content-Length': len(audioContent)
}
# Python 3.x使用http.client
conn = http.client.HTTPSConnection(host)
conn.request(method='POST', url=request, body=audioContent, headers=httpHeaders)
response = conn.getresponse()
print('Response status and response reason:')
print(response.status, response.reason)
body = response.read()
try:
print('Recognize response is:')
body = json.loads(body)
print(body)
status = body['status']
if status == 20000000:
result = body['flash_result']
# print(result)
# result = json.loads(result)
if 'sentences' in result:
for sentence in result['sentences']:
result_list.append(
AliRecognitionResult(sentence['text'], sentence['begin_time'], sentence['end_time']))
else:
print('Recognizer failed!')
except ValueError:
print('The response is not json format string')
conn.close()
return result_list
|