o dZh'@sddlmZddlZddlmZmZmZddlmZm Z ddl m Z ddl m Z mZddlmZddlmZdd lmZmZGd d d eZGd d d eZGdddeZedZeeefZdZGdddZddddd2ddZGd d!d!Zd3d&d'Z   ( (d4ed)d5d0d1Z!dS)6) annotationsN)DictLiteral TypedDict) HTTPErrorURLError) urlencode)Requesturlopen) NotRequired) AudioData) RequestErrorUnknownValueErrorc@eZdZUded<ded<dS) Alternativestr transcriptfloat confidenceN__name__ __module__ __qualname____annotations__rrT/var/www/auris/lib/python3.10/site-packages/speech_recognition/recognizers/google.pyr  rc@r)Resultlist[Alternative] alternativeboolfinalNrrrrrrrrc@r)GoogleResponsez list[Result]resultzNotRequired[int]Z result_indexNrrrrrr"rr")rz-http://www.google.com/speech-api/v2/recognizec@sLeZdZdd d ZdddZdddZd ddZd!ddZed"ddZ dS)#RequestBuilderendpointrkeylanguage filter_levelProfanityFilterLevelreturnNonecCs||_||_||_||_dSNr&r'r(r))selfr&r'r(r)rrr__init__%s zRequestBuilder.__init__ audio_datar r cCs@t|ts td|}||}||}t|||d}|S)Nz!``audio_data`` must be audio data)dataheaders) isinstancer ValueError build_url build_headers build_datar )r/r1urlr3 flac_datarequestrrrbuild2s   zRequestBuilder.buildcCs(td|j|j|jd}|jd|S)a* >>> builder = RequestBuilder(endpoint="http://www.google.com/speech-api/v2/recognize", key="awesome-key", language="en-US", filter_level=0) >>> builder.build_url() 'http://www.google.com/speech-api/v2/recognize?client=chromium&lang=en-US&key=awesome-key&pFilter=0' Zchromium)clientlangr'ZpFilter?)rr(r'r)r&)r/paramsrrrr6<szRequestBuilder.build_urlRequestHeaderscCs|j}dd|i}|S)z >>> builder = RequestBuilder(endpoint="", key="", language="", filter_level=1) >>> audio_data = AudioData(b"", 16_000, 1) >>> builder.build_headers(audio_data) {'Content-Type': 'audio/x-flac; rate=16000'} z Content-Typezaudio/x-flac; rate= sample_rate)r/r1Zrater3rrrr7LszRequestBuilder.build_headersbytescCs|j||jdd}|S)N)Z convert_rateZ convert_width)Z get_flac_datato_convert_raterC)r/r1r:rrrr8Ws  zRequestBuilder.build_datarCintcCs|dkrdSdS)zAudio samples must be at least 8 kHz >>> RequestBuilder.to_convert_rate(16_000) >>> RequestBuilder.to_convert_rate(8_000) >>> RequestBuilder.to_convert_rate(7_999) 8000 i@NrrBrrrrF^s zRequestBuilder.to_convert_rateN) r&rr'rr(rr)r*r+r,)r1r r+r )r+r)r1r r+rA)r1r r+rD)rCrGr+rG) rrrr0r<r6r7r8 staticmethodrFrrrrr%$s   r%en-US)r'r(r)r&rr' str | Noner(r)r*r+cCsHt|ts td|durt|tstd|durd}t||||dS)Nz``language`` must be a stringz$``key`` must be ``None`` or a stringz'AIzaSyBOti4mM-6x9WDnZIjIeyEU21OpBXqWBgwr.)r4rr5r%r.rrrcreate_request_builderjs rKc@s<eZdZdddZdd d Zedd dZedddZdS) OutputParsershow_allr with_confidencer+r,cCs||_||_dSr-rMrN)r/rMrNrrrr0s zOutputParser.__init__ response_textrcCsH||}|jr |S||d}|dd}|jr |d|fS|dS)Nrrg?r)convert_to_resultrMfind_best_hypothesisgetrN)r/rPZ actual_resultbest_hypothesisrrrrparses   zOutputParser.parsercCs`|dD]'}|s qt|d}t|dkr,t|ddgdkr&t|dSqt)a >>> response_text = '''{"result":[]} ... {"result":[{"alternative":[{"transcript":"one two three","confidence":0.49585345},{"transcript":"1 2","confidence":0.42899391}],"final":true}],"result_index":0} ... ''' >>> OutputParser.convert_to_result(response_text) {'alternative': [{'transcript': 'one two three', 'confidence': 0.49585345}, {'transcript': '1 2', 'confidence': 0.42899391}], 'final': True} >>> OutputParser.convert_to_result("") Traceback (most recent call last): ... speech_recognition.exceptions.UnknownValueError >>> OutputParser.convert_to_result('\n{"result":[]}') Traceback (most recent call last): ... speech_recognition.exceptions.UnknownValueError >>> OutputParser.convert_to_result('{"result":[{"foo": "bar"}]}') Traceback (most recent call last): ... speech_recognition.exceptions.UnknownValueError >>> OutputParser.convert_to_result('{"result":[{"alternative": []}]}') Traceback (most recent call last): ... speech_recognition.exceptions.UnknownValueError  r#rr)splitjsonloadslenrSr)rPliner#rrrrQs  zOutputParser.convert_to_result alternativesrrcCs4d|vr t|ddd}n|d}d|vrt|S)a >>> alternatives = [{"transcript": "one two three", "confidence": 0.42899391}, {"transcript": "1 2", "confidence": 0.49585345}] >>> OutputParser.find_best_hypothesis(alternatives) {'transcript': 'one two three', 'confidence': 0.42899391} >>> alternatives = [{"confidence": 0.49585345}] >>> OutputParser.find_best_hypothesis(alternatives) Traceback (most recent call last): ... speech_recognition.exceptions.UnknownValueError rcSs|dS)Nrr)rrrrsz3OutputParser.find_best_hypothesis..)r'rr)maxr)r\rTrrrrRs z!OutputParser.find_best_hypothesisN)rMr rNr r+r,)rPr)rPrr+r)r\rr+r)rrrr0rUrHrQrRrrrrrLs   $rLr;r timeoutrGc Csnzt||d}Wn'ty}ztd|jd}~wty/}ztd|jd}~ww|dS)Nr_zrecognition request failed: {}z!recognition connection failed: {}zutf-8)r rr formatreasonrreaddecode)r;r_responseerrrobtain_transcriptions rgF)r&r1r pfilterrMr rNc Cs>t||||d}||} t| |jd} t||d} | | S)aPerforms speech recognition on ``audio_data`` (an ``AudioData`` instance), using the Google Speech Recognition API. The Google Speech Recognition API key is specified by ``key``. If not specified, it uses a generic key that works out of the box. This should generally be used for personal or testing purposes only, as it **may be revoked by Google at any time**. To obtain your own API key, simply following the steps on the `API Keys `__ page at the Chromium Developers site. In the Google Developers Console, Google Speech Recognition is listed as "Speech API". The recognition language is determined by ``language``, an RFC5646 language tag like ``"en-US"`` (US English) or ``"fr-FR"`` (International French), defaulting to US English. A list of supported language tags can be found in this `StackOverflow answer `__. The profanity filter level can be adjusted with ``pfilter``: 0 - No filter, 1 - Only shows the first character and replaces the rest with asterisks. The default is level 0. Returns the most likely transcription if ``show_all`` is false (the default). Otherwise, returns the raw API response as a JSON dictionary. Raises a ``speech_recognition.UnknownValueError`` exception if the speech is unintelligible. Raises a ``speech_recognition.RequestError`` exception if the speech recognition operation failed, if the key isn't valid, or if there is no internet connection. r.r`rO)rKr<rgZoperation_timeoutrLrU) Z recognizerr1r'r(rhrMrNr&Zrequest_builderr;rPZ output_parserrrrrecognize_legacys  ri) r&rr'rJr(rr)r*r+r%)r;r r_rGr+r)NrIrFF)r1r r'rJr(rrhr*rMr rNr r&r)" __future__rrXtypingrrr urllib.errorrr urllib.parserurllib.requestr r Ztyping_extensionsr Zspeech_recognition.audior Zspeech_recognition.exceptionsr rrrr"r*rrAZENDPOINTr%rKrLrgrirrrrs<     I U