



import json
from typing import List, Optional, Tuple

from tokenizers import normalizers

from ...tokenization_utils_fast import PreTrainedTokenizerFast
from ...utils import logging
from .tokenization_squeezebert import SqueezeBertTokenizer


lowerCAmelCase__					  =  logging.get_logger(__name__)

lowerCAmelCase__					  =  {'''vocab_file''': '''vocab.txt''', '''tokenizer_file''': '''tokenizer.json'''}

lowerCAmelCase__					  =  {
    '''vocab_file''': {
        '''squeezebert/squeezebert-uncased''': (
            '''https://huggingface.co/squeezebert/squeezebert-uncased/resolve/main/vocab.txt'''
        ),
        '''squeezebert/squeezebert-mnli''': '''https://huggingface.co/squeezebert/squeezebert-mnli/resolve/main/vocab.txt''',
        '''squeezebert/squeezebert-mnli-headless''': (
            '''https://huggingface.co/squeezebert/squeezebert-mnli-headless/resolve/main/vocab.txt'''
        ),
    },
    '''tokenizer_file''': {
        '''squeezebert/squeezebert-uncased''': (
            '''https://huggingface.co/squeezebert/squeezebert-uncased/resolve/main/tokenizer.json'''
        ),
        '''squeezebert/squeezebert-mnli''': (
            '''https://huggingface.co/squeezebert/squeezebert-mnli/resolve/main/tokenizer.json'''
        ),
        '''squeezebert/squeezebert-mnli-headless''': (
            '''https://huggingface.co/squeezebert/squeezebert-mnli-headless/resolve/main/tokenizer.json'''
        ),
    },
}

lowerCAmelCase__					  =  {
    '''squeezebert/squeezebert-uncased''': 5_1_2,
    '''squeezebert/squeezebert-mnli''': 5_1_2,
    '''squeezebert/squeezebert-mnli-headless''': 5_1_2,
}


lowerCAmelCase__					  =  {
    '''squeezebert/squeezebert-uncased''': {'''do_lower_case''': True},
    '''squeezebert/squeezebert-mnli''': {'''do_lower_case''': True},
    '''squeezebert/squeezebert-mnli-headless''': {'''do_lower_case''': True},
}

class 				snake_case__(_UpperCamelCase      ):

					"""simple docstring"""






					lowercase_								=		VOCAB_FILES_NAMES
					lowercase_								=		PRETRAINED_VOCAB_FILES_MAP
					lowercase_								=		PRETRAINED_INIT_CONFIGURATION
					lowercase_								=		PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
					lowercase_								=		SqueezeBertTokenizer







					def __init__(							self :     List[str]				,       SCREAMING_SNAKE_CASE :     List[str]=None				,       SCREAMING_SNAKE_CASE :     List[Any]=None				,       SCREAMING_SNAKE_CASE :     int=True				,       SCREAMING_SNAKE_CASE :     Union[str, Any]="[UNK]"				,       SCREAMING_SNAKE_CASE :     List[Any]="[SEP]"				,       SCREAMING_SNAKE_CASE :     Any="[PAD]"				,       SCREAMING_SNAKE_CASE :     List[Any]="[CLS]"				,       SCREAMING_SNAKE_CASE :     Optional[int]="[MASK]"				,       SCREAMING_SNAKE_CASE :     int=True				,       SCREAMING_SNAKE_CASE :     List[str]=None				,       **SCREAMING_SNAKE_CASE :     List[Any]				,       ):
									super().__init__(
									    SCREAMING_SNAKE_CASE				,       tokenizer_file=SCREAMING_SNAKE_CASE				,       do_lower_case=SCREAMING_SNAKE_CASE				,       unk_token=SCREAMING_SNAKE_CASE				,       sep_token=SCREAMING_SNAKE_CASE				,       pad_token=SCREAMING_SNAKE_CASE				,       cls_token=SCREAMING_SNAKE_CASE				,       mask_token=SCREAMING_SNAKE_CASE				,       tokenize_chinese_chars=SCREAMING_SNAKE_CASE				,       strip_accents=SCREAMING_SNAKE_CASE				,       **SCREAMING_SNAKE_CASE				,       )

									lowercase__					:   Tuple													=					json.loads(self.backend_tokenizer.normalizer.__getstate__()					)
									if (
									    normalizer_state.get("lowercase"				,       SCREAMING_SNAKE_CASE					) != do_lower_case
									    or normalizer_state.get("strip_accents"				,       SCREAMING_SNAKE_CASE					) != strip_accents
									    or normalizer_state.get("handle_chinese_chars"				,       SCREAMING_SNAKE_CASE					) != tokenize_chinese_chars
									):
													lowercase__					:   Tuple													=					getattr(SCREAMING_SNAKE_CASE				,       normalizer_state.pop("type"					)					)
													lowercase__					:   Optional[int]													=					do_lower_case
													lowercase__					:   Union[str, Any]													=					strip_accents
													lowercase__					:   Union[str, Any]													=					tokenize_chinese_chars
													lowercase__					:   Optional[Any]													=					normalizer_class(**SCREAMING_SNAKE_CASE					)

									lowercase__					:   List[Any]													=					do_lower_case







					def 			snake_case					(							self :     Optional[int]				,       SCREAMING_SNAKE_CASE :     List[Any]				,       SCREAMING_SNAKE_CASE :     Any=None					):
									lowercase__					:   Union[str, Any]													=					[self.cls_token_id] + token_ids_a + [self.sep_token_id]

									if token_ids_a:
													output += token_ids_a + [self.sep_token_id]

									return output







					def 			snake_case					(							self :     Optional[int]				,       SCREAMING_SNAKE_CASE :     List[int]				,       SCREAMING_SNAKE_CASE :     Optional[List[int]] = None					):
									lowercase__					:   Dict													=					[self.sep_token_id]
									lowercase__					:   int													=					[self.cls_token_id]
									if token_ids_a is None:
													return len(cls + token_ids_a + sep					) * [0]
									return len(cls + token_ids_a + sep					) * [0] + len(token_ids_a + sep					) * [1]







					def 			snake_case					(							self :     Optional[int]				,       SCREAMING_SNAKE_CASE :     str				,       SCREAMING_SNAKE_CASE :     Optional[str] = None					):
									lowercase__					:   str													=					self._tokenizer.model.save(SCREAMING_SNAKE_CASE				,       name=SCREAMING_SNAKE_CASE					)
									return tuple(SCREAMING_SNAKE_CASE					)









import os
from shutil import copyfile
from typing import Any, Dict, List, Optional, Tuple

import sentencepiece as spm

from ...tokenization_utils import AddedToken, PreTrainedTokenizer
from ...utils import logging


lowerCAmelCase__					  =  logging.get_logger(__name__)

lowerCAmelCase__					  =  '''▁'''

lowerCAmelCase__					  =  {'''vocab_file''': '''sentencepiece.bpe.model''', '''monolingual_vocab_file''': '''dict.txt'''}

lowerCAmelCase__					  =  {
    '''vocab_file''': {
        '''vinai/bartpho-syllable''': '''https://huggingface.co/vinai/bartpho-syllable/resolve/main/sentencepiece.bpe.model''',
    },
    '''monolingual_vocab_file''': {
        '''vinai/bartpho-syllable''': '''https://huggingface.co/vinai/bartpho-syllable/resolve/main/dict.txt''',
    },
}

lowerCAmelCase__					  =  {'''vinai/bartpho-syllable''': 1_0_2_4}

class 				snake_case__(_UpperCamelCase      ):

					"""simple docstring"""






					lowercase_								=		VOCAB_FILES_NAMES
					lowercase_								=		PRETRAINED_VOCAB_FILES_MAP
					lowercase_								=		PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
					lowercase_								=		["""input_ids""", """attention_mask"""]







					def __init__(							self :     Optional[int]				,       SCREAMING_SNAKE_CASE :     List[str]				,       SCREAMING_SNAKE_CASE :     Optional[Any]				,       SCREAMING_SNAKE_CASE :     Optional[int]="<s>"				,       SCREAMING_SNAKE_CASE :     Optional[int]="</s>"				,       SCREAMING_SNAKE_CASE :     str="</s>"				,       SCREAMING_SNAKE_CASE :     List[str]="<s>"				,       SCREAMING_SNAKE_CASE :     Union[str, Any]="<unk>"				,       SCREAMING_SNAKE_CASE :     Tuple="<pad>"				,       SCREAMING_SNAKE_CASE :     List[str]="<mask>"				,       SCREAMING_SNAKE_CASE :     Optional[Dict[str, Any]] = None				,       **SCREAMING_SNAKE_CASE :     int				,       ):
									# Mask token behave like a normal word, i.e. include the space before it
									lowercase__					:   Dict													=					AddedToken(SCREAMING_SNAKE_CASE				,       lstrip=SCREAMING_SNAKE_CASE				,       rstrip=SCREAMING_SNAKE_CASE					) if isinstance(SCREAMING_SNAKE_CASE				,       SCREAMING_SNAKE_CASE					) else mask_token

									lowercase__					:   int													=					{} if sp_model_kwargs is None else sp_model_kwargs

									super().__init__(
									    bos_token=SCREAMING_SNAKE_CASE				,       eos_token=SCREAMING_SNAKE_CASE				,       unk_token=SCREAMING_SNAKE_CASE				,       sep_token=SCREAMING_SNAKE_CASE				,       cls_token=SCREAMING_SNAKE_CASE				,       pad_token=SCREAMING_SNAKE_CASE				,       mask_token=SCREAMING_SNAKE_CASE				,       sp_model_kwargs=self.sp_model_kwargs				,       **SCREAMING_SNAKE_CASE				,       )

									lowercase__					:   Dict													=					vocab_file
									lowercase__					:   Union[str, Any]													=					monolingual_vocab_file
									lowercase__					:   Union[str, Any]													=					spm.SentencePieceProcessor(**self.sp_model_kwargs					)
									self.sp_model.Load(str(SCREAMING_SNAKE_CASE					)					)

									# Load the reduced vocab

									# Keep order of special tokens for backward compatibility
									lowercase__					:   Any													=					{}
									lowercase__					:   int													=					0
									for token in [bos_token, pad_token, eos_token, unk_token, sep_token, cls_token]:
													if str(SCREAMING_SNAKE_CASE					) not in self.fairseq_tokens_to_ids:
																	lowercase__					:   Dict													=					cnt
																	cnt += 1
									with open(SCREAMING_SNAKE_CASE				,       "r"				,       encoding="utf-8"					) as f:
													for line in f.readlines():
																	lowercase__					:   int													=					line.strip().split()[0]
																	lowercase__					:   List[str]													=					len(self.fairseq_tokens_to_ids					)
									if str(SCREAMING_SNAKE_CASE					) not in self.fairseq_tokens_to_ids:
													lowercase__					:   Optional[Any]													=					len(self.fairseq_tokens_to_ids					)

									lowercase__					:   List[str]													=					{v: k for k, v in self.fairseq_tokens_to_ids.items()}







					def __getstate__(							self :     int					):
									lowercase__					:   Dict													=					self.__dict__.copy()
									lowercase__					:   Union[str, Any]													=					None
									lowercase__					:   Dict													=					self.sp_model.serialized_model_proto()
									return state







					def __setstate__(							self :     Union[str, Any]				,       SCREAMING_SNAKE_CASE :     Optional[Any]					):
									lowercase__					:   Union[str, Any]													=					d

									# for backward compatibility
									if not hasattr(self				,       "sp_model_kwargs"					):
													lowercase__					:   Dict													=					{}

									lowercase__					:   List[str]													=					spm.SentencePieceProcessor(**self.sp_model_kwargs					)
									self.sp_model.LoadFromSerializedProto(self.sp_model_proto					)







					def 			snake_case					(							self :     Optional[int]				,       SCREAMING_SNAKE_CASE :     List[int]				,       SCREAMING_SNAKE_CASE :     Optional[List[int]] = None					):

									if token_ids_a is None:
													return [self.cls_token_id] + token_ids_a + [self.sep_token_id]
									lowercase__					:   List[Any]													=					[self.cls_token_id]
									lowercase__					:   str													=					[self.sep_token_id]
									return cls + token_ids_a + sep + sep + token_ids_a + sep







					def 			snake_case					(							self :     Dict				,       SCREAMING_SNAKE_CASE :     List[int]				,       SCREAMING_SNAKE_CASE :     Optional[List[int]] = None				,       SCREAMING_SNAKE_CASE :     bool = False					):

									if already_has_special_tokens:
													return super().get_special_tokens_mask(
													    token_ids_a=SCREAMING_SNAKE_CASE				,       token_ids_a=SCREAMING_SNAKE_CASE				,       already_has_special_tokens=SCREAMING_SNAKE_CASE					)

									if token_ids_a is None:
													return [1] + ([0] * len(SCREAMING_SNAKE_CASE					)) + [1]
									return [1] + ([0] * len(SCREAMING_SNAKE_CASE					)) + [1, 1] + ([0] * len(SCREAMING_SNAKE_CASE					)) + [1]







					def 			snake_case					(							self :     str				,       SCREAMING_SNAKE_CASE :     List[int]				,       SCREAMING_SNAKE_CASE :     Optional[List[int]] = None					):
									lowercase__					:   Tuple													=					[self.sep_token_id]
									lowercase__					:   str													=					[self.cls_token_id]

									if token_ids_a is None:
													return len(cls + token_ids_a + sep					) * [0]
									return len(cls + token_ids_a + sep + sep + token_ids_a + sep					) * [0]







					@property
					def 			snake_case					(							self :     Optional[int]					):
									return len(self.fairseq_ids_to_tokens					)







					def 			snake_case					(							self :     List[Any]					):
									lowercase__					:   Any													=					{self.convert_ids_to_tokens(SCREAMING_SNAKE_CASE					): i for i in range(self.vocab_size					)}
									vocab.update(self.added_tokens_encoder					)
									return vocab







					def 			snake_case					(							self :     Union[str, Any]				,       SCREAMING_SNAKE_CASE :     str					):
									return self.sp_model.encode(SCREAMING_SNAKE_CASE				,       out_type=SCREAMING_SNAKE_CASE					)







					def 			snake_case					(							self :     List[str]				,       SCREAMING_SNAKE_CASE :     int					):
									if token in self.fairseq_tokens_to_ids:
													return self.fairseq_tokens_to_ids[token]
									else:
													return self.unk_token_id







					def 			snake_case					(							self :     List[str]				,       SCREAMING_SNAKE_CASE :     Any					):
									return self.fairseq_ids_to_tokens[index]







					def 			snake_case					(							self :     List[Any]				,       SCREAMING_SNAKE_CASE :     Union[str, Any]					):
									lowercase__					:   str													=					"".join(SCREAMING_SNAKE_CASE					).replace(SCREAMING_SNAKE_CASE				,       " "					).strip()
									return out_string







					def 			snake_case					(							self :     Optional[int]				,       SCREAMING_SNAKE_CASE :     str				,       SCREAMING_SNAKE_CASE :     Optional[str] = None					):
									if not os.path.isdir(SCREAMING_SNAKE_CASE					):
													logger.error(f"""Vocabulary path ({save_directory}) should be a directory"""					)
													return
									lowercase__					:   str													=					os.path.join(
									    SCREAMING_SNAKE_CASE				,       (filename_prefix + "-" if filename_prefix else "") + VOCAB_FILES_NAMES["vocab_file"]					)
									lowercase__					:   List[str]													=					os.path.join(
									    SCREAMING_SNAKE_CASE				,       (filename_prefix + "-" if filename_prefix else "") + VOCAB_FILES_NAMES["monolingual_vocab_file"]				,       )

									if os.path.abspath(self.vocab_file					) != os.path.abspath(SCREAMING_SNAKE_CASE					) and os.path.isfile(self.vocab_file					):
													copyfile(self.vocab_file				,       SCREAMING_SNAKE_CASE					)
									elif not os.path.isfile(self.vocab_file					):
													with open(SCREAMING_SNAKE_CASE				,       "wb"					) as fi:
																	lowercase__					:   str													=					self.sp_model.serialized_model_proto()
																	fi.write(SCREAMING_SNAKE_CASE					)

									if os.path.abspath(self.monolingual_vocab_file					) != os.path.abspath(
									    SCREAMING_SNAKE_CASE					) and os.path.isfile(self.monolingual_vocab_file					):
													copyfile(self.monolingual_vocab_file				,       SCREAMING_SNAKE_CASE					)
									elif not os.path.isfile(self.monolingual_vocab_file					):
													with open(SCREAMING_SNAKE_CASE				,       "w"				,       encoding="utf-8"					) as fp:
																	for token in self.fairseq_tokens_to_ids:
																					if token not in self.all_special_tokens:
																									fp.write(f"""{str(SCREAMING_SNAKE_CASE					)} \n"""					)

									return out_vocab_file, out_monolingual_vocab_file




