feat: upload

GPT_SoVITS/inference_webui.py

@@ -811,7 +811,6 @@ def get_tts_wav(
     sample_steps=8,
     if_sr=False,
     pause_second=0.3,
-    uploadParams=None
 ):
     global cache
     if ref_wav_path:
@@ -1056,12 +1055,63 @@ def get_tts_wav(
             audio_opt /= max_audio
     else:
         audio_opt = audio_opt.cpu().detach().numpy()
-    audioData = (audio_opt * 32767).astype(np.int16)
-    uploadAudio(opt_sr, audioData, uploadParams)
-    yield opt_sr, audioData
+    yield opt_sr, (audio_opt * 32767).astype(np.int16)
+
+
+def run(
+    ref_wav_path,
+    prompt_text,
+    prompt_language,
+    text,
+    text_language,
+    how_to_cut=i18n("不切"),
+    top_k=20,
+    top_p=0.6,
+    temperature=0.6,
+    ref_free=False,
+    speed=1,
+    if_freeze=False,
+    inp_refs=None,
+    sample_steps=8,
+    if_sr=False,
+    pause_second=0.3,
+    uploadParams=None,
+    errCallbackUrl=None
+):
+    try:
+        result = get_tts_wav(
+            ref_wav_path,
+            prompt_text,
+            prompt_language,
+            text,
+            text_language,
+            how_to_cut,
+            top_k,
+            top_p,
+            temperature,
+            ref_free,
+            speed,
+            if_freeze,
+            inp_refs,
+            sample_steps,
+            if_sr,
+            pause_second,
+        )
+        opt_sr, audio_data = next(result)
+        uploadAudio(opt_sr, audio_data, uploadParams)
+        return result
+    except Exception as e:
+        errCallback(errCallbackUrl, e)
+        yield 0, None
+
+
+def errCallback(errCallbackUrl, e):
+    requests.post(errCallbackUrl, data={"error": str(e)})
 
 
 def uploadAudio(opt_sr, audio_int16, uploadParams):
+    if not uploadParams:
+        return
     bio = io.BytesIO()
     write(bio, opt_sr, audio_int16)
     files = {
@@ -1237,8 +1287,10 @@ with gr.Blocks(title="GPT-SoVITS WebUI", analytics_enabled=False, js=js, css=css
             with gr.Column():
                 inp_ref = gr.Audio(label=i18n(
                     "请上传3~10秒内参考音频，超过会报错！"), type="filepath", scale=13)
-                uploadParams = gr.Textbox(label=i18n("结果上传参数"),
-                                          value="", lines=2, max_lines=2)
+                uploadParams = gr.Textbox(label=i18n("成功结果上传参数"),
+                                          value="", lines=1, max_lines=1)
+                errCallbackUrl = gr.Textbox(label=i18n("失败回调地址"),
+                                            value="", lines=1, max_lines=1)
             with gr.Column(scale=13):
                 ref_text_free = gr.Checkbox(
                     label=i18n("开启无参考文本模式。不填参考文本亦相当于开启。")
@@ -1388,7 +1440,8 @@ with gr.Blocks(title="GPT-SoVITS WebUI", analytics_enabled=False, js=js, css=css
                 sample_steps,
                 if_sr_Checkbox,
                 pause_second_slider,
-                uploadParams
+                uploadParams,
+                errCallbackUrl
             ],
             [output],
         )