o i @sZddlZddlZddlZddlmZmZmZmZddlm Z ddl m Z m Z ddl Z ddlZddlmZddlZddlmZddlZddlZdejd<e d d ZeZejejd d eeZeejd Ze edZ!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/websocket/chinese/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 9s  z+SpeechClientBridge.start..) r&qthreadr SpeechClient generatorstreaming_recognizer'process_responses_loopemptyprintget)r(r0r1clientstreamrequests responsesrrrstart3s  zSpeechClientBridge.startcCs d|_dS)NT)r&)r(rrr terminateBs zSpeechClientBridge.terminatecCs|jjt|dddS)NFblock)r%putbytes)r(bufferrrr add_requestEszSpeechClientBridge.add_requestcCs,|D]}|||j|j|jrdSqdS)N)r"r0r1r&)r(r<responserrrr5Hs z)SpeechClientBridge.process_responses_loopccs|js?|j}|durdS|g} z|jjdd}|dur"WdS||Wn tjy2Ynwqd|V|jrdSdS)NTFr?)r&r%r8appendr#Emptyjoin)r(chunkdatarrrr3Ns$  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_qc sz2 |IdH}td|||d|ddIdHtd|dtdIdHqtjyCtdYdSty_}ztj d |dd WYd}~dSd}~ww) NTzSending to client: rRrSrQzSuccessfully sent: g?zSend task was cancelled.zError in send_transcription: exc_info) r8rWrX send_jsonasynciosleepCancelledErrorwarning Exceptionerror)r]rb send_dataerrrsend_transcriptionvs& "rnz /wslang/zh/cs$|IdHtdt}ttt}t t |j |j |t t ||j}zzm |IdH}d|vr|d}td|zt|}|ddkrYtdWn?Wn<tjyltd|Yn+ty}ztd|WYd}~nd}~wwd |vr|d }|r||n q.Wn)tytd Ynty}ztjd |dd WYd}~nd}~wwWtd ||z|j dWn tjyYnwtddStd ||z|j dWn tjy Ynwtdw)NzWebSocket connection opened.TtextzReceived text message: finalz3Client confirmed 'final: True'. Closing connection.zReceived non-JSON text: z&Error processing client text message: rBzClient disconnected.z!Exception in websocket_endpoint: rczCleaning up resources...zWebSocket connection closed.)acceptrWrXjanusr$r!r'rarf create_task to_threadr=sync_qrnrbreceivejsonloadsr8JSONDecodeErrorrirjrkrDrr>cancelrAr#Full)r]r0bridge send_taskrKmessage json_datarmrrrwebsocket_endpointsv          r).rf threadingrYfastapirrrrfastapi.securityrgoogle.cloud.speechrrrrr# google.cloudr osr rWrwenviron oauth2_schemeapp basicConfigINFO getLoggerrLloggersetLevelDEBUGRATEintCHUNK AudioEncodingLINEAR16rr'r8r r!ra_AsyncQueueProxyrnr]rrrrrsP          3