Plotly Chart to PDF in Streamlit using st.download_button without saving the chart as HTML?

🎈 Using Streamlit
1

Just wondering if someone has worked this one out ?

2

Hi @horasio! :wave:

Have you tried adding pdfkit?

Best,
Charly

3

Did you have any issues using the method exeplained in the documentation

4

import plotly.io as pio
import plotly.express as pe
fig = pe.scatter(df, x = X, y = Y)
pio.write_image(fig, ‘Chart.pdf’, format=‘pdf’)

#please not fig is the chart that Plotly has produced.

You can put

pio.write_image(fig, ‘Chart.pdf’, format=‘pdf’)

inside st.button. But I want to put that line inside st.download_button so that the file gets downloaded to default download folder of the browser.

5

write_image allows you to write to a file-like object. You can write to an instance of io.BytesIO.

import io

f = io.BytesIO()
pio.write_image(fig, f, format="pdf")

And then pass its contents to download_button.

st.download_button("Download PDF", data=f.getvalue())

Alternatively you can use to_image that will return the bytes right away.

6

to_image

is the one I went with as it doesn’t download anything natively and where st.download_button works well

Thank you

7

check this post out !

hope it help !