输出Markdown (#160)

* 增加将历史记录以markdown保存的功能，

* 解决了中文乱码问题，优化了UI显示

* Update ChuanhuChatbot.py

* Update utils.py

---------

Co-authored-by: Chuan Hu <[email protected]>

ChuanhuChatbot.py

@@ -99,21 +99,27 @@ with gr.Blocks(css=customCSS) as demo:
 
                 with gr.Tab(label="保存/加载"):
                     with gr.Accordion(label="保存/加载对话历史记录", open=True):
+                        gr.Markdown("对话历史默认保存在history文件夹中。")
                         with gr.Column():
                             with gr.Row():
                                 with gr.Column(scale=6):
                                     saveFileName = gr.Textbox(
-                                        show_label=True, placeholder=f"在这里输入保存的文件名...", label="设置保存文件名", value="对话历史记录").style(container=True)
+                                        show_label=True, placeholder=f"设置文件名: 默认为.json，可选为.md", label="设置保存文件名", value="对话历史记录").style(container=True)
                                 with gr.Column(scale=1):
                                     saveHistoryBtn = gr.Button("💾 保存对话")
+                                    exportMarkdownBtn = gr.Button("📝 导出为Markdown")
                             with gr.Row():
                                 with gr.Column(scale=6):
                                     historyFileSelectDropdown = gr.Dropdown(label="从列表中加载对话", choices=get_history_names(plain=True), multiselect=False, value=get_history_names(plain=True)[0])
                                 with gr.Column(scale=1):
                                     historyRefreshBtn = gr.Button("🔄 刷新")
+                            with gr.Row():
+                                with gr.Column():
+                                    downloadFile = gr.File(interactive=True)
 
     gr.Markdown(description)
 
+    # Chatbot
     user_input.submit(predict, [keyTxt, systemPromptTxt, history, user_input, chatbot, token_count, top_p, temperature, use_streaming_checkbox, model_select_dropdown, use_websearch_checkbox], [chatbot, history, status_display, token_count], show_progress=True)
     user_input.submit(reset_textbox, [], [user_input])
 
@@ -129,20 +135,19 @@ with gr.Blocks(css=customCSS) as demo:
 
     reduceTokenBtn.click(reduce_token_size, [keyTxt, systemPromptTxt, history, chatbot, token_count, top_p, temperature, use_streaming_checkbox, model_select_dropdown], [chatbot, history, status_display, token_count], show_progress=True)
 
-    saveHistoryBtn.click(save_chat_history, [
-                  saveFileName, systemPromptTxt, history, chatbot], None, show_progress=True)
+    # Template
+    templateRefreshBtn.click(get_template_names, None, [templateFileSelectDropdown])
+    templateFileSelectDropdown.change(load_template, [templateFileSelectDropdown],  [promptTemplates, templateSelectDropdown], show_progress=True)
+    templateSelectDropdown.change(get_template_content, [promptTemplates, templateSelectDropdown, systemPromptTxt],  [systemPromptTxt], show_progress=True)
 
+    # S&L
+    saveHistoryBtn.click(save_chat_history, [saveFileName, systemPromptTxt, history, chatbot], downloadFile, show_progress=True)
     saveHistoryBtn.click(get_history_names, None, [historyFileSelectDropdown])
-
+    exportMarkdownBtn.click(export_markdown, [saveFileName, systemPromptTxt, history, chatbot], downloadFile, show_progress=True)
     historyRefreshBtn.click(get_history_names, None, [historyFileSelectDropdown])
-
     historyFileSelectDropdown.change(load_chat_history, [historyFileSelectDropdown, systemPromptTxt, history, chatbot],  [saveFileName, systemPromptTxt, history, chatbot], show_progress=True)
+    downloadFile.change(load_chat_history, [downloadFile, systemPromptTxt, history, chatbot],  [saveFileName, systemPromptTxt, history, chatbot])
 
-    templateRefreshBtn.click(get_template_names, None, [templateFileSelectDropdown])
-
-    templateFileSelectDropdown.change(load_template, [templateFileSelectDropdown],  [promptTemplates, templateSelectDropdown], show_progress=True)
-
-    templateSelectDropdown.change(get_template_content, [promptTemplates, templateSelectDropdown, systemPromptTxt],  [systemPromptTxt], show_progress=True)
 
 logging.info(colorama.Back.GREEN + "\n川虎的温馨提示：访问 http://localhost:7860 查看界面" + colorama.Style.RESET_ALL)
 # 默认开启本地服务器，默认可以直接从IP访问，默认不创建公开分享链接

utils.py

@@ -323,22 +323,42 @@ def delete_last_conversation(chatbot, history, previous_token_count):
     return chatbot, history, previous_token_count, construct_token_message(sum(previous_token_count))
 
 
-def save_chat_history(filename, system, history, chatbot):
+def save_file(filename, system, history, chatbot):
     logging.info("保存对话历史中……")
+    os.makedirs(HISTORY_DIR, exist_ok=True)
+    if filename.endswith(".json"):
+        json_s = {"system": system, "history": history, "chatbot": chatbot}
+        print(json_s)
+        with open(os.path.join(HISTORY_DIR, filename), "w") as f:
+            json.dump(json_s, f)
+    elif filename.endswith(".md"):
+        md_s = f"system: \n- {system} \n"
+        for data in history:
+            md_s += f"\n{data['role']}: \n- {data['content']} \n"
+        with open(os.path.join(HISTORY_DIR, filename), "w", encoding="utf8") as f:
+            f.write(md_s)
+    logging.info("保存对话历史完毕")
+    return os.path.join(HISTORY_DIR, filename)
+
+def save_chat_history(filename, system, history, chatbot):
     if filename == "":
         return
     if not filename.endswith(".json"):
         filename += ".json"
-    os.makedirs(HISTORY_DIR, exist_ok=True)
-    json_s = {"system": system, "history": history, "chatbot": chatbot}
-    logging.info(json_s)
-    with open(os.path.join(HISTORY_DIR, filename), "w") as f:
-        json.dump(json_s, f, ensure_ascii=False, indent=4)
-    logging.info("保存对话历史完毕")
+    return save_file(filename, system, history, chatbot)
+
+def export_markdown(filename, system, history, chatbot):
+    if filename == "":
+        return
+    if not filename.endswith(".md"):
+        filename += ".md"
+    return save_file(filename, system, history, chatbot)
 
 
 def load_chat_history(filename, system, history, chatbot):
     logging.info("加载对话历史中……")
+    if type(filename) != str:
+        filename = filename.name
     try:
         with open(os.path.join(HISTORY_DIR, filename), "r") as f:
             json_s = json.load(f)