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