o ]h5 @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> Tzko-KRcommand_and_search)encodingsample_rate_hertzaudio_channel_count use_enhanced language_codemodelenable_spoken_punctuation)configinterim_resultsz/healthcs ddiS)Nstatusokrrr(/home/air/websocket/korean/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_qcsZz |IdH}||d|ddIdHtdIdHqtjy,YdSw)NTrRrSrQg?)r8 send_jsonasynciosleepCancelledError)r]rb send_datarrrsend_transcriptiontsrhz /wslang/ko/c s@|IdHtdt}ttt}tt |j |j |tt ||j }z> |IdH}d|vrL|d}td|||IdHnd|vrg|d}tdt|d|rf||n q,ty~||tdYdSty}z||td |WYd}~dSd}~ww) NopenedTtextzReceived text: rBzReceived audio bytes: z bytesclosedz Exception: )acceptr7janusr$r!r'rard create_task to_threadr=sync_qrhrbreceiverWrX send_textrVrDrr>cancel Exception)r]r0bridge send_taskrKmessageerrrwebsocket_endpoints@   ry)-rd threadingrYfastapirrrrfastapi.securityrgoogle.cloud.speechrrrmr# google.cloudr osr rWenviron oauth2_schemeapp basicConfigINFO getLoggerrLloggersetLevelDEBUGRATEintCHUNK AudioEncodingLINEAR16rr'r8r r!ra_AsyncQueueProxyrhr]ryrrrrsN          3