Update TextGen/router.py

TextGen/router.py

@@ -1,5 +1,6 @@
 from pydantic import BaseModel
-
+from fastapi import FastAPI, HTTPException, Query
+from fastapi.responses import FileResponse
 from fastapi.middleware.cors import CORSMiddleware
 
 from langchain.chains import LLMChain
@@ -11,6 +12,12 @@ from langchain_google_genai import (
 )
 from TextGen import app
 
+my_hf_token=os.environ["HF_TOKEN"]
+
+tts_client = Client("https://jofthomas-xtts.hf.space/--replicas/sxv98/",hf_token=my_hf_token)
+
+
+
 class Generate(BaseModel):
     text:str
 
@@ -52,4 +59,29 @@ def api_home():
 
 @app.post("/api/generate", summary="Generate text from prompt", tags=["Generate"], response_model=Generate)
 def inference(input_prompt: str):
-    return generate_text(prompt=input_prompt)
+    return generate_text(prompt=input_prompt)
+
+@app.get("/generate_wav")
+async def generate_wav(text: str, language: str = "en"):
+    try:
+        # 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
+            "./narator_out.wav",  # str (filepath on your computer (or URL) of file) in 'Reference Audio' Audio component
+            "./narator_out.wav",  # 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
+            True,  # bool in 'Agree' Checkbox component
+            fn_index=1
+        )
+
+        # Get the path of the generated wav file
+        wav_file_path = result[1]
+
+        # Return the generated wav file as a response
+        return FileResponse(wav_file_path, media_type="audio/wav", filename="output.wav")
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))