




'''simple docstring'''







from pathlib import PurePosixPath
from typing import Optional

import fsspec
from fsspec import AbstractFileSystem
from huggingface_hub.hf_api import DatasetInfo

from ..utils.file_utils import get_authentication_headers_for_url
from ..utils.hub import hf_hub_url


class 			_a  (	__a ):
       __a				:     Dict   				=					""""""
       __a				:     str   				=					"""hf-legacy"""  # "hf://"" is reserved for hffs
       def __init__(	self						:	str   ,				lowercase						:	Optional[DatasetInfo] = None   ,				lowercase						:	Optional[str] = None   ,				**lowercase						:	Tuple   ,				):



           '''simple docstring'''

           super().__init__(self   ,				**lowercase				)
           UpperCAmelCase			       =		repo_info
           UpperCAmelCase			       =		token
           UpperCAmelCase			       =		None
       def     A      (	self						:	Optional[int]				):



           '''simple docstring'''

           if self.dir_cache is None:
               UpperCAmelCase			       =		{}
               for hf_file in self.repo_info.siblings:
                   # TODO(QL): add sizes
                   UpperCAmelCase			       =		{
                       '''name''': hf_file.rfilename,
                       '''size''': None,
                       '''type''': '''file''',
                   }
                   self.dir_cache.update(
                       {
                           str(lowercase				): {'''name''': str(lowercase				), '''size''': None, '''type''': '''directory'''}
                           for d in list(PurePosixPath(hf_file.rfilename				).parents				)[:-1]
                       }				)
       def     A      (	self						:	Tuple   ,				lowercase						:	str   ,				lowercase						:	str = "rb"   ,				**lowercase						:	Dict   ,				):



           '''simple docstring'''

           if not isinstance(self.repo_info   ,				lowercase				):
               raise NotImplementedError(f"Open is only implemented for dataset repositories, but got {self.repo_info}"				)
           UpperCAmelCase			       =		hf_hub_url(self.repo_info.id   ,				lowercase   ,				revision=self.repo_info.sha				)
           return fsspec.open(
               lowercase   ,				mode=lowercase   ,				headers=get_authentication_headers_for_url(lowercase   ,				use_auth_token=self.token				)   ,				client_kwargs={'''trust_env''': True}   ,				).open()
       def     A      (	self						:	List[str]   ,				lowercase						:	Optional[int]   ,				**lowercase						:	List[Any]				):



           '''simple docstring'''

           self._get_dirs()
           UpperCAmelCase			       =		self._strip_protocol(lowercase				)
           if path in self.dir_cache:
               return self.dir_cache[path]
           else:
               raise FileNotFoundError(lowercase				)







       def     A      (	self						:	List[Any]   ,				lowercase						:	Union[str, Any]   ,				lowercase						:	Optional[Any]=False   ,				**lowercase						:	Any				):



           '''simple docstring'''

           self._get_dirs()
           UpperCAmelCase			       =		PurePosixPath(path.strip('''/'''				)				)
           UpperCAmelCase			       =		{}
           for p, f in self.dir_cache.items():
               UpperCAmelCase			       =		PurePosixPath(p.strip('''/'''				)				)
               UpperCAmelCase			       =		p.parent
               if root == path:
                   UpperCAmelCase			       =		f
           UpperCAmelCase			       =		list(paths.values()				)
           if detail:
               return out
           else:
               return sorted(f['''name'''] for f in out				)








# 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 	UpperCAmelCase_					(					UpperCamelCase    ):

   '''simple docstring'''




   __A :      Dict     		=		"openai/whisper-base"
   __A :      str     		=		(
       "This is a tool that transcribes an audio into text. It takes an input named `audio` and returns the "
       "transcribed text."
   )
   __A :      Any     		=		"transcriber"
   __A :      Any     		=		WhisperProcessor
   __A :      int     		=		WhisperForConditionalGeneration

   __A :      Any     		=		["audio"]
   __A :      List[str]     		=		["text"]






   def _snake_case    (		self						,				__A ):






     """simple docstring"""





     return self.pre_processor(__A						,				return_tensors="pt" ).input_features






   def _snake_case    (		self						,				__A ):






     """simple docstring"""





     return self.model.generate(inputs=__A )






   def _snake_case    (		self						,				__A ):






     """simple docstring"""





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



