i AdZddlZddlZddlZddlmZddlmZmZm Z ddl Z dddddd d d Z d d d dZ dedefdZde eeefdee j"eeffdZ d#de eeej(fdededee j"effdZdefdZdefdZ d$de j"dedede eeej(fdededdfdZ d%de eeej(fde j"dedeeddf d Z d&de eeej(fdedee j"effd!Z d%de eeej(fde j"dedeeddf d"Zy)'aBAudio I/O utilities using miniaudio. This module provides functions for reading and writing audio files. - Reading: Uses miniaudio to support WAV, MP3, FLAC, and Vorbis formats. Uses ffmpeg for M4A/AAC format support. - Writing: Uses miniaudio for WAV and ffmpeg for MP3, FLAC, OGG, Opus, and Vorbis encoding. N)Path)OptionalTupleUnionwavmp3flacvorbism4a)rrr oggr r aacSIGNED16SIGNED32FLOAT32)int16int32float32datareturnc|dddk(r |dddk(ry|ddd k(s|d d d k(s|d d d k(ry|dddk(ry|dddk(ry|dddk(rytd)z6Detect audio format from bytes data using magic bytes.NsRIFF sWAVErsID3rssrsfLaCr OggSr ftypr z(Unable to detect audio format from bytes) ValueError)rs g/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/mlx_audio/audio_io.py_detect_format_from_bytesr $s BQx7tAbzW4 bqV Qq [ 8D1IDecode audio using ffmpeg (for formats not supported by miniaudio like M4A). Args: input_data: Path to the audio file or raw bytes data. Returns: Tuple of (samples as int16 numpy array, sample_rate, nchannels). Raises: RuntimeError: If ffmpeg is not found or decoding fails. ffmpegNz ======================================== ffmpeg not found! ======================================== ffmpeg is required for M4A/AAC audio decoding. Install ffmpeg: macOS: brew install ffmpeg Ubuntu: sudo apt install ffmpeg ffprobez1ffprobe not found (usually installed with ffmpeg)z-vquietz -print_formatjsonz -show_streamsz-select_streamsza:0-ipipe:0Tinputcapture_output)r,rzffprobe failed: streamszNo audio streams found in file sample_rateiDchannelsr-fs16lez-acodec pcm_s16le-ar-acpipe:1zffmpeg decoding failed: dtype)shutilwhich RuntimeError isinstancebytes subprocessrunstr returncodestderrdecoder'loadsstdoutgetintnp frombufferr) r" ffmpeg_path ffprobe_path probe_cmd probe_resultr' probe_infostreamr. nchannels decode_cmd decode_resultsampless r_decode_ffmpegrS5sq,,x(K 2  << *LNOO*e$            "~~            O  "~~iE !#-l.A.A.H.H.J-KLMML//6689J >>) $;<<  "1 %Ffjj67KFJJz1-.I*e$             N   #      O          N   #z$G 1$5m6J6J6Q6Q6S5TUVVmmM00AG K **r!file always_2dr7c$d}t|ttfr9t|jj dj }|dvrbd}n_t|t jrE|jd|jd}|jd|ddd k(s|d dd k(rd}|rOt|t jr"|jd|j}n|}t|\}}} ndd l } t|ttfrL| jt|} | jt|| j| j } nt|t jr|jd|j} t!| }|d k(r| j#| } nS|dk(r| j%| } n<|dk(r| j'| } n%|dk(r| j)| } nt+d|| j-| | j| j } nt/dt1|| j}| j} t3j4| j6t2j8}| dkDr|j;d| }|dvr|j=|dz }n|dk(rn|j=|}|r&|j>dk(r|d d t2j@f}||fS)aRead an audio file using miniaudio (or ffmpeg for M4A/AAC). Args: file: Path to the audio file or a BytesIO object. always_2d: If True, always return a 2D array (samples, channels). dtype: Data type for the output array. Supports 'float32', 'float64', 'int16'. Returns: Tuple of (audio_data, sample_rate). audio_data is a numpy array with shape (samples,) for mono or (samples, channels) for multi-channel. F.)r r r opusTrrrrrNr)rOr.rrr r zUnsupported format: zUnsupported file type: r6)rfloat64g@r)!r;r?rsuffixlstriplowerioBytesIOseekreadrS miniaudio get_file_info decode_filerOr.r wav_get_info mp3_get_info flac_get_infovorbis_get_inforrB TypeErrortyperGarrayrRrreshapeastypendimnewaxis)rTrUr7 use_ffmpegextheaderr"rRr.rOrcinfodecodedrfmts rrbrbs"J$d $4j&&s+113 / /J D"** % ! 2 ! !A;' !VBQZ7%:J dBJJ ' IIaLJJ*8*D'i  dS$K (**3t95D++D .. ,,,G bjj ) IIaL99;D+D1Ce| --d3 --d3 ..t4 006 #7u!=>>&&.. ,,'G 5d4j\BC C)) %% ((7??"((;1}//"i0 &&..''1 '  ..'W\\Q&!RZZ-( K r!c0tjdduS)z+Check if ffmpeg is available on the system.r$N)r8r9r!r_check_ffmpeg_availablerys << ! --r!cJtjd}| td|S)zGet the path to ffmpeg executable. Returns: Path to ffmpeg executable. Raises: RuntimeError: If ffmpeg is not found. r$a5 ======================================== ffmpeg not found! ======================================== ffmpeg is required for MP3/FLAC encoding and M4A/AAC decoding. Install ffmpeg: macOS: brew install ffmpeg Ubuntu: sudo apt install ffmpeg Alternatively, use WAV format which doesn't require ffmpeg. )r8r9r:)rIs r_get_ffmpeg_pathr{s3,,x(K L   r! sampleraterOoutputformatbitratec t}|jdk(r|j}n|jj}|ddddt |dt |ddg }|d k(r|j d |gn2|d k(r|j d d d |gn|dvr|j d dg|dk(rd}|j d|gt |tjr|jdtj||d} | jdk7r&td| jj|j!| j"|j%dy|jt |tj||d} | jdk7r&td| jjy)a}Encode audio using ffmpeg. Args: data: Audio data as int16 numpy array (samples,) or (samples, channels) samplerate: Sample rate in Hz nchannels: Number of channels output: Output file path or BytesIO object format: Output format (mp3, flac, ogg, opus, vorbis, etc.) bitrate: Audio bitrate for lossy formats (default: 128k) rYz-yr0r1r3r4r(r)rz-b:arXz-c:alibopus)r r r r r r5Tr*rzffmpeg failed: N)r{rotobytesflattenr?extendr;r_r`appendr=r>r@r:rArBwriterDra) rr|rOr}r~rrI pcm_bytescmdresults r_encode_ffmpegr7s$#$K yyA~LLN LLN**,      J  I  C FG$% 6  FIvw78 $ $ FF#$JJf~&"**% 8      !1E1E1G0HIJ J V]]# A 3v;      !1E1E1G0HIJ J "r!cddl}|Kt|ttfr3t|jj dj }nd}|j }t|tjsst|dr0t|ds$tj|j}n7t|drtj|}ntj|}|jtjtjfvr:tj |dd}|d zj#tj$}n<|jtj$k7r|j#tj$}|j&d k(rd }|}n|j(d }|j+}|d k(s|d k(rt|j-}t|t.j0r#|j3||j5dyt7|d 5}|j3|dddy|dk(rddl } | jd|j} |j9d|||j:j<| } t|t.j0rddl} | jAdd5} | jB}ddd |jE| t7|d5}|j3|jGddd|j5dddl$}|jK|y|jEt|| y|dvrItMs.ddl'}|jQd|jSdtTdtW|||||ytYd|#1swYyxYw#1swYxYw#1swYxYw#ddl$}|jKwxYw)a/Write audio data to a file. Args: file: Path to the output file or a BytesIO object. data: Audio data as numpy array. Shape can be (samples,) for mono or (samples, channels) for multi-channel. samplerate: Sample rate in Hz. format: Output format. Supports 'wav', 'flac', 'mp3', 'ogg', 'opus', 'vorbis'. If None, inferred from file extension. Note: WAV uses miniaudio for encoding. MP3, FLAC, OGG, Opus, and Vorbis use ffmpeg (must be installed: brew install ffmpeg). rNrWrtolist __array__gg?irYpcmrawwbhr})namerOr. sample_formatrRz.wavF)r\deleterb)r rr rXr zffmpeg is required for z encoding but was not found. Install with: brew install ffmpeg (macOS) or sudo apt install ffmpeg (Ubuntu). Falling back will fail - consider using WAV format instead.r) stacklevelr~zUnsupported output format: )-rcr;r?rr\r]r^rGndarrayhasattrrlrasarrayr7rr[cliprnrroshaperrr_r`rraopenDecodedSoundFile SampleFormatrtempfileNamedTemporaryFilerwav_write_filerbosunlinkrywarningswarnupperRuntimeWarningrr)rTrr|r~rcrO samples_flatrfrl samples_arraysoundrtmptmp_pathrrs rrrs0(~ dS$K ($Z&&--c288:FF \\^F dBJJ ' 4 "74+E88DKKM*D T; '::d#D88D>D zzbjj"**--wwtT3'u $$RXX. rxx {{288$ yyA~  JJqM ||~  &E/ ((* dBJJ ' JJy ! IIaLdD!Q ""! 5 C)<)<)>? **"#0099! +  dBJJ ' ,,F5,IS88J $((59(D)QJJqvvx(* !  (#  $ $SY 6 ; ;&(  MM)&,,.)9:NN   tZDH6vh?@@e"!,JI*) (#s<'O O O%; OO%O  OO"O%%O<ct||dS)aRead audio file (soundfile-compatible API). This is a drop-in replacement for soundfile.read(). Args: file: Path to audio file or BytesIO object. always_2d: If True, always return 2D array. Returns: Tuple of (data, samplerate). r[)rUr7)rb)rTrUs rsf_readrs   ;;r!c"t||||y)a+Write audio file (soundfile-compatible API). This is a drop-in replacement for soundfile.write(). Args: file: Path to output file or BytesIO object. data: Audio data as numpy array. samplerate: Sample rate in Hz. format: Output format (wav, flac, mp3). rN)r)rTrr|r~s rsf_writers  $j0r!)Fr[)r128k)N)F)__doc__r_r8r=pathlibrtypingrrrnumpyrG _FORMAT_MAP_SAMPLE_FORMAT_MAPr<r?r rrFrSr`boolrbryr{rrrrrxr!rrsl ))        EEEcE"|+c4&'|+ 2::sC  |+B_ T2::% &_ _  _  2::s? _ D.. #B LK **LKLKLK #tRZZ' ( LK  LK  LK LKf! sA T2::% &sA **sAsA SM sA  sAr< T2::% &<< 2::s?<,! 1 T2::% &1 **11 SM 1  1r!