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")