Create ai_technical_doc.py

tools/ai_technical_doc.py

@@ -0,0 +1,69 @@
+# tools/ai_technical_doc.py
+import datetime
+import re
+from fpdf import FPDF
+from langdetect import detect
+
+# === PDF Export Function ===
+def export_text_to_pdf(text, output_path=None, language="en"):
+    if output_path is None:
+        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+        output_path = f"technical_documentation_{timestamp}.pdf"
+
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_auto_page_break(auto=True, margin=15)
+
+    pdf.set_font("Arial", 'B', 16)
+    pdf.set_text_color(0, 51, 102)
+    title = "Technical Documentation - AI Act (Art. 11)" if language == "en" else "Documentation Technique - AI Act"
+    pdf.cell(0, 15, title, ln=True, align='C')
+    pdf.ln(10)
+
+    pdf.set_font("Arial", '', 12)
+    pdf.set_text_color(0, 0, 0)
+    for line in text.strip().split('\n'):
+        line = line.strip()
+        if line.startswith("## "):
+            section = line.replace("## ", "").strip()
+            pdf.set_font("Arial", 'B', 13)
+            pdf.set_text_color(30, 30, 120)
+            pdf.ln(8)
+            pdf.cell(0, 10, section, ln=True)
+            pdf.set_font("Arial", '', 12)
+            pdf.set_text_color(0, 0, 0)
+        elif line.startswith("- **"):
+            match = re.match(r"- \*\*(.+?)\*\*: (.+)", line)
+            if match:
+                label, value = match.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, value)
+        elif line == "---":
+            pdf.line(10, pdf.get_y(), 200, pdf.get_y())
+            pdf.ln(5)
+        else:
+            pdf.multi_cell(0, 10, line)
+    pdf.output(output_path)
+    return output_path
+
+
+# === Questions ===
+QUESTIONS = [
+    ("system_name", "What is the name of your AI system?"),
+    ("provider", "Who is the provider or developer of the system?"),
+    ("intended_purpose", "What is the intended purpose of the system?"),
+    ("architecture", "Describe the system architecture."),
+    ("training_data", "What kind of training data is used?"),
+    ("testing_methodology", "How was the system tested and validated?"),
+    ("performance_metrics", "What are the system's performance metrics?"),
+    ("risk_management", "What risk management measures were taken?"),
+    ("cybersecurity", "What cybersecurity measures are in place?"),
+    ("human_oversight", "How is human oversight implemented?"),
+    ("versioning", "How is version control maintained?"),
+    ("recordkeeping", "How are logs and records maintained?")
+]
+
+def get_questions():
+    return QUESTIONS