Ai<" dZddlZddlmZmZddlmZddlmZddl m Z ddl m Z d d l mZmZeed Zd ed edefdZd eded ede dedef dZded edeefdZdS)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_secondsreturnc|tjdd5}||jd|j}dddn #1swxYwY t ||d ddl}||S#t$rYSwxYw# ddl}||w#t$rYwwxYwxYw)NFz.wav)deletesuffixwav)format) fail_on_emptyr)tempfileNamedTemporaryFileexportnamer osremove Exception)r rtmppathrs (src\gzzm\services\transcribe_pipeline.pytranscribe_segmentr s"  $E& A A ASsx...x$T=NNN  III IIdOOOO    D    III IIdOOOO    D sS$AA A B%A?? B  B B;B+*B;+ B85B;7B88B; offset_secspeaker_registryspeaker_devicec  )*tt|ddpd}tt|ddpd}t||}|d}|dpd} |dpg} |d pg} g} t t d } | s9t|}t||||| \}}| D] } t|d d pd }t|d|p|}n#t$rd }d }YnwxYw|ddpd}|d}|t|o|dtv}| t||zdt||zd|t|||t|dnddn2| D].} t|d d pd *t|d*p*)n#t$rd *d )YnwxYw|ddpd}|d}|t|o|dtv})*fd| D}|stdtt*dz}t!t#|tt)dz}|||}t|}t||||| \}}| t|*zdt|)zd|t|||t|dnddg}g}|D]g} | dpdpd}!| | t#|!dkr$|!| vs |!tvr| |g}h|r| |t%|D]\}"}#t|#dd d pd }$t|#dd|$p|$}%dd|#D}&|"t#|dz k}'t|'o|}(tdtt|$dz}t!t#|tt|%dz}||kr3td|dz }t!t#||dz}|||}t|}t||||| \}}| t||$zdt||%zd|&t|(||t|dndd0|| | tt#|dz dS)z6Run ASR -> chunk-level speaker matching for one chunk. frame_ratei>channelsr languagetext time_stampstime_stamps_tokensu，,、;；:：-—) pcm_bytes sample_rater&registrydevicestartend sentence_endN)r0r2r(r3rlsimcg|]\}t|ddpddz kr.t|ddpddzkZ|]S)r0r1gMbP?r2)floatget).0ts_ends_starts r z*process_chunk_pipeline..psz"""w,,344$FF!%%s++2s33ut|CCDCCric>g|]}|ddpdS)r(r))r:)r;xs rr?z*process_chunk_pipeline..s+!G!G!Ga!%%"3"3"9r!G!G!Gr@(g@@)r'r(r* duration_sec)intgetattrr r:stripsetlistr r r9rbool _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_textr3sentence_end_flagtokens_in_sentencestart_msend_mssub_segsub_pcmrl_subsim_subgroups cur_groupr<txtgigrpg_startg_endg_textis_lastgrp_sentence_endr=r>s+ @@rprocess_chunk_pipelinerr%s!gg|U;;DuEEK77J227a88H$Wm<> h!6::%$($6$6"2;2G5A...T      0!Z Z H  Wc : : AcBBhll5'::EgFF    ! VR006BI ( ^ < <  ($()Qy} 7Q$R$R!"""""!""" & q#eGdN&;&;"<"<==S\\3uUT\/B/B+C+CDD!(6/24W==">% +%-) ###!&zG';Q!?!?$Z%%7;; )(,->(?(?$4;4GuWa000T    (*F$&I' # #uuV}}*1r  ###s88q==c-B&B&BcYFVFVMM),,, "I ) i((($V,,! ! CA 7C 8 8 ?C@@c"gkk%99DWEE!G!G3!G!G!GHHNNPPVq1#'(E4E#F#F q#eGdN&;&;"<"<==S\\3uUT\/B/B+C+CDDX%%"1hm44H Wx"}==F!(6/24W==">% +%-) ###!&zG';Q!?!?$Z%%7;; &(,-=(>(>$4;4GuWa000T     1! Hc'll++f4   s& A EE('E(A I**I;:I; file_pathcNt||}g} |D])}|tj|* |D],} ddl}||#t $rY)wxYwn4#|D],} ddl}||#t $rY)wxYwwxYw|S)Nr)rrMr from_filerrr)rsrchunksoutprs rsplit_file_to_segmentsrys "9m < rsc555555............UUUUUUUU C^$$ % %   S T     _ ___& _  _  ____Dc#$|BTr@