(HJjBdZddlZddlZddlZddlmZddlmZmZm Z ddl Z dddddd d d d Z d d ddZ dedefdZde eeefdee j"eeffdZ d$de eeej(fdededee j"effdZdefdZdefdZ d%de j"dedede eeej(fdededdfd Z d&de eeej(fde j"dedeeddf d!Z d'de eeej(fdedee j"effd"Z d&de eeej(fde j"dedeeddf d#Zy)(a]Audio 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, OGG, Opus, and WebM format support. - Writing: Uses miniaudio for WAV and ffmpeg for MP3, FLAC, OGG, Opus, Vorbis, and WebM encoding. N)Path)OptionalTupleUnionwavmp3flacvorbism4awebm)rrr oggr r aacr SIGNED16SIGNED32FLOAT32)int16int32float32datareturnc|dddk(r |dddk(ry|ddd k(s|d d d k(s|d d d k(ry|dddk(ry|dddk(ry|dddk(ry|dddk(rytd)z6Detect audio format from bytes data using magic bytes.NsRIFF sWAVErsID3rssrsfLaCr OggSr ftypr Eߣr z(Unable to detect audio format from bytes) ValueError)rs \/Users/ahmed/devFolder/claude-voice/.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/WebM 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_infostreamr0 nchannels decode_cmd decode_resultsampless r!_decode_ffmpegrU9sq,,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_2dr9c4d}t|ttfr9t|jj dj }|dvrjd}ngt|t jrM|jd|jd}|jd|ddd k(s|d dd 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!| }|dk(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/OGG/Opus/WebM). 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 rr opusr TrrrrrNrr)rQr0rrr r zUnsupported format: zUnsupported file type: r8)rfloat64g@r)!r=rArsuffixlstriplowerioBytesIOseekreadrU miniaudio get_file_info decode_filerQr0r" wav_get_info mp3_get_info flac_get_infovorbis_get_infor rD TypeErrortyperIarrayrTrreshapeastypendimnewaxis)rVrWr9 use_ffmpegextheaderr$rTr0rQreinfodecodedrfmts r!rdrds"J$d $4j&&s+113 7 7J D"** % ! 2 ! 1QK7 "bqzW$bqz00J 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)r:r;r#r!_check_ffmpeg_availabler{s << ! --r#cJtjd}| td|S)zGet the path to ffmpeg executable. Returns: Path to ffmpeg executable. Raises: RuntimeError: If ffmpeg is not found. r&a? ======================================== ffmpeg not found! ======================================== ffmpeg is required for MP3/FLAC/WebM encoding and M4A/AAC/WebM decoding. Install ffmpeg: macOS: brew install ffmpeg Ubuntu: sudo apt install ffmpeg Alternatively, use WAV format which doesn't require ffmpeg. )r:r;r<)rKs r!_get_ffmpeg_pathr}#s3,,x(K L   r# sampleraterQoutputformatbitratec t}|jdk(r|j}n|jj}|ddddt |dt |ddg }|d k(r|j d |gnM|d k(r|j d d d |gn2|dk(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) r[z-yr2r3r5r6r*r+rz-b:arZz-c:alibopusr )r r r r r r7Tr,rzffmpeg failed: N)r}rqtobytesflattenrAextendr=rarbappendr?r@rBr<rCrDwriterFrc) rr~rQrrrrK pcm_bytescmdresults r!_encode_ffmpegr?s$#$K yyA~LLN LLN**,      J  I  C FG$% 6  FIvw78 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', 'webm'. If None, inferred from file extension. Note: WAV uses miniaudio for encoding. MP3, FLAC, OGG, Opus, Vorbis, and WebM use ffmpeg (must be installed: brew install ffmpeg). rNrYrtolist __array__gg?ir[pcmrawwbhr)namerQr0 sample_formatrTz.wavF)r^deleterb)r rr rZr r 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) stacklevelrzUnsupported output format: )-rer=rArr^r_r`rIndarrayhasattrrnrasarrayr9rr]cliprprrqshaperrrarbrrcopenDecodedSoundFile SampleFormatrtempfileNamedTemporaryFilerwav_write_filerdosunlinkr{warningswarnupperRuntimeWarningrr )rVrr~rrerQ samples_flatrfrn samples_arraysoundrtmptmp_pathrrs r!rrs0(~ 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 C C&(  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])rWr9)rd)rVrWs r!sf_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)rVrr~rs r!sf_writers  $j0r#)Fr])r128k)N)F)__doc__rar:r?pathlibrtypingrrrnumpyrI _FORMAT_MAP_SAMPLE_FORMAT_MAPr>rAr"rrHrUrbboolrdr{r}rrrrrzr#r!rso ))          EEEcE(|+c4&'|+ 2::sC  |+Bc T2::% &c c  c  2::s? c L.. #B NK **NKNKNK #tRZZ' ( NK  NK  NK NKj! sA T2::% &sA **sAsA SM sA  sAr< T2::% &<< 2::s?<,! 1 T2::% &1 **11 SM 1  1r#