增加对话记忆按键

app.py

@@ -40,16 +40,20 @@ provider_dict = {
 with gr.Blocks() as demo:
   chatbot = gr.Chatbot([[None, None]], label='AI')
   msg = gr.Textbox(value="", label='')
-  clear = gr.Button("Clear")
+  with gr.Row():
+    clear = gr.Button("清空对话")
+    remember = gr.Button("记住最近一次对话")
   system_msg = gr.Textbox(value="你是一名助手，可以解答问题。", label='系统提示')
   with gr.Row():
     model_name = gr.Dropdown(['gpt-3.5-turbo', 'gpt-4'], value='gpt-3.5-turbo', label='模型')
     provider_name  =  gr.Dropdown(provider_dict.keys(), value='GetGpt', label='提供者')
 
+  memory_msg = gr.Textbox(value="", label='记忆')
+
   def user(user_message, history):
       return gr.update(value="", interactive=False), history + [[user_message, None]]
 
-  def bot(history, model_name, provider_name, system_msg):
+  def bot(history, model_name, provider_name, system_msg, memory_msg):
       history[-1][1] = ''
       if len(system_msg)>3000:
          system_msg = system_msg[:1500] + system_msg[-1500:]
@@ -59,6 +63,10 @@ with gr.Blocks() as demo:
       ```
       {}
       ```
+      历史对话中的关键信息：
+      ```
+      {}
+      ```
       用户最新的话：
       ```
       {}
@@ -69,17 +77,52 @@ with gr.Blocks() as demo:
                                           provider=provider_dict[provider_name], 
                                           messages=[{"role": "user", 
                                                      "content": prompt.format(system_msg,
+                                                                              memory_msg,
                                                                               history[-1][0])}],
                                           stream=True)
       for c in bot_msg:
           history[-1][1] += c
           yield history
+
+  def update_memory(memory_msg, history, model_name, provider_name):
+     if history==None:
+        return gr.update(value=memory_msg, interactive=False)
+     prompt = """
+     请你记录对话内容，尤其是用户提供的各种关键信息。已知的对话内容总结如下：
+     ```
+     {}
+     ```
+     最新的对话内容如下：
+     ```
+     用户:{}
+     AI:{}
+     ```
+     总结格式如下：
+     ```
+     关键信息总结：XXX
+     ```
+     请对已知对话和最新对话的关键信息进行总结，以应对后续的聊天：
+     """
+     memory = g4f.ChatCompletion.create(model=model_name,
+                                            provider=provider_dict[provider_name],
+                                            messages=[
+                                               {
+                                                  'role':'user',
+                                                  'content': prompt.format(memory_msg,
+                                                                           history[-1][0],
+                                                                           history[-1][1])
+                                               }
+                                            ])
+     return gr.update(value=memory, interactive=False)
   
   response = msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-      bot, [chatbot, model_name, provider_name, system_msg], chatbot
+      bot, [chatbot, model_name, provider_name, system_msg, memory_msg], chatbot
   )
   response.then(lambda: gr.update(interactive=True), None, [msg], queue=False)
-  clear.click(lambda: None, None, chatbot, queue=False)
+  remember.click(
+     update_memory, [memory_msg, chatbot, model_name, provider_name], memory_msg
+  )
+  clear.click(lambda: (None, None), None, [chatbot, memory_msg], queue=False)
 
 demo.title = "AI Chat"
 demo.queue()