=i<* dZddlZddlZddlZddlZddlmZmZmZddl m Z ddl m Z m Z mZddlmZddlZddlmZeej+j,dd z Z ej1d d 5Zej4eZdddeej=d dj?Z ejBe"ee ejFejHe%Z&e&jOe"ee ejFe(ej=ddZ)e(ej=ddZ*e(ej=ddZ+deeeeee,ffdZ-de(fdZ.de(dee,de/fdZ0d#dede1deeeeee,e(ffdZ2d$ded e1d!e/de,fd"Z3y#1swYxYw#e$riZY*wxYw)%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|ddxs#t|tr|jdnd}t|ddxs#t|tr|jdnd}t|ddxs't|tr|jdndxsg}|||fS)z=Normalize transcribe result to (language, text, time_stamps).languageNtext time_stamps)getattr isinstancedictget)firstrtext_valts_lists /app/gzzm/transcribe.py_extract_transcribe_payloadr%(suj$/oZX]_cMdEIIj4IjnHufd+eZPUW[E\ &0AbdHe]D1zR\]bdhRieii 6NoszxzG Xw &&c t|tr3|jd|jd|jdd}n+t|dd}| t|dd}| t|dd} t |xsdS#t $rYywxYw)Nendfinishend_time)rrr rfloat Exception)ts_itemend_vals r$_extract_end_secondsr00s'4 ++eW[[7;;zSV;W%XY'5$/ ?gx6G ?gz37GW^$$ s0A?? B  B chunk_durationr#c|tkry|sytd|Dd}|dkry|td|z }td||z }|tkxs |tkDS)NFc32K|]}t|ywNr0.0tss r$ z6_should_fallback_for_partial_result..EsB'B+B/'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_resultrD?sj:: B'BCPKa 3un#==NsN[89L 9 9 g\Lg=ggr& chunk_pathfallback_secondsc t}| tdtj|}t |dz } |j |dd}|rW|d}t |\}}} |s| r?t|| r-td| Dd } tjd ||| n||| |fStjd |td t|dz} d} g} g}tdt || D]1}|||| z}tj dd5}|j#|j$d|j$}ddd |j dd}|s t'j(||d}t |\}}}| xs|} |r| j+||dz }|D]_}t-|t.rt1|j3d|j3d|j3dd}t1|j3d|j3d|j3d|}|j3d|j3dd}nt5|dd}t5|dd}|t7|dr t5|d}|t7|dr t5|d}t1|xsd}t1|xs|}t5|dt5|dd}|j+t9||zdt9||zd|db t'j(|4| dj;| ||fS#t$rtjd |YwxYw#1swY%xYw#t$rYwxYw#t$r@tjdY t'j(|#t$rYwxYwwxYw#t$rYwxYw# t'j(w#t$rYwwxYwxYw) 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_stampsrc32K|]}t|ywr4r5r6s r$r9z6_run_transcribe_chunk_with_fallback..as&R'B';B'?'r: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%rDr>loggerwarningr- exceptionintrangetempfileNamedTemporaryFileexportnameosremoveappendrrr,r rhasattrroundjoin)rErFMODELsegr1resultsr!rr"r#rA 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 KE }566   ,CX&Nd""dW["\ AJE*Ee*L 'Hh76~wO"%&R'&R\_"`KNNw"&# $XwFFQS]^a-./$6K%)O LIaS;7 lL;$>?  ( (f E NN7<<N 6||HF' **D]a*bKB  (#C$AI.I).T +Hh -9O##H-%.J!b$'!"&&"&&"&&WXBY2Z"[\EubffXrvvjRW?X.Y Z[C!vvfbffVR.@AH!(Wd!;J&r5$7H!)gb,.G%,R%> 'GB ,C#*2z#:!*"34E 1E2C&r672vr3JKH  "5:#5q9 z!115$"!"2  (#U8\ BGGL19n LLm dVXbcdF ER       BH M   (#         (#  sA,L=)L=*M# N!M07FNO =M M #M- 0 M=<M=O O!N88 OOO  O OOPO43P4 P =P?P Ptmp_path chunk_seconds fail_on_emptyct}| tdtj}t||}tj dt ||g}d}d}g} |D]} t| |dkDrtdtd|dznd \} } } }|xs| }| r|j| | D]_}t|trt|jd |jd |jd d}t|jd|jd|jd|}|jd|jdd}nt|d d}t|dd}|t!|d r t|d }|t!|dr t|d}t|xsd}t|xs|}t|dt|dd}| jt#||zdt#||zd|db||z }tj|z }|su|rtdt |d|dtj%dt ||dt#|dt |ggd|D]} t'j(|Sdj-|}t/| |}t1|}|dj-|j3t#|dt |||d|D]} t'j(|S#t*$rYwxYw#t*$rY6wxYw#|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+rr)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 KE }566))+K'-@K LL3S5E{SE"HL"$OE%JDgHUXYHYQB 0B(C!D_aE AJ M>  -:H Z(#b$'!"&&"&&"&&WXBY2Z"[\EubffXrvvjRW?X.Y Z[C!vvfbffVR.@AH!(Wd!;J&r5$7H!)gb,.G%,R%> 'GB ,C#*2z#:!*"34E 1E2C&r672vr3JKH&&"5<#7; |!3Q7$(!$, N *LA&DYY[;. "%Z[^_j[kZlltvAuBBC$DEE NNUWZ[fWg h$!*a0k*!&( .A  ! GGEN >PYZ89KL!HHUO))+j!,+&-"4  A  !   9  A  !   s\!H,ML)+A ML8) L54L58 MMM5M$#M5$ M0 -M5/M0 0M5)r)T)4__doc__rcrloggingr_typingrrrmodelrutils.audio_utilsrr r pydubr jsonpathlibr __file__resolveparents _config_pathopen_fload _GZZM_CONFIGr-strr upper LOG_LEVEL basicConfigrr getLogger__name__rZsetLevelr,r=r?r@rr%r0boolrDr]r}rr&r$rs ((tt H~%%'//25GG    3  1R tyy}  2    f5 6 < < > ''9gllCD   8 $GLL9:&+<+;+;