=i<* dZddlZddlZddlZddlZddlmZmZmZddl m Z ddl m Z m Z mZddlmZddlZddlmZeejdd z Z ed d 5ZejeZdddn #1swxYwYn #e$riZYnwxYweed dZ ej!e"ee ej#ej$e%Z&e&'e"ee ej#e(eddZ)e(eddZ*e(eddZ+deeeeee,ffdZ-de(fdZ.de(dee,de/fdZ0d%dede1deeeeee,e(ffdZ2d&d!ed"e1d#e/de,fd$Z3dS)'u文件转写核心逻辑。 职责概览： - 对输入音频进行分块转写并拼接文本与时间戳。 - 当单块转写失败/空结果时自动降级为更小子块重试。 - 提供可配置的空结果容错模式（fail_on_empty）。 N)ListOptionalTuple) get_model)split_audio_to_chunks!insert_punctuations_into_segmentsmerge_sentences_from_tokens) AudioSegment)Pathzgzzm_config.jsonrzutf-8)encoding log_levelINFO)level&asr_min_chunk_sec_for_partial_fallback!asr_partial_result_coverage_ratiog?asr_partial_result_tail_gap_secreturnc~t|ddp+t|tr|dnd}t|ddp+t|tr|dnd}t|ddp-t|tr|dndpg}|||fS)z=Normalize transcribe result to (language, text, time_stamps).languageNtext time_stamps)getattr isinstancedictget)firstrtext_valts_lists src\gzzm\transcribe.py_extract_transcribe_payloadr%(suj$//oZX]_cMdMd4nEIIj4I4I4IjnHufd++eZPUW[E\E\0d &0A0A0AbdHe]D11zR\]bdhRiRi6seii 6N6N6NoszxzG Xw &&c ^t|tr?|d|d|dd}n7t|dd}|t|dd}|t|dd} t |pdS#t $rYdSwxYw)Nendfinishend_time)rrr rfloat Exception)ts_itemend_vals r$_extract_end_secondsr00s'4  8++eW[[7;;zSV;W;W%X%XYY'5$// ?gx66G ?gz377GW^$$$ sss B B,+B,chunk_durationr#c|tkrdS|sdStd|Dd}|dkrdS|td|z }td||z }|tkp |tkS)NFc34K|]}t|VdSNr0.0tss r$ z6_should_fallback_for_partial_result..Es+BBB+B//BBBBBBr&r+defaultrTgMbP?)"MIN_CHUNK_SEC_FOR_PARTIAL_FALLBACKmaxPARTIAL_RESULT_COVERAGE_RATIOPARTIAL_RESULT_TAIL_GAP_SEC)r1r# max_end_seccoverage_ratio tail_gap_secs r$#_should_fallback_for_partial_resultrC?s:::u uBB'BBBCPPPKat 3un#=#==NsN[899L 9 9 g\Lg=ggr& chunk_pathfallback_secondscn t}|tdtj|}t |dz } ||dd}|rn|d}t |\}}} |s| rOt|| r9td| Dd } t d ||| n||| |fSt d |n+#t$rt d |YnwxYwtd t|dz} d} g} g}tdt || D]B}|||| z}tjdd5}||jd|j}dddn #1swxYwY ||dd}|s' t'j|#t$rYwxYw|d}t |\}}}| p|} |r| ||dz }|D]}t-|t.rt1|d|d|dd}t1|d|d|d|}|d|dd}nt5|dd}t5|dd}| t7|drt5|d}| t7|drt5|d}t1|pd}t1|p|}t5|dt5|dd}|t9||zdt9||zd|dnR#t$rEt d|Y t'j|#t$rYwxYwwxYw t'j| #t$rYwxYw# t'j|w#t$rYwwxYwxYw| d| ||fS) zTranscribe one chunk; if it fails, split it into smaller pieces and retry. Returns (language, text, time_stamps, chunk_duration_sec). N模型尚未加载完成g@@T)audiorreturn_time_stampsrc34K|]}t|VdSr4r5r6s r$r9z6_run_transcribe_chunk_with_fallback..as+&R&RB';B'?'?&R&R&R&R&R&Rr&r+r:u]检测到疑似截断结果，降级为小分片重试: chunk=%s duration=%.3fs max_end=%.3fsu8切片转写结果为空，降级为小分片重试：%su;模型对切片转写失败，降级为小分片重试：%sriFz.wav)deletesuffixwav)formatstartbegin start_timer(r)r*rwordrrPr(ru#降级子切片仍转写失败：%s)r RuntimeErrorr from_filelen transcriber%rCr=loggerwarningr- exceptionintrangetempfileNamedTemporaryFileexportnameosremoveappendrrr,r rhasattrroundjoin)rErFMODELsegr1resultsr!rr"r#r@ fallback_msmerged_language merged_texts merged_ts sub_start_mssub_segsub_tmpsub_path sub_results sub_firstsub_langsub_text sub_ts_list sub_offsetr8rPr(text_seg start_attrend_attrs r$#_run_transcribe_chunk_with_fallbackr}Ns KKE }5666   , ,CXX&Nd""dW["\\  GAJE*Ee*L*L 'Hh G7 G6~wOO G"%&R&R'&R&R&R\_"`"`"`KNNw"&# $XwFFQS]^^^^ dddVXbcccccda-..//$6K%)O LIaS;77,, lL;$>>?  (f E E E $ NN7<N 6 6 6|H $ $ $ $ $ $ $ $ $ $ $ $ $ $ $' **D]a*bbK B  (####    E$AI.I).T.T +Hh -9O .##H---%.J!  b$'' L!"&&"&&"&&WXBYBY2Z2Z"["[\\EubffXrvvjRW?X?X.Y.Y Z Z[[C!vvfbffVR.@.@AAHH!(Wd!;!;J&r5$77H!)gb,.G.G)%,R%>%> 'GB ,C,C'#*2z#:#:!*"344E 1E22C&r672vr3J3JKKH  "5:#5q99 z!1155$""! *      BH M M M   (####      + 2  (####      (####     BGGL119n LLsBC+C++%DD$F00F4 7F4 <O9G-- G:9G:=G;O98Q49%QQ4 P66 QQQQ4 Q## Q10Q14R6R  R R RR RTtmp_path chunk_seconds fail_on_emptycDt}|tdtj}t||}tdt ||g}d}d}g} |D] } t| |dkr!tdtd|dznd \} } } }|p| }| r| | | D]}t|trt|d |d |d d}t|d|d|d|}|d|dd}nt|d d}t|dd}| t!|d rt|d }| t!|drt|d}t|pd}t|p|}t|dt|dd}| t#||zdt#||zd|d||z }"tj|z }|s|r#tdt |d|dtdt ||dt#|dt |ggd|D]'} t'j|#t*$rY$wxYwSd|}t/| |}t1|}|d|t#|dt |||d|D]'} t'j|#t*$rY$wxYwS#|D]'} t'j|#t*$rY$wxYwwxYw)aTranscribe a file at `tmp_path` using the global model. Returns a dict compatible with the previous JSONResponse payload. This function will remove any temporary chunk files it created, but does not remove `tmp_path` (caller is responsible). NrHu 切分后得到 %d 个 chunk: %sr+rrrD)rFrPrQrRr(r)r*rrSrrTrUu5识别失败: 未从任何切片获得文本 (chunks=z, paths=)u8本次转写未获得文本，返回空结果: chunks=%d)rrtime_secchunksrtime_stamps_tokens )rrVtimerrZdebugrXr}r=minrerrr,r rrfrgr[rcrdr-rhr r strip)r~rrri start_infer chunk_pathstextsr start_offsetall_time_stampsrE chunk_lang chunk_text chunk_ts_listr1r8rPr(rzr{r| infer_timep full_textrmerged_time_stampss r$transcribe_audio_filers KKE }5666)++K'-@@K LL3S5E5E{SSSE"HL"$OE% + +JDgHUXYHYHYQB 0B(C(C!D!D!D_aEEE AJ M>  -:H ) Z(((#  b$'' L!"&&"&&"&&WXBYBY2Z2Z"["[\\EubffXrvvjRW?X?X.Y.Y Z Z[[C!vvfbffVR.@.@AAHH!(Wd!;!;J&r5$77H!)gb,.G.G)%,R%>%> 'GB ,C,C'#*2z#:#:!*"344E 1E22C&r672vr3J3JKKH&&"5<#7;; |!3Q77$(( N *LLY[[;.   E"$D[^_j[k[k$D$DvA$D$D$DEEE NNUWZ[fWgWg h h h$!*a00k**!&( .  A  !      GGENN >PYZZ89KLL!HHUOO))++j!,,+&&-"4     A  !        A  !      s\5J,O3&L;; MM Ar?rr%r0boolrCr]r}rr&r$rs ((((((((((tttttttttt tH~~%%''/25GG    3  1 1%R ty}} %%%%%%%%%%%%%%%LLL C   f55 6 6 < < > > '''9glCCDDDD  8 $ $GL99:::&+U<+;+;