Update legal_document_analysis.py

legal_document_analysis.py

@@ -394,12 +394,18 @@ def generate_pdf_analysis(document_text, summary, detected_clauses, hidden_oblig
 
     pdf.ln(10)
 
-    # Footer
-    pdf.set_y(-15)
-    pdf.set_font("Arial", 'I', 8)
-    pdf.cell(0, 10, f'Page {pdf.page_no()}', 0, 0, 'C')
+    # Save PDF to a temporary file
+    temp_pdf_path = "legal_document_analysis.pdf"
+    pdf.output(temp_pdf_path, 'F')
 
-    return pdf
+    # Load the PDF into a BytesIO object
+    with open(temp_pdf_path, "rb") as f:
+        pdf_buffer = io.BytesIO(f.read())
+
+    # Optionally, delete the temporary file
+    os.remove(temp_pdf_path)
+
+    return pdf_buffer
 
 # Function to handle chatbot interaction
 def chatbot_query(user_input):
@@ -604,9 +610,7 @@ def display_legal_analysis_page():
 
             # Download PDF Analysis Button
             st.subheader("Download Analysis as PDF")
-            pdf_buffer = io.BytesIO()
-            pdf = generate_pdf_analysis(document_text, summary, detected_clauses, hidden_obligations, detected_risks, risk_assessment_matrix, risk_level_distribution, risks_by_type, stacked_bar_chart, risk_heatmap)
-            pdf.output(pdf_buffer, 'F')
+            pdf_buffer = generate_pdf_analysis(document_text, summary, detected_clauses, hidden_obligations, detected_risks, risk_assessment_matrix, risk_level_distribution, risks_by_type, stacked_bar_chart, risk_heatmap)
             pdf_buffer.seek(0)
 
             # Add download button for PDF