Lib/python/miko-recognize-long.py - master - mikopbx/ModuleSpeechRecognize

Lib/python/miko-recognize-long.py

26 строк1 KB

boffart

init

02 окт 2025, 12:39

02 окт 2025, 12:39107ee1d

#! /usr/bin/env python3
from tinkoff.cloud.stt.v1 import stt_pb2_grpc
from auth import authorization_metadata
from audio import audio_open_read
from common import build_recognition_request, make_channel, print_recognition_response, BaseRecognitionParser
import json
from google.protobuf.json_format import MessageToDict
from tinkoff.cloud.stt.v1 import stt_pb2

def main():
    args = BaseRecognitionParser().parse_args()
    if args.encoding == stt_pb2.RAW_OPUS:
        raise ValueError("RAW_OPUS encoding is not supported by this script")
    with audio_open_read(args.audio_file, args.encoding, args.rate, args.num_channels, args.chunk_size,
                         args.pyaudio_max_seconds) as reader:
        try:
            stub = stt_pb2_grpc.SpeechToTextStub(make_channel(args))
            metadata = authorization_metadata(args.api_key, args.secret_key, "tinkoff.cloud.stt")
            response = stub.LongRunningRecognize(build_recognition_request(args, reader), metadata=metadata)
        except Exception as inst:
            print(json.dumps(str(inst)))
        else:
            print(response.id)

if __name__ == "__main__":
    main()