Update legal_document_analysis.py

legal_document_analysis.py

@@ -372,16 +372,26 @@ def generate_pdf_analysis(document_text, summary, detected_clauses, hidden_oblig
         pdf.cell(0, 10, f"{risk['phrase']}: {risk['summary']} (Risk Level: {risk['risk_level']})", ln=True)
     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.ln(60)
+    # Save images temporarily and add them to the PDF
+    def save_base64_image(image_str, filename):
+        with open(filename, "wb") as img_file:
+            img_file.write(base64.b64decode(image_str))
+
+    # Save images
+    image_filenames = [
+        "risk_assessment_matrix.png",
+        "risk_level_distribution.png",
+        "risks_by_type.png",
+        "stacked_bar_chart.png",
+        "risk_heatmap.png"
+    ]
 
-    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.ln(60)
+    images = [risk_assessment_matrix, risk_level_distribution, risks_by_type, stacked_bar_chart, risk_heatmap]
+    
+    for img_str, filename in zip(images, image_filenames):
+        save_base64_image(img_str, filename)
+        pdf.image(filename, x=10, y=pdf.get_y(), w=90)  # Use the saved image file
 
-    pdf.image(base64_to_image(risk_heatmap), x=10, y=pdf.get_y(), w=190)  # Fit image to width
     pdf.ln(10)
 
     # Footer
@@ -456,7 +466,7 @@ def send_pdf_via_email(pdf_buffer, recipient_email):
     pdf_buffer.seek(0)
     pdf_attachment.write(pdf_buffer.read())
     pdf_attachment.seek(0)
-    
+
     part = MIMEApplication(pdf_attachment.read(), Name='legal_document_analysis.pdf')
     part['Content-Disposition'] = 'attachment; filename="legal_document_analysis.pdf"'
     msg.attach(part)