Ai<" dZddlZddlmZmZddlmZddlmZddl m Z ddl m Z d d l mZmZeed Zd ed edefdZd eded ede dedef dZded edeefdZy)u统一转写流水线服务。 职责概览： - 对单个音频 chunk 执行 ASR、说话人匹配和时间偏移叠加。 - 提供 HTTP 文件路径下的分段读取能力，供 /transcribe 顺序消费。 N)DictList) AudioSegment)split_audio_to_chunks)SpeakerRegistry)transcribe_audio_file)audio_segment_to_pcm_bytesidentify_speaker_id_from_pcmu 。.!?！？segment chunk_secondsreturnchtjdd5}|j|jd|j}ddd t |d ddl}|j |S#1swY.xYw#t$rYSwxYw# ddl}|j w#t$rYwwxYwxYw)NFz.wav)deletesuffixwav)format) fail_on_emptyr)tempfileNamedTemporaryFileexportnamer osremove Exception)r rtmppathrs )/app/gzzm/services/transcribe_pipeline.pytranscribe_segmentr s  $ $E& ASsxx.xx B$T=N   IIdO B A       IIdO   sM*A/ B A;/A8; BB B1 B"!B1" B.+B1-B..B1 offset_secspeaker_registryspeaker_devicec L tt|ddxsd}tt|ddxsd}t||}|jd}|jdxsdj } |jdxsg} |jd xsg} g} t t d } | st|}t||||| \}}| D]} t|jd d xsd }t|jd|xs|}|jddxsd}|jd}|t|xr |dtv}| jt||zdt||zd|t||| t|dnddn| D]} t|jd d xsd }t|jd|xs|}|jddxsd}|jd}|t|xr |dtv}| Dcgc]N}t|jd d xsd |dz k\r't|jdd xsd |dzkr|P}}|stdtt|dz}t!t#|tt|dz}|||}t|}t||||| \}} | jt||zdt||zd|t||| t| dnddg}!g}"|D][}|jdxsdxsd}#|"j|t#|#dk(s<|#| vs |#tvsI|!j|"g}"]|"r|!j|"t%|!D]\}$}%t|%djd d xsd }&t|%djd|&xs|&}'dj'|%D(cgc]}(|(jddxsdc}(j })|$t#|!dz k(}*t|*xr|}+tdtt|&dz}t!t#|tt|'dz}||kr'td|dz }t!t#||dz}|||}t|}t||||| \}} | jt||&zdt||'zd|)t|+|| t| dndd|| | tt#|dz dS#t$rd }d }YkwxYw#t$rd }d }YwxYwcc}wcc}(w)z6Run ASR -> chunk-level speaker matching for one chunk. frame_ratei>channelsr languagetext time_stampstime_stamps_tokensu，,、;；:：-—) pcm_bytes sample_rater&registrydevicestartgend sentence_endN)r0r1r(r2rlsimgMbP?ri(g@@)r'r(r* duration_sec)intgetattrr getstripsetlistr r floatrbool _SENT_ENDappendroundmaxminlen enumeratejoin),r r!rr"r#r-r& transcribedr'r(chunk_tstoken_tsenrichedINTERNAL_SPLIT_PUNCTSchunk_pcm_byteschunk_rl chunk_simsentence sent_startsent_end sent_textr2s_starts_endsentence_end_flagttokens_in_sentencestart_msend_mssub_segsub_pcmrl_subsim_subgroups cur_grouptxtgigrpg_startg_endxg_textis_lastgrp_sentence_ends, rprocess_chunk_pipelinerl%sgg|U;DuEK77J27a8H$Wm h!6:%$($6"2;2G5A.T  !0!H  Wc : AcBhll5':EgF! VR06BI ( ^ <  ($()Qy} 7Q$R! ""!Aw,34$F!%%s+2s3ut|C! "&q#eGdN&;"<=S\3uUT\/B+CD!(624W=">% +%-) #!&zG';Q!?$Z%%7; )(,->(?$4;4GuWa0T  (*F$&I'uuV}*1r  #s8q=c-B&BcYFVMM), "I (  i($V,CA 7C 8 ?C@c"gkk%9DWE3!G3a!%%"3"9r"93!GHNNPVq1#'(E4E#F q#eGdN&;"<=S\3uUT\/B+CDX%"1hm4H Wx"}=F!(624W=">% +%-) #!&zG';Q!?$Z%%7; &(,-=(>$4;4GuWa0T  1-s!zc'l+f4  a   0  "^"Hs1>U4.>V1AVV! 4 VV VV file_pathc6t||}g} |D]&}|jtj|( |D]} ddl}|j ||S#t $rY(wxYw#|D]'} ddl}|j |#t $rY%wxYwwxYw)Nr)rrCr from_filerrr)rmrchunksoutprs rsplit_file_to_segmentsrss "9m r?rBr:r r@strrlrsrr~s5..U ^$ %   S T  _ ___& _  _  _Dc#$|BTr}