Update TextGen/router.py

TextGen/router.py

@@ -20,8 +20,11 @@ from gradio_client import Client
 class Message(BaseModel):
  npc: str | None = None
  input: str | None = None
-
-
+ 
+class VoiceMessage(BaseModel):
+ npc: str | None = None
+ input: str | None = None
+ language: str | None = "en"
 song_base_api=os.environ["VERCEL_API"]
 
 my_hf_token=os.environ["HF_TOKEN"]
@@ -73,14 +76,18 @@ def api_home():
 def inference(message: Message):
  return generate_text(prompt=message.input)
 
-
+#Dummy function for now
+def determine_vocie_from_npc(npc):
+ return "./voices/blacksmith.mp3"
+ 
 @app.get("/generate_wav")
-async def generate_wav(text: str, language: str = "en",voice:str="./voices/blacksmith.mp3"):
+async def generate_wav(message:VoiceMessage):
  try:
+ voice=determine_vocie_from_npc(message.npc)
  # Use the Gradio client to generate the wav file
  result = tts_client.predict(
- text, # str in 'Text Prompt' Textbox component
- language, # str in 'Language' Dropdown component
+ message.input, # str in 'Text Prompt' Textbox component
+ message.language, # str in 'Language' Dropdown component
  voice, # str (filepath on your computer (or URL) of file) in 'Reference Audio' Audio component
  voice, # str (filepath on your computer (or URL) of file) in 'Use Microphone for Reference' Audio component
  False, # bool in 'Use Microphone' Checkbox component