o h@sddlZddlZddlZddlmZmZmZmZddlm Z ddl m Z m Z ddl Z ddlZddlmZddlZddlmZdejd<e d d ZeZd Zeed ZGd ddZddZdede jfddZedddedefddZ dS)N) APIRouter WebSocketDependsWebSocketDisconnect)OAuth2PasswordBearer)RecognitionConfigStreamingRecognitionConfig)speech)loggerzmedical_memo.jsonGOOGLE_APPLICATION_CREDENTIALStoken)tokenUrli> c@s<eZdZddZddZddZddZd d Zd d Zd S)SpeechClientBridgecCs ||_t|_d|_||_dS)NF) _on_responsequeueQueue_queue_endedstreaming_config)selfr on_responser%/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).0contentrrr -s  z+SpeechClientBridge.start..) rqthreadr SpeechClient generatorstreaming_recognizerprocess_responses_loopemptyprintget)rr r!clientstreamrequests responsesrrrstart's  zSpeechClientBridge.startcCs d|_dS)NT)r)rrrr terminate6s zSpeechClientBridge.terminatecCs|jjt|dddS)NFblock)rputbytes)rbufferrrr add_request9szSpeechClientBridge.add_requestcCs,|D]}|||j|j|jrdSqdS)N)rr r!r)rr,responserrrr%<s z)SpeechClientBridge.process_responses_loopccs|js?|j}|durdS|g} z|jjdd}|dur"WdS||Wn tjy2Ynwqd|V|jrdSdS)NTFr/)rrr(appendrEmptyjoin)rchunkdatarrrr#Bs$  zSpeechClientBridge.generatorN) __name__ __module__ __qualname__rr-r.r4r%r#rrrrr s rcCsd}|jsdS|jd}|jsdS|jdj}d|t|}|js?tj||dtjt|}| d|ddSt d||| d|ddS)Nr  Fis_final transcriptz==>T) results alternativesrClenrBsysstdoutwriteflushr1r')r5r websocketnum_chars_printedresultrCoverwrite_charsrrron_transcription_responseSs   rOrKasync_qcsJz |IdH}||d|ddIdHqtjy$YdSw)NTrBrCrA)r( send_jsonasyncioCancelledError)rKrP send_datarrrsend_transcriptionhsrUz/yokohama/api/wsja-JPlangc s|IdHztjtjjjt|ddd}tj|dd}Wn ty>}z|jdd|dIdHWYd}~dSd}~wwt }t |t }t j|j|j|fd }|tt||j}zPz |IdH} | rq|| qctytd Ynty}ztd |WYd}~nd}~wwW|||r|td dS|||r|td w) u // WebSocket接続を処理し、指定された言語でリアルタイム音声認識を実行します。 // ex: /ws?lang=ko-KR N latest_longT)encodingsample_rate_hertz language_codemodelenable_automatic_punctuation)configinterim_resultsizUnsupported language: )codereason)targetargsz&WebSocket connection closed by client.zWebSocket error: zWebSocket resources cleaned up.)acceptr r AudioEncodingLINEAR16RATEr ExceptionclosejanusrrrO threadingThreadr-sync_qrR create_taskrUrP receive_bytesr4rr infoerrorr.cancelis_aliver9) rKrWdynamic_configdynamic_streaming_configer bridge bridge_thread send_taskr;rrrwebsocket_endpointts^     rz)rV)!rRrkrGfastapirrrrfastapi.securityrgoogle.cloud.speechrrrjr google.cloudr osr^r environ oauth2_schemerouterrgintCHUNKrrO_AsyncQueueProxyrUrKstrrzrrrrs*       3