Refactor voice recording and transcription to allow for specifying the language.

This commit is contained in:
Paul Gauthier 2023-08-14 14:11:03 -03:00
parent f4e1c1010b
commit 23579d4741
2 changed files with 9 additions and 4 deletions

View file

@ -14,9 +14,14 @@ from .dump import dump # noqa: F401
class Commands:
def __init__(self, io, coder):
def __init__(self, io, coder, voice_language=None):
self.io = io
self.coder = coder
if voice_language == 'auto':
voice_language=None
self.voice_language = voice_language
self.tokenizer = tiktoken.encoding_for_model(coder.main_model.name)
def is_command(self, inp):
@ -460,7 +465,7 @@ class Commands:
history = "\n".join(history)
dump(history)
text = v.record_and_transcribe(history)
text = v.record_and_transcribe(history, self.voice_language)
if text:
self.io.add_to_input_history(text)
print()