Automatic stress by default, override manually with +

app.py

@@ -12,13 +12,12 @@ import torch
 
 
 class StressOption(Enum):
-    ManualStress = "Наголоси вручну"
-    AutomaticStress = "Автоматичні наголоси (Beta)"
+    AutomaticStress = "Автоматичні наголоси"
 
 
 class VoiceOption(Enum):
-    MaleVoice = "Микита (чоловічий)"
     FemaleVoice = "Олена (жіночий)"
+    MaleVoice = "Микита (чоловічий)"
 
 
 def download(url, file_name):
@@ -105,7 +104,7 @@ iface = gr.Interface(
     ],
     title="🐸💬🇺🇦 - Coqui TTS",
     theme="huggingface",
-    description="Україномовний🇺🇦 TTS за допомогою Coqui TTS (для наголосу використовуйте + перед голосною)",
+    description="Україномовний🇺🇦 TTS за допомогою Coqui TTS (щоб вручну поставити наголос, використовуйте + перед голосною)",
     article="Якщо вам подобається, підтримайте за посиланням: [SUPPORT LINK](https://send.monobank.ua/jar/48iHq4xAXm),  "
     + "Github: [https://github.com/robinhad/ukrainian-tts](https://github.com/robinhad/ukrainian-tts)   \n"
     + "Model training - [Yurii Paniv @robinhad](https://github.com/robinhad)   \n"
@@ -114,19 +113,19 @@ iface = gr.Interface(
     + f'<center><img src="{badge}" alt="visitors badge"/></center>',
     examples=[
         [
-            "Введ+іть, б+удь л+аска, сво+є р+ечення.",
+            "Введіть, будь ласка, своє речення.",
             VoiceOption.FemaleVoice.value,
-            StressOption.ManualStress.value,
+            StressOption.AutomaticStress.value,
         ],
         [
-            "Введ+іть, б+удь л+аска, сво+є р+ечення.",
+            "Введіть, будь ласка, своє речення.",
             VoiceOption.MaleVoice.value,
-            StressOption.ManualStress.value,
+            StressOption.AutomaticStress.value,
         ],
         [
-            "Введіть, будь ласка, своє речення.",
+            "Вв+едіть, будь ласка, св+оє реч+ення.",
             VoiceOption.MaleVoice.value,
-            StressOption.ManualStress.value,
+            StressOption.AutomaticStress.value,
         ],
         [
             "Привіт, як тебе звати?",

stress.py

@@ -1,13 +1,23 @@
+import numpy as np
 from ukrainian_word_stress import Stressifier, StressSymbol
 
 stressify = Stressifier(stress_symbol=StressSymbol.CombiningAcuteAccent)
 
 
-def sentence_to_stress(sentence):
-    stressed = stressify(sentence).replace(StressSymbol.CombiningAcuteAccent, "+")
+def sentence_to_stress(sentence: str) -> str:
+    # save custom stress positions
+    all_stresses = []
+    orig_words = sentence.split(" ")
+    for i in range(0, len(orig_words)):
+        if "+" in orig_words[i]:
+            all_stresses.append(i)
+
+    # add stress before vowel
+    stressed = stressify(sentence.replace("+", "")).replace(StressSymbol.CombiningAcuteAccent, "+")
     new_stressed = ""
     start = 0
     last = 0
+
     while True:
         plus_position = stressed.find("+", start)
         if plus_position != -1:
@@ -19,6 +29,13 @@ def sentence_to_stress(sentence):
         else:
             new_stressed += stressed[last:]
             break
+    
+    # replace already stressed words
+    if len(all_stresses) > 0:
+        words = new_stressed.split(" ")
+        for stressed in all_stresses:
+            words[stressed] = orig_words[stressed]
+        return " ".join(words)
     return new_stressed
 
 
@@ -31,6 +48,10 @@ if __name__ == "__main__":
     print(sentence_to_stress(sentence))
     sentence = "Не тільки в Україні таке може бути."
     print(sentence_to_stress(sentence))
+    sentence = "Не тільки в +Укра+їні т+аке може бути."
+    print(sentence_to_stress(sentence))
+    sentence = "два + два"
+    print(sentence_to_stress(sentence))
     sentence = "Н тльк в крн тк мж бт."
     print(sentence_to_stress(sentence))
     sentence = "Н тльк в крн тк мж бт."