Update TextGen/router.py

TextGen/router.py

@@ -1,7 +1,7 @@
 import os
 import time
 from pydantic import BaseModel
-from fastapi import FastAPI, HTTPException, Query
+from fastapi import FastAPI, HTTPException, Query, Request
 from fastapi.responses import FileResponse
 from fastapi.middleware.cors import CORSMiddleware
 
@@ -67,15 +67,37 @@ def api_home():
 def inference(input_prompt: str):
     return generate_text(prompt=input_prompt)
 
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.responses import FileResponse
+import json
+
+app = FastAPI()
+
 @app.get("/generate_wav")
-async def generate_wav(text: str, language: str = "en"):
+async def generate_wav(request: Request):
     try:
+        body = await request.json()
+        text = body.get("text")
+        language = body.get("language", "en")
+        voice_choice = body.get("voice_choice", "./blacksmith.mp3")
+
+        valid_voices = [
+            "./blacksmith.mp3",
+            "./female.wav",
+            "./female.mp3",
+            "./narator_out.wav",
+            "./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
-            "./blacksmith.mp3",  # str (filepath on your computer (or URL) of file) in 'Reference Audio' Audio component
-            "./blacksmith.mp3",  # str (filepath on your computer (or URL) of file) in 'Use Microphone for Reference' Audio 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
             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
@@ -92,6 +114,7 @@ async def generate_wav(text: str, language: str = "en"):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+
 @app.get("/generate_song")
 async def generate_song(text: str):
     try: