添加gpt问答接口

app.py

@@ -3,47 +3,6 @@ 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)
@@ -65,6 +24,19 @@ 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
 
@@ -93,7 +65,8 @@ def get_text(text, language_str, hps):
 
 def infer(text, key, sdp_ratio, noise_scale, noise_scale_w, length_scale, sid):
     global net_g
-    message = gpt_35_api_stream(key, text)
+    message = gpt_35_api(key, text)
+    print(message)
     bert, phones, tones, lang_ids = get_text(message, "ZH", hps)
     with torch.no_grad():
         x_tst=phones.to(device).unsqueeze(0)
@@ -166,7 +139,7 @@ if __name__ == "__main__":
                 """)
                 text = gr.TextArea(label="Text", placeholder="Input Text Here",
                                       value="虚拟主播是什么？")
-                text = gr.TextArea(label="Key", placeholder="请输入上面提示中获取的gpt key",
+                key = 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混合比')

requirements.txt

@@ -0,0 +1,18 @@
+librosa==0.9.1
+matplotlib
+numpy
+numba
+phonemizer
+scipy
+tensorboard
+torch
+torchvision
+Unidecode
+amfm_decompy
+jieba
+transformers
+pypinyin
+cn2an
+gradio
+av
+openai==0.28