File size: 958 Bytes
1866014 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
"""
Expand dates and numbers into words with nikud
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 phonikud.log import log
class Expander:
def __init__(self):
self.dictionary = Dictionary()
def expand_text(self, text: str):
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)
text = " ".join(words)
text = self.dictionary.expand_text(text)
return text
|