Apply stress model for words with vowel_count > 1

app.py

@@ -60,7 +60,7 @@ def tts(text: str, stress: str):
         with torch.no_grad():
             wavs = synthesizer.tts(text)
             synthesizer.save_wav(wavs, fp)
-        return fp.name
+        return fp.name, text
 
 
 iface = gr.Interface(
@@ -75,7 +75,7 @@ iface = gr.Interface(
             choices=[option.value for option in StressOption],
         ),
     ],
-    outputs=gr.outputs.Audio(label="Output"),
+    outputs=[gr.outputs.Audio(label="Output"), gr.outputs.Textbox(label="Наголошений текст")],
     title="🐸💬🇺🇦 - Coqui TTS",
     theme="huggingface",
     description="Україномовний🇺🇦 TTS за допомогою Coqui TTS (для наголосу використовуйте + перед голосною)",

requirements.txt

@@ -1 +1 @@
-TTS==0.6.2
+TTS==0.7.1

stress.py

@@ -11,7 +11,10 @@ replace_accents = importer.load_pickle("uk-accentor", "replace_accents")
 # Back to CPU
 # accentor.cpu()
 
-alphabet = "абгґдеєжзиіїйклмнопрстуфхцчшщьюя"
+vowels = "аеєиіїоуюя"
+consonants = "бвгґджзйклмнпрстфхцчшщь"
+special = "'"
+alphabet = vowels + consonants + special
 
 def accent_word(word):
     with torch.no_grad():
@@ -44,7 +47,14 @@ def sentence_to_stress(sentence):
         if not all(first_word_sep) or len(element) == 0:
             continue
         else:
-            new_list[word_index] = accent_word(new_list[word_index])
+            vowels_in_words = list(map(lambda letter: letter in vowels, new_list[word_index]))
+            if vowels_in_words.count(True) == 0:
+                continue
+            elif vowels_in_words.count(True) == 1:
+                vowel_index = vowels_in_words.index(True)
+                new_list[word_index] = new_list[word_index][0:vowel_index] + "+" + new_list[word_index][vowel_index::]
+            else:
+                new_list[word_index] = accent_word(new_list[word_index])
 
     return "".join(new_list)
 
@@ -52,10 +62,12 @@ def sentence_to_stress(sentence):
 if __name__ == "__main__":
     sentence = "Кам'янець-Подільський - місто в Хмельницькій області України, центр Кам'янець-Подільської міської об'єднаної територіальної громади і Кам'янець-Подільського району."
     print(sentence_to_stress(sentence))
+    sentence = "Привіт, як тебе звати?"
+    print(sentence_to_stress(sentence))
     #test_words1 = ["словотворення", "архаїчний", "програма", "а-ля-фуршет"]
 
     stressed_words = accentor.predict(["привіт"], mode='stress')
     plused_words = [replace_accents(x) for x in stressed_words]
 
     print('With stress:', stressed_words)
-    print('With pluses:', plused_words)
+    print('With pluses:', plused_words)