Update legal_document_analysis.py

legal_document_analysis.py

@@ -6,7 +6,6 @@ from langchain_groq import ChatGroq
 from docx import Document
 import matplotlib.pyplot as plt
 import io
-import base64
 import tempfile
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
@@ -369,7 +368,11 @@ def generate_pdf_analysis(document_text, summary, detected_clauses, hidden_oblig
     pdf.set_font("Arial", 'I', 8)
     pdf.cell(0, 10, f'Page {pdf.page_no()}', 0, 0, 'C')
 
-    return pdf
+    # Save the PDF to a temporary file
+    pdf_file_path = tempfile.mktemp(suffix=".pdf")
+    pdf.output(pdf_file_path, 'F')
+
+    return pdf_file_path  # Return the path to the saved PDF
 
 # Function to handle chatbot interaction
 def chatbot_query(user_input):
@@ -572,9 +575,13 @@ def display_legal_analysis_page():
 
             # Download PDF Analysis Button
             st.subheader("Download Analysis as PDF")
+            pdf_file_path = generate_pdf_analysis(document_text, summary, detected_clauses, hidden_obligations, detected_risks, risk_assessment_matrix_path, risk_level_distribution_path, risks_by_type_path, stacked_bar_chart_path, risk_heatmap_path)
+
+            # Read PDF into BytesIO for download
             pdf_buffer = io.BytesIO()
-            pdf = generate_pdf_analysis(document_text, summary, detected_clauses, hidden_obligations, detected_risks, risk_assessment_matrix_path, risk_level_distribution_path, risks_by_type_path, stacked_bar_chart_path, risk_heatmap_path)
-            pdf.output(pdf_buffer, 'F')
+            with open(pdf_file_path, 'rb') as f:
+                pdf_buffer.write(f.read())
+            
             pdf_buffer.seek(0)
 
             # Add download button for PDF