Update app.py

app.py

@@ -18,6 +18,7 @@ from text.symbols import symbols
 from text import text_to_sequence
 import unicodedata
 from scipy.io.wavfile import write
+import openai
 
 def get_text(text, hps):
     text_norm = text_to_sequence(text, hps.data.text_cleaners)
@@ -88,20 +89,50 @@ def selection(speaker):
     elif speaker == "派蒙":
         spk = 16
         return spk
-    
-def sle(language,tts_input0):
+def friend_chat(text,key,call_name,indentity):
+  call_name = call_name
+  openai.api_key = key
+  identity = identity
+  start_sequence = '\n'+str(call_name)+':'
+  restart_sequence = "\nYou: "
+  all_text = identity + restart_sequence
+  if 1 == 1:
+     prompt0 = text #当期prompt
+  if text == 'quit':
+     return prompt0
+  prompt = identity + prompt0 + start_sequence
+
+  response = openai.Completion.create(
+    model="text-davinci-003",
+    prompt=prompt,
+    temperature=0.5,
+    max_tokens=1000,
+    top_p=1.0,
+    frequency_penalty=0.5,
+    presence_penalty=0.0,
+    stop=["\nYou:"]
+  )
+  return response['choices'][0]['text'].strip()
+def is_japanese(string):
+        for ch in string:
+            if ord(ch) > 0x3040 and ord(ch) < 0x30FF:
+                return True
+        return False    
+def sle(language,text,tts_input2,call_name,tts_input3):
     if language == "中文":
-        tts_input1 = "[ZH]" + tts_input0.replace('\n','。').replace(' ',',') + "[ZH]"
-        return tts_input1
-    if language == "英文":
-        tts_input1 = "[EN]" + tts_input0.replace('\n','.').replace(' ',',') + "[EN]"
+        tts_input1 = "[ZH]" + text.replace('\n','。').replace(' ',',') + "[ZH]"
         return tts_input1
+    if language == "对话":
+        text = friend_chat(text,tts_input2,call_name,tts_input3).replace('\n','。').replace(' ',',')
+        text = f"[JA]{text}[JA]" if is_japanese(text) else f"[ZH]{text}[ZH]"
+        return text
     elif language == "日文":
-        tts_input1 = "[JA]" + tts_input0.replace('\n','。').replace(' ',',') + "[JA]"
+        tts_input1 = "[JA]" + text.replace('\n','。').replace(' ',',') + "[JA]"
         return tts_input1
-def infer(language,text,speaker_id, n_scale= 0.667,n_scale_w = 0.8, l_scale = 1 ):
+def infer(language,text,tts_input2,tts_input3,speaker_id,n_scale= 0.667,n_scale_w = 0.8, l_scale = 1 ):
+    speaker_name = speaker_id
     speaker_id = int(selection(speaker_id))
-    stn_tst = get_text(sle(language,text), hps_ms)
+    stn_tst = get_text(sle(language,text,tts_input2,speaker_name,tts_input3), hps_ms)
     with torch.no_grad():
         x_tst = stn_tst.unsqueeze(0).to(dev)
         x_tst_lengths = torch.LongTensor([stn_tst.size(0)]).to(dev)
@@ -112,11 +143,10 @@ def infer(language,text,speaker_id, n_scale= 0.667,n_scale_w = 0.8, l_scale = 1
         spending_time = "推理时间："+str(t2-t1)+"s" 
         print(spending_time)
     return (hps_ms.data.sampling_rate, audio) 
-lan = ["中文","日文","英文(摆啦，有需要可以放17号位重新训练)"]
+lan = ["中文","日文","对话"]
 idols = ["高咲侑（误）","歩夢","かすみ","しずく","果林","愛","彼方","せつ菜","璃奈","栞子","エマ","ランジュ","ミア","三色绘恋1","三色绘恋2","派蒙"]
 
 
-
 dev = torch.device("cpu")
 hps_ms = utils.get_hparams_from_file("config.json")
 net_g_ms = SynthesizerTrn(
@@ -138,14 +168,16 @@ with app:
 
         with gr.TabItem("Basic"):
 
-            tts_input1 = gr.TextArea(label="牺牲质量的轻量化vits，如需用于创作和我说一声", value="一次審査、二次審査、それぞれの欄に記入をお願いします。")
-            language = gr.Dropdown(label="选择语言",choices=lan, value="日文", interactive=True)
+            tts_input1 = gr.TextArea(label="输入你的文本", value="一次審査、二次審査、それぞれの欄に記入をお願いします。")
+            tts_input2 = gr.TextArea(label="如需使用openai，输入你的openai-key", value="官网")
+            tts_input3 = gr.TextArea(label="写上你给她的设定", value="恶魔系学妹。")
+            language = gr.Dropdown(label="选择合成方式",choices=lan, value="日文", interactive=True)
             para_input1 = gr.Slider(minimum= 0.01,maximum=1.0,label="更改噪声比例", value=0.667)
             para_input2 = gr.Slider(minimum= 0.01,maximum=1.0,label="更改噪声偏差", value=0.8)
             para_input3 = gr.Slider(minimum= 0.1,maximum=10,label="更改时间比例", value=1)
             tts_submit = gr.Button("Generate", variant="primary")
             speaker1 = gr.Dropdown(label="选择说话人",choices=idols, value="かすみ", interactive=True)
             tts_output2 = gr.Audio(label="Output")
-            tts_submit.click(infer, [language,tts_input1,speaker1,para_input1,para_input2,para_input3], [tts_output2])
+            tts_submit.click(infer, [language,tts_input1,tts_input2,tts_input3,speaker1,para_input1,para_input2,para_input3], [tts_output2])
     #app.launch(share=True)
     app.launch()