将TexSoup树转换回LaTeX

from TexSoup import TexSoup

# 假设我们已经有一个TexSoup树对象tree
# 这里省略了解析LaTeX文档的步骤

def convert_to_latex(node):
    if isinstance(node, TexSoup.data.RArg):
        return '{' + node.string + '}'
    elif isinstance(node, TexSoup.data.Raw):
        return node.string
    elif isinstance(node, TexSoup.data.Environment):
        return '\\begin{' + node.name + '}\n' + ''.join(convert_to_latex(child) for child in node.contents) + '\n\\end{' + node.name + '}'
    elif isinstance(node, TexSoup.data.Command):
        return '\\' + node.name + ''.join(convert_to_latex(arg) for arg in node.args)

latex_code = ''.join(convert_to_latex(node) for node in tree.contents)
print(latex_code)