
    qiH	                         d Z ddlmZ ddlmZ ddlmZmZmZ ddl	m
Z
mZ ddlmZ  G d ded	
      Ze G d de             ZdgZy)z
Processor class for TrOCR.
   )BatchFeature)
ImageInput)ProcessingKwargsProcessorMixinUnpack)PreTokenizedInput	TextInput)auto_docstringc                       e Zd Zi Zy)TrOCRProcessorKwargsN)__name__
__module____qualname__	_defaults     \/opt/pipecat/venv/lib/python3.12/site-packages/transformers/models/trocr/processing_trocr.pyr   r      s    Ir   r   F)totalc            
            e Zd Zd	 fd	Ze	 	 d	dedz  deez  ee   z  ee   z  dz  de	e
   defd       Zed        Z xZS )
TrOCRProcessorNc                 &    t         |   ||       y )N)super__init__)selfimage_processor	tokenizerkwargs	__class__s       r   r   zTrOCRProcessor.__init__   s    )4r   imagestextr   returnc                     ||t        d       | j                  t        fd| j                  j                  i|}| | j
                  |fi |d   }| | j                  |fi |d   }|S |S d   d<   |S )NzBYou need to specify either an `images` or `text` input to process.tokenizer_init_kwargsimages_kwargstext_kwargs	input_idslabels)
ValueError_merge_kwargsr   r   init_kwargsr   )r   r   r    r   output_kwargsinputs	encodingss          r   __call__zTrOCRProcessor.__call__"   s     >dlabb*** 
"&.."<"<
 
 )T))&SM/4RSF&tL}]/KLI<M^(5F8Mr   c                 :    | j                   j                  }|dgz   S )Nr'   )r   model_input_names)r   image_processor_input_namess     r   r0   z TrOCRProcessor.model_input_names?   s!    &*&:&:&L&L#*hZ77r   )NN)r   r   r   r   r
   r   r	   r   listr   r   r   r.   propertyr0   __classcell__)r   s   @r   r   r      s    5  %)aeT! ++d9o=EV@WWZ^^ -.	
 
 8 8 8r   r   N)__doc__image_processing_utilsr   image_utilsr   processing_utilsr   r   r   tokenization_utils_baser   r	   utilsr
   r   r   __all__r   r   r   <module>r<      sS    3 % H H C #+5  $8^ $8 $8N 
r   