# text_encoder.py

from src.sbv2.text.symbols import symbols

_symbol_to_id = {s: i for i, s in enumerate(symbols)}
_id_to_symbol = {i: s for i, s in enumerate(symbols)}

def text_to_sequence(text):
    sequence = [_symbol_to_id[s] for s in text if s in _symbol_to_id]
    return sequence

def sequence_to_text(sequence):
    return ''.join([_id_to_symbol[i] for i in sequence if i in _id_to_symbol])