Update app.py

app.py

@@ -55,28 +55,25 @@ def process_text(inputs):
 # #from streamlit_quill import Delta, Renderer
 # from quill.delta import Delta
 
-import base64
-from io import BytesIO
-from delta import Delta, html
-from weasyprint import HTML
-
 def save_as_pdf(text):
-    # convert quill editor Delta object to HTML
-    delta = Delta(text)
-    html_str = html.render(delta)
-    
-    # convert HTML to PDF
+    # convert quill editor HTML to PDF
     pdf_file = BytesIO()
-    HTML(string=html_str).write_pdf(pdf_file)
-    
+    HTML(string=text).write_pdf(pdf_file, stylesheets=[CSS(string="""
+        .ql-editor {
+            font-size: 16px;
+            line-height: 1.6;
+        }
+        """)], style_tags=True)
+
     # 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)
 
 
+
 def form_page():
 
     st.markdown("### Your custom obituary writer :pencil:")
@@ -171,23 +168,16 @@ def editor_page():
 
     st.write("Use the editor below to edit the obituary:")
 
-    # Get the initial text to be edited
-    initial_text = st.session_state.output_text
-
-    # Create a Delta object from the edited text
-    edited_delta = Delta(st.text_input("Edited text", value=initial_text))
-
-    # Render the Delta object using the Renderer class
-    edited_html = Renderer().render(edited_delta)
+    quill_text = st.session_state.edited_text
+    edited_text = st_quill(quill_text)
 
     st.write("Here is the edited obituary:")
-    st.write(edited_html, unsafe_allow_html=True)
+    st.write(edited_text)
 
-    # Store the edited text in session state
-    st.session_state.edited_text = edited_html
+    st.session_state.edited_text = edited_text
 
     if st.button("Save as PDF"):
-        # Save the edited text as a PDF
+        # Save the output text as a PDF
         save_as_pdf(st.session_state.edited_text)
         st.write("The custom obituary has been saved as a PDF.")
 
@@ -210,6 +200,7 @@ def editor_page():
         </style>
     """, unsafe_allow_html=True)
 
+