Update app.py

app.py

@@ -87,99 +87,54 @@ with open('dict.txt', 'r', encoding='utf-8') as f:
             base_word = remove_combining_chars(line_normalized.replace('+', '').lower())
             custom_dict[base_word] = line_normalized
 
-def synthesise(text, speed, steps, 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")
-
-    # Pre-process the text
+def transform_text(text, apply_custom_dict, add_pauses):
     text = normalize_text(text)
-    text = replace_with_custom_dict(text, custom_dict)
+    if apply_custom_dict:
+        text = replace_with_custom_dict(text, custom_dict)
     text = convert_accented_text(text)
-    # Uncomment the next line if you want to add pauses
-    # text = add_pauses(text)
+    if add_pauses:
+        text = add_pauses(text)
+    return text
 
+def synthesise(transformed_text, speed, steps, progress=gr.Progress()):
+    if transformed_text.strip() == "":
+        raise gr.Error("Ви повинні ввести текст")
+    if len(transformed_text) > 50000:
+        raise gr.Error("Текст повинен бути менше 50 000 символів")
+    
     print("*** saying ***")
-    print(text)
+    print(transformed_text)
     print("*** end ***")
-
-    return 24000, inference(text, progress, speed=speed, alpha=1.0, diffusion_steps=steps, embedding_scale=1.0)[0]
+    
+    return 24000, inference(transformed_text, progress, speed=speed, alpha=1.0, diffusion_steps=steps, embedding_scale=1.0)[0]
 
 if __name__ == "__main__":
-    i = gr.Interface(
-        fn=synthesise,
-        description=description,
-        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)
-        ],
-        outputs=[
-            gr.Audio(
-                label="Audio:",
-                autoplay=False,
-                streaming=False,
-                type="numpy",
-            ),
-        ],
-        allow_flagging='never',
-        cache_examples=False,
-        title='StyleTTS2 Ukrainian Demo',
-        examples=[
-            [
-                '''Мені+ трина+дцятий мина+ло;:::
-Я пас ягня+та за село+м. ... ... 
-Чи то так со+нечко сія+ло,:::
-Чи так мені+ чого+ було+ –:::
-Мені+ так лю+бо,::: лю+бо ста+ло,:::
-Нена+че в Бо+га. ... ... . ... ... . ... ... 
-Уже+ прокли+кали до па+ю,::: 
-А я собі+ у бур’яні+ 
-Молю+ся Бо+гу,::: і не зна+ю,:::
-Чого+ мале+нькому мені+
-Тоді+ так при+язно моли+лось,:::
-Чого+ так ве+село було+. ... ... 
-Госпо+днє не+бо і село+,:::
-Ягня+,::: здає+ться,::: весели+лось,:::
-І со+нце грі+ло –::: не пекло+. ... ... 
-Та не до+вго со+нце грі+ло,:::
-Не до+вго моли+лось;::: 
-Запекло+,::: почервоні+ло
-І рай запали+ло. ... ... 
-Мов проки+нувся,::: –::: дивлю+ся:::
-Село+ почорні+ло,:::
-Бо+же не+бо голубеє –:::
-І те помарні+ло. ... ... 
-Погля+нув я на ягня+та –:::
-Не мої+ ягня+та;:::
-Оберну+вся я на ха+ти –:::
-Нема+ в мене+ ха+ти. ... ... 
-Не дав мені+ Бог нічо+го! ... ...
-І хли+нули сльо+зи. ... ... . ... ... . ... ... 
-Тяжкі+ сльо+зи. ... ... . ... ... . ... ...  А ді+вчи+на,:::
-При самі+й доро+зі,:::
-Недале+ко ко+ло мене+
-Пло+скінь вибира+ла,:::
-Та й почу+ла,::: що я пла+чу:::
-Прийшла+,::: привіта+ла,:::
-Утира+ла мої+ сльо+зи,:::
-І поцілува+ла. ... ... . ... ... . ... ...  
-Нена+че со+нце засія+ло,:::
-Нена+че все на сві+ті ста+ло
-Моє+::: лани+,::: гаї+,::: сади+. ... ... . ... ... . ... ... 
-І ми,::: жарту+ючи,::: погна+ли
-Чужі+ ягня+та до води+. ... ... 
-Бридня+! ... .... ... ... . ... ...  А й до+сі,::: як згада+ю,:::
-То се+рце пла+че та боли+ть,:::
-Чому+ Госпо+дь не дав дожить
-Мало+го ві+ку у ті+м ра+ю. ... ... 
-Уме+р би,::: орючи+ на ни+ві,:::
-Нічо+го б на сві+ті не знав. ... ... 
-Не був би в сві+ті юроди+вим. ... ... 
-Люде+й і [Бо+га] не прокля+в! ... ...''', 1.0, 6
-            ]
-        ],
-    )
-    i.queue(max_size=20, default_concurrency_limit=4)
-    i.launch(share=False, server_name="0.0.0.0")
+    with gr.Blocks() as demo:
+        gr.Markdown(description)
+        with gr.Row():
+            text_input = gr.Textbox(label='Text:', lines=5, max_lines=10)
+            transformed_text_output = gr.Textbox(label='Transformed Text:', lines=5, max_lines=10, interactive=False)
+        with gr.Row():
+            apply_custom_dict_checkbox = gr.Checkbox(label='Замінити слова за словником', value=True)
+            add_pauses_checkbox = gr.Checkbox(label='Додати паузи', value=False)
+        with gr.Row():
+            speed_slider = gr.Slider(label='Швидкість:', maximum=1.3, minimum=0.7, value=1.0)
+            steps_slider = gr.Slider(label='Кількість кроків дифузії:', minimum=3, maximum=20, step=1, value=3)
+        generate_button = gr.Button('Згенерувати аудіо')
+        audio_output = gr.Audio(label="Audio:", autoplay=False, streaming=False, type="numpy")
+        
+        def update_transformed_text(text, apply_custom_dict, add_pauses):
+            transformed_text = transform_text(text, apply_custom_dict, add_pauses)
+            return transformed_text
+        
+        # Set up live update of transformed text
+        text_input.change(fn=update_transformed_text, inputs=[text_input, apply_custom_dict_checkbox, add_pauses_checkbox], outputs=transformed_text_output)
+        apply_custom_dict_checkbox.change(fn=update_transformed_text, inputs=[text_input, apply_custom_dict_checkbox, add_pauses_checkbox], outputs=transformed_text_output)
+        add_pauses_checkbox.change(fn=update_transformed_text, inputs=[text_input, apply_custom_dict_checkbox, add_pauses_checkbox], outputs=transformed_text_output)
+        
+        def generate_audio(transformed_text, speed, steps):
+            return synthesise(transformed_text, speed, steps)
+        
+        generate_button.click(fn=generate_audio, inputs=[transformed_text_output, speed_slider, steps_slider], outputs=audio_output)
+        
+    demo.launch(share=False, server_name="0.0.0.0")