利用AI大模型和Mermaid生成流程图

小江爱学习~

已于 2025-01-10 16:35:45 修改

阅读量3.3k

点赞数 15

文章标签： java python

于 2025-01-10 16:34:49 首次发布

本文链接：https://blog.csdn.net/m0_62458145/article/details/145061101

版权

核心点1：利用大模型生成流程图的语句（Code）

确定业务流程：
- 用户需要明确要绘制的业务流程，包括主要步骤、决策点以及各步骤之间的关系。
- 将确定的业务流程以文字形式描述出来。
生成Mermaid代码：
- 将描述好的业务流程交给大模型（如ChatGPT），让其生成相应的Mermaid代码。
- 大模型会根据输入的业务流程描述，自动生成符合Mermaid语法的代码。

例如，假设我们要绘制一个用户登录的业务流程图，内容包括用户登录、检查用户权限、显示内容或权限错误以及用户注销等步骤。描述业务流程如下：

用户登录
检查用户权限
如果权限有效，显示内容；否则显示权限错误
用户可以选择注销

将上述业务流程描述交给ChatGPT，生成如下Mermaid代码：

PlainText



graph TB
A[用户登录]-->B[检查用户权限]
B--权限有效-->C[显示内容]
B--权限无效-->D[显示权限错误]
C-->E[用户操作]

核心点2：利用Mermaid的开源API将Code转换成流程图片

渲染流程图：
- 将生成的Mermaid代码放入支持Mermaid的环境中，如在线免费网站（Online FlowChart & Diagrams Editor - Mermaid Live Editor），进行渲染生成流程图。
- 用户可以根据自己的需求对生成的流程图进行调整和优化。
下载或引用图片：
- 将生成的流程图按照自己需要的格式下载下来，或者直接引用当前图片的链接，将其嵌入到文档中。

Mermaid API的详细使用

Mermaid提供了丰富的API方法，使得在网页中创建和展示各种类型的图表变得非常方便。以下是一些关键步骤：

初始化图表：使用mermaid.initialize方法可以初始化一个流程图，该方法接受一个可选的配置对象作为参数，用于自定义图表的样式和行为。
添加节点和消息：使用mermaidAPI.add方法可以向流程图中添加节点或消息。节点定义字符串可以使用-->表示节点之间的连接关系。
渲染图表：使用mermaidAPI.render方法可以将流程图渲染到指定的元素中。渲染后，图表将自动根据元素的大小进行布局和缩放，以适应不同的屏幕尺寸。

示例代码与流程

以下是一个完整的示例流程，展示了如何使用大模型和Mermaid API生成业务流程图：

确定业务流程并描述：例如，用户登录、检查权限、显示内容或权限错误、用户注销等步骤。
使用大模型生成Mermaid代码：将业务流程描述交给ChatGPT或其他大模型，生成Mermaid代码。
在Mermaid环境中渲染：将生成的Mermaid代码粘贴到支持Mermaid的在线网站（如Online FlowChart & Diagrams Editor - Mermaid Live Editor）中进行渲染。
下载或引用图片：根据需要下载生成的流程图图片，或直接引用其链接嵌入到文档中。

注意事项

确保大模型能够准确理解并生成符合Mermaid语法的代码。
根据实际需求调整和优化生成的流程图。
注意Mermaid API的版本兼容性，确保使用的API方法在当前版本中有效。

通过以上步骤，你可以高效地利用大模型和Mermaid实现业务流程图的快速生成，并将其应用于各种文档和项目中。

代码：

# Mermaid Live Editor API
    api_url = "https://mermaid.ink/img/"
    full_url = f"{api_url}{base64_str}"

    # 下载图片
    response = requests.get(full_url)
    if response.status_code == 200:
        # 确保目录存在
        os.makedirs(upload_dir, exist_ok=True)

        # 保存图片
        with open(output_path, 'wb') as f:
            f.write(response.content)
        print(f"图片已保存到: {output_path}")
        print(f"访问链接: {full_url}")
        return output_path
    else:
        print(f"错误: {response.status_code}")
        print(f"URL: {full_url}")
        return None