| from fpdf import FPDF |
| import pandas as pd |
|
|
| class ComplianceCert: |
| def __init__(self, csv_path): |
| self.df = pd.read_csv(csv_path) |
| |
| def _format_risk(self, trend): |
| return { |
| "↑": ("HIGH", (255,0,0)), |
| "→": ("MEDIUM", (255,165,0)), |
| "↓": ("LOW", (0,128,0)) |
| }.get(trend, ("UNKNOWN", (0,0,0))) |
|
|
| def generate_pdf(self, output_path="compliance_cert.pdf"): |
| pdf = FPDF() |
| pdf.add_page() |
| pdf.set_font("Arial", 'B', 16) |
| pdf.cell(0, 10, "Agentic AI Deployment Compliance Certificate", ln=1, align='C') |
| pdf.set_font("Arial", size=10) |
| col_widths = [40, 30, 30, 30, 40] |
| headers = ["Component", "Risk Trend", "Last Audit", "Automation", "Compliance"] |
| for i, header in enumerate(headers): |
| pdf.cell(col_widths[i], 10, header, border=1) |
| pdf.ln() |
| for _, row in self.df.iterrows(): |
| risk_text, risk_color = self._format_risk(row['Predicted_Risk_Trend']) |
| pdf.cell(col_widths[0], 10, row['Component'], border=1) |
| pdf.set_text_color(*risk_color) |
| pdf.cell(col_widths[1], 10, risk_text, border=1) |
| pdf.set_text_color(0,0,0) |
| pdf.cell(col_widths[2], 10, str(row['Last_Audit_Date']), border=1) |
| pdf.cell(col_widths[3], 10, f"Level {row['Automation_Level']}", border=1) |
| pdf.cell(col_widths[4], 10, row['Compliance_Frameworks'].split(',')[0], border=1) |
| pdf.ln() |
| pdf.output(output_path) |
|
|
| if __name__ == "__main__": |
| cert = ComplianceCert("master_plan_v3.2.csv") |
| cert.generate_pdf() |
|
|
