Update app.py

app.py

@@ -92,9 +92,13 @@ def save_as_pdf(html_string):
     with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as pdffile:
         # Convert the HTML file to a PDF file using WeasyPrint
         weasyprint.HTML(tmpfile.name).write_pdf(pdffile)
-        # Return the file name of the PDF file
-        return pdffile.name
-
+        # Read the PDF file contents and encode as a base64 string
+        pdf_contents = pdffile.read()
+        b64_pdf = base64.b64encode(pdf_contents).decode()
+        # Generate download link for the PDF file
+        href = f'<a href="data:application/octet-stream;base64,{b64_pdf}" download="document.pdf">Download PDF</a>'
+        # Display the download link using st.write
+        st.write(href, unsafe_allow_html=True)