Update app.py

app.py

@@ -85,39 +85,21 @@ from bs4 import BeautifulSoup
 import pdfkit
 
 def save_as_pdf(text):
-    # Save the edited text as a temporary HTML file
-    with open("temp.html", "w", encoding="utf-8") as f:
+    # save the quill editor text to a temporary HTML file
+    with open('temp.html', 'w') as f:
         f.write(text)
 
-    # Configure options for PDF generation
-    options = {
-        "page-size": "Letter",
-        "margin-top": "0.75in",
-        "margin-right": "0.75in",
-        "margin-bottom": "0.75in",
-        "margin-left": "0.75in",
-        "encoding": "UTF-8",
-        "no-outline": None
-    }
-
-    # Generate the PDF from the HTML file
-    pdfkit.from_file("temp.html", "output.pdf", options=options)
-
-    # Read the generated PDF file
-    with open("output.pdf", "rb") as f:
-        pdf_data = f.read()
+    # convert the HTML file to a PDF with pdfkit
+    pdfkit.from_file('temp.html', 'output.pdf')
 
-    # Encode the PDF data to base64
-    b64 = base64.b64encode(pdf_data).decode("utf-8")
+    # encode the PDF to base64
+    with open('output.pdf', 'rb') as f:
+        b64 = base64.b64encode(f.read()).decode('utf-8')
 
-    # Generate a download link for the PDF
+    # 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)
 
-    # Delete the temporary HTML and PDF files
-    os.remove("temp.html")
-    os.remove("output.pdf")
-
 
 
 
@@ -221,14 +203,34 @@ def editor_page():
     edited_text = st_quill(quill_text)
 
     st.write("Here is the edited obituary:")
-    #st.write(edited_text)
 
-    st.session_state.edited_text = edited_text
+    st.session_state.edited_text = edited_text.get('html')
 
     if st.button("Save as PDF"):
-        # Save the edited text as a PDF
-        save_as_pdf(edited_text.get("content"))
+        # Save the output text as a PDF
+        save_as_pdf(edited_text.get('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%;
+            }
+            {edited_text.get('css')}
+        </style>
+    """, unsafe_allow_html=True)
+