fix random_voice_number for suno

gabrielchua · Oct 1, 2024 · 2cf7afe · 2cf7afe
1 parent 3b75e3f
commit 2cf7afe
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/app.py b/app.py
@@ -65,8 +65,8 @@ def generate_podcast(
 
     text = ""
 
-    # Choose random number from 0 to 9
-    random_voice_number = random.randint(0, 9) # this is for suno model
+    # Choose random number from 0 to 8
+    random_voice_number = random.randint(0, 8) # this is for suno model
 
     if not use_advanced_audio and language in NOT_SUPPORTED_IN_MELO_TTS:
         raise gr.Error(ERROR_MESSAGE_NOT_SUPPORTED_IN_MELO_TTS)

diff --git a/utils.py b/utils.py
@@ -148,9 +148,11 @@ def generate_podcast_audio(
 
 def _use_suno_model(text: str, speaker: str, language: str, random_voice_number: int) -> str:
     """Generate advanced audio using Bark."""
+    host_voice_num = str(random_voice_number)
+    guest_voice_num = str(random_voice_number + 1)
     audio_array = generate_audio(
         text,
-        history_prompt=f"v2/{language}_speaker_{random_voice_number if speaker == 'Host (Jane)' else random_voice_number + 1}",
+        history_prompt=f"v2/{language}_speaker_{host_voice_num if speaker == 'Host (Jane)' else guest_voice_num}",
     )
     file_path = f"audio_{language}_{speaker}.mp3"
     write_wav(file_path, SAMPLE_RATE, audio_array)