如何在Streamlit中将Markdown文本转换并下载为PDF报告？

Question

我有一段在Streamlit中生成的Markdown文本，如下所示：

st_md = '''
compare mongodb to other no sql databases

Uploaded Files: []

 Here is a comparison of MongoDB to some other major NoSQL databases:

- MongoDB is a document database. It stores data in flexible JSON-like documents rather than rows and columns like an RDBMS. Other document databases include CouchDB and Amazon DocumentDB.

...

advantages and disadvantages of mongodb to other no sql ds

Uploaded Files: []

 Here are some key advantages and disadvantages of MongoDB compared to other NoSQL databases:
...
'''

我想将st_md的内容下载为一个PDF文件。我尝试使用download_button方法，但收到错误提示说PDF文件已损坏。我遗漏了什么？

st.download_button(
    label="Download data as pdf",
    data=st_md,
    file_name='test.pdf',
)

Matchu · Answer

st.download_button函数设计用于下载文件，但它可能无法直接处理Markdown文本到PDF的转换。为了实现您的目标，您可以使用reportlab库将Markdown文本转换为PDF文件。

import streamlit as st
from reportlab.lib.pagesizes import letter
from reportlab.pdfgen import canvas
from io import BytesIO
import base64

# 您的Markdown文本内容
st_md = '''
# 您的Markdown内容
'''

# 将Markdown转换为PDF的函数
def markdown_to_pdf(markdown_text):
    buffer = BytesIO()
    # 创建一个画布对象，设置页面大小为信纸尺寸
    p = canvas.Canvas(buffer, pagesize=letter)
    # 在画布上绘制Markdown文本（需自行调整位置）
    p.drawString(100, 800, markdown_text)
    # 保存PDF到缓冲区
    p.save()
    # 将缓冲区指针重置回开始位置
    buffer.seek(0)
    return buffer

# 获取PDF缓冲区
pdf_buffer = markdown_to_pdf(st_md)

# 显示下载按钮
st.download_button(
    label="以PDF格式下载数据",
    data=pdf_buffer,
    file_name='test.pdf',
)

# 可选：在Streamlit中显示PDF内容
st.image(pdf_buffer, format='pdf')

# 关闭缓冲区以释放资源
pdf_buffer.close()

Bite code · Answer

根据Chathura Abeywickrama的建议，我使用以下代码将Markdown内容转换成了PDF：

import markdown2
import pdfkit

# 将Markdown内容转换为HTML
html = markdown2.markdown(st_md)

# 从HTML生成PDF文件
pdfkit.from_string(html, 'example.pdf')

然后，在download_button中使用生成的PDF文件：

# 打开已生成的PDF文件并以二进制模式读取
with open("example.pdf", "rb") as f:
    # 在Streamlit中显示下载按钮，用户可以点击下载该PDF文件
    st.download_button(label="下载对话记录", data=f, file_name='example.pdf')