
    sh                     p   S SK r S SKrS SKrS SKrS SKrS SKJrJrJr  S SK	r	S SK
J
r
  SSKJrJrJr  SSKJrJrJrJr  SSKJrJr  SSKJr  SS	KJr  S
SSSSSSSSSSSSSS.rSSSSSSSSSS S!S!S"S"S.rS#\S$\S%\S&\\\4   4S' jrS&\\   4S( jr    S-S)\S*\\\\	RB                  4      S+\S%\S&\4
S, jjr"g).    N)ListOptionalUnion)tqdm   )
load_audiolog_mel_spectrogrampad_or_trim)DecodingOptionsDecodingResultdecodedetect_language)ModelDimensionsWhisper)
transcribe)__version__zhttps://openaipublic.azureedge.net/main/whisper/models/d3dd57d32accea0b295c96e26691aa14d8822fac7d9d27d5dc00b4ca2826dd03/tiny.en.ptzhttps://openaipublic.azureedge.net/main/whisper/models/65147644a518d12f04e32d6f3b26facc3f8dd46e5390956a9424a650c0ce22b9/tiny.ptzhttps://openaipublic.azureedge.net/main/whisper/models/25a8566e1d0c1e2231d1c762132cd20e0f96a85d16145c3a00adf5d1ac670ead/base.en.ptzhttps://openaipublic.azureedge.net/main/whisper/models/ed3a0b6b1c0edf879ad9b11b1af5a0e6ab5db9205f891f668f8b0e6c6326e34e/base.ptzhttps://openaipublic.azureedge.net/main/whisper/models/f953ad0fd29cacd07d5a9eda5624af0f6bcf2258be67c92b79389873d91e0872/small.en.ptzhttps://openaipublic.azureedge.net/main/whisper/models/9ecf779972d90ba49c06d968637d720dd632c55bbf19d441fb42bf17a411e794/small.ptzhttps://openaipublic.azureedge.net/main/whisper/models/d7440d1dc186f76616474e0ff0b3b6b879abc9d1a4926b7adfa41db2d497ab4f/medium.en.ptzhttps://openaipublic.azureedge.net/main/whisper/models/345ae4da62f9b3d59415adc60127b97c714f32e89e936602e85993674d08dcb1/medium.ptzhttps://openaipublic.azureedge.net/main/whisper/models/e4b87e7e0bf463eb8e6956e646f1e277e901512310def2c24bf0e11bd3c28e9a/large-v1.ptzhttps://openaipublic.azureedge.net/main/whisper/models/81f7c96c852ee8fc832187b0132e569d6c3065a3252ed18e56effd0b6a73e524/large-v2.ptzhttps://openaipublic.azureedge.net/main/whisper/models/e5b1a55b89c1367dacf97e3e19bfd829a01529dbfdeefa8caeb59b3f1b81dadb/large-v3.ptzhttps://openaipublic.azureedge.net/main/whisper/models/aff26ae408abcba5fbf8813c21e62b0941638c5f6eebfb145be0c9839262a19a/large-v3-turbo.pt)ztiny.entinyzbase.enbasezsmall.ensmallz	medium.enmediumzlarge-v1zlarge-v2zlarge-v3largezlarge-v3-turboturbos%   ABzY8J1N>@0{>%R00Bk>$p{7v037`oCl~+#00s#   ABzY8bu8Lr0{>%RKn9Fp%m@SkK7Kt=7ytkOs%   ABzY8;40c<0{>%RzzG;p*o+Vo09|#PsxSZm00s(   ABzY8KQ!870{>%RzyTQH3`Q^yNP!>##QT-<FaQ7ms4   ABzY8>?_)10{>%RpeA61k&I|OI3I$65C{;;pbCHh0B{qLQ;+}v00s1   ABzY8DmU6=0{>%Rpa?J`kvJ6qF(V^F86#Xh7JUGMK}P<N0000s>   ABzY8usPae0{>%R7<zz_OvQ{)4kMa0BMw6u5rT}kRKX;$NfYBv00*Hl@qhsU00s,   ABzY8B0Jh+0{>%R7}kK1fFL7w6%<-Pf*t^=N)Qr&0RR9s2   ABzY8r9j$a0{>%R7#4sLmoOs{s)o3~84-RPdcFk!JR<kSfC2yjsK   ABzY8zd+h!0{>%R7=D0pU<_bnWW*tkYAhobTNnu$jnkEkXqp)j;w1Tzk)UH3X%SZd&fFZ2fC2yjs9   ABzY8gWO1E0{>%R7(9S+Kn!D~%ngiGaR?*L!iJG9p-nab0JQ=-{D1-g00s(   ABzY8j^C+e0{>%RARaKHP%t(lGR*)0g!tONPyhe`urlroot	in_memoryreturnc           
      ^   [         R                  " USS9  U R                  S5      S   n[         R                  R	                  U[         R                  R                  U 5      5      n[         R                  R                  U5      (       a2  [         R                  R                  U5      (       d  [        U S35      e[         R                  R                  U5      (       aq  [        US5       nUR                  5       nS S S 5        [        R                  " W5      R                  5       U:X  a  U(       a  U$ U$ [        R                  " U S35        [         R"                  R%                  U 5       n[        US5       n['        [)        UR+                  5       R-                  S	5      5      S
SSSS9 n	 UR                  S5      n
U
(       d  O-UR/                  U
5        U	R1                  [3        U
5      5        MF  S S S 5        S S S 5        S S S 5        [        US5      R                  5       n[        R                  " U5      R                  5       U:w  a  [        S5      eU(       a  U$ U$ ! , (       d  f       GNs= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       N= f)NT)exist_ok/z! exists and is not a regular filerbzH exists, but the SHA256 checksum does not match; re-downloading the filewbzContent-LengthP   iBi   )totalncolsunit
unit_scaleunit_divisori    zeModel has been downloaded but the SHA256 checksum does not not match. Please retry loading the model.)osmakedirssplitpathjoinbasenameexistsisfileRuntimeErroropenreadhashlibsha256	hexdigestwarningswarnurllibrequesturlopenr   intinfogetwriteupdatelen)r   r   r   expected_sha256download_targetfmodel_bytessourceoutputloopbuffers              d/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/whisper/__init__.py	_downloadrL   6   s   KKt$iinR(Oggll4)9)9#)>?O	ww~~o&&rww~~o/N/No..OPQQ	ww~~o&&/4(A&&(K )>>+&002oE"+;@@MM"##kl 
			$_d0Kvfkkm''(89:
 T*V$CK( 
 1L	$  -224K~~k",,./As
 	
 $;88? )(
 
 1L0K	$	$sI   -I*1J>4J2AI<:JJ*
I9<
J
J
J	J
J,c                  <    [        [        R                  5       5      $ )z%Returns the names of available models)list_MODELSkeys     rK   available_modelsrS   b   s    rR   namedevicedownload_rootc                    Uc'  [         R                  R                  5       (       a  SOSnUcr  [        R                  R                  [        R                  R                  S5      S5      n[        R                  R                  [        R                  " SU5      S5      nU [        ;   a  [        [        U    X#5      n[        U    nOc[        R                  R                  U 5      (       a&  U(       a  [        U S5      R                  5       OU nSnO[        S	U  S
[        5        35      eU(       a  [         R"                  " U5      O[        US5       n[         R$                  S:  a  SS0O0 n[         R&                  " U4SU0UD6n	SSS5        A[)        S0 W	S   D6n
[+        U
5      nUR-                  U	S   5        Ub  UR/                  U5        UR1                  U5      $ ! , (       d  f       Na= f)a!  
Load a Whisper ASR model

Parameters
----------
name : str
    one of the official model names listed by `whisper.available_models()`, or
    path to a model checkpoint containing the model dimensions and the model state_dict.
device : Union[str, torch.device]
    the PyTorch device to put the model into
download_root: str
    path to download the model files; by default, it uses "~/.cache/whisper"
in_memory: bool
    whether to preload the model weights into host memory

Returns
-------
model : Whisper
    The Whisper ASR model instance
Ncudacpu~z.cacheXDG_CACHE_HOMEwhisperr!   zModel z not found; available models = z1.13weights_onlyTmap_locationdimsmodel_state_dictrQ   )torchrX   is_availabler*   r-   r.   
expandusergetenvrO   rL   _ALIGNMENT_HEADSr1   r3   r4   r2   rS   ioBytesIOr   loadr   r   load_state_dictset_alignment_headsto)rT   rU   rV   r   defaultcheckpoint_filealignment_headsfpkwargs
checkpointr_   models               rK   
load_modelrs   g   s   6 ~ ::2244%'',,rww11#6ARYY/?%I9Uw#GDM=L*40			5>$tT*//1DTF9:J:L9MN
 	

 (1

?#d?D6QQ	+0+<+<+F.$'BZZBB6B
 	R 	0Z/0DDME	*%789"!!/288F 	RQs   4G""
G0)NNF)#r5   rf   r*   r:   r8   typingr   r   r   ra   r   audior   r	   r
   decodingr   r   r   r   rr   r   r   r   versionr   rO   re   strboolbytesrL   rS   rU   rs   rQ   rR   rK   <module>r{      sP    	 	   ( (   ? ? N N + "   T N T N V P X R V V V S b Y( 8277GAR=E^LIA8 $)93 )9c )9d )9uUCZ7H )9X $s)   26	:
:U3,-.: : 	:
 :rR   