Merge pull request #95 from danielj-n/main

Prevent latex formatting from running on code blocks and inline code

Author: npiv

chatblade/printer.py

@@ -87,15 +87,31 @@ def extract_messages(messages, args):
         print(message.content.strip())
 
 def format_latex(msg):
+    # Replace code blocks and inline code with markers. Use null delimiters to
+    # hopefully avoid any overlap with anything chatgpt could ever output.
+    code_block_pattern = re.compile(r"```[\w]*\n.*?\n```", re.DOTALL)
+    code_blocks = re.findall(code_block_pattern, msg)
+    msg = re.sub(code_block_pattern, "\0CODE_BLOCK\0", msg)
+    code_inline_pattern = re.compile(r"`.*?`", re.DOTALL)
+    code_inlines = re.findall(code_inline_pattern, msg)
+    msg = re.sub(code_inline_pattern, "\0CODE_INLINE\0", msg)
+
     converter = LatexNodes2Text()
     msg = converter.latex_to_text(msg)
 
     # do no change code blocks to smart quotes, this will break the markdown
     # parser.
     msg = msg.replace("“", "``")
 
+    # Restore the code blocks and inline code at the markers
+    for code_block in code_blocks:
+        msg = msg.replace("\0CODE_BLOCK\0", code_block, 1)
+    for code_inline in code_inlines:
+        msg = msg.replace("\0CODE_INLINE\0", code_inline, 1)
+
     return msg
 
+
 def detect_and_format_message(msg, cutoff=None, theme=None):
     # convert any latex markup to ASCII.
     msg = format_latex(msg)