KdZddlmZddlmZddlmZddlZddlmZddlm Z ddl m Z dd l m Z dd l m Z dd lmZdd lmZdd lmZdZdZdZGdde j.ZGddeZGddeZdZefdZdZdZy)z(Wrapper for interacting with speech API.)absolute_import)division)unicode_literalsN) storage_util)apis) exceptions)log) properties) console_io)files)urllibspeechv1zl[{}] is not a valid format for result output. Must be a Google Cloud Storage URI (format: gs://bucket/file).ceZdZdZy)ErrorzExceptions for this module.N__name__ __module__ __qualname____doc__0lib/googlecloudsdk/command_lib/ml/speech/util.pyrr,s#rrceZdZdZy)AudioExceptionzRaised if audio is not found.Nrrrrrr0s%rrceZdZdZy)UriFormatErrorz&Error if the specified URI is invalid.Nrrrrrr4s.rrcPtjt|}|j}tj j |rtj||_ |Stjj|r ||_ |Stdj|)zGDetermine whether path to audio is local, set RecognitionAudio message.z}Invalid audio source [{}]. The source must either be a local path or a Google Cloud Storage URL (such as gs://bucket/object).)rGetMessagesModule SPEECH_APIRecognitionAudioospathisfiler ReadBinaryFileContentscontentrObjectReference IsStorageUrlurirformat)r#versionmessagesaudios rGetRecognitionAudioFromPathr.8s  # #J 8(  # # %%WW^^D006EM ,##006EI ,  FFLf G rcfd}|S)zFReturns a hook to get the RecognitionAudio message for an API version.ct|S)aJDetermine whether path to audio is local, build RecognitionAudio message. Args: path: str, the path to the audio. Raises: AudioException: If audio is not found locally and does not appear to be Google Cloud Storage URL. Returns: speech_v1_messages.RecognitionAudio, the audio message. )r.)r#r+s rGetAudioFromPathz&GetAudioHook..GetAudioFromPathKs 'tW 55rr)r+r1s` r GetAudioHookr2Is 6 rc|r=tjj|sttj ||S)zValidates given output URI against validator function. Args: output_uri: str, the output URI for the analysis. Raises: UriFormatError: if the URI is not valid. Returns: str, The same output_uri. )rr'r(rOUTPUT_ERROR_MESSAGEr*) output_uris rValidateOutputUrir6\s7 44AA*M -44Z@ AA rc6tjs2tjjj j ry|jj}|sytjj|jdr|ddn||jj|jj|jj |jj"|jj$|jj&d}dj)dt+|j-D}ddd |d d f}tjj/|}t0j2j5d j7|y) zGPrint Url to the Speech-to-text UI console for given recognize request.Nzgs://)r-encodingmodellocalesamplingchannelsenhanced;c3K|]6\}}|r/dt|jvrdj||8yw) unspecifiedz{}={}N)strlowerr*).0keyvalues r z&MaybePrintSttUiLink..s?L(? e-s5z7G7G7I"INN3.(?s<>httpszconsole.cloud.google.comz/speech/transcriptions/createz*Try this using the Speech-to-Text UI at {})r IsRunFromShellScriptr VALUEScoredisable_promptsGetBoolr-r)r parse quote_plus startswithconfigr9r: languageCodesampleRateHertzaudioChannelCount useEnhancedjoinsorteditems urlunparser statusPrintr*)request audio_uripayloadparams url_tuple target_urls rMaybePrintSttUiLinkrcmsA%%',,446 mm)   ,, ! !(33Grls /&' 7,*#*2* 6 $J  $&U&/U/",&"!Gr