modify api implementation

webui.py

@@ -15,10 +15,6 @@ 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",)
@@ -27,8 +23,6 @@ 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:
@@ -38,14 +32,14 @@ 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(...)):
+def tts_api(voice, text):
     try:
         output_path = infer(voice, text)
-        return FileResponse(output_path, media_type="audio/wav", filename=output_path.split("/")[-1])
+        with open(output_path, "rb") as f:
+            audio_bytes = f.read()
+        return (200, {}, audio_bytes)
     except Exception as e:
-        return JSONResponse(status_code=500, content={"error": str(e)})
-
-app.post("/api/tts")(tts_api)
+        return (500, {"error": str(e)}, None)
 
 def gen_single(prompt, text):
     output_path = infer(prompt, text)
@@ -55,7 +49,6 @@ def update_prompt_audio():
     update_button = gr.update(interactive=True)
     return update_button
 
-
 with gr.Blocks() as demo:
     mutex = threading.Lock()
     gr.HTML('''
@@ -85,11 +78,25 @@ with gr.Blocks() as demo:
                      inputs=[prompt_audio, input_text_single],
                      outputs=[output_audio])
 
-gr.mount_gradio_app(app, demo, path="/")
+    # 新增API接口，供外部HTTP调用
+    api_iface = gr.Interface(
+        fn=lambda voice, text: infer(voice, text),
+        inputs=[gr.Audio(label="voice", type="filepath"), gr.Textbox(label="text")],
+        outputs=gr.Audio(label="result", type="filepath"),
+        allow_flagging="never",
+        description="API接口: 传入参考音频和文本，返回合成音频。"
+    )
+
+    # 将API接口隐藏在UI中，只暴露HTTP接口
+    with demo:
+        gr.TabItem("API接口", visible=False)
+        api_iface.render()
+
+    # 移除 add_api_route 和 mount_gradio_app，Spaces 不支持
 
 def main():
     tts.load_normalizer()
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    demo.launch(server_name="0.0.0.0", server_port=7860)
 
 if __name__ == "__main__":
     main()