添加gpt

app.py

@@ -24,7 +24,18 @@ from text import cleaned_text_to_sequence, get_bert
 from text.cleaner import clean_text
 import gradio as gr
 import webbrowser
+import openai
 
+# openai.log = "debug"
+openai.api_base = "https://api.chatanywhere.com.cn/v1"
+
+
+# 非流式响应
+
+def gpt_35_api(gptkey, message):
+    openai.api_key = "sk-" + gptkey
+    completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[{"role": "user", "content": message}])
+    return completion.choices[0].message.content
 
 net_g = None
 
@@ -51,8 +62,10 @@ def get_text(text, language_str, hps):
 
     return bert, phone, tone, language
 
-def infer(text, sdp_ratio, noise_scale, noise_scale_w, length_scale, sid):
+def infer(text, key, sdp_ratio, noise_scale, noise_scale_w, length_scale, sid):
     global net_g
+    message = gpt_35_api(key, text)
+    print(message)
     bert, phones, tones, lang_ids = get_text(text, "ZH", hps)
     with torch.no_grad():
         x_tst=phones.to(device).unsqueeze(0)
@@ -67,9 +80,9 @@ def infer(text, sdp_ratio, noise_scale, noise_scale_w, length_scale, sid):
         del x_tst, tones, lang_ids, bert, x_tst_lengths, speakers
         return audio
 
-def tts_fn(text, speaker, sdp_ratio, noise_scale, noise_scale_w, length_scale):
+def tts_fn(text, key, speaker, sdp_ratio, noise_scale, noise_scale_w, length_scale):
     with torch.no_grad():
-        audio = infer(text, sdp_ratio=sdp_ratio, noise_scale=noise_scale, noise_scale_w=noise_scale_w, length_scale=length_scale, sid=speaker)
+        audio = infer(text, key,sdp_ratio=sdp_ratio, noise_scale=noise_scale, noise_scale_w=noise_scale_w, length_scale=length_scale, sid=speaker)
     return "Success", (hps.data.sampling_rate, audio)
 
 
@@ -113,10 +126,14 @@ if __name__ == "__main__":
         with gr.Row():
             with gr.Column():
                 gr.Markdown(value="""
-                【AI贝拉】在线语音合成（Bert-Vits2）\n
+                【AI贝拉】在线语音对话版（Bert-Vits2 + gpt）\n
                 作者：Xz乔希 https://space.bilibili.com/5859321\n
+                集成作者：碎语碎念 https://space.bilibili.com/4269384\n
                 声音归属：贝拉kira https://space.bilibili.com/672353429\n
                 Bert-VITS2项目：https://github.com/Stardust-minus/Bert-VITS2\n
+                GPT_API_free项目：https://github.com/chatanywhere/GPT_API_free\n
+                本项目中的apiKey可以免费从https://github.com/chatanywhere/GPT_API_free\n
+                获取，参考项目文档即可！\n
                 使用本模型请严格遵守法律法规！\n
                 发布二创作品请标注本项目作者及链接、作品使用Bert-VITS2 AI生成！\n                
                 """)
@@ -131,26 +148,11 @@ if __name__ == "__main__":
             with gr.Column():
                 text_output = gr.Textbox(label="Message")
                 audio_output = gr.Audio(label="Output Audio")
-                gr.Markdown(value="""
-                【AI向晚】https://huggingface.co/spaces/XzJosh/Ava-Bert-VITS2\n
-                【AI嘉然】https://huggingface.co/spaces/XzJosh/Diana-Bert-VITS2\n
-                【AI乃琳】https://huggingface.co/spaces/XzJosh/Eileen-Bert-VITS2\n
-                【AI珈乐】https://huggingface.co/spaces/XzJosh/Carol-Bert-VITS2\n
-                【AI塔菲】https://huggingface.co/spaces/XzJosh/Taffy-Bert-VITS2\n
-                【AI东雪莲】https://huggingface.co/spaces/XzJosh/Azuma-Bert-VITS2\n
-                【AI奶绿】https://huggingface.co/spaces/XzJosh/LAPLACE-Bert-VITS2\n
-                【AI尼奈】https://huggingface.co/spaces/XzJosh/nine1-Bert-VITS2\n
-                【AI电棍】https://huggingface.co/spaces/XzJosh/otto-Bert-VITS2\n
-                【AI七海】https://huggingface.co/spaces/XzJosh/Nana7mi-Bert-VITS2\n
-                【AI阿梓】https://huggingface.co/spaces/XzJosh/Azusa-Bert-VITS2\n
-                【AI星瞳】https://huggingface.co/spaces/XzJosh/XingTong-Bert-VITS2\n
-                【AI剑魔】https://huggingface.co/spaces/XzJosh/Aatrox-Bert-VITS2\n
-                 """)
         btn.click(tts_fn,
                 inputs=[text, speaker, sdp_ratio, noise_scale, noise_scale_w, length_scale],
                 outputs=[text_output, audio_output])
-    
+
 #    webbrowser.open("http://127.0.0.1:6006")
 #    app.launch(server_port=6006, show_error=True)
-        
+
     app.launch(show_error=True)

requirements.txt

@@ -15,3 +15,4 @@ pypinyin
 cn2an
 gradio
 av
+openai==0.28