Upload app.py

app.py

@@ -11,7 +11,8 @@ from langchain_community.document_loaders import PyPDFLoader
 from langchain.vectorstores import FAISS
 from langchain_openai import OpenAIEmbeddings
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-import pdfkit  # Added import for pdfkit
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
 
 # Function to process PDF, run Q&A, and return results
 def process_pdf(api_key, uploaded_file, questions_path, prompt_path, display_placeholder):
@@ -79,6 +80,25 @@ def process_pdf(api_key, uploaded_file, questions_path, prompt_path, display_pla
 
     return qa_results
 
+# Function to create a PDF using reportlab
+def create_pdf(content):
+    buffer = BytesIO()
+    pdf = canvas.Canvas(buffer, pagesize=letter)
+    pdf.setFont("Helvetica", 10)
+    
+    # Start position for writing text
+    text = pdf.beginText(40, 750)
+
+    for line in content.split("\n"):
+        text.textLine(line)
+    
+    pdf.drawText(text)
+    pdf.showPage()
+    pdf.save()
+
+    buffer.seek(0)
+    return buffer
+
 # Streamlit app layout
 st.title("Climate Policy Summary Tool")
 
@@ -106,9 +126,8 @@ if st.button("Generate") and api_key and uploaded_file:
             st.download_button("Download as Markdown", markdown_output, file_name="results.md")
 
             # Create a PDF file for the user to download
-            html_content = markdown_output.replace("### ", "<h3>").replace("**", "<b>").replace("\n", "<br>")
-            pdf_output = pdfkit.from_string(html_content, False)
-            st.download_button("Download as PDF", pdf_output, file_name="results.pdf")
+            pdf_output = create_pdf(markdown_output)
+            st.download_button("Download as PDF", data=pdf_output, file_name="results.pdf")
 
         except Exception as e:
             st.error(f"An error occurred: {e}")