o qmiI@s~ddlZddlZddlZddlZddlZddlZddlmZddlm Z ej r/ddl m Z m Z GdddZGd d d ZdS) N) HTTPStatus)JobStatusV1Response)#AsyncSpeechToTextTranslateJobClientSpeechToTextTranslateJobClientc@eZdZdeddfddZedefddZ d'd ejed e de fd d Z d(de d e de fddZdejejeeffddZdejeejejeejfffddZdede fddZde fddZde fddZde fddZde fd d!Zde fd"d#Zde fd$d%Zd&S))AsyncSpeechToTextTranslateJobjob_idclientrcC||_||_dS)aa Initialise the asynchronous speech-to-text-translate-translate job. Parameters ---------- job_id : str The unique job identifier returned from a previous job initialisation. client : AsyncSpeechToTextTranslateJobClient The async client instance used to create the job. !!! important This must be the **same client instance** that was used to initialise the job originally, as it contains the subscription key and configuration required to authenticate and manage the job. N_job_id_clientselfr r r]/home/ubuntu/.local/lib/python3.10/site-packages/sarvamai/speech_to_text_translate_job/job.py__init__s z&AsyncSpeechToTextTranslateJob.__init__returncC|jSzt Returns the job ID associated with this job instance. Returns ------- str rrrrrr , z$AsyncSpeechToTextTranslateJob.job_idN@ file_pathstimeoutc s|jj|jdd|DdIdH}tj|d}tj|d4IdHY}|D]J}tj|}|j |j }t |d} |j || ddd d IdH} Wdn1sSwY| jtjksd| jtjkrotd |d | jq%WdIdHd S1IdHswYd S)a Upload input audio files for the speech-to-text-translate job. Parameters ---------- file_paths : Sequence[str] List of full paths to local audio files. timeout : float, optional The maximum time to wait for the upload to complete (in seconds), by default 60.0 Returns ------- bool True if all files are uploaded successfully. cSg|]}tj|qSrospathbasename.0prrr Lz>AsyncSpeechToTextTranslateJob.upload_files..r filesNrrb BlockBlob audio/wavzx-ms-blob-typez Content-TypecontentheadersUpload failed for : T)rget_upload_linksrhttpxTimeout AsyncClientr r!r" upload_urlsfile_urlopenputread status_coderIM_USEDOK RuntimeError) rrr upload_linksclient_timeoutsessionr! file_nameurlfresponserrr upload_files7s<      z*AsyncSpeechToTextTranslateJob.upload_filesX poll_intervalcsrt} |IdH}|j}|dvr|St||kr0td|jd|dt|IdHq) Polls job status until it completes or fails. Parameters ---------- poll_interval : int, optional Time in seconds between polling attempts (default is 5). timeout : int, optional Maximum time to wait for completion in seconds (default is 600). Returns ------- JobStatusV1Response Final job status. Raises ------ TimeoutError If the job does not complete within the given timeout. TN>failed completedJob  did not complete within seconds.) asyncioget_event_looptime get_status job_statelower TimeoutErrorrsleeprrKrstartstatusstaterrrwait_until_completees  z1AsyncSpeechToTextTranslateJob.wait_until_completecs$|IdH}dd|jpgDS) Get the mapping of input files to their corresponding output files. Returns ------- List[Dict[str, str]] List of mappings with keys 'input_file' and 'output_file'. NcS<g|]}|jr|jr|jdkr|jdj|jdjdqSSuccessr) input_file output_fileinputsoutputsr]rDr$detailrrrr&  zEAsyncSpeechToTextTranslateJob.get_output_mappings..rU job_detailsr job_statusrrrget_output_mappingss  z1AsyncSpeechToTextTranslateJob.get_output_mappingsc s|IdH}ggd}|jpgD]K}|jrt|jdkrqz3|jdj|j|j|jr9t|jdkr9|jdjndd}|jdkrJ|d|n|d|Wqt t fy]Yqw|S)- Get detailed results for each file in the batch job. Returns ------- Dict[str, List[Dict[str, Any]]] Dictionary with 'successful' and 'failed' keys, each containing a list of file details. Each file detail includes: - 'file_name': Name of the input file - 'status': Status of processing ('Success' or 'Failed') - 'error_message': Error message if failed (None if successful) - 'output_file': Name of output file if successful (None if failed) N successfulrMrrDr\ error_messagerdrbrrrM rUrlrflenrDr]rtrgappend IndexErrorAttributeErrorrrnresultsri file_inforrrget_file_resultss4   z.AsyncSpeechToTextTranslateJob.get_file_results output_dirc s0|IdH}dd|D}|jj|j|dIdH}tj|ddt4IdHa}|D]R}|j|dj }| |IdH}|j t j ksK|j t jkrXtd|dd |j tj||d d } t| d } | |jWdn1szwYq-WdIdHdS1IdHswYdS) ~ Download output files to the specified directory. Parameters ---------- output_dir : str Local directory where outputs will be saved. Returns ------- bool True if all files downloaded successfully. Raises ------ RuntimeError If a file fails to download. NcSg|]}|dqSrdrr$mrrrr&zBAsyncSpeechToTextTranslateJob.download_outputs..r(Texist_okrdDownload failed for r3rc.jsonwb)rorget_download_linksrr makedirsr5r7 download_urlsr9getr=rr>r?r@r!joinr:writer0) rr~mappings file_namesdownload_linksrCrrErG output_pathrFrrrdownload_outputss4    z.AsyncSpeechToTextTranslateJob.download_outputscs|j|jIdHS)v Retrieve the current status of the job. Returns ------- JobStatusV1Response NrrUrrrrrrUsz(AsyncSpeechToTextTranslateJob.get_statuscs|jj|jdIdHS) Start the speech-to-text-translate job processing. Returns ------- JobStatusV1Response r Nrr[rrrrrr[sz#AsyncSpeechToTextTranslateJob.startcs.z |IdHWdStjyYdSw)f Check if the job exists in the system. Returns ------- bool NTFrUr5HTTPStatusErrorrrrrexists sz$AsyncSpeechToTextTranslateJob.existscs|IdHj}|dvS)o Check if the job is either completed or failed. Returns ------- bool N>rMrNrUrVrW)rr]rrr is_completesz)AsyncSpeechToTextTranslateJob.is_completec|IdHjdkS)h Check if the job completed successfully. Returns ------- bool NrNrrrrr is_successful$z+AsyncSpeechToTextTranslateJob.is_successfulcr)\ Check if the job has failed. Returns ------- bool NrMrrrrr is_failed.rz'AsyncSpeechToTextTranslateJob.is_failedNrrIrJ__name__ __module__ __qualname__strrpropertyr typingSequencefloatboolrHintrr^ListDictroAnyr}rrUr[rrrrrrrrr sF   / $ 1*    r c@r))SpeechToTextTranslateJobr r rcCr )aK Initialise the synchronous speech-to-text-translate job. Parameters ---------- job_id : str The unique job identifier returned from a previous job initialisation. client : SpeechToTextTranslateJobClient The client instance used to create the job. !!! important This must be the **same client instance** that was used to initialise the job originally, as it contains the subscription key and configuration required to authenticate and manage the job. Nr rrrrr:s z!SpeechToTextTranslateJob.__init__rcCrrrrrrrr OrzSpeechToTextTranslateJob.job_idrrrc Cs|jj|jdd|Dd}tj|d}tj|d^}|D]R}tj|}|j |j }t |\} } | dur8d} t |d} |j|| d| d d } Wdn1sSwY| jtjksd| jtjkrotd |d | jqWdd S1s{wYd S)a Upload input audio files for the speech-to-text-translate job. Parameters ---------- file_paths : Sequence[str] List of full paths to local audio files. timeout : float, optional The maximum time to wait for the upload to complete (in seconds), by default 60.0 Returns ------- bool True if all files are uploaded successfully. cSrrrr#rrrr&or'z9SpeechToTextTranslateJob.upload_files..r(r*Nr-r+r,r.r/r2r3T)rr4rr5r6Clientr r!r"r8r9 mimetypes guess_typer:r;r=rr>r?r@) rrrrArBr r!rDrE content_type_rFrGrrrrHZs>      z%SpeechToTextTranslateJob.upload_filesrIrJrKcCs\t} |}|j}|dvr|St||kr(td|jd|dt|q)rLT>rMrNrOrPrQ)rT monotonicrUrVrWrXrrYrZrrrr^s  z,SpeechToTextTranslateJob.wait_until_completecCs|}dd|jp gDS)r_cSr`rarerhrrrr&rjz@SpeechToTextTranslateJob.get_output_mappings..rkrmrrrros z,SpeechToTextTranslateJob.get_output_mappingsc Cs|}ggd}|jp gD]K}|jrt|jdkrqz3|jdj|j|j|jr5t|jdkr5|jdjndd}|jdkrF|d|n|d|Wqt t fyYYqw|S)rprqrNrsrbrrrMrurzrrrr}s2   z)SpeechToTextTranslateJob.get_file_resultsr~c Cs|}dd|D}|jj|j|d}tj|ddt[}|D]O}|j|dj }| |}|j t j ks=|j t jkrJtd|dd|j tj||d d } t| d } | |jWd n1slwYq"Wd dS1s}wYdS) rcSrrrrrrrr&rz=SpeechToTextTranslateJob.download_outputs..r(Trrdrr3rcrrN)rorrrr rr5rrr9rr=rr>r?r@r!rr:rr0) rr~rrrr rrErGrrFrrrrs2      z)SpeechToTextTranslateJob.download_outputscCs|j|jS)rrrrrrrUsz#SpeechToTextTranslateJob.get_statuscCs|jj|jdS)rrrrrrrr['szSpeechToTextTranslateJob.startcCs&z|WdStjyYdSw)rTFrrrrrr1s zSpeechToTextTranslateJob.existscCs|jdvS)r>rMrNrrrrrr?z$SpeechToTextTranslateJob.is_completecC|jdkS)rrNrrrrrrIrz&SpeechToTextTranslateJob.is_successfulcCr)rrMrrrrrrSrz"SpeechToTextTranslateJob.is_failedNrrrrrrrr9s>  2 $ 1*    r)rRrr rTrr5httprtypesr TYPE_CHECKINGr rrr rrrrrs  )