devops/encoach_backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix repeated voices in listening

Browse Source
This commit is contained in:
Cristiano Ferreira
2024-03-26 23:35:26 +00:00
parent 08f05ac3e0
commit f1d2ec3bf8
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
helper/exercises.py
View File

@@ -278,6 +278,7 @@ def generate_listening_1_conversation(topic: str):
GEN_QUESTION_TEMPERATURE GEN_QUESTION_TEMPERATURE
) )
chosen_voices = []
name_to_voice = {} name_to_voice = {}
for segment in processed['conversation']: for segment in processed['conversation']:
if 'voice' not in segment: if 'voice' not in segment:
@@ -285,10 +286,17 @@ def generate_listening_1_conversation(topic: str):
if name in name_to_voice: if name in name_to_voice:
voice = name_to_voice[name] voice = name_to_voice[name]
else: else:
if segment['gender'].lower() == 'male': voice = None
voice = random.choice(MALE_NEURAL_VOICES)['Id'] while voice is None:
else: if segment['gender'].lower() == 'male':
voice = random.choice(FEMALE_NEURAL_VOICES)['Id'] available_voices = MALE_NEURAL_VOICES
else:
available_voices = FEMALE_NEURAL_VOICES
chosen_voice = random.choice(available_voices)['Id']
if chosen_voice not in chosen_voices:
voice = chosen_voice
chosen_voices.append(voice)
name_to_voice[name] = voice name_to_voice[name] = voice
segment['voice'] = voice segment['voice'] = voice
return response, processed return response, processed