From b4d933aa3908e8cd9dd78aa9f208a5deaa58f726 Mon Sep 17 00:00:00 2001
From: Routhleck <1310722434@qq.com>
Date: Mon, 28 Aug 2023 22:06:50 +0800
Subject: [PATCH] Update markdown_process.py

---
 markdown_process.py | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/markdown_process.py b/markdown_process.py
index ab346a2..cce39c1 100644
--- a/markdown_process.py
+++ b/markdown_process.py
@@ -51,15 +51,17 @@ def replace_img_url_2(content):
 
 
 def add_newlines_around_formula(content):
-    # 在 $$ 前后添加空行，确保不重复添加空行，并且公式块内部的内容不发生改变
-    pattern_start = r'(?<!\n)\n\$\$'
-    replacement_start = r'\n\n$$'
-    content = re.sub(pattern_start, replacement_start, content)
+    # 在整个 $$ ... $$ 公式块前后添加空行，确保不重复添加空行
+
+    # 处理公式块前的换行
+    content = re.sub(r'(?<=\S)\n\$\$$', '\n\n$$', content)
+    # 处理公式块后的换行
+    content = re.sub(r'\$\$\$(?=\S)', '$$\n\n', content)
+    # 处理公式块前没有换行的情况
+    content = re.sub(r'(?<=[^\n])\$\$$', '\n\n$$', content)
+    # 处理公式块后没有换行的情况
+    content = re.sub(r'\$\$(?=[^\n])', '$$\n\n', content)
     
-    pattern_end = r'\$\$(?!\n)'
-    replacement_end = r'$$\n'
-    content = re.sub(pattern_end, replacement_end, content)
-
     return content
 
 
@@ -75,7 +77,7 @@ def main(input_path, out_path):
     md_content = prepend_markdown_header(md_content)
     md_content = replace_img_url_1(md_content)
     md_content = replace_img_url_2(md_content)
-    md_content = add_newlines_around_formula(md_content)
+    # md_content = add_newlines_around_formula(md_content)
 
     # 将修改后的内容写入新文件
     with open(out_path, "w", encoding="utf-8") as file: