Update app.py

app.py

@@ -77,23 +77,23 @@ def process_text(inputs):
 #     st.markdown(href, unsafe_allow_html=True)
 
 
-def save_as_pdf(text):
-    # create a temporary HTML file
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".html") as tmpfile:
-        tmpfile.write(text.encode())
-        tmpfile.flush()
-
-        # generate a PDF from the HTML
-        pdf_file = BytesIO()
-        weasyprint.HTML(tmpfile.name).write_pdf(pdf_file)
-
-        # 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)
+import tempfile
+import weasyprint
+from bs4 import BeautifulSoup
 
+def save_as_pdf(html_string):
+    # Parse HTML string into BeautifulSoup object
+    soup = BeautifulSoup(html_string, 'html.parser')
+    # Generate temporary file name with .html extension
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".html") as tmpfile:
+        # Write the prettified HTML string to the temporary file
+        tmpfile.write(soup.prettify().encode())
+    # Generate temporary file name with .pdf extension
+    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