o tgY @sRddlZddlZddlZddlmZmZmZmZddlm Z ddl m Z m Z ddl Z ddlZddlmZddlZddlmZddlZdejd<e d d ZeZejejd d eeZeejd ZeedZ e e j!j"d ddddddZ#e e#ddZ$e%dddZ&GdddZ'ddZ(dede j)fddZ*e+d defd!d"Z,dS)#N) APIRouter WebSocketDependsWebSocketDisconnect)OAuth2PasswordBearer)RecognitionConfigStreamingRecognitionConfig)speech)FastAPIzmedical_memo.jsonGOOGLE_APPLICATION_CREDENTIALStoken)tokenUrlz)%(asctime)s - %(levelname)s - %(message)s)levelformati> Tzja-JPcommand_and_search)encodingsample_rate_hertzaudio_channel_count use_enhanced language_codemodelenable_spoken_punctuation)configinterim_resultsz/healthcs ddiS)Nstatusokrrr/home/air/api/stt_router.py health_check&sr c@s<eZdZddZddZddZddZd d Zd d Zd S)SpeechClientBridgecCs ||_t|_d|_||_dS)NF) _on_responsequeueQueue_queue_endedstreaming_config)selfr' on_responserrr__init__+s  zSpeechClientBridge.__init__cCsrd|_||_||_t}|}dd|D}||j|}|||j s7t |j |j r)dSdS)NFcss|] }tj|dVqdS)) audio_contentN)r StreamingRecognizeRequest).0contentrrr 7s  z+SpeechClientBridge.start..) r&qthreadr SpeechClient generatorstreaming_recognizer'process_responses_loopemptyprintget)r(r0r1clientstreamrequests responsesrrrstart1s  zSpeechClientBridge.startcCs d|_dS)NT)r&)r(rrr terminate@s zSpeechClientBridge.terminatecCs|jjt|dddS)NFblock)r%putbytes)r(bufferrrr add_requestCszSpeechClientBridge.add_requestcCs,|D]}|||j|j|jrdSqdS)N)r"r0r1r&)r(r<responserrrr5Fs z)SpeechClientBridge.process_responses_loopccs|js?|j}|durdS|g} z|jjdd}|dur"WdS||Wn tjy2Ynwqd|V|jrdSdS)NTFr?)r&r%r8appendr#Emptyjoin)r(chunkdatarrrr3Ls$  zSpeechClientBridge.generatorN) __name__ __module__ __qualname__r*r=r>rDr5r3rrrrr!*s r!cCsd}|jsdS|jd}|jsdS|jdj}d|t|}|jsGtd|tj ||dtj t|}| d|ddSt d||td|| d |ddS) Nr zPartial transcription:  Fis_final transcriptz==>zFinal transcription: T) results alternativesrSlenrRlogginginfosysstdoutwriteflushrAr7)rEr0 websocketnum_chars_printedresultrSoverwrite_charsrrron_transcription_response]s"   rar]async_qcsJz |IdH}||d|ddIdHqtjy$YdSw)NTrRrSrQ)r8 send_jsonasyncioCancelledError)r]rb send_datarrrsend_transcriptiontsrgz /ws/websocketc sN|IdHtdt}t}ttt}t j |j |j |fd}| t t||j}z> |IdH}d|vrS|d}td|||IdHnd|vrn|d}tdt|d|rm||n q3ty||td YdSty}z||td |WYd}~dSd}~ww) Nopened)targetargsTtextzReceived text: rBzReceived audio bytes: z bytesclosedz Exception: )acceptr7rdget_event_loopjanusr$r!r'ra threadingThreadr=sync_q create_taskrgrbreceiverWrX send_textrVrDrr>cancel Exception) r]loopr0bridge bridge_thread send_taskrKmessageerrrwebsocket_endpointsD   r~)-rdrprYfastapirrrrfastapi.securityrgoogle.cloud.speechrrror# google.cloudr osr rWenviron oauth2_schemeapp basicConfigINFO getLoggerrLloggersetLevelDEBUGRATEintCHUNK AudioEncodingLINEAR16rr'r8r r!ra_AsyncQueueProxyrgr]r~rrrrsN          3