o di=@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlZeejje ejejdedZdZd d Zdd d Zd dZddZddZddZdS)N)upload_to_azure_blob)send_openai_analysis)get_db_connection) load_dotenv)levelSARVAM_SUBSCRIPTION_KEYzstorage/audio/c Csz|di}|dg}|s|ddgdddWSg}t}|D]M}|dd}|||d d }|||dd|d d|d dd }t|dkrfd|vp\d|v} | radnd|d<nd|d<||q"|r{tdd|Dnd} d dd|D} t dt|dt|d| dd| p|dd|t|t | ddWSt y} zt d| |ddgdddWYd} ~ Sd} ~ ww)a5 Parse Sarvam API response with diarization Returns: dict with: - full_transcript: Complete transcript text - speaker_segments: List of segments with speaker, text, timestamps - num_speakers: Number of unique speakers - duration: Total duration in seconds diarized_transcriptentries transcriptr)full_transcriptspeaker_segments num_speakersduration speaker_idunknown_ start_time_secondsend_time_seconds)speakerrtext start_timeend_time speaker_0 speaker_1agentcustomerroleNcSg|]}|dqS)r.0sr!r!Uz-parse_diarized_transcript..cSr )rr!r"r!r!r%r&Xr'u✅ Parsed diarization: z speakers, z segments, z.1fr$u'❌ Error parsing diarized transcript: )getsetaddreplacetitlelenlowerappendmaxjoinlogginginforound Exceptionerror) result diarized_datar r speakers_setentryr speaker_namesegmentis_agentrr er!r!r%parse_diarized_transcriptsV          ( r?cCst}|tjj}|r|d}|d}nd}d}|d|d|}|D]H}z#|d}|d}tj t d |d } t |} | j d krZtd |d | j Wq*t| d} | | jWdn1spwYtj| std|Wq*t} | std|Wq*| d} | d}| dddddtj| d| ddd}t|| std|Wq*t| std| Wq*t| |}|std| Wq*t|}|d|d||dt|d|d|d | | d!d"t|df |t d#|t!||zt"| t d$|Wnt#yO}zt$d%| d&|WYd}~nd}~wwWq*t#ys}ztd'|dd&t|WYd}~q*d}~ww|%|%d(d)iS)*N_calls_sarvamresponse 7987_calls7987_sarvamresponseSELECT * FROM z WHERE status = 0callidfileUrl translate_.wav'❌ Failed to download file for callid . Status code: wb+❌ File missing after download for callid /❌ Failed to initialize Sarvam job for callid job_idoutput_storage_pathinput_storage_path?r/.❌ Failed to upload file to Azure for callid )❌ Failed to start Sarvam job for jobid #❌ Failed to get result for jobid z INSERT INTO z (callid, transcript, speaker_segments, num_speakers, duration, request_id, language, raw_response, status, created_at, updated_at) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, NOW(), NOW()) r r rr language_coderu.✅ Translated transcript inserted for callid u#🧹 Removed local file for callid u⚠️ Could not delete file : ❌ Exception for callid messagezTranslation batch completed)&rcursorpymysqlcursors DictCursorexecutefetchallospathr1 AUDIO_DIRrequestsr( status_coder2r6openwritecontentexistsinit_sarvam_jobsplitrstripbasenamerstart_sarvam_jobpoll_sarvam_statusr?jsondumpsstrcommitr3rremover5warningclose)bidconnr\ calls_tablesarvam_response_tablecallscallcall_idfile_url local_fileresponsefjobrO output_url azure_urlr7r8r>r!r!r%send_to_sarvamns       >      "*rc Cstdd}z#tjd|idd}|jdkr|WStd|jd|jWdStjj yG}ztd t |WYd}~dSd}~ww) Napplication/jsonAPI-Subscription-Keyz Content-Typez7https://api.sarvam.ai/speech-to-text-translate/job/initTheadersrqverifyu❌ Job init failed:  - u'❌ Request exception during job init: ) SUBSCRIPTION_KEYrepostrfrqr2r6r exceptionsRequestExceptionrs)rrr>r!r!r%rks  rkc Cstdd}|ddid}z!tjd||dd}|jdkrWdStd |jd |jWd StjjyL}ztd t |WYd}~d Sd}~ww) Nrrwith_diarizationT)rOjob_parametersz2https://api.sarvam.ai/speech-to-text-translate/jobrrIu ❌ Failed to start Sarvam job: rFu(❌ Request exception during job start: ) rrerrfr2r6rrrrs)rOrdatarr>r!r!r%ros" roc Csdti}tdD]<}tdztjd|d|dd}|jdkr|}|d r\t|d d kr\t d |d |d d|d d d|d dkoZ|d d ddk}nt d |d |d d|d dk}|rt d|dtdd|vr| dd\}}| dd|} n| dd} tj| dd} t d| d|| jdkr| j rz| } t d|d| | WWStjyt d|d | j YWdSwt d!|d"| j WdSt d#|d$| jWdSt d%|d&|d n t d'|d(|jWqtjjyE} zt d)|d| WYd} ~ dSd} ~ wwt d*|dS)+Nrx z3https://api.sarvam.ai/speech-to-text-translate/job/z/statusT)rrrI job_detailsrzJob z state: job_staterstate CompletedSuccessz - no job details availableu✅ Job z+ completed successfully, fetching result...rRrTrSz/0.json?z/0.json)ru✅ Final response job :u✅ Final JSON for job rYu"❌ Failed to decode JSON for job z. Raw response: u❌ Empty response for job z . Response: u2❌ Failed to get content from output URL for job z. Response status: u❌ Job z5 not completed or failed, retrying... Current state: u'❌ Failed to check job status for job rKu/❌ Exception while polling job status for job u❌ Timeout polling job )rrangetimesleeprer(rfrqr-r2r3rlrmrJSONDecodeErrorr6rrrv) rOrrattemptrr job_successbase_urlquery final_urlfinal_responseparsedr>r!r!r%rpsZ  ,"     rpc Cst}|tjj}|d}|d}|d|d|f|}|s>td|d|d|d| | d St |t rH| d nd}|d krjtd|d |d|d |d | | d SzzY|d}t jtd|d} t |} | jdkrtd|d| jWW| | d St| d} | | jWdn1swYt j| std|WW| | d St} | std|WW| | d S| d} | d}| ddd ddt j| d| ddd}t|| s.td|WW| | d St| sGtd| WW| | d St| |}|sctd | WW| | d St|}|d!|d"||d#t !|d$|d%|d&| | d'd(t"|df |d)|d*|f|#t$d+|t%||t j| rt &| t$d,|WW| | d-St'y}ztd.|d/t"|WYd}~W| | d Sd}~ww| | w)0Nr@rArDz WHERE callid = %su⚠️ Call ID z not found in table z (bid=z).Fstatusrz found in table z ) but status=z, expected 0. Skipping.rFrGrHrIrJrKrLrMrNrOrPrQrRrSrTrUrVrWz INSERT INTO z (callid, transcript, speaker_segments, num_speakers, duration, request_id, language, raw_response, status, created_at, updated_at) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, NOW(), NOW()) r r rrrXrzUPDATE z! SET status = 1 WHERE callid = %suA✅ Translated transcript inserted and status updated for callid u)🧹 Deleted local audio file for callid TrZrY)(rr\r]r^r_r`fetchoner2rvrw isinstancedictr(rbrcr1rdrerfr6rgrhrirjrkrlrmrnrrorpr?rqrrrsrtr3rrur5)r~rxryr\rzr{r}rrrrrrrOrrr7r8r>r!r!r%process_call_ids    D   = 8 > 0  ,  '         r)N)rbrer2rr]rq azure_uploadr openai_helperr db_configrdotenvrurllib3disable_warningsrInsecureRequestWarning basicConfigINFOgetenvrrdr?rrkrorprr!r!r!r%s,       OY 4