:jbddlZddlmZddlmZddlmZmZddlm Z ddl m Z GddeZ y) N)Pipeline) AudioFile) AnnotationSegment)Client) DiarizeOutputc eZdZdZddededzffd ZdeedefdZ dd e d e dzd e dzd e dzde f d Z xZS)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") Nmodeltokenc t|||_|xs tjj dd|_t|j |_y)NPYANNOTEAI_API_KEY) super__init__r osenvirongetr r_client)selfr r kwargs __class__s |/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/pyannote/audio/pipelines/pyannoteai/sdk.pyrz SDK.__init__4sA  Hbjjnn-A4H djj) diarizationreturnct}t|D]$\}}t|d|d}|d}||||f<&|jdS)Nstartend)rrspeakerstring)r enumerater rename_tracks)rr annotationtturnsegmentrs r _deserializezSDK._deserialize;s_\  - -GAtDMtE{CG9oG%,Jwz " -''11rfile num_speakers min_speakers max_speakersc <|jj|}|jj||||d|jd}|jj |}|j |dd}|j |dd} t || S)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_outputr1r2s rapplyz SDK.applyEsR,,T2 %% %%%**& \\**62 *.*;*;JxrIs'. #,-!Fc (c r