Update utils.py

utils.py

@@ -145,3 +145,37 @@ def greedy_search(input_ids: torch.Tensor,
                 del probs_sum
                 gc.collect()
                 return 
+
+def convert_to_markdown(text):
+    text = text.replace("$","$")
+    def replace_leading_tabs_and_spaces(line):
+        new_line = []
+        
+        for char in line:
+            if char == "\t":
+                new_line.append("	")
+            elif char == " ":
+                new_line.append(" ")
+            else:
+                break
+        return "".join(new_line) + line[len(new_line):]
+
+    markdown_text = ""
+    lines = text.split("\n")
+    in_code_block = False
+
+    for line in lines:
+        if in_code_block is False and line.startswith("```"):
+            in_code_block = True
+            markdown_text += f"{line}\n"
+        elif in_code_block is True and line.startswith("```"):
+            in_code_block = False
+            markdown_text += f"{line}\n"
+        elif in_code_block:
+            markdown_text += f"{line}\n"
+        else:
+            line = replace_leading_tabs_and_spaces(line)
+            line = re.sub(r"^(#)", r"\\\1", line)
+            markdown_text += f"{line}  \n"
+
+    return markdown_text