Update app.py

app.py

@@ -85,16 +85,19 @@ from bs4 import BeautifulSoup
 import pdfkit
 
 def save_as_pdf(text):
-    # save the quill editor text to a temporary HTML file
-    with open('temp.html', 'w') as f:
-        f.write(text)
-
-    # convert the HTML file to a PDF with pdfkit
-    pdfkit.from_file('temp.html', 'output.pdf')
+    # convert quill editor HTML to PDF
+    pdf_file = BytesIO()
+    options = {
+        'page-size': 'Letter',
+        'margin-top': '0.75in',
+        'margin-right': '0.75in',
+        'margin-bottom': '0.75in',
+        'margin-left': '0.75in',
+    }
+    pdfkit.from_string(text, pdf_file, options=options)
 
     # encode the PDF to base64
-    with open('output.pdf', 'rb') as f:
-        b64 = base64.b64encode(f.read()).decode('utf-8')
+    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>'
@@ -202,13 +205,20 @@ def editor_page():
     quill_text = st.session_state.output_text
     edited_text = st_quill(quill_text)
 
+    # Extract the HTML and CSS content from the quill editor
+    soup = BeautifulSoup(edited_text['html'], 'html.parser')
+    style_tag = soup.new_tag('style')
+    style_tag.string = edited_text['css']
+    soup.head.append(style_tag)
+    html_content = str(soup)
+
     st.write("Here is the edited obituary:")
 
-    st.session_state.edited_text = edited_text.get('html')
+    st.session_state.edited_text = html_content
 
     if st.button("Save as PDF"):
         # Save the output text as a PDF
-        save_as_pdf(edited_text.get('html'))
+        save_as_pdf(html_content)
         st.write("The custom obituary has been saved as a PDF.")
 
     # Add some custom CSS to style the editor
@@ -227,7 +237,6 @@ def editor_page():
             .ql-editor {
                 height: 100%;
             }
-            {edited_text.get('css')}
         </style>
     """, unsafe_allow_html=True)