o 0’hã @sddlZddlZddlZddlmZmZmZmZddlm Z ddl mZmZddl Z ddlZddlmZddlZdejd<e dd ZeƒZd ZeedƒZeejjd dd ddd dZeed dZGdd„dƒZdd„Zdede jfdd„Ze d¡defdd„ƒZ!dS)éN)Ú APIRouterÚ WebSocketÚDependsÚWebSocketDisconnect)ÚOAuth2PasswordBearer)ÚRecognitionConfigÚStreamingRecognitionConfig)Úspeechzmedical_memo.jsonÚGOOGLE_APPLICATION_CREDENTIALSÚtoken)ÚtokenUrli€>é éTzja-JPÚcommand_and_search)ÚencodingÚsample_rate_hertzÚaudio_channel_countÚuse_enhancedÚ language_codeÚmodelÚenable_spoken_punctuation)ÚconfigÚinterim_resultsc@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd S)ÚSpeechClientBridgecCs ||_t ¡|_d|_||_dS)NF)Ú_on_responseÚqueueÚQueueÚ_queueÚ_endedÚstreaming_config)ÚselfrÚon_response©r"ú%/home/air/yokohama/back/stt_router.pyÚ__init__ s zSpeechClientBridge.__init__cCsrd|_||_||_t ¡}| ¡}dd„|Dƒ}| |j|¡}| |¡|j ¡s7t |j ¡ƒ|j ¡r)dSdS)NFcss|] }tj|dVqdS))Ú audio_contentN)r ÚStreamingRecognizeRequest)Ú.0Úcontentr"r"r#Ú ,s € ÿ ÿz+SpeechClientBridge.start..)rÚqÚthreadr ÚSpeechClientÚ generatorÚstreaming_recognizerÚprocess_responses_loopÚemptyÚprintÚget)r r*r+ÚclientÚstreamÚrequestsÚ responsesr"r"r#Ústart&sþ ÿzSpeechClientBridge.startcCs d|_dS)NT)r)r r"r"r#Ú terminate5s zSpeechClientBridge.terminatecCs|jjt|ƒdddS)NF©Úblock)rÚputÚbytes)r Úbufferr"r"r#Úadd_request8szSpeechClientBridge.add_requestcCs,|D]}| ||j|j¡|jrdSqdS)N)rr*r+r)r r6Úresponser"r"r#r/;sÿþz)SpeechClientBridge.process_responses_loopccs‚|js?|j ¡}|durdS|g} z|jjdd}|dur"WdS| |¡Wn tjy2Ynwqd |¡V|jrdSdS)NTFr9ó)rrr2ÚappendrÚEmptyÚjoin)r ÚchunkÚdatar"r"r#r-As$€ ÿúózSpeechClientBridge.generatorN) Ú__name__Ú __module__Ú__qualname__r$r7r8r>r/r-r"r"r"r#rsrcCs¢d}|jsdS|jd}|jsdS|jdj}d|t|ƒ}|js?tj ||d¡tj ¡t|ƒ}| d|dœ¡dSt d||ƒ| d|dœ¡dS)Nrú ú F©Úis_finalÚ transcriptz==>T)ÚresultsÚalternativesrMÚlenrLÚsysÚstdoutÚwriteÚflushr;r1)r?r*Ú websocketÚnum_chars_printedÚresultrMÚoverwrite_charsr"r"r#Úon_transcription_responseRs rYrUÚasync_qcÃsJz | ¡IdH}| |d|ddœ¡IdHqtjy$YdSw)NTrLrMrK)r2Ú send_jsonÚasyncioÚCancelledError)rUrZÚ send_datar"r"r#Úsend_transcriptiongs€þþÿr_z/wsc Ãs(| ¡IdHtdƒt ¡}t ¡}tttƒ}t j |j|j|fd}| ¡t t||jƒ¡}z+ | ¡IdH}d|vrK|d}| |¡IdHnd|vr[|d}|rZ| |¡n q3tyr| ¡| ¡tdƒYdSty“}z| ¡| ¡td|›ƒWYd}~dSd}~ww)NÚopened)ÚtargetÚargsTÚtextr<ÚclosedzException: )Úacceptr1r\Úget_event_loopÚjanusrrrrYÚ threadingÚThreadr7Úsync_qÚcreate_taskr_rZÚreceiveÚ send_textr>rr8ÚcancelÚ Exception) rUÚloopr*ÚbridgeÚ bridge_threadÚ send_taskrEÚmessageÚer"r"r#Úwebsocket_endpointss@€ €ö€ýrv)"r\rhrQÚfastapirrrrÚfastapi.securityrÚgoogle.cloud.speechrrrgrÚgoogle.cloudr ÚosÚenvironÚ oauth2_schemeÚrouterÚRATEÚintÚCHUNKÚ AudioEncodingÚLINEAR16rrrrYÚ_AsyncQueueProxyr_rUrvr"r"r"r#Ús< ù 3