



'''simple docstring'''
import unittest
from pathlib import Path

from transformers import is_vision_available, load_tool
from transformers.testing_utils import get_tests_dir

from .test_tools_common import ToolTesterMixin


if is_vision_available():
					from PIL import Image



class _A	(     unittest.TestCase						,					__SCREAMING_SNAKE_CASE     ):





		def 				__A							(  self					)							->  Union[str, Any]:

					'''simple docstring'''
					__UpperCAmelCase :   str    					=  load_tool("""image-question-answering"""					)
					self.tool.setup()
					__UpperCAmelCase :   int    					=  load_tool("""image-question-answering"""			,							remote=__UpperCAmelCase					)





		def 				__A							(  self					)							->  Any:

					'''simple docstring'''
					__UpperCAmelCase :   Any    					=  Image.open(Path(get_tests_dir("""fixtures/tests_samples/COCO"""					)					) / """000000039769.png"""					)
					__UpperCAmelCase :   Union[str, Any]    					=  self.tool(__UpperCAmelCase			,							"""How many cats are sleeping on the couch?"""					)
					self.assertEqual(__UpperCAmelCase			,							"""2"""					)





		def 				__A							(  self					)							->  Tuple:

					'''simple docstring'''
					__UpperCAmelCase :   int    					=  Image.open(Path(get_tests_dir("""fixtures/tests_samples/COCO"""					)					) / """000000039769.png"""					)
					__UpperCAmelCase :   List[Any]    					=  self.remote_tool(__UpperCAmelCase			,							"""How many cats are sleeping on the couch?"""					)
					self.assertEqual(__UpperCAmelCase			,							"""2"""					)





		def 				__A							(  self					)							->  Optional[int]:

					'''simple docstring'''
					__UpperCAmelCase :   Tuple    					=  Image.open(Path(get_tests_dir("""fixtures/tests_samples/COCO"""					)					) / """000000039769.png"""					)
					__UpperCAmelCase :   Union[str, Any]    					=  self.tool(image=__UpperCAmelCase			,							question="""How many cats are sleeping on the couch?"""					)
					self.assertEqual(__UpperCAmelCase			,							"""2"""					)





		def 				__A							(  self					)							->  Optional[int]:

					'''simple docstring'''
					__UpperCAmelCase :   List[Any]    					=  Image.open(Path(get_tests_dir("""fixtures/tests_samples/COCO"""					)					) / """000000039769.png"""					)
					__UpperCAmelCase :   List[Any]    					=  self.remote_tool(image=__UpperCAmelCase			,							question="""How many cats are sleeping on the couch?"""					)
					self.assertEqual(__UpperCAmelCase			,							"""2"""					)









'''simple docstring'''
import os
import warnings
from pathlib import Path
from typing import Optional, Union

from .dynamic_module_utils import custom_object_save
from .tokenization_utils_base import PreTrainedTokenizerBase
from .utils import PushToHubMixin, copy_func, direct_transformers_import, logging


_UpperCamelCase									=      logging.get_logger(__name__)

# Dynamically import the Transformers module to grab the attribute classes of the processor form their names.
_UpperCamelCase									=      direct_transformers_import(Path(__file__).parent)


_UpperCamelCase									=      {
    '''AutoTokenizer''': '''PreTrainedTokenizerBase''',
    '''AutoFeatureExtractor''': '''FeatureExtractionMixin''',
    '''AutoImageProcessor''': '''ImageProcessingMixin''',
}



class _A	(     __SCREAMING_SNAKE_CASE     ):
		_SCREAMING_SNAKE_CASE       :     str       =      ["feature_extractor", "tokenizer"]
		# Names need to be attr_class for attr in attributes
		_SCREAMING_SNAKE_CASE       :     Optional[int]       =      None
		_SCREAMING_SNAKE_CASE       :     str       =      None
		_SCREAMING_SNAKE_CASE       :     List[str]       =      None





		def __init__(  self			,							*__UpperCAmelCase			,							**__UpperCAmelCase					)							->  Optional[int]:

					'''simple docstring'''
					# Sanitize args and kwargs
					for key in kwargs:
								if key not in self.attributes:
											raise TypeError(f'Unexpected keyword argument {key}.'					)
					for arg, attribute_name in zip(__UpperCAmelCase			,							self.attributes					):
								if attribute_name in kwargs:
											raise TypeError(f'Got multiple values for argument {attribute_name}.'					)
								else:
											__UpperCAmelCase :   int    					=  arg

					if len(__UpperCAmelCase					) != len(self.attributes					):
								raise ValueError(
								    f'This processor requires {len(self.attributes					)} arguments: {", ".join(self.attributes					)}. Got '
								    f'{len(__UpperCAmelCase					)} arguments instead.'					)

					# Check each arg is of the proper class (this will also catch a user initializing in the wrong order)
					for attribute_name, arg in kwargs.items():
								__UpperCAmelCase :   int    					=  getattr(self			,							f'{attribute_name}_class'					)
								# Nothing is ever going to be an instance of "AutoXxx", in that case we check the base class.
								__UpperCAmelCase :   str    					=  AUTO_TO_BASE_CLASS_MAPPING.get(__UpperCAmelCase			,							__UpperCAmelCase					)
								if isinstance(__UpperCAmelCase			,							__UpperCAmelCase					):
											__UpperCAmelCase :   str    					=  tuple(getattr(__UpperCAmelCase			,							__UpperCAmelCase					) for n in class_name if n is not None					)
								else:
											__UpperCAmelCase :   Optional[Any]    					=  getattr(__UpperCAmelCase			,							__UpperCAmelCase					)

								if not isinstance(__UpperCAmelCase			,							__UpperCAmelCase					):
											raise ValueError(
											    f'Received a {type(__UpperCAmelCase					).__name__} for argument {attribute_name}, but a {class_name} was expected.'					)

								setattr(self			,							__UpperCAmelCase			,							__UpperCAmelCase					)





		def __repr__(  self					)							->  List[str]:

					'''simple docstring'''
					__UpperCAmelCase :   Any    					=  [f'- {name}: {repr(getattr(self			,							__UpperCAmelCase					)					)}' for name in self.attributes]
					__UpperCAmelCase :   List[Any]    					=  """\n""".join(__UpperCAmelCase					)
					return f'{self.__class__.__name__}:\n{attributes_repr}'





		def 				__A							(  self			,							__UpperCAmelCase			,							__UpperCAmelCase = False			,							**__UpperCAmelCase					)							->  Any:

					'''simple docstring'''
					os.makedirs(__UpperCAmelCase			,							exist_ok=__UpperCAmelCase					)

					if push_to_hub:
								__UpperCAmelCase :   Optional[int]    					=  kwargs.pop("""commit_message"""			,							__UpperCAmelCase					)
								__UpperCAmelCase :   List[Any]    					=  kwargs.pop("""repo_id"""			,							save_directory.split(os.path.sep					)[-1]					)
								__UpperCAmelCase :   Any    					=  self._create_repo(__UpperCAmelCase			,							**__UpperCAmelCase					)
								__UpperCAmelCase :   List[str]    					=  self._get_files_timestamps(__UpperCAmelCase					)
					# If we have a custom config, we copy the file defining it in the folder and set the attributes so it can be
					# loaded from the Hub.
					if self._auto_class is not None:
								__UpperCAmelCase :   Union[str, Any]    					=  [getattr(self			,							__UpperCAmelCase					) for attribute_name in self.attributes]
								__UpperCAmelCase :   Dict    					=  [(a.init_kwargs if isinstance(__UpperCAmelCase			,							__UpperCAmelCase					) else a) for a in attrs]
								custom_object_save(self			,							__UpperCAmelCase			,							config=__UpperCAmelCase					)

					for attribute_name in self.attributes:
								__UpperCAmelCase :   Optional[int]    					=  getattr(self			,							__UpperCAmelCase					)
								# Include the processor class in the attribute config so this processor can then be reloaded with the
								# `AutoProcessor` API.
								if hasattr(__UpperCAmelCase			,							"""_set_processor_class"""					):
											attribute._set_processor_class(self.__class__.__name__					)
								attribute.save_pretrained(__UpperCAmelCase					)

					if self._auto_class is not None:
								# We added an attribute to the init_kwargs of the tokenizers, which needs to be cleaned up.
								for attribute_name in self.attributes:
											__UpperCAmelCase :   str    					=  getattr(self			,							__UpperCAmelCase					)
											if isinstance(__UpperCAmelCase			,							__UpperCAmelCase					):
														del attribute.init_kwargs["auto_map"]

					if push_to_hub:
								self._upload_modified_files(
								    __UpperCAmelCase			,							__UpperCAmelCase			,							__UpperCAmelCase			,							commit_message=__UpperCAmelCase			,							token=kwargs.get("""use_auth_token"""					)			,							)





		@classmethod
		def 				__A							(  cls			,							__UpperCAmelCase			,							__UpperCAmelCase = None			,							__UpperCAmelCase = False			,							__UpperCAmelCase = False			,							__UpperCAmelCase = None			,							__UpperCAmelCase = "main"			,							**__UpperCAmelCase			,							)							->  Optional[int]:

					'''simple docstring'''
					__UpperCAmelCase :   List[str]    					=  cache_dir
					__UpperCAmelCase :   Tuple    					=  force_download
					__UpperCAmelCase :   Optional[int]    					=  local_files_only
					__UpperCAmelCase :   str    					=  revision

					__UpperCAmelCase :   int    					=  kwargs.pop("""use_auth_token"""			,							__UpperCAmelCase					)
					if use_auth_token is not None:
								warnings.warn(
								    """The `use_auth_token` argument is deprecated and will be removed in v5 of Transformers."""			,							__UpperCAmelCase					)
								if token is not None:
											raise ValueError(
											    """`token` and `use_auth_token` are both specified. Please set only the argument `token`."""					)
								__UpperCAmelCase :   Any    					=  use_auth_token

					if token is not None:
								# change to `token` in a follow-up PR
								__UpperCAmelCase :   Dict    					=  token

					__UpperCAmelCase :   Optional[int]    					=  cls._get_arguments_from_pretrained(__UpperCAmelCase			,							**__UpperCAmelCase					)
					return cls(*__UpperCAmelCase					)





		@classmethod
		def 				__A							(  cls			,							__UpperCAmelCase="AutoProcessor"					)							->  Dict:

					'''simple docstring'''
					if not isinstance(__UpperCAmelCase			,							__UpperCAmelCase					):
								__UpperCAmelCase :   List[Any]    					=  auto_class.__name__

					import transformers.models.auto as auto_module

					if not hasattr(__UpperCAmelCase			,							__UpperCAmelCase					):
								raise ValueError(f'{auto_class} is not a valid auto class.'					)

					__UpperCAmelCase :   Optional[Any]    					=  auto_class





		@classmethod
		def 				__A							(  cls			,							__UpperCAmelCase			,							**__UpperCAmelCase					)							->  List[str]:

					'''simple docstring'''
					__UpperCAmelCase :   Union[str, Any]    					=  []
					for attribute_name in cls.attributes:
								__UpperCAmelCase :   List[Any]    					=  getattr(cls			,							f'{attribute_name}_class'					)
								if isinstance(__UpperCAmelCase			,							__UpperCAmelCase					):
											__UpperCAmelCase :   str    					=  tuple(getattr(__UpperCAmelCase			,							__UpperCAmelCase					) if n is not None else None for n in class_name					)
											__UpperCAmelCase :   int    					=  kwargs.get("""use_fast"""			,							__UpperCAmelCase					)
											if use_fast and classes[1] is not None:
														__UpperCAmelCase :   int    					=  classes[1]
											else:
														__UpperCAmelCase :   Optional[Any]    					=  classes[0]
								else:
											__UpperCAmelCase :   str    					=  getattr(__UpperCAmelCase			,							__UpperCAmelCase					)

								args.append(attribute_class.from_pretrained(__UpperCAmelCase			,							**__UpperCAmelCase					)					)
					return args





		@property
		def 				__A							(  self					)							->  Tuple:

					'''simple docstring'''
					__UpperCAmelCase :   Optional[Any]    					=  getattr(self			,							self.attributes[0]					)
					return getattr(__UpperCAmelCase			,							"""model_input_names"""			,							__UpperCAmelCase					)


_UpperCamelCase									=      copy_func(ProcessorMixin.push_to_hub)
if ProcessorMixin.push_to_hub.__doc__ is not None:
					_UpperCamelCase									=      ProcessorMixin.push_to_hub.__doc__.format(
					    object='''processor''', object_class='''AutoProcessor''', object_files='''processor files'''
					)




