feat: 点击“新的对话”按钮清空的历史显示

ChuanhuChatbot.py

@@ -62,7 +62,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                     cancelBtn = gr.Button(value="", variant="secondary", visible=False, elem_id="cancel_btn")
             with gr.Row():
                 emptyBtn = gr.Button(
-                    i18n("🧹 新的对话"),
+                    i18n("🧹 新的对话"), elem_id="empty_btn"
                 )
                 retryBtn = gr.Button(i18n("🔄 重新生成"))
                 delFirstBtn = gr.Button(i18n("🗑️ 删除最旧对话"))

assets/custom.js

@@ -15,6 +15,7 @@ var appTitleDiv = null;
 var chatbot = null;
 var chatbotWrap = null;
 var apSwitch = null;
+var empty_botton = null;
 var messageBotDivs = null;
 var renderLatex = null;
 var shouldRenderLatex = false;
@@ -36,6 +37,7 @@ function gradioLoaded(mutations) {
             chatbotWrap = document.querySelector('#chuanhu_chatbot > .wrap');
             apSwitch = document.querySelector('.apSwitch input[type="checkbox"]');
             renderLatex = document.querySelector("#render_latex_checkbox > label > input");
+            empty_botton = document.getElementById("empty_btn")
 
             if (gradioContainer && apSwitch) {  // gradioCainter 加载出来了没?
                 adjustDarkMode();
@@ -58,6 +60,9 @@ function gradioLoaded(mutations) {
                 shouldRenderLatex = renderLatex.checked;
                 updateMathJax();
             }
+            if (empty_botton) {
+                emptyHistory();
+            }
         }
     }
 }
@@ -354,6 +359,17 @@ function loadHistoryHtml() {
     }
 }
 
+function emptyHistory (){
+    empty_botton.addEventListener("click", function () {
+        localStorage.removeItem("chatHistory");
+        historyMessages = chatbotWrap.querySelector('.history-message');
+        if (historyMessages) {
+            chatbotWrap.removeChild(historyMessages);
+            console.log("History Cleared");
+        }
+    });
+}
+
 // 监视页面内部 DOM 变动
 var observer = new MutationObserver(function (mutations) {
     gradioLoaded(mutations);