Fix wav check and upload in script

This commit is contained in:
Aleksey Filippov 2022-01-14 14:12:58 +04:00
parent f03d307115
commit 8b05336a3c

27
main.py
View File

@ -1,37 +1,26 @@
#!/usr/bin/env python3
import json
import struct
import sys
import wave
from io import BytesIO
from urllib.request import Request, urlopen
from scipy.io import wavfile
from scipy.io.wavfile import read as read_wav
def stt(wav_file: str, url: str) -> str:
print('Connecting to \'{}\'...'.format(url))
request = Request('{}/stt'.format(url), data=_load_wav(wav_file), headers={'Content-Type': 'audio/wav'})
print(f'Connecting to \'{url}\'...')
request = Request(f'{url}/stt', data=_load_wav(wav_file), headers={'Content-Type': 'audio/wav'})
result = json.loads(urlopen(request).read().decode('utf-8'))
if not ('code' in result and 'text' in result):
raise RuntimeError('Wrong reply from server: {}'.format(result))
return result['text'] if not result['code'] else 'Server error: [{code}]: {text}'.format(**result)
raise RuntimeError(f'Wrong reply from server: {result}')
return result.text if not result.code else f'Server error: [{result.text}]: {result.code}'
def _load_wav(wav_file, convert_rate=16000, convert_width=2, channels=1):
def _load_wav(wav_file):
_check_wav(wav_file)
with wave.open(wav_file, 'rb') as in_:
src_data = in_.readframes(in_.getnframes())
with BytesIO() as file:
with wave.open(file, 'wb') as out:
out.setframerate(convert_rate)
out.setsampwidth(convert_width)
out.setnchannels(channels)
out.writeframes(src_data)
result = file.getvalue()
with open(wav_file, 'rb') as file:
result = file.read()
return result
@ -53,7 +42,7 @@ def _main():
exit(1)
file = sys.argv[1]
server = 'http://127.0.0.1:8086' if len(sys.argv) < 3 else sys.argv[2]
print('Result: {}'.format(stt(file, server)))
print(f'Result: {stt(file, server)}')
if __name__ == '__main__':