Update app.py

app.py

@@ -77,11 +77,17 @@ def process_text(inputs):
 
 
 def save_as_pdf(text):
-    # convert quill editor HTML to PDF
+    # create a temporary file to write the HTML string
     with tempfile.NamedTemporaryFile(delete=False, suffix=".html") as tmpfile:
-        tmpfile.write(text.encode("utf-8"))
-        tmpfile.flush()
-        pdf_file = HTML(tmpfile.name).write_pdf()
+        # write the HTML string to the temporary file
+        tmpfile.write(text.encode())
+
+    # create a PDF file using the temporary HTML file and the CSS styles
+    options = {
+        "quiet": "",
+        "enable-local-file-access": ""
+    }
+    pdf_file = pdfkit.from_file(tmpfile.name, False, options=options)
 
     # encode the PDF to base64
     b64 = base64.b64encode(pdf_file).decode('utf-8')
@@ -92,6 +98,7 @@ def save_as_pdf(text):
 
 
 
+
 def form_page():
 
     st.markdown("### Your custom obituary writer :pencil:")
@@ -191,33 +198,46 @@ def editor_page():
     quill_text = st.session_state.output_text
     edited_text = st_quill(quill_text)
 
-    st.write("Here is the edited obituary:")
-    st.write(edited_text)
+    # get the CSS styles from the edited text
+    css = edited_text.get("css")
+    css += """
+        .ql-editor {
+            font-size: 14px;
+            line-height: 1.5;
+        }
+    """
 
+    st.write("Here is the edited obituary:")
+    st.markdown(edited_text.get("html"), unsafe_allow_html=True)
+    
+    #st.write(edited_text)
     st.session_state.edited_text = edited_text
 
+
+    #st.session_state.edited_text = edited_text
+
     if st.button("Save as PDF"):
         # Save the output text as a PDF
-        save_as_pdf(st.session_state.edited_text)
+        save_as_pdf(edited_text.get("html") + f"<style>{css}</style>")
         st.write("The custom obituary has been saved as a PDF.")
 
     # Add some custom CSS to style the editor
-    st.markdown("""
+    st.markdown(f"""
         <style>
-            #toolbar {
+            #toolbar {{
                 background-color: #f3f3f3;
                 border-radius: 5px;
                 padding: 5px;
-            }
-            .ql-container {
+            }}
+            .ql-container {{
                 border-radius: 5px;
                 border: 1px solid #ccc;
                 height: 400px;
-            }
-            .ql-editor {
+            }}
+            .ql-editor {{
                 height: 100%;
-            }
-            {edited_text.get('css')}
+                {css}
+            }}
         </style>
     """, unsafe_allow_html=True)