Update tools/ai_act_generator.py

tools/ai_act_generator.py

@@ -22,19 +22,56 @@ def export_text_to_pdf(text, output_path=None, language="fr"):
     if output_path is None:
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         output_path = f"ai_act_register_{timestamp}.pdf"
+
     pdf = FPDF()
     pdf.add_page()
     pdf.set_auto_page_break(auto=True, margin=15)
-    pdf.set_font("Arial", size=12)
 
+    # Title
+    pdf.set_font("Arial", 'B', 16)
+    pdf.set_text_color(0, 51, 102)
     title = "AI Act Compliance Register" if language == "en" else "Registre de Conformité AI Act"
-    pdf.set_font("Arial", 'B', 14)
-    pdf.cell(0, 10, title, ln=True)
+    pdf.cell(0, 15, title, ln=True, align='C')
     pdf.ln(10)
-    pdf.set_font("Arial", size=12)
 
-    for line in text.split('\n'):
-        pdf.multi_cell(0, 10, line)
+    # Subtitle
+    pdf.set_font("Arial", 'I', 12)
+    pdf.set_text_color(90, 90, 90)
+    subtitle = f"Generated on {datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+    pdf.cell(0, 10, subtitle, ln=True, align='C')
+    pdf.ln(5)
+
+    # Reset style for body
+    pdf.set_font("Arial", '', 12)
+    pdf.set_text_color(0, 0, 0)
+
+    # Parse and format sections
+    for line in text.strip().split('\n'):
+        line = line.strip()
+        if line.startswith("## "):
+            section_title = line.replace("## ", "").strip()
+            pdf.set_font("Arial", 'B', 13)
+            pdf.set_text_color(30, 30, 120)
+            pdf.ln(8)
+            pdf.cell(0, 10, section_title, ln=True)
+            pdf.ln(2)
+            pdf.set_font("Arial", '', 12)
+            pdf.set_text_color(0, 0, 0)
+        elif line.startswith("- **"):
+            key_value = re.match(r"- \*\*(.+?)\*\*: (.+)", line)
+            if key_value:
+                label, answer = key_value.groups()
+                pdf.set_font("Arial", 'B', 12)
+                pdf.cell(0, 10, f"{label}:", ln=True)
+                pdf.set_font("Arial", '', 12)
+                pdf.multi_cell(0, 10, f"{answer}")
+                pdf.ln(2)
+        elif line.strip() == "---":
+            pdf.line(10, pdf.get_y(), 200, pdf.get_y())
+            pdf.ln(5)
+        else:
+            pdf.multi_cell(0, 10, line)
+            pdf.ln(2)
 
     pdf.output(output_path)
     return output_path