Update app.py

app.py

@@ -33,6 +33,36 @@ def is_english(string):
         else:
             return False
 
+def to_html(chat_history):
+    chat_html = ""
+    for item in chat_history:
+        if item['role'] == 'user':
+            chat_html += f"""
+                <div style="margin-bottom: 20px;">
+                    <div style="text-align: right; margin-right: 20px;">
+                        <span style="background-color: #4CAF50; color: black; padding: 10px; border-radius: 10px; display: inline-block; max-width: 80%; word-wrap: break-word;">
+                            {item['content']}
+                        </span>
+                    </div>
+                </div>
+            """
+        else:
+            chat_html += f"""
+                <div style="margin-bottom: 20px;">
+                    <div style="text-align: left; margin-left: 20px;">
+                        <span style="background-color: white; color: black; padding: 10px; border-radius: 10px; display: inline-block; max-width: 80%; word-wrap: break-word;">
+                            {item['content']}
+                        </span>
+                    </div>
+                </div>
+            """
+    output_html = f"""
+        <div style="height: 400px; overflow-y: scroll; padding: 10px;">
+            {chat_html}
+        </div>
+    """
+    return output_html
+
 def extrac(text):
     text = re.sub("<[^>]*>","",text)
     result_list = re.split(r'\n', text)
@@ -76,7 +106,7 @@ def chatgpt(text):
                 del messages[-2:]
             with open('log.pickle', 'wb') as f:
                 pickle.dump(messages, f)
-            return reply
+            return reply,messages
     except:
         messages.append({"role": "user", "content": text},)
         chat = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
@@ -88,7 +118,7 @@ def chatgpt(text):
             del messages[-2:]
         with open('log.pickle', 'wb') as f:
             pickle.dump(messages, f)
-        return reply
+        return reply,messages
 
 def get_symbols_from_json(path):
     assert os.path.isfile(path)
@@ -122,12 +152,12 @@ def get_text(text,hps_ms):
 
 def create_tts_fn(net_g,hps,speaker_id):
     speaker_id = int(speaker_id)
-    def tts_fn(history,is_gpt,api_key,is_audio,audiopath,repeat_time,text, language, extract, n_scale= 0.667,n_scale_w = 0.8, l_scale = 1 ):
-        repeat_time = int(repeat_time)
+    def tts_fn(is_gpt,api_key,is_audio,audiopath,repeat_time,text, language, extract, n_scale= 0.667,n_scale_w = 0.8, l_scale = 1 ):
+        repeat_ime = int(repeat_time)
         if is_gpt:
             openai.api_key = api_key
-            text = chatgpt(text)
-        history = history + [(text, None)]
+            text,messages = chatgpt(text)
+            htm = to_html(messages)
         if not extract:
             print(text)
             t1 = time.time()
@@ -149,7 +179,7 @@ def create_tts_fn(net_g,hps,speaker_id):
                         os.system(cmd)
             except:
                 pass
-            return history,file_path,(hps.data.sampling_rate,audio)
+            return (hps.data.sampling_rate, audio),file_path,htm
         else:
             a = ['【','[','(','（']
             b = ['】',']',')','）']
@@ -192,7 +222,7 @@ def create_tts_fn(net_g,hps,speaker_id):
                     pass
                 
             file_path = "subtitles.srt"
-            return history,file_path,(hps.data.sampling_rate, np.concatenate(audio_fin))
+            return history,file_path,(hps.data.sampling_rate, np.concatenate(audio_fin)),file_path,htm
     return tts_fn
 
 if __name__ == '__main__':
@@ -241,7 +271,7 @@ if __name__ == '__main__':
                                             f'<img style="width:auto;height:400px;" src="file/image/{name}.png">' 
                                             '</div>'
                                         )
-                                    chatbot = gr.Chatbot(elem_id="History")
+                                    output_UI = gr.outputs.HTML()
                                 with gr.Row():
                                     with gr.Column(scale=0.85):
                                         input1 = gr.TextArea(label="Text", value=example,lines = 1)    
@@ -262,6 +292,6 @@ if __name__ == '__main__':
                                     audio_input1 = gr.Checkbox(value=False, label="修改音频路径(live2d)")
                                     audio_input2 = gr.TextArea(label="音频路径",lines=1,value = '#参考 D:/app_develop/live2d_whole/2010002/sounds/temp.wav')
                                  
-                        btnVC.click(tts_fn, inputs=[chatbot,api_input1,api_input2,audio_input1,audio_input2,audio_input3,input1,input2,input3,input4,input5,input6], outputs=[chatbot,output2,output1])
+                        btnVC.click(tts_fn, inputs=[api_input1,api_input2,audio_input1,audio_input2,audio_input3,input1,input2,input3,input4,input5,input6], outputs=[output1,output2,output_UI])
                 
     app.launch()