
    Z j5                     F    S r SSKJr  SSKJr  \ " S S\5      5       rS/rg)z$
Speech processor class for Whisper
   )ProcessorMixin)auto_docstringc                   T   ^  \ rS rSrU 4S jrSS jr\S 5       rS	S\4S jjr	Sr
U =r$ )
WhisperProcessor   c                 $   > [         TU ]  X5        g )N)super__init__)selffeature_extractor	tokenizer	__class__s      /root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/models/whisper/processing_whisper.pyr
   WhisperProcessor.__init__   s    *6    c                 6    U R                   R                  XUS9$ )N)tasklanguageno_timestamps)r   get_decoder_prompt_ids)r   r   r   r   s       r   r   'WhisperProcessor.get_decoder_prompt_ids   s    ~~44$an4oor   c                 L   UR                  SS 5      nUR                  SS 5      nUR                  SS 5      n[        U5      S:  a
  US   nUSS  nUc  Uc  [        S5      eUb  U R                  " U/UQ7SU0UD6nUb  U R                  " U40 UD6nUc  W$ Uc  W$ WS   WS'   U$ )	Naudiosampling_ratetext       zAYou need to specify either an `audio` or `text` input to process.	input_idslabels)poplen
ValueErrorr   r   )r   argskwargsr   r   r   inputs	encodingss           r   __call__WhisperProcessor.__call__   s    

7D)

?D9zz&$'t9q=GE8D=T\`aa++E`D``Y_`Ft6v6I<M](5F8Mr   r   c                 4    U R                   R                  XS9$ )N)return_tensors)r   get_prompt_ids)r   r   r*   s      r   r+   WhisperProcessor.get_prompt_ids8   s    ~~,,T,QQr    )NNT)np)__name__
__module____qualname____firstlineno__r
   r   r   r'   strr+   __static_attributes____classcell__)r   s   @r   r   r      s6    7p  2R3 R Rr   r   N)__doc__processing_utilsr   utilsr   r   __all__r-   r   r   <module>r:      s;    / # "R~ "R "RJ 
r   