[voice] add support for whisper-1 model

This commit is contained in:
wanggang
2023-03-08 11:32:27 +08:00
parent 882e6c3576
commit d7a8854fa1
4 changed files with 13 additions and 7 deletions

View File

@@ -4,19 +4,21 @@ google voice service
"""
import json
import openai
from config import conf
from common.log import logger
from voice.voice import Voice
class OpenaiVoice(Voice):
def __init__(self):
pass
openai.api_key = conf().get('open_ai_api_key')
def voiceToText(self, voice_file):
logger.debug(
'[Openai] voice file name={}'.format(voice_file))
file = open(voice_file, "rb")
reply = openai.Audio.transcribe("whisper-1", file)
json_dict = json.loads(reply)
text = json_dict['text']
text = reply["text"]
logger.info(
'[Openai] voiceToText text={} voice file name={}'.format(text, voice_file))
return text