
    Z jH	                         S r SSKJr  SSKJr  SSKJrJrJr  SSK	J
r
Jr  SSKJr   " S S\S	S
9r\ " S S\5      5       rS/rg)z
Processor class for TrOCR.
   )BatchFeature)
ImageInput)ProcessingKwargsProcessorMixinUnpack)PreTokenizedInput	TextInput)auto_docstringc                       \ rS rSr0 rSrg)TrOCRProcessorKwargs    N)__name__
__module____qualname____firstlineno__	_defaults__static_attributes__r       {/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/models/trocr/processing_trocr.pyr   r      s    Ir   r   F)totalc            
          ^  \ rS rSrSU 4S jjr\  SS\S-  S\\-  \	\   -  \	\   -  S-  S\
\   S\4S jj5       r\S	 5       rS
rU =r$ )TrOCRProcessor   Nc                 $   > [         TU ]  X5        g )N)super__init__)selfimage_processor	tokenizerkwargs	__class__s       r   r   TrOCRProcessor.__init__   s    4r   imagestextr!   returnc                 
   Uc  Uc  [        S5      eU R                  " [        4SU R                  R                  0UD6nUb  U R
                  " U40 US   D6nUb  U R                  " U40 US   D6nUc  W$ Uc  W$ WS   WS'   U$ )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__TrOCRProcessor.__call__"   s     >dlabb** 
"&.."<"<
 
 ))&SM/4RSFtL}]/KLI<M^(5F8Mr   c                 :    U R                   R                  nUS/-   $ )Nr,   )r   model_input_names)r   image_processor_input_namess     r   r6    TrOCRProcessor.model_input_names?   s!    &*&:&:&L&L#*hZ77r   r   )NN)r   r   r   r   r   r
   r   r	   r   listr   r   r   r3   propertyr6   r   __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>rC      sS    3 % H H C #+5  $8^ $8 $8N 
r   