Streamlit应用中实现点击按钮才通过Reportlab生成PDF下载的问题

嘿，我来帮你搞定这个问题！你现在的困扰应该是Streamlit应用一加载，Reportlab就自动把PDF生成到本地文件夹里，而你想要改成只有用户点击下载按钮时，才生成PDF并让用户自行下载对吧？

其实核心思路很简单：把PDF生成的逻辑和Streamlit的下载按钮绑定，让生成动作只在按钮点击时触发，而且把生成的PDF存在内存缓冲区里，直接通过下载按钮返回给用户，根本不会写到你的本地文件夹里。

我给你整理了完整的实现步骤和代码，你可以直接参考：

第一步：把PDF生成逻辑封装成函数 #

咱们把生成PDF的代码放到一个函数里，这样就不会在页面加载时自动执行，只有调用这个函数的时候才会生成PDF：

import streamlit as st
import io
from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import letter
from reportlab.platypus import Paragraph
from reportlab.lib.styles import getSampleStyleSheet

def generate_pdf():
    # 用内存缓冲区代替本地文件，避免自动保存到本地
    buffer = io.BytesIO()
    doc = canvas.Canvas(buffer, pagesize=letter)
    styles = getSampleStyleSheet()
    
    # 这里写你原本的PDF内容逻辑，比如添加标题
    pdf_title = "Results"
    title_paragraph = Paragraph(pdf_title, styles["Title"])
    # 调整标题的显示位置和换行规则
    title_paragraph.wrapOn(doc, letter[0] - 100, letter[1] - 100)
    title_paragraph.drawOn(doc, 50, letter[1] - 100)
    
    # 你还可以继续添加更多内容，比如普通文本、表格之类的
    doc.drawString(100, 600, "这是测试内容，你可以换成自己的业务数据")
    
    # 把PDF内容保存到缓冲区
    doc.save()
    # 把缓冲区指针移到开头，不然读取内容时会从结尾开始，导致PDF是空的
    buffer.seek(0)
    return buffer

第二步：用Streamlit下载按钮触发生成和下载 #

在你的主应用逻辑里，添加下载按钮，只有点击按钮时才会调用上面的生成函数，然后把缓冲区的内容作为下载文件返回给用户：

st.header("你的应用标题")

# 添加下载按钮，触发PDF生成和下载
if st.download_button(
    label="点击下载Results PDF",
    data=generate_pdf(),
    file_name="results.pdf",
    mime="application/pdf"
):
    st.success("PDF已生成，正在下载！")

为什么这样改就生效？ #

• 之前的代码会自动生成到本地，是因为你在页面加载时就执行了PDF生成逻辑，现在把这部分逻辑封装到函数里，只有按钮点击才会触发。
• 用io.BytesIO()内存缓冲区代替本地文件，生成的PDF全程在内存里流转，不会写到你的本地文件夹，直接通过下载按钮传给用户的浏览器。
• buffer.seek(0)这一步可不能忘，不然读取缓冲区内容时会从末尾开始，导致下载的PDF是空文件哦。

你可以把这段代码替换你现有的相关部分，测试一下，应该就能实现你想要的效果啦！

备注：内容来源于stack exchange，提问作者rhettley02