Merge eed2095a42927ea0a9c6bd548eaf4926b6c99f8f into 9da7e17efe05041e31d3c3f42c8730ae890397f2

api_v2.py

@@ -112,9 +112,12 @@ import wave
 import signal
 import numpy as np
 import soundfile as sf
+import shutil
 from fastapi import FastAPI, Request, HTTPException, Response
 from fastapi.responses import StreamingResponse, JSONResponse
 from fastapi import FastAPI, UploadFile, File
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFiles
 import uvicorn
 from io import BytesIO
 from tools.i18n.i18n import I18nAuto
@@ -141,6 +144,7 @@ if config_path in [None, ""]:
     config_path = "GPT-SoVITS/configs/tts_infer.yaml"
 
 tts_config = TTS_Config(config_path)
+print("以下为TTS_CONFIG配置, 如需修改请查看/GPT_SoVITS/configs/tts_infer.yaml")
 print(tts_config)
 tts_pipeline = TTS(tts_config)
 
@@ -459,7 +463,84 @@ async def set_sovits_weights(weights_path: str = None):
         return JSONResponse(status_code=400, content={"message": f"change sovits weight failed", "Exception": str(e)})
     return JSONResponse(status_code=200, content={"message": "success"})
 
+APP.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # 允许所有域名的请求
+    allow_credentials=True,
+    allow_methods=["*"],  # 允许所有方法
+    allow_headers=["*"],  # 允许所有请求头
+)
 
+@APP.get("/info")
+async def get_info():
+    try:
+        gpt_weights_dir_v2 = 'GPT_weights_v2'
+        sovits_weights_dir_v2 = 'SoVITS_weights_v2'
+        gpt_weights_dir = 'GPT_weights'
+        sovits_weights_dir = 'SoVITS_weights'
+        
+        gpt_filenames = []
+        sovits_filenames = []
+        
+        for dir in [gpt_weights_dir_v2, gpt_weights_dir]:
+            if os.path.exists(dir):
+                gpt_filenames.extend([f"{dir}/{f}" for f in os.listdir(dir) if os.path.isfile(os.path.join(dir, f))])
+        
+        for dir in [sovits_weights_dir_v2, sovits_weights_dir]:
+            if os.path.exists(dir):
+                sovits_filenames.extend([f"{dir}/{f}" for f in os.listdir(dir) if os.path.isfile(os.path.join(dir, f))])
+        
+        if not gpt_filenames:
+            return JSONResponse(status_code=404, content={"message": "No GPT weights files found"})
+        if not sovits_filenames:
+            return JSONResponse(status_code=404, content={"message": "No SoVITS weights files found"})
+        
+        return JSONResponse(status_code=200, content={
+            "gpt_weights_files": gpt_filenames,
+            "sovits_weights_files": sovits_filenames,
+            "server_port": port
+        })
+    except Exception as e:
+        return JSONResponse(status_code=500, content={"message": f"Error retrieving weights info", "error": str(e)})
+
+@APP.post("/tts")
+async def tts_post_endpoint(request: TTS_Request):
+    req = request.model_dump()
+    print("\nProcessed request (req):")
+    print(f"Type: {type(req)}")
+    print("Content:")
+    for key, value in req.items():
+        print(f"  {key}: {value}")
+    
+    return await tts_handle(req)
+
+@APP.post("/upload_file")
+async def upload_file(file: UploadFile = File(...)):
+    try:
+        # Create a temporary directory if it doesn't exist
+        temp_dir = "temp_files"
+        os.makedirs(temp_dir, exist_ok=True)
+
+        # Define the path to save the uploaded file
+        file_path = os.path.join(temp_dir, file.filename)
+
+        # Save the uploaded file to the temporary directory
+        with open(file_path, "wb") as buffer:
+            shutil.copyfileobj(file.file, buffer)
+
+        return JSONResponse(status_code=200, content={"message": "File uploaded successfully", "file_path": file_path})
+    except Exception as e:
+        return JSONResponse(status_code=500, content={"message": "File upload failed", "error": str(e)})
+
+APP.mount("/", StaticFiles(directory="dist", html=True), name="static")
+print("--------------------------------")
+print(f"前端界面已在 http://{host}:{port} 开启。")
+print("目前的前端版本只适配默认端口9880, 更改api端口会导致前端页面无法工作, 但不影响后端api运行。")
+print("在前端界面中上传的音频文件将会保存在 ./temp_files 目录下，如有需要请手动删除。")
+print("请至少运行一遍webui.py, 放好模型, 再运行本API, 以确保存放模型的文件夹SoVITS_weights和GPT_weights存在。")
+print("如遇配置错误，请检查命令行上方输出的配置详情，并修改文件/GPT_SoVITS/configs/tts_infer.yaml")
+print("如果运行环境是mac, 请将tts_infer.yaml内custom条目下的device改为cpu, is_half改为false")
+print("--------------------------------")
 
 if __name__ == "__main__":
     try:

dist/index.html

@@ -0,0 +1,13 @@
+<!doctype html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>Vite Project</title>
+    <script type="module" crossorigin src="/assets/index-BXQvAA72.js"></script>
+    <link rel="stylesheet" crossorigin href="/assets/index-Dl43Gj3X.css">
+  </head>
+  <body>
+    <div id="app"></div>
+  </body>
+</html>