Fix to keep prompted emotion for every split line

This commit is contained in:
mrq 2023-02-05 06:55:09 +00:00
parent 79e0b85602
commit c7f85dbba2

10
app.py
View File

@ -16,11 +16,6 @@ def generate(text, delimiter, emotion, prompt, voice, mic_audio, preset, seed, c
else: else:
voices = [] voices = []
if emotion == "Custom" and prompt.strip() != "":
text = f"[{prompt},] {text}"
elif emotion != "None":
text = f"[I am really {emotion.lower()},] {text}"
if voice == "microphone": if voice == "microphone":
if mic_audio is None: if mic_audio is None:
raise gr.Error("Please provide audio from mic when choosing `microphone` as a voice input") raise gr.Error("Please provide audio from mic when choosing `microphone` as a voice input")
@ -76,6 +71,11 @@ def generate(text, delimiter, emotion, prompt, voice, mic_audio, preset, seed, c
audio_cache = {} audio_cache = {}
for line, cut_text in enumerate(texts): for line, cut_text in enumerate(texts):
if emotion == "Custom" and prompt.strip() != "":
cut_text = f"[{prompt},] {cut_text}"
elif emotion != "None":
cut_text = f"[I am really {emotion.lower()},] {cut_text}"
print(f"[{str(line+1)}/{str(len(texts))}] Generating line: {cut_text}") print(f"[{str(line+1)}/{str(len(texts))}] Generating line: {cut_text}")
gen, additionals = tts.tts(cut_text, **settings ) gen, additionals = tts.tts(cut_text, **settings )