o qmi?I@s~ddlZddlZddlZddlZddlZddlZddlmZddlm Z ej r/ddl m Z m Z GdddZGd d d ZdS) N) HTTPStatus)JobStatusV1Response)AsyncSpeechToTextJobClientSpeechToTextJobClientc@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))AsyncSpeechToTextJobjob_idclientrcC||_||_dS)aD Initialize the asynchronous speech-to-text job. Parameters ---------- job_id : str The unique job identifier returned from a previous job initialization. client : AsyncSpeechToTextJobClient The async client instance used to create the job. !!! important This must be the **same client instance** that was used to initialize the job originally, as it contains the subscription key and configuration required to authenticate and manage the job. N_job_id_clientselfr r rS/home/ubuntu/.local/lib/python3.10/site-packages/sarvamai/speech_to_text_job/job.py__init__ zAsyncSpeechToTextJob.__init__returncC|jSzt Returns the job ID associated with this job instance. Returns ------- str rrrrrr % zAsyncSpeechToTextJob.job_idN@ file_pathstimeoutc s*|jj|jdd|DdIdH}tj|d}tj|d4IdHf}|D]W}tj|}|j |j }t |d%} t |\} } | durFd} |j|| d| d d IdH} Wdn1s`wY| jtjksq| jtjkr|td |d | jq%WdIdHd S1IdHswYd S) Upload input audio files for the speech-to-text 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 Ez5AsyncSpeechToTextJob.upload_files..r filesNrrb audio/wav BlockBlobzx-ms-blob-typez Content-TypecontentheadersUpload failed for : T)rget_upload_linksrhttpxTimeout AsyncClientr"r#r$ upload_urlsfile_urlopen mimetypes guess_typeputread status_coderIM_USEDOK RuntimeError) rrr upload_linksclient_timeoutsessionr# file_nameurlf content_type_responserrr upload_files0sB      z!AsyncSpeechToTextJob.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 TimeoutErrorrsleeprrQrstartstatusstaterrrwait_until_completeas  z(AsyncSpeechToTextJob.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_fileinputsoutputsrcrHr&detailrrrr(  z.r[ job_detailsr job_statusrrrget_output_mappingss  z(AsyncSpeechToTextJob.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 successfulrSrrHrb error_messagerjrhrxrS r[rrrllenrHrcrzrmappend IndexErrorAttributeErrorrrtresultsro file_inforrrget_file_resultss4   z%AsyncSpeechToTextJob.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|]}|dqSrjrr&mrrrr(z9AsyncSpeechToTextJob.download_outputs..r*Texist_okrjDownload failed for r5ri.jsonwb)rurget_download_linksrr"makedirsr7r9 download_urlsr;getrArrBrCrDr#joinr<writer2) rrmappings file_namesdownload_linksrGrrIrM output_pathrJrrrdownload_outputss4    z%AsyncSpeechToTextJob.download_outputscs|j|jIdHS)v Retrieve the current status of the job. Returns ------- JobStatusV1Response Nrr[rrrrrr[szAsyncSpeechToTextJob.get_statuscs|jj|jdIdHS)w Start the speech-to-text job processing. Returns ------- JobStatusV1Response r NrrarrrrrraszAsyncSpeechToTextJob.startcs.z |IdHWdStjyYdSw)f Check if the job exists in the system. Returns ------- bool NTFr[r7HTTPStatusErrorrrrrexistsszAsyncSpeechToTextJob.existscs|IdHj}|dvS)o Check if the job is either completed or failed. Returns ------- bool N>rSrTr[r\r])rrcrrr is_completesz AsyncSpeechToTextJob.is_completec|IdHjdkS)h Check if the job completed successfully. Returns ------- bool NrTrrrrr is_successful z"AsyncSpeechToTextJob.is_successfulcr)\ Check if the job has failed. Returns ------- bool NrSrrrrr is_failed*rzAsyncSpeechToTextJob.is_failedNrrOrP__name__ __module__ __qualname__strrpropertyr typingSequencefloatboolrNintrrdListDictruAnyrrr[rarrrrrrrrr s>  2 $ 1*    r c@r))SpeechToTextJobr r rcCr )a8 Initialize the synchronous speech-to-text job. Parameters ---------- job_id : str The unique job identifier returned from a previous job initialization. client : SpeechToTextJobClient The client instance used to create the job. !!! important This must be the **same client instance** that was used to initialize the job originally, as it contains the subscription key and configuration required to authenticate and manage the job. Nr rrrrr6rzSpeechToTextJob.__init__rcCrrrrrrrr KrzSpeechToTextJob.job_idrrrc Cs|jj|jdd|Dd}tj|d}tj|dQ}|D]E}tj|}|j |j }t |d} |j || dddd } Wd n1sFwY| j tjksW| j tjkrbtd |d | j qWd d S1snwYd S)rcSr rr!r%rrrr(jr)z0SpeechToTextJob.upload_files..r*r,r-r/r.r0r1Nr4r5T)rr6rr7r8Clientr"r#r$r:r;r<r?rArrBrCrD) rrrrErFr r#rHrIrJrMrrrrNVs8      zSpeechToTextJob.upload_filesrOrPrQcCs\t} |}|j}|dvr|St||kr(td|jd|dt|q)rRT>rSrTrUrVrW)rZ monotonicr[r\r]r^rr_r`rrrrds  z#SpeechToTextJob.wait_until_completecCs|}dd|jp gDS)recSrfrgrkrnrrrr(rpz7SpeechToTextJob.get_output_mappings..rqrsrrrrus z#SpeechToTextJob.get_output_mappingsc Cs|}ggd}|jp gD]K}|jrt|jdkrqz3|jdj|j|j|jr5t|jdkr5|jdjndd}|jdkrF|d|n|d|Wqt t fyYYqw|S)rvrwrNryrhrxrSr{rrrrrs2   z SpeechToTextJob.get_file_resultsrc 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(rz4SpeechToTextJob.download_outputs..r*Trrjrr5rirrN)rurrrr"rr7rrr;rrArrBrCrDr#rr<rr2) rrrrrr rrIrMrrJrrrrs2      z SpeechToTextJob.download_outputscCs|j|jS)rrrrrrr[szSpeechToTextJob.get_statuscCs|jj|jdS)rrrrrrrraszSpeechToTextJob.startcCs&z|WdStjyYdSw)rTFrrrrrr)s zSpeechToTextJob.existscCs|jdvS)r>rSrTrrrrrr7zSpeechToTextJob.is_completecC|jdkS)rrTrrrrrrArzSpeechToTextJob.is_successfulcCr)rrSrrrrrrKrzSpeechToTextJob.is_failedNrrrrrrrr5s>  . $ 1*    r)rXr=r"rZrr7httprtypesr TYPE_CHECKINGr rrr rrrrrs  (