i-( SrSSKrSSKrSSKrSSKrSSKrSSKrSSK J r SSK J r J r Jr \R"\5rS\4SjrS\S\S\R*4S jrS \S\S\R*4S jrS \R*S \S\ 4SjrS\ S\R*4SjrS!S \R*S\S\S\S\S\R*4 SjjrS"S \ S\S\S\S\ 4 SjjrS#S \R*S\S\S \S\R*4 SjjrS$S \R*S\S\S\S\S\R*4 SjjrS%S\ \R*S \S\S\R*4S jjr!g)&aAudio I/O and processing utilities. Provides functions for loading, resampling, silence removal, chunking, cross-fading, and format conversion. All public functions in this module operate on **numpy float32 arrays** with shape ``(C, T)`` (channels-first). N) AudioSegment)detect_leading_silencedetect_nonsilentsplit_on_silence audio_pathc[R"USSS9upURU4$![aD SSKnUR USSS9upUR S:XaU[RSS24nX4s$f=f) u-Load audio from a file path, returning (data, sample_rate). Tries two backends in order: 1. soundfile — covers WAV/FLAC/OGG etc., no ffmpeg needed. 2. librosa — covers MP3/M4A etc. via audioread + ffmpeg. Returns: (data, sample_rate) where data is a numpy float32 array of shape (C, T). float32Tdtype always_2drNFsrmono) sfreadT Exceptionlibrosaloadndimnpnewaxis)rdatarrs 1/mnt/workspace/OmniVoice/omnivoice/utils/audio.py load_waveformr-su 77:Y$Gvvrz << t%<@ 99> A &Dxs%(A A65A6 sampling_ratereturnc[U5up#URSS:a[R"USSS9nX1:wa@[R R [R"U5X1S9R5nU$)zLoad a waveform from file and resample to the target rate. Parameters: audio_path: path of the audio. sampling_rate: target sampling rate. Returns: Numpy float32 array of shape (1, T). rrTaxiskeepdims orig_freqnew_freq) rshapermean torchaudio functionalresampletorch from_numpynumpy)rrrrs r load_audior.EstZ(HD zz!}qwwt!d3 $$--   T "b. %'  Krawc[R"U5n[R"USSS9up4URnURSS:a[R"USSS 9nXA:wa@[RR!["R$"U5XAS 9R'5nU$![ aR SSKnURS5 URUSSS9up4URS:XaU[RSS24nNf=f) zLoad audio from in-memory bytes and resample. Parameters: raw: raw audio file bytes (e.g. from WebDataset). sampling_rate: target sampling rate. Returns: Numpy float32 array of shape (1, T). r Tr rNFr rr r#)ioBytesIOrrrrrseekrrrrr&r'r(r)r*r+r,r-)r0rbufrrrs rload_audio_bytesr6[s **S/C '773i4@vv zz!}qwwt!d3 $$--   T "b. %'  K '  <<5<9 99> A &D 's$B,,ADDaudio sample_ratecUS-RSS5R[R5nURSS:aUR R 5n[UR5SUURSS9$)zFConvert a numpy float32 array of shape (C, T) to a pydub AudioSegment.@iirr)r sample_width frame_ratechannels) clipastyperint16r&rflattenrtobytes)r7r8 audio_ints rnumpy_to_audiosegmentrEst&&vu5<rreshaper)rFrs raudiosegment_to_numpyrLsg 88D--/ 0 7 7 Cg MD }}BJJM"" <<DMM * , ,,r/mid_sillead_sil trail_silc[X5nUS:a.[UUSUSS9n[R"SS9nUHnXW- nM [ XSUS5n[ U5$)akRemove middle silences longer than *mid_sil* ms and trim edge silences. Parameters: audio: numpy array with shape (C, T). sampling_rate: sampling rate of the audio. mid_sil: middle-silence threshold in ms (0 to skip). lead_sil: kept leading silence in ms. trail_sil: kept trailing silence in ms. Returns: Numpy array with shape (C, T'). r )min_silence_lensilence_thresh keep_silence seek_step)duration)rErrsilentremove_silence_edgesrL)r7rrMrNrOwavenon_silent_segssegs rremove_silencer]sl& ! 6D{* #   ""A."C KD#  3 ?D  &&r/silence_thresholdc[XS9n[SXA- 5nXSnUR5n[XS9n[SXB- 5nXSnUR5nU$)z:Remove edge silences, keeping *lead_sil* / *trail_sil* ms.)r^rN)rmaxreverse)r7rNrOr^ start_idxs rrYrYse'uRIAy+,I * E MMOE&uRIAy,-I * E MMOE Lr/ pad_duration fade_durationcURSS:XaU$[X#-5n[X-5nUR5nUS:a[XFRSS-5nUS:a[R "SSU[R S9[RSS24nUSSU24==U-ss'[R "SSU[R S9[RSS24n USU*S24==U -ss'US:aE[R"URSU4URS9n [R"XU /SS9nU$) a9Apply fade-in/out and pad with silence to prevent clicks. Args: audio: numpy array of shape (C, T). pad_duration: silence padding duration per side (seconds). fade_duration: fade curve duration (seconds). sample_rate: audio sampling rate. Returns: Processed numpy array of shape (C, T_new). rHrr;rr N.r!) r&intcopyminrlinspacer rzerosr concatenate) r7rcrdr8 fade_samples pad_samples processedkfade_infade_outsilences rfade_and_pad_audiorus)" {{2! }23Ll01K Ia  oob1Q6 7 q5kk!Q[U5S:XaUS$[X!-5nUS-nUnUSR5nUSSGHWnU/n[XFRS5n U S:aO[ R "SSU [ RS9[ RSS24n USSU *S24==U -ss'UR[ R"USRSU4[ RS95 UR5n [XKRS5n U S:aK[ R "SSU [ RS9[ RSS24n U SSU 24==U -ss'URU 5 [ R"USS9nGMZ U$) aConcatenate audio chunks with silence gaps and cross-fade at boundaries. Args: chunks: list of numpy arrays, each (C, T). sample_rate: audio sample rate. silence_duration: total silence gap duration in seconds. Returns: Merged numpy array (C, T_total). rrNrHrf.rg) r{rhrirjr&rrkr rappendrlrm)rr8rtotal_nfade_n silence_nmergedchunkpartsfout_nw_outrrfin_nw_ins rcross_fade_chunksr-sj 6{aay"01G \FI AY^^ FV\\"-. A:KK1fBJJ? A NE "IcF78m $ - $ RXXvayq19=RZZPQ**,FMM"-. 19;;q!U"**=bjj!mLD C%K D (  WB/#& Mr/),rzr)rzrrQ)皙?ri])g.@g@g4@)g333333?)"__doc__r2loggingr-r soundfilerr+r(pydubr pydub.silencerrr getLogger__name__loggerstrrrhndarrayr.bytesr6rErLr]floatrYrurlistrr/rrs$  TT   8 $c03srzz,% L  # , - -- #' ::#'#'#' #'  #' ZZ #'P"     , ) ::))) ) ZZ )^ .* ::.*.*.* .*  .* ZZ .*h"*  ***ZZ *r/