iVddlZddlmZddlmZddlmZmZddlm Z GddeZ y) NPipeline) AudioFile) AnnotationSegment) DiarizeOutputc |eZdZdZd dedzffd ZdeedefdZ dde d e dzd e dzd e dzde f d Z xZS)LocalaWrapper around official pyannoteAI on-premise package Parameters ---------- token : str, optional pyannoteAI API key created from https://dashboard.pyannote.ai. Defaults to using `PYANNOTEAI_API_KEY` environment variable. Usage ----- >>> from pyannote.audio.pipelines.pyannoteai.local import Local >>> pipeline = Local(token="{PYANNOTEAI_API_KEY}") >>> speaker_diarization = pipeline("/path/to/your/audio.wav") Ntokenc t|ddlm}|xs tj j dd|_||j|_y)NrrPYANNOTEAI_API_KEY) super__init__pyannoteai.localrosenvirongetr _pipeline)selfr kwargs_LocalPipeline __class__s ~/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pyannote/audio/pipelines/pyannoteai/local.pyrzLocal.__init__0s= ?Hbjjnn-A4H ' 3 diarizationreturnct}t|D]$\}}t|d|d}|d}||||f<&|jdS)Nstartend)rr speakerstring)r enumerater rename_tracks)rr annotationtturnsegmentr!s r _deserializezLocal._deserialize8sZ\  -GAtDMtE{CG9oG%,Jwz ".''11rfile num_speakers min_speakers max_speakersc rd|vr%|jj|df|||d|}nZd|vrK|d}t|dr|jd}|jj||ddf|||d|}n t d |j |d }|j |d } t || S) a Speaker diarization using on-premise pyannoteAI package. Parameters ---------- file : AudioFile Processed file. num_speakers : int, optional Force number of speakers to diarize. If not provided, the number of speakers will be determined automatically. min_speakers : int, optional Not supported yet. Minimum number of speakers. Has no effect when `num_speakers` is provided. max_speakers : int, optional Not supported yet. Maximum number of speakers. Has no effect when `num_speakers` is provided. Returns ------- output : DiarizeOutput DiarizeOutput object containing both regular and exclusive speaker diarization results. audio)r+r,r-waveformnumpyT)force sample_rate)r0r3z7AudioFile must provide either 'audio' or 'waveform' keyrexclusive_diarization)speaker_diarizationexclusive_speaker_diarization)rdiarizehasattrr1 ValueErrorr)r ) rr*r+r,r-r predictionsr0r5r6s rapplyz Local.applyAs: d?0$..00W )))   K4 J'Hx)#>>>50$..00%d=6IJ)))   KVW W*.*;*;K  > Dj> Dj > Dj >  > rr ) rpyannote.audiorpyannote.audio.core.ior pyannote.corerrr5r r rrrIs$. #,-/_ H_ r