o pi@s\ddlZddlmZddlmZddlmZmZddlm Z ddl m Z GdddeZ dS) N)Pipeline) AudioFile) AnnotationSegment)Client) DiarizeOutputc sxeZdZdZddededBffdd Zdeed efd d Z   dd e d e dBde dBde dBd e f ddZ ZS)SDKa.Wrapper around official pyannoteAI API client Parameters ---------- model : str, optional pyannoteAI speaker diarization model. Defaults to "precision-2". 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.sdk import SDK >>> pipeline = SDK(token="{PYANNOTEAI_API_KEY}") >>> speaker_diarization = pipeline("/path/to/your/audio.wav") precision-2Nmodeltokenc s4t||_|ptjdd|_t|j|_dS)NPYANNOTEAI_API_KEY) super__init__r osenvirongetr r_client)selfr r kwargs __class__[/home/ubuntu/.local/lib/python3.10/site-packages/pyannote/audio/pipelines/pyannoteai/sdk.pyr4s z SDK.__init__ diarizationreturncCsJt}t|D]\}}t|d|dd}|d}||||f<q|dS)Nstartend)rrspeakerstring)r enumerater rename_tracks)rr annotationtturnsegmentrrrr _deserialize;s  zSDK._deserializefile num_speakers min_speakers max_speakersc Csd|j|}|jj||||d|jdd}|j|}||dd}||dd} t|| dS)alSpeaker diarization using pyannoteAI web API This method will upload `file`, initiate a diarization job, retrieve its output, and deserialize the latter into a good old pyannote.core.Annotation instance. 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 ------- speaker_diarization : Annotation Speaker diarization result (when successful) Raises ------ PyannoteAIFailedJob If the job failed PyannoteAICanceledJob If the job was canceled HTTPError If something else went wrong FT)r'r(r) confidencer exclusiveoutputrexclusiveDiarization)speaker_diarizationexclusive_speaker_diarization)ruploaddiarizer retriever%r) rr&r'r(r) media_urljob_id job_outputr.r/rrrapplyEs" ) z SDK.apply)r N)NNN)__name__ __module__ __qualname____doc__strrlistdictrr%rintrr6 __classcell__rrrrr!s$ r) rpyannote.audiorpyannote.audio.core.ior pyannote.corerrpyannoteai.sdkr,pyannote.audio.pipelines.speaker_diarizationrrrrrrs