Update app.py

app.py

@@ -1,24 +1,65 @@
 import gradio as gr
 from infer import inference
+import unicodedata
+import regex
 
+# Основное описание
 description = '''
 Програма може не коректно визначати деякі наголоси і не перетворює цифри, акроніми і різні скорочення в словесну форму.
 Якщо наголос не правильний, використовуйте символ + після наголошеного складу.
 Також дуже маленькі речення можуть крешати, тому пишіть щось більше а не одне-два слова.
 '''
 
-def synthesise(text, speed, steps, progress=gr.Progress()):
+# Дополнительные функции для обработки текста
+def normalize_text(text):
+    return unicodedata.normalize('NFC', text)
+
+def add_pauses(text):
+    text = text.replace(':', ':::')
+    text = text.replace(',', ',:::')
+    text = text.replace(';', ';:::')
+    text = text.replace('—', '—:::')
+    text = text.replace('–', '–:::')
+    text = text.replace('.', '. ... ... ')
+    text = text.replace('!', '! ... ...')
+    text = text.replace('?', '? ... ...')
+    return text
+
+def convert_accented_text(text):
+    result = ""
+    for char in text:
+        decomposed = unicodedata.normalize('NFD', char)
+        if any('COMBINING ACUTE ACCENT' in unicodedata.name(c, '') for c in decomposed):
+            base_char = ''.join([c for c in decomposed if 'COMBINING ACUTE ACCENT' not in unicodedata.name(c, '')])
+            result += unicodedata.normalize('NFC', base_char) + "+"
+        else:
+            result += unicodedata.normalize('NFC', char)
+    return result
+
+def preprocess_text(text, add_pauses, convert_accent):
+    if convert_accent:
+        text = convert_accented_text(text)
+    if add_pauses:
+        text = add_pauses(text)
+    return text
+
+# Функция синтеза
+def synthesise(text, speed, steps, add_pauses, convert_accent, progress=gr.Progress()):
     if text.strip() == "":
         raise gr.Error("You must enter some text")
     if len(text) > 50000:
         raise gr.Error("Text must be <50k characters")
     
+    # Применяем обработку в зависимости от флагов чекбоксов
+    text = preprocess_text(text, add_pauses, convert_accent)
+    
     print("*** saying ***")
     print(text)
     print("*** end ***")
     
     return 24000, inference(text, progress, speed=speed, alpha=1.0, diffusion_steps=steps, embedding_scale=1.0)[0]
 
+# Запуск интерфейса
 if __name__ == "__main__":
     i = gr.Interface(
         fn=synthesise,
@@ -26,7 +67,9 @@ if __name__ == "__main__":
         inputs=[
             gr.Text(label='Text:', lines=5, max_lines=10),
             gr.Slider(label='Швидкість:', maximum=1.3, minimum=0.7, value=1.0),
-            gr.Slider(label='Кількість кроків дифузії:', minimum=3, maximum=20, step=1, value=3)
+            gr.Slider(label='Кількість кроків дифузії:', minimum=3, maximum=20, step=1, value=3),
+            gr.Checkbox(label="Add pauses", value=False),
+            gr.Checkbox(label="Convert accented text", value=False)
         ],
         outputs=[
             gr.Audio(
@@ -40,59 +83,13 @@ if __name__ == "__main__":
         cache_examples=False,
         title='StyleTTS2 Ukrainian Demo',
         examples=[
+            ["Решта окупантів звернула на Вокзальну — центральну вулицю Бучі...", 1.0, 6, False, False],
+            ["Одна дівчинка стала королевою Франції. Звали її Анна...", 1.0, 6, True, True],
             [
-                '''Мені+ трина+дцятий мина+ло;:::
-Я пас ягня+та за село+м. ... ... 
-Чи то так со+нечко сія+ло,:::
-Чи так мені+ чого+ було+ –:::
-Мені+ так лю+бо,::: лю+бо ста+ло,:::
-Нена+че в Бо+га. ... ... . ... ... . ... ... 
-Уже+ прокли+кали до па+ю,::: 
-А я собі+ у бур’яні+ 
-Молю+ся Бо+гу,::: і не зна+ю,:::
-Чого+ мале+нькому мені+
-Тоді+ так при+язно моли+лось,:::
-Чого+ так ве+село було+. ... ... 
-Госпо+днє не+бо і село+,:::
-Ягня+,::: здає+ться,::: весели+лось,:::
-І со+нце грі+ло –::: не пекло+. ... ... 
-
-Та не до+вго со+нце грі+ло,:::
-Не до+вго моли+лось;::: 
-Запекло+,::: почервоні+ло
-І рай запали+ло. ... ... 
-Мов проки+нувся,::: –::: дивлю+ся:::
-Село+ почорні+ло,:::
-Бо+же не+бо голубеє –:::
-І те помарні+ло. ... ... 
-Погля+нув я на ягня+та –:::
-Не мої+ ягня+та;:::
-Оберну+вся я на ха+ти –:::
-Нема+ в мене+ ха+ти. ... ... 
-Не дав мені+ Бог нічо+го! ... ...
-І хли+нули сльо+зи. ... ... . ... ... . ... ... 
-Тяжкі+ сльо+зи. ... ... . ... ... . ... ...  А ді+вчи+на,:::
-При самі+й доро+зі,:::
-Недале+ко ко+ло мене+
-Пло+скінь вибира+ла,:::
-Та й почу+ла,::: що я пла+чу:::
-Прийшла+,::: привіта+ла,:::
-Утира+ла мої+ сльо+зи,:::
-І поцілува+ла. ... ... . ... ... . ... ...  
-Нена+че со+нце засія+ло,:::
-Нена+че все на сві+ті ста+ло
-Моє+::: лани+,::: гаї+,::: сади+. ... ... . ... ... . ... ... 
-І ми,::: жарту+ючи,::: погна+ли
-Чужі+ ягня+та до води+. ... ... 
-
-Бридня+! ... .... ... ... . ... ...  А й до+сі,::: як згада+ю,:::
-То се+рце пла+че та боли+ть,:::
-Чому+ Госпо+дь не дав дожить
-Мало+го ві+ку у ті+м ра+ю. ... ... 
-Уме+р би,::: орючи+ на ни+ві,:::
-Нічо+го б на сві+ті не знав. ... ... 
-Не був би в сві+ті юроди+вим. ... ... 
-Люде+й і [Бо+га] не прокля+в! ... ...''', 1.0, 6
+                '''Одного разу, в невеликому місті, в якому все йшло за своїм розпорядком, 
+                з’явився мандрівник з далекої країни. Він розповів жителям про дивовижні землі, 
+                що знаходяться далеко за горизонтом, і про скарби, що чекають на тих, хто наважиться 
+                на небезпечну подорож...''', 1.0, 6, True, False
             ]
         ],
     )