o Ox{i @s*ddlZddlZddlZddlZddlmZmZmZmZm Z ddl m Z ddl m Z mZddlmZddlZddlZddlZddlZdejd<e Zejejdd eeZd Zeed Ze e jj d d d ddd dZ!ee!d dZ"GdddZ#ddZ$dedej%fddZ&e'ddefddZ(dS)N) APIRouter WebSocketDependsWebSocketDisconnectFastAPI)OAuth2PasswordBearer)RecognitionConfigStreamingRecognitionConfig)speechzmedical_memo.jsonGOOGLE_APPLICATION_CREDENTIALSz)%(asctime)s - %(levelname)s - %(message)s)levelformati> Tzja-JPcommand_and_search)encodingsample_rate_hertzaudio_channel_count use_enhanced language_codemodelenable_spoken_punctuation)configinterim_resultsc@s<eZdZddZddZddZddZd d Zd d Zd S)SpeechClientBridgecCs ||_t|_d|_||_dS)NF) _on_responsequeueQueue_queue_endedstreaming_config)selfr on_responser#*/home/air/websocket/japanese/stt_router.py__init__(s  zSpeechClientBridge.__init__c Cstdd|_||_||_z%t}td|}dd|D}||j|}td| |Wn+t y]}ztd|t |j dd t|d WYd}~nd}~wwtd dS) Nu.[DEBUG] SpeechClientBridge 스레드 시작됨Fu)[DEBUG] Google SpeechClient 생성 성공css|] }tj|dVqdS)) audio_contentN)r StreamingRecognizeRequest).0contentr#r#r$ :s  z+SpeechClientBridge.start..uC[DEBUG] Google API 스트리밍 연결 성공, 응답 대기 중...u. [CRITICAL ERROR] Bridge start 내부 에러: TzError: is_final transcriptu+[DEBUG] SpeechClientBridge 스레드 종료)printrqthreadr SpeechClient generatorstreaming_recognizer process_responses_loop Exception traceback print_excputstr)r!r/r0clientstreamrequests responseser#r#r$start.s*( zSpeechClientBridge.startcCs d|_dS)NT)r)r!r#r#r$ terminateLs zSpeechClientBridge.terminatecCs|jjt|dddS)NFblock)rr8bytes)r!bufferr#r#r$ add_requestOszSpeechClientBridge.add_requestc Csnz|D]}|||j|j|jrWdSqWdSty6}ztd|tWYd}~dSd}~ww)Nu> [CRITICAL ERROR] Google API 응답 처리 중 에러 발생: )rr/r0rr5r.r6r7)r!r=responser>r#r#r$r4Rsz)SpeechClientBridge.process_responses_loopccs|js?|j}|durdS|g} z|jjdd}|dur"WdS||Wn tjy2Ynwqd|V|jrdSdS)NTFrA)rrgetappendrEmptyjoin)r!chunkdatar#r#r$r2]s$  zSpeechClientBridge.generatorN) __name__ __module__ __qualname__r%r?r@rEr4r2r#r#r#r$r's rcCst|jsdS|jd}|jsdS|jdj}|js)td||d|ddStd||d|ddS)Nrz [Partial] Fr+z[Final] T)results alternativesr-r,r.r8)rFr/ websocketresultr-r#r#r$on_transcription_responsens  rUrSasync_qc s|z |IdH}||d|ddIdHqtjy$YdSty=}ztd|WYd}~dSd}~ww)NTr,r-r+u*[ERROR] 클라이언트로 전송 실패: )rH send_jsonasyncioCancelledErrorr5r.)rSrV send_datar>r#r#r$send_transcriptionsr[z /wslang/ja/c s0|IdHtdt}ttt}tt |j |j |}tt ||j }z2 |IdH}d|vrK|d}td|||IdHnd|vr[|d}|rZ||n q,tyrtd||YdSty}ztd|t||WYd}~dSd}~ww)Nu&>> 클라이언트 웹소켓 연결됨Ttextz[Received Text] rCu >> 클라이언트 연결 끊김u*[ERROR] 웹소켓 엔드포인트 에러: )acceptr.janusrrr rUrX create_task to_threadr?sync_qr[rVreceive send_textrErr@cancelr5r6r7)rSr/bridge bridge_task send_taskrMmessager>r#r#r$websocket_endpoints@   ri))rX threadingsysr6fastapirrrrrfastapi.securityrgoogle.cloud.speechrr google.cloudr r^rosloggingenvironapp basicConfigINFO getLoggerrNloggerRATEintCHUNK AudioEncodingLINEAR16rr rrU_AsyncQueueProxyr[rSrir#r#r#r$sB      G