添加gpt问答接口

app.py

@@ -1,9 +1,49 @@
 import sys, os
-
 if sys.platform == "darwin":
     os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 
 import logging
+import openai
+
+# openai.log = "debug"
+openai.api_key = "sk-"
+openai.api_base = "https://api.chatanywhere.com.cn/v1"
+
+
+
+# 非流式响应
+# completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[{"role": "user", "content": "Hello world!"}])
+# print(completion.choices[0].message.content)
+
+def gpt_35_api_stream(key, messages: list):
+    openai.api_key = "sk-" + key
+    """为提供的对话消息创建新的回答 (流式传输)
+
+    Args:
+        messages (list): 完整的对话消息
+        api_key (str): OpenAI API 密钥
+
+    Returns:
+        tuple: (results, error_desc)
+    """
+    try:
+        response = openai.ChatCompletion.create(
+            model='gpt-3.5-turbo',
+            messages=messages,
+            stream=True,
+        )
+        completion = {'role': '', 'content': ''}
+        for event in response:
+            if event['choices'][0]['finish_reason'] == 'stop':
+                print(f'收到的完成数据: {completion}')
+                break
+            for delta_k, delta_v in event['choices'][0]['delta'].items():
+                print(f'流响应数据: {delta_k} = {delta_v}')
+                completion[delta_k] += delta_v
+        messages.append(completion)  # 直接在传入参数 messages 中追加消息
+        return True, ''
+    except Exception as err:
+        return False, f'OpenAI API 异常: {err}'
 
 logging.getLogger("numba").setLevel(logging.WARNING)
 logging.getLogger("markdown_it").setLevel(logging.WARNING)
@@ -51,9 +91,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
-    bert, phones, tones, lang_ids = get_text(text, "ZH", hps)
+    message = gpt_35_api_stream(text)
+    bert, phones, tones, lang_ids = get_text(message, "ZH", hps)
     with torch.no_grad():
         x_tst=phones.to(device).unsqueeze(0)
         tones=tones.to(device).unsqueeze(0)
@@ -67,9 +108,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)
 
 
@@ -114,42 +155,30 @@ if __name__ == "__main__":
             with gr.Column():
                 gr.Markdown(value="""
                 【AI嘉然①】在线语音合成（Bert-Vits2）\n
-                作者：Xz乔希 https://space.bilibili.com/5859321\n
+                音声作者：Xz乔希 https://space.bilibili.com/5859321\n
+                集成作者：碎语碎念 https://space.bilibili.com/4269384\n
                 声音归属：嘉然今天吃什么 https://space.bilibili.com/672328094\n
                 Bert-VITS2项目：https://github.com/Stardust-minus/Bert-VITS2\n
-                【AI嘉然②】https://huggingface.co/spaces/XzJosh/Jiaran-Bert-VITS2\n
-                【AI嘉然③】https://huggingface.co/spaces/XzJosh/ranran-Bert-VITS2\n
+                GPT_API_free项目：https://github.com/chatanywhere/GPT_API_free\n
+                本项目中的apiKey可以免费从https://github.com/chatanywhere/GPT_API_free获取，参考项目文档即可！\n
                 使用本模型请严格遵守法律法规！\n
                 发布二创作品请标注本项目作者及链接、作品使用Bert-VITS2 AI生成！\n                
                 """)
                 text = gr.TextArea(label="Text", placeholder="Input Text Here",
                                       value="大家好我是嘉然戴安娜，关注嘉然，顿顿解馋，谢谢！")
+                text = gr.TextArea(label="Key", placeholder="请输入上面提示中获取的gpt key",
+                                   value="key")
                 speaker = gr.Dropdown(choices=speakers, value=speakers[0], label='Speaker')
                 sdp_ratio = gr.Slider(minimum=0.1, maximum=1, value=0.2, step=0.01, label='SDP/DP混合比')
                 noise_scale = gr.Slider(minimum=0.1, maximum=1, value=0.5, step=0.01, label='感情调节')
                 noise_scale_w = gr.Slider(minimum=0.1, maximum=1, value=0.9, step=0.01, label='音素长度')
                 length_scale = gr.Slider(minimum=0.1, maximum=2, value=1, step=0.01, label='生成长度')
                 btn = gr.Button("点击生成", variant="primary")
-            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/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/Carol-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/Ava-Bert-VITS2\n
-                 """)
         btn.click(tts_fn,
-                inputs=[text, speaker, sdp_ratio, noise_scale, noise_scale_w, length_scale],
+                inputs=[text, key, 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)