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()