Spaces:

amoldwalunj
/

streamlit_custom_obituary

Runtime error

App Files Files Community

amoldwalunj commited on Mar 22, 2023

Commit

f4e74e3

1 Parent(s): 4be8c4a

Update app.py

Browse files

Files changed (1) hide show

app.py +10 -30

app.py CHANGED Viewed

@@ -92,13 +92,8 @@ 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)
-        # 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)
@@ -207,29 +202,14 @@ def editor_page():
     st.session_state.edited_text = edited_text
-    if st.button("Save as PDF"):
-        # Save the output text as a PDF
-        save_as_pdf(f"<html><head><style>{edited_text}</style></head><body>{edited_text}</body></html>")
-        st.write("The custom obituary has been saved as a PDF.")
-    # Add some custom CSS to style the editor
-    st.markdown("""
-        <style>
-            #toolbar {
-                background-color: #f3f3f3;
-                border-radius: 5px;
-                padding: 5px;
-            }
-            .ql-container {
-                border-radius: 5px;
-                border: 1px solid #ccc;
-                height: 400px;
-            }
-            .ql-editor {
-                height: 100%;
-            }
-        </style>
-    """, unsafe_allow_html=True)

     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
     st.session_state.edited_text = edited_text
+    if st.button("Download as PDF"):
+        # Save edited text as HTML string
+        html_string = f"<html><head><style>{edited_text.get('css', '')}</style></head><body>{edited_text.get('content', '')}</body></html>"
+        # Save HTML string as PDF file and generate download link
+        with st.spinner("Generating PDF..."):
+            pdf_file = save_as_pdf(html_string)
+        st.markdown(f'<a href="{pdf_file}" download="edited_text.pdf">Download PDF</a>', unsafe_allow_html=True)