Paillat-dev/viralfactory
1
0
Fork 0
mirror of https://github.com/Paillat-dev/viralfactory.git synced 2026-01-02 01:06:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix import, fix option indices in init, formatting

Browse Source
This commit is contained in:
Paillat
2024-02-15 17:52:19 +01:00
parent c57817df85
commit 67237a5ecf
1 changed files with 31 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/engines/TTSEngine/CoquiTTSEngine.py
View File

@@ -1,6 +1,6 @@
import gradio as gr
import TTS
from TTS.api import TTS
import os
import torch
@@ -95,9 +95,9 @@ class CoquiTTSEngine(BaseTTSEngine):
def __init__(self, options: list):
super().__init__()
self.voice = options[0][0]
self.language = options[1][0]
self.to_force_duration = options[2][0]
self.voice = options[0]
self.language = options[1]
self.to_force_duration = options[2]
self.duration = options[3]
os.environ["COQUI_TOS_AGREED"] = "1"
@@ -117,7 +117,9 @@ class CoquiTTSEngine(BaseTTSEngine):
Returns:
float: The time taken to synthesize the speech with whispering effect.
"""
self.tts.tts_to_file(text=text, file_path=path, lang=self.language, speaker=self.voice)
self.tts.tts_to_file(
text=text, file_path=path, lang=self.language, speaker=self.voice
)
if self.to_force_duration:
self.force_duration(float(self.duration), path)
return self.time_with_whisper(path)
@@ -139,10 +141,18 @@ class CoquiTTSEngine(BaseTTSEngine):
),
]
duration_checkbox = gr.Checkbox(label="Force duration", info="Force the duration of the generated audio to be at most the specified value", value=False)
duration = gr.Number(label="Duration [s]", value=57, step=1, minimum=10, visible=False)
duration_checkbox = gr.Checkbox(
label="Force duration",
info="Force the duration of the generated audio to be at most the specified value",
value=False,
)
duration = gr.Number(
label="Duration [s]", value=57, step=1, minimum=10, visible=False
)
duration_switch = lambda x: gr.update(visible=x)
duration_checkbox.change(duration_switch, inputs=[duration_checkbox], outputs=[duration])
duration_checkbox.change(
duration_switch, inputs=[duration_checkbox], outputs=[duration]
)
options.append(duration_checkbox)
options.append(duration)