?iUddlZddlZddlZddlmZddlmZmZddlZddl m Z m Z m Z m Z ddlmZddlmZddlmZdaeeed<ed e fd Ze d d e Zeej2ddZdededeefdZej;de dfde defdZedk(rddl Z e jBdddyy)N)asynccontextmanager)ListOptional)FastAPIFile HTTPException UploadFile) JSONResponse) AudioSegment) Qwen3ASRModelMODEL_cKtdtj}tjj s t dt jddtjdddddd ttjd  a tj|z }td |dddyw)Nu-【1 开始加载 Qwen3-ASR-0.6B 模型...】u?未检测到 GPU（CUDA），请确认容器已正确启用 GPUzQwen/Qwen3-ASR-0.6Bz /app/hf_cachecudaT zQwen/Qwen3-ForcedAligner-0.6Bzcuda:0)dtype device_map) cache_dir torch_dtyperlow_cpu_mem_usageuse_safetensorsmax_inference_batch_sizemax_new_tokensforced_alignerforced_aligner_kwargsu!✅ 模型加载完成！耗时: z.2fu 秒 ) printtimetorchr is_available RuntimeErrorr from_pretrainedfloat16dictbfloat16r )r start_load load_times /app/qwen_asr_demo.pylifespanr)s 9:J :: " " $\]]  ) )!MM!"6"..  E j(I -i_F CD sBz Qwen3 ASR APIz1.0.0)titleversionr)ASR_CHUNK_SECONDS15 file_path chunk_secondsreturncdtj|}td|dz}g}tdt ||D]d}||z}|||}t j dd5}|j|jd|j|jdddf|S#1swYrxYw) NrirF.wavdeletesuffixwav)format) r from_filemaxrangelentempfileNamedTemporaryFileexportnameappend) r.r/audiochunk_mschunksstartendsegmenttmps r(split_audio_to_chunksrH0s  " "9 -E1m$t+HFq#e*h/hc"  ( (f E NN388EN 2 MM#(( #F E0 MF Es !9B&&B/ z /transcribe.fileclKt tdd|jxsdj}|j ds|j ds tdd|j drdnd}t j d | 5}|j|jd{|j}ddd tj}tt}g}d}|D]M} tj| d } | s|xs| d j}|j| d j Otj|z } |s td dt#|dj%|j't)| dt+|d t-j.| t3j5dgD]} t-j.| S7J#1swY:xYw#t0$rYRwxYw#t0$rYFwxYw#t6$rYSwxYw# t-j.n#t0$rYnwxYw t3j5dgD]'} t-j.| #t0$rY%wxYww#t6$rYwwxYwxYww)Niu模型尚未加载完成) status_codedetailr2z.mp3iu仅支持 wav 或 mp3 文件Fr3)rAlanguageriu 识别失败 )rNtexttime_secrC chunk_paths)r rfilenamelowerendswithr<r=writereadr?rrH CHUNK_SECONDS transcriberNr@rQr joinstriproundr;osremoveOSErrorlocalsget Exception) rIrTr5rGtmp_path start_inferrStextsrN chunk_pathresults infer_times r(transcribe_audiorj=sg }4NOO #**,H   f %):):6)B4RSS((0VfF  $ $E& AS  #$88 B%iik +HmD "&%J&&Z$&GG#:wqz':': WQZ__- & YY[;. CG G$--/!*a0k*     IIh  $hll="= IIj)>E$ B A@         IIh     $hll="= IIj)>    sB J4G;.G8 /G;J4 AH5BH5,HH& H5H&7J48G;;HJ4 HJ4HJ4 H# H&"H##H&& H2/J41H22J45J17I  J1 IJ1IJ1J";JJ" J J"J J"!J1" J.+J1-J..J11J4__main__zqwen_asr_demo:appz0.0.0.0i@)hostport)"r^rr< contextlibrtypingrrrfastapirrrr fastapi.responsesr pydubr qwen_asrr r __annotations__r)appintgetenvrYstrrHpostrj__name__uvicornrunr(rs *! <<*"!%x % g  4OWxHIBII1489  S  c -.23i22L22j zGKK#)$?r~