Update legal_document_analysis.py

legal_document_analysis.py

@@ -15,6 +15,7 @@ from fpdf import FPDF
 import getpass
 import pandas as pd
 import seaborn as sns
+import tempfile
 
 # Load environment variables from .env file
 load_dotenv()
@@ -337,6 +338,13 @@ def plot_risk_heatmap(detected_risks):
 def base64_to_image(data):
     return io.BytesIO(base64.b64decode(data))
 
+# Function to save base64 images to temporary files
+def save_base64_image(base64_data):
+    img_data = base64.b64decode(base64_data)
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
+        temp_file.write(img_data)
+        return temp_file.name
+
 # Function to generate PDF document with improved aesthetics
 def 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 = FPDF()
@@ -371,15 +379,15 @@ def generate_pdf_analysis(document_text, summary, detected_clauses, hidden_oblig
     pdf.ln(10)
 
     # Add visualizations for risks
-    pdf.image(base64_to_image(risk_assessment_matrix), x=10, y=pdf.get_y(), w=90)
-    pdf.image(base64_to_image(risk_level_distribution), x=110, y=pdf.get_y()-50, w=90)  # Position next to the first image
+    pdf.image(save_base64_image(risk_assessment_matrix), x=10, y=pdf.get_y(), w=90)
+    pdf.image(save_base64_image(risk_level_distribution), x=110, y=pdf.get_y()-50, w=90)  # Position next to the first image
     pdf.ln(60)
 
-    pdf.image(base64_to_image(risks_by_type), x=10, y=pdf.get_y(), w=90)
-    pdf.image(base64_to_image(stacked_bar_chart), x=110, y=pdf.get_y()-50, w=90)  # Position next to the previous image
+    pdf.image(save_base64_image(risks_by_type), x=10, y=pdf.get_y(), w=90)
+    pdf.image(save_base64_image(stacked_bar_chart), x=110, y=pdf.get_y()-50, w=90)  # Position next to the previous image
     pdf.ln(60)
 
-    pdf.image(base64_to_image(risk_heatmap), x=10, y=pdf.get_y(), w=190)  # Fit image to width
+    pdf.image(save_base64_image(risk_heatmap), x=10, y=pdf.get_y(), w=190)  # Fit image to width
     pdf.ln(10)
 
     # Footer
@@ -441,7 +449,6 @@ 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