
"""simple docstring"""






# Copyright 2023 The HuggingFace Inc. team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from ..models.whisper import WhisperForConditionalGeneration, WhisperProcessor
from .base import PipelineTool



class 			lowercase					( lowercase__		):
			lowercase        =     '''openai/whisper-base'''
			lowercase        =     (
			    '''This is a tool that transcribes an audio into text. It takes an input named `audio` and returns the '''
			    '''transcribed text.'''
			)
			lowercase        =     '''transcriber'''
			lowercase        =     WhisperProcessor
			lowercase        =     WhisperForConditionalGeneration

			lowercase        =     ['''audio''']
			lowercase        =     ['''text''']






			def 					UpperCAmelCase							(self  :	int      ,SCREAMING_SNAKE_CASE_  :	List[Any]			)       ->	Union[str, Any]:





				"""simple docstring"""


				return self.pre_processor(SCREAMING_SNAKE_CASE_      ,return_tensors='''pt'''			).input_features






			def 					UpperCAmelCase							(self  :	str      ,SCREAMING_SNAKE_CASE_  :	Tuple			)       ->	List[str]:





				"""simple docstring"""


				return self.model.generate(inputs=SCREAMING_SNAKE_CASE_			)






			def 					UpperCAmelCase							(self  :	Optional[int]      ,SCREAMING_SNAKE_CASE_  :	Optional[Any]			)       ->	List[str]:





				"""simple docstring"""


				return self.pre_processor.batch_decode(SCREAMING_SNAKE_CASE_      ,skip_special_tokens=SCREAMING_SNAKE_CASE_			)[0]




"""simple docstring"""






import warnings

from ...processing_utils import ProcessorMixin
from ...tokenization_utils_base import BatchEncoding



class 			lowercase					( lowercase__		):
			lowercase        =     ['''image_processor''', '''tokenizer''']
			lowercase        =     '''CLIPImageProcessor'''
			lowercase        =     ('''XLMRobertaTokenizer''', '''XLMRobertaTokenizerFast''')






			def __init__(self  :	Union[str, Any]      ,SCREAMING_SNAKE_CASE_  :	int=None      ,SCREAMING_SNAKE_CASE_  :	str=None      ,**SCREAMING_SNAKE_CASE_  :	Union[str, Any]			)       ->	int:





				"""simple docstring"""


				lowerCAmelCase												=  None
				if "feature_extractor" in kwargs:
					warnings.warn(
					    '''The `feature_extractor` argument is deprecated and will be removed in v5, use `image_processor`'''
					    ''' instead.'''      ,SCREAMING_SNAKE_CASE_      ,)
					lowerCAmelCase												=  kwargs.pop('''feature_extractor'''			)

				lowerCAmelCase												=  image_processor if image_processor is not None else feature_extractor
				if image_processor is None:
					raise ValueError('''You need to specify an `image_processor`.'''			)
				if tokenizer is None:
					raise ValueError('''You need to specify a `tokenizer`.'''			)

				super().__init__(SCREAMING_SNAKE_CASE_      ,SCREAMING_SNAKE_CASE_			)






			def __call__(self  :	Union[str, Any]      ,SCREAMING_SNAKE_CASE_  :	List[str]=None      ,SCREAMING_SNAKE_CASE_  :	Optional[int]=None      ,SCREAMING_SNAKE_CASE_  :	Union[str, Any]=None      ,**SCREAMING_SNAKE_CASE_  :	Any			)       ->	List[Any]:





				"""simple docstring"""


				if text is None and images is None:
					raise ValueError('''You have to specify either text or images. Both cannot be none.'''			)

				if text is not None:
					lowerCAmelCase												=  self.tokenizer(SCREAMING_SNAKE_CASE_      ,return_tensors=SCREAMING_SNAKE_CASE_      ,**SCREAMING_SNAKE_CASE_			)

				if images is not None:
					lowerCAmelCase												=  self.image_processor(SCREAMING_SNAKE_CASE_      ,return_tensors=SCREAMING_SNAKE_CASE_      ,**SCREAMING_SNAKE_CASE_			)

				if text is not None and images is not None:
					lowerCAmelCase												=  image_features.pixel_values
					return encoding
				elif text is not None:
					return encoding
				else:
					return BatchEncoding(data=dict(**SCREAMING_SNAKE_CASE_			)      ,tensor_type=SCREAMING_SNAKE_CASE_			)






			def 					UpperCAmelCase							(self  :	List[str]      ,*SCREAMING_SNAKE_CASE_  :	Union[str, Any]      ,**SCREAMING_SNAKE_CASE_  :	Tuple			)       ->	Optional[Any]:





				"""simple docstring"""


				return self.tokenizer.batch_decode(*SCREAMING_SNAKE_CASE_      ,**SCREAMING_SNAKE_CASE_			)






			def 					UpperCAmelCase							(self  :	Union[str, Any]      ,*SCREAMING_SNAKE_CASE_  :	int      ,**SCREAMING_SNAKE_CASE_  :	Union[str, Any]			)       ->	Any:





				"""simple docstring"""


				return self.tokenizer.decode(*SCREAMING_SNAKE_CASE_      ,**SCREAMING_SNAKE_CASE_			)






			@property
			def 					UpperCAmelCase							(self  :	List[Any]			)       ->	Dict:





				"""simple docstring"""


				lowerCAmelCase												=  self.tokenizer.model_input_names
				lowerCAmelCase												=  self.image_processor.model_input_names
				return list(dict.fromkeys(tokenizer_input_names + image_processor_input_names			)			)


