Update TextGen/router.py

TextGen/router.py

@@ -69,30 +69,14 @@ def inference(input_prompt: str):
 
 
 @app.get("/generate_wav")
-async def generate_wav(request: Request):
+async def generate_wav(text: str, language: str = "en",voice:str="./voices/blacksmith.mp3):
     try:
-        body = await request.json()
-        text = body.get("text")
-        language = body.get("language", "en")
-        voice_choice = body.get("voice_choice", "./blacksmith.mp3")
-
-        valid_voices = [
-            "./voices/blacksmith.mp3",
-            "./voices/female.wav",
-            "./voices/female.mp3",
-            "./voices/narator_out.wav",
-            "./voices/blacksmith2.mp3"
-        ]
-
-        if voice_choice not in valid_voices:
-            raise HTTPException(status_code=400, detail="Invalid voice choice")
-
         # 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
-            voice_choice,  # str (filepath on your computer (or URL) of file) in 'Reference Audio' Audio component
-            voice_choice,  # str (filepath on your computer (or URL) of file) in 'Use Microphone for Reference' Audio 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
             False,  # bool in 'Cleanup Reference Voice' Checkbox component
             False,  # bool in 'Do not use language auto-detect' Checkbox component