我们看别人代码时,总希望有流程图,这样可以一目了然,不过自己写的代码,又不想花几个小时去画流程图。有没有更好的方法呢?
方法就是用ChatGPT等大模型自动生成流程图,并用python等语言实现自动输出流程图。
对下面的代码生成流程图,并用Mermaid语法输出 。
1、生成流程图的 Mermaid语法
2、把Mermaid语法的流程图输出图片
要使用Python或Java生成Mermaid语法输出的流程图图片,您可以使用以下方法:
mermaid-cli 工具来将Mermaid代码转换为图片。首先,安装mmdc(mermaid-cli的执行程序)。mmdc。- import os
-
- def generate_mermaid_image(mermaid_code, output_path):
- with open("temp.mmd", "w") as file:
- file.write(mermaid_code)
- os.system(f"mmdc -i temp.mmd -o {output_path}")
- os.remove("temp.mmd")
-
- mermaid_code = """
- graph TD;
- A-->B;
- A-->C;
- B-->D;
- C-->D;
- """
-
- generate_mermaid_image(mermaid_code, "output.png")