Update app.py

app.py

@@ -5,8 +5,9 @@ import os
 import pickle
 import torch
 import markdown
-from xhtml2pdf import pisa
+from weasyprint import HTML
 import io
+from io import BytesIO
 from grobidmonkey import reader
 
 from transformers import pipeline
@@ -175,29 +176,21 @@ def create_pdf_from_markdown_strings(markdown_strings):
 
     return html_pages
 
-def html_to_pdf(html_content):
-    # Create a temporary file
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
-        pdfkit.from_string(html_content, tmp_file.name)
-        return tmp_file.name
-
-    # Convert combined HTML to PDF directly into a file
-    pdfkit.from_string(combined_html, output_file, options=options)
+def generate_pdf(html):
+    # Create a PDF from the HTML string
+    pdf_file = BytesIO()
+    HTML(string=html).write_pdf(pdf_file)
+    pdf_file.seek(0)
+    return pdf_file
 
 html_content = create_pdf_from_markdown_strings(st.session_state.summ_text)
-
-
-if html_content:
-    pdf_path = html_to_pdf(html_content)
+pdf_file = generate_pdf(html_content)
     
-    with open(pdf_path, "rb") as pdf_file:
-        st.download_button(
-            label="Download PDF",
-            data=pdf_file,
-            file_name="converted.pdf",
-            mime="application/pdf"
-        )
-
-    # Remove the temporary file
-    os.remove(pdf_path)
+# Provide download link
+st.download_button(
+    label="Download PDF",
+    data=pdf_file,
+    file_name="sample.pdf",
+    mime="application/pdf"
+)