Update bin_public/utils/utils.py

bin_public/utils/utils.py

@@ -336,21 +336,38 @@ 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):
-    # 初步想法，不自己取名，统一叫data.json，    待实现
+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(f"./data.json", "w") as f:
-        json.dump(json_s, f, ensure_ascii=False, indent=4)
-    logging.info("保存对话历史完毕")
+    return save_file(filename, system, history, chatbot)
+
 
-    return f"./data.json"
+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):