| """ | |
| Expand dates and numbers into words with niqqud | |
| This happens before phonemization | |
| """ | |
| from .numbers import num_to_word | |
| from .dates import date_to_word | |
| from .time_to_word import time_to_word | |
| from .dictionary import Dictionary | |
| from mishkal.log import log | |
| class Expander: | |
| def __init__(self): | |
| self.dictionary = Dictionary() | |
| def expand_text(self, text: str): | |
| text = self.dictionary.expand_text(text) | |
| words = [] | |
| for source_word in text.split(): | |
| try: | |
| word = date_to_word(source_word) | |
| if word == source_word: | |
| word = time_to_word(word) | |
| if word == source_word: | |
| word = num_to_word(word) | |
| words.append(word) | |
| except Exception as e: | |
| log.error(f"Failed to expand {word} with error: {e}") | |
| words.append(source_word) | |
| return " ".join(words) | |