Update app.py

Add function to clear memory

app.py

@@ -1,5 +1,6 @@
 import g4f
 import gradio as gr
+from gradio import ChatInterface
 from g4f.Provider import (
     Ails,
     You,
@@ -92,8 +93,9 @@ def change_prompt(prompt_set_name, prompt_name):
 def user(user_message, history):
     return gr.update(value="", interactive=False), history + [[user_message, None]]
 
-def bot(message, history, model_name, provider_name, system_msg, agent):
-    response = ''
+def bot(history, model_name, provider_name, system_msg, agent):
+    history[-1][1] = ''
+    message = history[-1][0]
 
     if len(system_msg)>3000:
         system_msg = system_msg[:2000] + system_msg[-1000:]
@@ -126,22 +128,36 @@ def bot(message, history, model_name, provider_name, system_msg, agent):
     prompt = PromptTemplate(
                         input_variables=["chat_history", "human_input"], template=new_template
                         )
-    llm_chain = LLMChain(
-                        llm=llm,
-                        prompt=prompt,
-                        verbose=False,
-                        memory=memory,
-                    )
-    bot_msg = llm_chain.run(message)
+
+    prev_memory = ''
+    if len(history)>1 and history[-2][1]!=None:
+        memory.chat_memory.add_user_message(history[-2][0])
+        memory.chat_memory.add_ai_message(history[-2][1]) 
+        prev_memory = memory.load_memory_variables({})['chat_history']
+    
+    prompt = new_template.format(
+        chat_history = prev_memory,
+        human_input = message
+    )
+
+    bot_msg = llm._call(prompt=prompt)
+
     for c in bot_msg:
-        response += c
-    return response
+        history[-1][1] += c
+        yield history
 
-def empty_chat():
+def empty_fn():
     global memory
-    memory = ConversationBufferWindowMemory(k=6, memory_key="chat_history")
+    memory = ConversationBufferWindowMemory(k=10, memory_key="chat_history")
     return None
 
+def undo_fn(history):
+    return history[:-1]
+
+def retry_fn(history):
+    history[-1][1] = None
+    return history
+
 prompt_set_list = {}
 for prompt_file in os.listdir("prompt_set"):
     key = prompt_file
@@ -165,7 +181,7 @@ with gr.Blocks() as demo:
     The following is a conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.
     {{chat_history}}
     Human: {{human_input}}
-    Chatbot:"""
+    AI:"""
 
     memory = ConversationBufferWindowMemory(k=6, memory_key="chat_history")
     with gr.Row():
@@ -173,18 +189,42 @@ with gr.Blocks() as demo:
         provider = gr.Dropdown(available_dict['gpt-3.5-turbo'], value='AiService', label='提供者', min_width=20)
         agent = gr.Dropdown(['系统提示', '维基百科', 'duckduckgo'], value='系统提示', label='Agent')
     system_msg = gr.Textbox(value="你是一名助手，可以解答问题。", label='系统提示')
-    gr.ChatInterface(bot,
-                     additional_inputs=[
-                                model_name,
-                                provider,
-                                system_msg,
-                                agent]
-                    )
+    
+    chatbot = gr.Chatbot([[None, None]], label='AI')
+    with gr.Group():
+        with gr.Row():
+            textbox = gr.Textbox(
+                                    container=False,
+                                    show_label=False,
+                                    label="请输入：",
+                                    scale=7,
+                                    autofocus=True,
+                                )
+            submit = gr.Button('发送', scale=1, variant="primary", min_width=150,)
+    with gr.Row():
+        retry = gr.Button('🔄 重试')
+        undo = gr.Button('↩️ 撤销')
+        clear = gr.Button("🗑️ 清空")
+
     with gr.Row():
         default_prompt_set = "1 中文提示词.json"
         prompt_set_name = gr.Dropdown(prompt_set_list.keys(), value=default_prompt_set, label='提示词集合')
         prompt_name = gr.Dropdown(prompt_set_list[default_prompt_set].keys(), label='提示词', min_width=5, container=True)
 
+    textbox.submit(user, [textbox, chatbot], [textbox, chatbot], queue=False).then(
+        bot, [chatbot, model_name, provider, system_msg, agent], chatbot
+    ).then(lambda: gr.update(interactive=True), None, [textbox], queue=False)
+
+    response = submit.click(user, [textbox, chatbot], [textbox, chatbot], queue=False).then(
+        bot, [chatbot, model_name, provider, system_msg, agent], chatbot
+    ).then(lambda: gr.update(interactive=True), None, [textbox], queue=False)
+
+    retry.click(retry_fn, [chatbot], [chatbot]).then(
+        bot, [chatbot, model_name, provider, system_msg, agent], chatbot
+    )
+    undo.click(undo_fn, [chatbot], [chatbot], queue=False)
+    clear.click(empty_fn, None, [chatbot], queue=False)
+
     prompt_set_name.select(change_prompt_set, prompt_set_name, prompt_name)
     model_name.select(change_model, model_name, provider)
     prompt_name.select(change_prompt, [prompt_set_name, prompt_name], system_msg)