From 5cbd2cad157f51ed27eb4cc3198dad77aa02c6b0 Mon Sep 17 00:00:00 2001
From: ikechan8370 <geyinchibuaa@gmail.com>
Date: Sun, 5 Mar 2023 15:44:59 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BC=98=E5=8C=96tts=E5=AF=B9=E9=85=8D?=
 =?UTF-8?q?=E7=BD=AE=E7=9A=84=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 utils/tts.js | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/utils/tts.js b/utils/tts.js
index 270a3fa..aaf3725 100644
--- a/utils/tts.js
+++ b/utils/tts.js
@@ -1,6 +1,6 @@
 import { Config } from './config.js'
 import fetch from 'node-fetch'
-import random from 'random'
+import _ from 'lodash'
 let proxy
 if (Config.proxy) {
   try {
@@ -24,7 +24,6 @@ const newFetch = (url, options = {}) => {
 
   return fetch(url, mergedOptions)
 }
-const space = Config.ttsSpace
 
 function randomNum (minNum, maxNum) {
   switch (arguments.length) {
@@ -48,6 +47,12 @@ export async function generateAudio (text, speaker = '随机', language = '中
       noiseScale, noiseScaleW, lengthScale
     ]
   }
+  let space = Config.ttsSpace
+  if (space.endsWith('api/generate')) {
+    let trimmedSpace = _.trimEnd(space, '/api/generate')
+    logger.warn(`vits api 当前为${space}，已校正为${trimmedSpace}`)
+    space = trimmedSpace
+  }
   logger.info(`正在使用接口${space}/api/generate`)
   let response = await newFetch(`${space}/api/generate`, {
     method: 'POST',