Add application file

accentor.py

@@ -1,4 +1,4 @@
-from ukrainian_word_stress import StressSymbol
+from ukrainian_word_stress import StressSymbol, Stressifier
 from ukrainian_accentor_transformer import Accentor
 
 def stress_replace_and_shift(stressed: str):
@@ -22,15 +22,32 @@ def stress_replace_and_shift(stressed: str):
             break
     return new_stressed
 
+stressify = Stressifier(stress_symbol=StressSymbol.CombiningAcuteAccent)
 accentor_transformer = Accentor()
 
-def accentification(sentence: str):
-
-    sentence = sentence.replace("+", "")
+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
 
-    accented_sentence = accentor_transformer(sentence)
 
-    return accented_sentence
+
+# def accentification(sentence: str):
+#
+#     sentence = sentence.replace("+", "")
+#     sentence = sentence.replace(
+#         StressSymbol.CombiningAcuteAccent, ""
+#     )
+#
+#     accented_sentence = accentor_transformer(sentence)
+#
+#     return accented_sentence