Update TextGen/router.py

TextGen/router.py

@@ -20,7 +20,7 @@ from TextGen import app
 from gradio_client import Client, handle_file
 from typing import List
 from elevenlabs.client import ElevenLabs
-from elevenlabs import stream
+from elevenlabs import Voice, VoiceSettings, stream
 
 
 Eleven_client = ElevenLabs(
@@ -74,9 +74,9 @@ main_npcs={
     "Bard":"./voices/Bard_voice.mp3"
 }
 main_npcs_elevenlabs={
-    "Blacksmith":"",
-    "Herbalist":"",
-    "Bard":""
+    "Blacksmith":"yYdk7n49vTsUKiXxnosS",
+    "Herbalist":"143zSsxc4O5ifS97lPCa",
+    "Bard":"143zSsxc4O5ifS97lPCa"
 }
 main_npc_system_prompts={
     "Blacksmith":"You are a blacksmith in a video game",
@@ -193,11 +193,11 @@ def determine_elevenLav_voice_from_npc(npc,genre):
         return main_npcs[npc]
     else:
         if genre =="Male":
-            "./voices/default_male.mp3"
+            "bIHbv24MWmeRgasZH58o"
         if genre=="Female":
-            return"./voices/default_female.mp3"
+            return"pFZP5JQG7iQjIQuC4Bku"
         else:
-            return "./voices/narator_out.wav"    
+            return "TX3LPaxmHKxFdv7VOQHJ"    
 
 @app.post("/generate_wav")
 async def generate_wav(message: VoiceMessage):
@@ -240,8 +240,15 @@ def generate_voice_eleven(message: VoiceMessage = None):
         Last_message = message
 
     def audio_stream():
+        this_voice_id=determine_elevenLav_voice_from_npc(message.npc, message.genre)
+        
         # Generate the audio stream from ElevenLabs
-        for chunk in Eleven_client.generate(text=message.input, stream=True):
+        for chunk in Eleven_client.generate(text=message.input,
+                                            voice=Voice(
+                                                voice_id=this_voice_id,
+                                                settings=VoiceSettings(stability=0.71, similarity_boost=0.5, style=0.0, use_speaker_boost=True)
+                                            ),
+                                            stream=True):
             yield chunk
 
     return StreamingResponse(audio_stream(), media_type="audio/mpeg")