Update legal_document_analysis.py

legal_document_analysis.py

@@ -395,7 +395,12 @@ 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 PDF to a BytesIO object
+    pdf_buffer = io.BytesIO()
+    pdf.output(pdf_buffer, 'F')  # Save to BytesIO
+    pdf_buffer.seek(0)
+
+    return pdf_buffer
 
 # Function to handle chatbot interaction
 def chatbot_query(user_input):
@@ -449,6 +454,7 @@ def send_feedback(feedback_content):
 def send_pdf_via_email(pdf_buffer, recipient_email):
     sender_email = os.getenv("SENDER_EMAIL")
     password = os.getenv("EMAIL_PASS")
+
     msg = MIMEMultipart()
     msg['From'] = sender_email
     msg['To'] = recipient_email
@@ -597,10 +603,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.seek(0)
+            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)
 
             # Add download button for PDF
             st.download_button(