feat: 加入 Markdown 下载功能

ChuanhuChatbot.py

@@ -82,6 +82,9 @@ with gr.Blocks(theme=small_and_beautiful_theme) as demo:
                             with gr.Column(min_width=42, scale=1):
                                 historyDownloadBtn = gr.Button(
                                     i18n("⏬"), elem_id="gr-history-download-btn")
+                            with gr.Column(min_width=42, scale=1):
+                                historyMarkdownDownloadBtn = gr.Button(
+                                    i18n("⤵️"), elem_id="gr-history-mardown-download-btn")
                     with gr.Row():
                         with gr.Column(scale=6):
                             saveFileName = gr.Textbox(
@@ -665,6 +668,8 @@ with gr.Blocks(theme=small_and_beautiful_theme) as demo:
                         saveFileName, systemPromptTxt, chatbot])
     historyDownloadBtn.click(None, [
                              user_name, historySelectList], None, _js='(a,b)=>{return downloadHistory(a,b);}')
+    historyMarkdownDownloadBtn.click(None, [
+                                     user_name, historySelectList], None, _js='(a,b)=>{return downloadHistoryMarkdown(a,b);}')
     historySearchTextbox.input(
         filter_history,
         [user_name, historySearchTextbox],

modules/utils.py

@@ -332,23 +332,26 @@ def construct_assistant(text):
 
 def save_file(filename, system, history, chatbot, user_name):
     os.makedirs(os.path.join(HISTORY_DIR, user_name), exist_ok=True)
+    if filename.endswith(".md"):
+        filename = filename[:-3]
     if not filename.endswith(".json") and not filename.endswith(".md"):
         filename += ".json"
-    if filename.endswith(".json"):
-        json_s = {"system": system, "history": history, "chatbot": chatbot}
-        if "/" in filename or "\\" in filename:
-            history_file_path = filename
-        else:
-            history_file_path = os.path.join(HISTORY_DIR, user_name, filename)
-        logging.info(f"保存文件，文件名为{filename}，用户为{user_name}, 文件路径为{history_file_path}")
-        with open(history_file_path, "w", encoding='utf-8') as f:
-            json.dump(json_s, f, ensure_ascii=False)
-    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, user_name, filename), "w", encoding="utf8") as f:
-            f.write(md_s)
+
+    json_s = {"system": system, "history": history, "chatbot": chatbot}
+    if "/" in filename or "\\" in filename:
+        history_file_path = filename
+    else:
+        history_file_path = os.path.join(HISTORY_DIR, user_name, filename)
+    with open(history_file_path, "w", encoding='utf-8') as f:
+        json.dump(json_s, f, ensure_ascii=False)
+
+    filename = filename.split("/")[-1]
+    filename_md = filename[:-5] + ".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, user_name, filename_md), "w", encoding="utf8") as f:
+        f.write(md_s)
     return os.path.join(HISTORY_DIR, user_name, filename)
 
 

web_assets/javascript/utils.js

@@ -22,7 +22,17 @@ function downloadHistory(username, historyname) {
     } else {
         fileUrl = `/file=./history/${username}/${historyname}`;
     }
-    downloadFile(fileUrl, historyname);
+    downloadFile(fileUrl + ".json", historyname + ".json");
+}
+
+function downloadHistoryMarkdown(username, historyname) {
+    let fileUrl;
+    if (username === null) {
+        fileUrl = `/file=./history/${historyname}`;
+    } else {
+        fileUrl = `/file=./history/${username}/${historyname}`;
+    }
+    downloadFile(fileUrl + ".md", historyname + ".md");
 }
 
 function downloadFile(fileUrl, filename="") {