File size: 794 Bytes
3f53a9f |
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 |
import num2words
from .number_names import NUMBER_NAMES
import re
def add_diacritics(words: str):
new_words = []
for word in words.split():
if NUMBER_NAMES.get(word):
new_words.append(NUMBER_NAMES[word])
elif NUMBER_NAMES.get(word[1:]):
# With Vav or Bet
new_words.append(NUMBER_NAMES[word[0]] + NUMBER_NAMES[word[1:]])
else:
new_words.append(word)
return " ".join(new_words)
def num_to_word(maybe_number: str) -> str:
def replace_number(match):
num = match.group()
words = num2words.num2words(num, lang="he", ordinal=False)
return add_diacritics(words)
# Replace all whole numbers in the string
result = re.sub(r"\d+", replace_number, maybe_number)
return result
|