
'''simple docstring'''



import argparse
import os
import re

import packaging.version


_lowercase          =						"""examples/"""
_lowercase          =						{
    """examples""": (re.compile(R"""^check_min_version\(\"[^\"]+\"\)\s*$""", re.MULTILINE), """check_min_version(\"VERSION\")\n"""),
    """init""": (re.compile(R"""^__version__\s+=\s+\"([^\"]+)\"\s*$""", re.MULTILINE), """__version__ = \"VERSION\"\n"""),
    """setup""": (re.compile(R"""^(\s*)version\s*=\s*\"[^\"]+\",""", re.MULTILINE), R"""\1version=\"VERSION\","""),
    """doc""": (re.compile(R"""^(\s*)release\s*=\s*\"[^\"]+\"$""", re.MULTILINE), """release = \"VERSION\"\n"""),
}
_lowercase          =						{
    """init""": """src/diffusers/__init__.py""",
    """setup""": """setup.py""",
}
_lowercase          =						"""README.md"""


def A (__lowerCamelCase :Any					,      __lowerCamelCase :Optional[int]					,      __lowerCamelCase :Optional[Any]						):
						with open(__lowerCamelCase					,      """r"""					,      encoding="""utf-8"""					,      newline="""\n"""						) as f:
												_lowerCAmelCase    		=					f.read()
						_lowerCAmelCase		,       _lowerCAmelCase    		=					REPLACE_PATTERNS[pattern]
						_lowerCAmelCase    		=					replace.replace("""VERSION"""					,      __lowerCamelCase						)
						_lowerCAmelCase    		=					re_pattern.sub(__lowerCamelCase					,      __lowerCamelCase						)
						with open(__lowerCamelCase					,      """w"""					,      encoding="""utf-8"""					,      newline="""\n"""						) as f:
												f.write(__lowerCamelCase						)


def A (__lowerCamelCase :Optional[int]						):
						for folder, directories, fnames in os.walk(__lowerCamelCase						):
												# Removing some of the folders with non-actively maintained examples from the walk
												if "research_projects" in directories:
																		directories.remove("""research_projects"""						)
												if "legacy" in directories:
																		directories.remove("""legacy"""						)
												for fname in fnames:
																		if fname.endswith(""".py"""						):
																								update_version_in_file(os.path.join(__lowerCamelCase					,      __lowerCamelCase						)					,      __lowerCamelCase					,      pattern="""examples"""						)


def A (__lowerCamelCase :Union[str, Any]					,      __lowerCamelCase :Union[str, Any]=False						):
						for pattern, fname in REPLACE_FILES.items():
												update_version_in_file(__lowerCamelCase					,      __lowerCamelCase					,      __lowerCamelCase						)
						if not patch:
												update_version_in_examples(__lowerCamelCase						)


def A ():
						_lowerCAmelCase    		=					"""🤗 Transformers currently provides the following architectures"""
						_lowerCAmelCase    		=					"""1. Want to contribute a new model?"""
						with open(__lowerCamelCase					,      """r"""					,      encoding="""utf-8"""					,      newline="""\n"""						) as f:
												_lowerCAmelCase    		=					f.readlines()

						# Find the start of the list.
						_lowerCAmelCase    		=					0
						while not lines[start_index].startswith(_start_prompt						):
												start_index += 1
						start_index += 1

						_lowerCAmelCase    		=					start_index
						# Update the lines in the model list.
						while not lines[index].startswith(_end_prompt						):
												if lines[index].startswith("""1."""						):
																		_lowerCAmelCase    		=					lines[index].replace(
																		    """https://huggingface.co/docs/diffusers/main/model_doc"""					,      """https://huggingface.co/docs/diffusers/model_doc"""					,      )
												index += 1

						with open(__lowerCamelCase					,      """w"""					,      encoding="""utf-8"""					,      newline="""\n"""						) as f:
												f.writelines(__lowerCamelCase						)


def A ():
						with open(REPLACE_FILES["""init"""]					,      """r"""						) as f:
												_lowerCAmelCase    		=					f.read()
						_lowerCAmelCase    		=					REPLACE_PATTERNS["""init"""][0].search(__lowerCamelCase						).groups()[0]
						return packaging.version.parse(__lowerCamelCase						)


def A (__lowerCamelCase :Union[str, Any]=False						):
						_lowerCAmelCase    		=					get_version()
						if patch and default_version.is_devrelease:
												raise ValueError("""Can't create a patch version from the dev branch, checkout a released version!"""						)
						if default_version.is_devrelease:
												_lowerCAmelCase    		=					default_version.base_version
						elif patch:
												_lowerCAmelCase    		=					f'{default_version.major}.{default_version.minor}.{default_version.micro + 1}'
						else:
												_lowerCAmelCase    		=					f'{default_version.major}.{default_version.minor + 1}.0'

						# Now let's ask nicely if that's the right one.
						_lowerCAmelCase    		=					input(f'Which version are you releasing? [{default_version}]'						)
						if len(__lowerCamelCase						) == 0:
												_lowerCAmelCase    		=					default_version

						print(f'Updating version to {version}.'						)
						global_version_update(__lowerCamelCase					,      patch=__lowerCamelCase						)


def A ():
						_lowerCAmelCase    		=					get_version()
						_lowerCAmelCase    		=					f'{current_version.major}.{current_version.minor + 1}.0.dev0'
						_lowerCAmelCase    		=					current_version.base_version

						# Check with the user we got that right.
						_lowerCAmelCase    		=					input(f'Which version are we developing now? [{dev_version}]'						)
						if len(__lowerCamelCase						) == 0:
												_lowerCAmelCase    		=					dev_version

						print(f'Updating version to {version}.'						)
						global_version_update(__lowerCamelCase						)


#    print("Cleaning main README, don't forget to run `make fix-copies`.")
#    clean_main_ref_in_model_list()


if __name__ == "__main__":
						_lowercase          =						argparse.ArgumentParser()
						parser.add_argument("""--post_release""", action="""store_true""", help="""Whether this is pre or post release.""")
						parser.add_argument("""--patch""", action="""store_true""", help="""Whether or not this is a patch release.""")
						_lowercase          =						parser.parse_args()
						if not args.post_release:
												pre_release_work(patch=args.patch)
						elif args.patch:
												print("""Nothing to do after a patch :-)""")
						else:
												post_release_work()


'''simple docstring'''



from collections import UserDict
from typing import Union

import numpy as np
import requests

from ..utils import (
    add_end_docstrings,
    logging,
)
from .audio_classification import ffmpeg_read
from .base import PIPELINE_INIT_ARGS, Pipeline


_lowercase          =						logging.get_logger(__name__)




@add_end_docstrings(_SCREAMING_SNAKE_CASE					)
class     UpperCAmelCase_	(						_SCREAMING_SNAKE_CASE					):






						'''simple docstring'''






						def __init__(   self       ,  **_lowercase ):

												"""simple docstring"""



												super().__init__(**_lowercase )

												if self.framework != "pt":
																		raise ValueError(F'The {self.__class__} is only available in PyTorch.' )
												# No specific FOR_XXX available yet




						def __call__(   self       ,  _lowercase       ,  **_lowercase ):

												"""simple docstring"""



												return super().__call__(_lowercase       ,  **_lowercase )




						def    _lowercase      (   self       ,  **_lowercase ):

												"""simple docstring"""



												_lowerCAmelCase    		=					{}
												if "candidate_labels" in kwargs:
																		_lowerCAmelCase    		=					kwargs["""candidate_labels"""]
												if "hypothesis_template" in kwargs:
																		_lowerCAmelCase    		=					kwargs["""hypothesis_template"""]

												return preprocess_params, {}, {}




						def    _lowercase      (   self       ,  _lowercase       ,  _lowercase=None       ,  _lowercase="This is a sound of {}." ):

												"""simple docstring"""



												if isinstance(_lowercase       ,  _lowercase ):
																		if audio.startswith("""http://""" ) or audio.startswith("""https://""" ):
																								# We need to actually check for a real protocol, otherwise it's impossible to use a local file
																								# like http_huggingface_co.png
																								_lowerCAmelCase    		=					requests.get(_lowercase ).content
																		else:
																								with open(_lowercase       ,  """rb""" ) as f:
																														_lowerCAmelCase    		=					f.read()

												if isinstance(_lowercase       ,  _lowercase ):
																		_lowerCAmelCase    		=					ffmpeg_read(_lowercase       ,  self.feature_extractor.sampling_rate )

												if not isinstance(_lowercase       ,  np.ndarray ):
																		raise ValueError("""We expect a numpy ndarray as input""" )
												if len(audio.shape ) != 1:
																		raise ValueError("""We expect a single channel audio input for ZeroShotAudioClassificationPipeline""" )

												_lowerCAmelCase    		=					self.feature_extractor(
												    [audio]       ,  sampling_rate=self.feature_extractor.sampling_rate       ,  return_tensors="""pt""" )
												_lowerCAmelCase    		=					candidate_labels
												_lowerCAmelCase    		=					[hypothesis_template.format(_lowercase ) for x in candidate_labels]
												_lowerCAmelCase    		=					self.tokenizer(_lowercase       ,  return_tensors=self.framework       ,  padding=_lowercase )
												_lowerCAmelCase    		=					[text_inputs]
												return inputs




						def    _lowercase      (   self       ,  _lowercase ):

												"""simple docstring"""



												_lowerCAmelCase    		=					model_inputs.pop("""candidate_labels""" )
												_lowerCAmelCase    		=					model_inputs.pop("""text_inputs""" )
												if isinstance(text_inputs[0]       ,  _lowercase ):
																		_lowerCAmelCase    		=					text_inputs[0]
												else:
																		# Batching case.
																		_lowerCAmelCase    		=					text_inputs[0][0]

												_lowerCAmelCase    		=					self.model(**_lowercase       ,  **_lowercase )

												_lowerCAmelCase    		=					{
												    """candidate_labels""": candidate_labels,
												    """logits""": outputs.logits_per_audio,
												}
												return model_outputs




						def    _lowercase      (   self       ,  _lowercase ):

												"""simple docstring"""



												_lowerCAmelCase    		=					model_outputs.pop("""candidate_labels""" )
												_lowerCAmelCase    		=					model_outputs["""logits"""][0]

												if self.framework == "pt":
																		_lowerCAmelCase    		=					logits.softmax(dim=0 )
																		_lowerCAmelCase    		=					probs.tolist()
												else:
																		raise ValueError("""`tf` framework not supported.""" )

												_lowerCAmelCase    		=					[
												    {"""score""": score, """label""": candidate_label}
												    for score, candidate_label in sorted(zip(_lowercase       ,  _lowercase )       ,  key=lambda _lowercase : -x[0] )
												]
												return result
