add api

webui.py

@@ -15,6 +15,10 @@ import gradio as gr
 from indextts.infer import IndexTTS
 from tools.i18n.i18n import I18nAuto
 
+from fastapi import FastAPI, UploadFile, Form
+from fastapi.responses import FileResponse, JSONResponse
+import uvicorn
+
 i18n = I18nAuto(language="zh_CN")
 MODE = 'local'
 snapshot_download("IndexTeam/IndexTTS-1.5",local_dir="checkpoints",)
@@ -23,6 +27,8 @@ tts = IndexTTS(model_dir="checkpoints", cfg_path="checkpoints/config.yaml")
 os.makedirs("outputs/tasks",exist_ok=True)
 os.makedirs("prompts",exist_ok=True)
 
+app = FastAPI()
+
 @spaces.GPU
 def infer(voice, text,output_path=None):
     if not tts:
@@ -32,6 +38,15 @@ def infer(voice, text,output_path=None):
     tts.infer(voice, text, output_path)
     return output_path
 
+def tts_api(voice: str = Form(...), text: str = Form(...)):
+    try:
+        output_path = infer(voice, text)
+        return FileResponse(output_path, media_type="audio/wav", filename=output_path.split("/")[-1])
+    except Exception as e:
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
+app.post("/api/tts")(tts_api)
+
 def gen_single(prompt, text):
     output_path = infer(prompt, text)
     return gr.update(value=output_path,visible=True)
@@ -70,11 +85,11 @@ with gr.Blocks() as demo:
                      inputs=[prompt_audio, input_text_single],
                      outputs=[output_audio])
 
+gr.mount_gradio_app(app, demo, path="/")
 
 def main():
     tts.load_normalizer()
-    demo.queue(20)
-    demo.launch(server_name="0.0.0.0")
+    uvicorn.run(app, host="0.0.0.0", port=8000)
 
 if __name__ == "__main__":
     main()