Spaces:
Running
Running
File size: 1,349 Bytes
fe5a4b7 7ad0f22 fe5a4b7 |
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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
from ukrainian_word_stress import Stressifier, StressSymbol
from ukrainian_accentor_transformer import Accentor
accentors = ["model", "vocab", "none"]
def stress_replace_and_shift(stressed: str):
stressed = stressed.replace(
StressSymbol.CombiningAcuteAccent, "+"
)
new_stressed = ""
start = 0
last = 0
# shift stress symbol by one "при+віт" -> "пр+ивіт"
while True:
plus_position = stressed.find("+", start)
if plus_position != -1:
new_stressed += (
stressed[last : plus_position - 1] + "+" + stressed[plus_position - 1]
)
start = plus_position + 1
last = start
else:
new_stressed += stressed[last:]
break
return new_stressed
stressify = Stressifier(stress_symbol=StressSymbol.CombiningAcuteAccent)
accentor_transformer = Accentor()
def accentification(sentence: str, mode: str):
if (mode != "none"):
sentence = sentence.replace("+", "")
sentence = sentence.replace(
StressSymbol.CombiningAcuteAccent, ""
)
if (mode == "vocab"):
accented_sentence = stressify(sentence)
elif (mode == "model"):
accented_sentence = accentor_transformer(sentence)
else:
accented_sentence = sentence
return accented_sentence
|