Update TextGen/router.py

TextGen/router.py

@@ -25,6 +25,8 @@ class VoiceMessage(BaseModel):
     npc: str | None  = None
     input: str | None = None
     language: str | None = "en"
+    genre:str | None = "Male"
+    
 song_base_api=os.environ["VERCEL_API"]
 
 my_hf_token=os.environ["HF_TOKEN"]
@@ -32,7 +34,10 @@ my_hf_token=os.environ["HF_TOKEN"]
 tts_client = Client("https://jofthomas-xtts.hf.space/",hf_token=my_hf_token)
 
 
-
+main_npcs={
+    "Blacksmith":"./voices/blacksmith.mp3",
+    "Herbalist":"./voices/female.wav"
+}
 class Generate(BaseModel):
     text:str
 
@@ -77,13 +82,21 @@ def inference(message: Message):
     return generate_text(prompt=message.input)
 
 #Dummy function for now
-def determine_vocie_from_npc(npc):
-    return "./voices/blacksmith.mp3"
+def determine_vocie_from_npc(npc,genre):
+    if npc in main_npcs:
+        return main_npcs[npc]
+    else:
+        if genre =="Male":
+            "./voices/blacksmith.mp3"
+        if genre=="Female":
+            return"./voices/female.wav"
+        else:
+            return "./voices/narator_out.wav"
     
 @app.post("/generate_wav")
 async def generate_wav(message:VoiceMessage):
     try:
-        voice=determine_vocie_from_npc(message.npc)
+        voice=determine_vocie_from_npc(message.npc, message.genre)
         # Use the Gradio client to generate the wav file
         result = tts_client.predict(
             message.input,  # str in 'Text Prompt' Textbox component