Spaces:

amoldwalunj
/

streamlit_custom_obituary

Runtime error

amoldwalunj commited on Mar 22, 2023

Commit

54b4c8b

•

1 Parent(s): 98a7c2c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -95,18 +95,15 @@ def save_as_pdf(text):
         'margin-bottom': '0.75in',
         'margin-left': '0.75in',
     }
-    try:
-        pdfkit.from_string(text, pdf_file, options=options)
-    except OSError as e:
-        st.error("Error generating PDF. Please check if wkhtmltopdf is installed.")
-        return
     # encode the PDF to base64
     b64 = base64.b64encode(pdf_file.getvalue()).decode('utf-8')
-    # generate a download link for the PDF
-    href = f'<a href="data:application/octet-stream;base64,{b64}" download="output.pdf">Download PDF</a>'
-    st.markdown(href, unsafe_allow_html=True)
@@ -224,11 +221,15 @@ def editor_page():
         new_html = f"""<html><head><style>{css}</style></head><body>{soup}</body></html>"""
         # Save the output text as a PDF
-        try:
-            save_as_pdf(new_html)
-            st.write("The custom obituary has been saved as a PDF.")
-        except Exception as e:
-            st.error("Error saving PDF. " + str(e))
     # Add some custom CSS to style the editor
     st.markdown("""

         'margin-bottom': '0.75in',
         'margin-left': '0.75in',
     }
+    pdfkit.from_string(text, pdf_file, options=options)
     # encode the PDF to base64
     b64 = base64.b64encode(pdf_file.getvalue()).decode('utf-8')
+    # generate the filename
+    filename = 'output.pdf'
+    return b64, filename
         new_html = f"""<html><head><style>{css}</style></head><body>{soup}</body></html>"""
         # Save the output text as a PDF
+        pdf_data, filename = save_as_pdf(new_html)
+        # display a download button for the PDF file
+        st.download_button(
+            label='Download PDF',
+            data=pdf_data,
+            file_name=filename,
+            mime='application/pdf'
+        )
     # Add some custom CSS to style the editor
     st.markdown("""