增加了删除早期历史记录的按钮

ChuanhuChatbot.py

@@ -88,7 +88,8 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                     "🧹 新的对话",
                 )
                 retryBtn = gr.Button("🔄 重新生成")
-                delLastBtn = gr.Button("🗑️ 删除一条对话")
+                delFirstBtn = gr.Button("🗑️ 删除早期历史")
+                delLastBtn = gr.Button("🗑️ 删除最后对话")
                 reduceTokenBtn = gr.Button("♻️ 总结对话")
 
         with gr.Column():
@@ -301,6 +302,12 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     ).then(**end_outputing_args)
     retryBtn.click(**get_usage_args)
 
+    delFirstBtn.click(
+        delete_first_conversation,
+        [history, token_count],
+        [history, token_count, status_display],
+    )
+
     delLastBtn.click(
         delete_last_conversation,
         [chatbot, history, token_count],

modules/utils.py

@@ -115,7 +115,11 @@ def convert_mdtext(md_text):
 
 
 def convert_asis(userinput):
-    return f"<p style=\"white-space:pre-wrap;\">{html.escape(userinput)}</p>"+ALREADY_CONVERTED_MARK
+    return (
+        f'<p style="white-space:pre-wrap;">{html.escape(userinput)}</p>'
+        + ALREADY_CONVERTED_MARK
+    )
+
 
 def detect_converted_mark(userinput):
     if userinput.endswith(ALREADY_CONVERTED_MARK):
@@ -154,6 +158,17 @@ def construct_token_message(token, stream=False):
     return f"Token 计数: {token}"
 
 
+def delete_first_conversation(history, previous_token_count):
+    if history:
+        del history[:2]
+        del previous_token_count[0]
+    return (
+        history,
+        previous_token_count,
+        construct_token_message(sum(previous_token_count)),
+    )
+
+
 def delete_last_conversation(chatbot, history, previous_token_count):
     if len(chatbot) > 0 and standard_error_msg in chatbot[-1][1]:
         logging.info("由于包含报错信息，只删除chatbot记录")
@@ -417,8 +432,14 @@ def cancel_outputing():
     logging.info("中止输出……")
     shared.state.interrupt()
 
+
 def transfer_input(inputs):
     # 一次性返回，降低延迟
     textbox = reset_textbox()
     outputing = start_outputing()
-    return inputs, gr.update(value=""), gr.Button.update(visible=False), gr.Button.update(visible=True)
+    return (
+        inputs,
+        gr.update(value=""),
+        gr.Button.update(visible=False),
+        gr.Button.update(visible=True),
+    )