o pi@sPddlZddlmZddlmZddlmZmZddlm Z GdddeZ dS) NPipeline) AudioFile) AnnotationSegment) DiarizeOutputc steZdZdZddedBffdd Zdeedefdd Z   dd e d e dBd e dBd e dBde f ddZ ZS)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 s:tddlm}|ptjdd|_||j|_dS)NrrPYANNOTEAI_API_KEY) super__init__pyannoteai.localrosenvirongetr _pipeline)selfr kwargs_LocalPipeline __class__]/home/ubuntu/.local/lib/python3.10/site-packages/pyannote/audio/pipelines/pyannoteai/local.pyr 0s  zLocal.__init__ diarizationreturncCsJt}t|D]\}}t|d|dd}|d}||||f<q|dS)Nstartend)rrspeakerstring)r enumerater rename_tracks)rr annotationtturnsegmentrrrr _deserialize8s  zLocal._deserializefile num_speakers min_speakers max_speakersc Ksd|vr|jj|df|||d|}n+d|vr<|d}t|dr(|jdd}|jj||ddf|||d|}ntd ||d }||d } t|| d 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)r,r/z7AudioFile must provide either 'audio' or 'waveform' keyrexclusive_diarization)speaker_diarizationexclusive_speaker_diarization)rdiarizehasattrr- ValueErrorr&r) rr'r(r)r*r predictionsr,r1r2rrrapplyAs@    z Local.apply)N)NNN)__name__ __module__ __qualname____doc__strr listdictrr&rintrr7 __classcell__rrrrr s$ r ) rpyannote.audiorpyannote.audio.core.ior pyannote.corerrr1rr rrrrs