Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import io
 import openai
 import datetime
 import time
@@ -6,9 +7,9 @@ import gradio as gr
 import json
 from jinja2 import Template
 import requests
-#from pdfminer.high_level import extract_text
 import fitz
-import pdfkit
+from xhtml2pdf import pisa
+from io import BytesIO
 
 # Initialize OpenAI
 openai.api_key = os.environ.get('OPENAI_API_KEY')
@@ -346,20 +347,33 @@ def pdf_to_text(contract_file_cmpt, contract_text_tbox, file_name_tbox):
 
     return file_text, redline_file_name
 
+def convert_html_to_pdf(source_html, output_filename):
+    # Result file stream
+    result_file = open(output_filename, "w+b")
+
+    # Convert HTML to PDF
+    pisa_status = pisa.CreatePDF(
+        BytesIO(source_html.encode("UTF-8")),  # HTML content
+        dest=result_file)  # File handle to receive the result
+
+    # Close the result file
+    result_file.close()
+
+    # Return True on success and False on errors
+    return pisa_status.err
+
+
 def download_pdf(compliance_comments,contract_redline_html,file_name_tbox):
-    
-    #config = pdfkit.configuration(wkhtmltopdf="/tmp/pip-ephem-wheel-cache-4svwmn1h/wheels/c4/fe/24/47a8dabc269dcd143551749339e4d8bbc2ca2002ee03943027")
-    #contract_redline_comments = "<h2><u>Contract Redline:</u></h2><br>"+contract_redline_html + "<br><h2><u>Compliance Comments:</u></h2><br>"+compliance_comments
-    global pdf_download
-    #pdf_download = pdfkit.from_string(contract_redline_comments,None,configuration=config)
-    with open('NILI Contract Redline.pdf', 'rb') as file:
-        pdf_download = file.read()
-
-    #with open("Champion Contract Test.pdf", 'wb') as file:
-        #file.write(pdf_download)
-    
-    #return {pdf_download_file: file_name_tbox}
-    return {pdf_download_file: "NILI Contract Redline.pdf"}
+    #config = pdfkit.configuration(wkhtmltopdf="/usr/local/bin/wkhtmltopdf")
+
+    contract_redline_comments = "<h2><u>Contract Redline:</u></h2><br>"+contract_redline_html + "<br><h2><u>Compliance Comments:</u></h2><br>"+compliance_comments
+
+    #global pdf_download
+    #pdf_download = pdfkit.from_string(contract_redline_comments,file_name_tbox,configuration=config)
+
+    convert_html_to_pdf(contract_redline_comments, file_name_tbox)
+
+    return {pdf_download_file: file_name_tbox}
 
 # Gradio UI
 CIMStheme = gr.themes.Soft().set(