Update tools/ai_act_generator.py

tools/ai_act_generator.py

@@ -1,82 +1,135 @@
 # NEW TOOL: Generate AI Act Compliance Register
 from smolagents import tool
 
-@tool
-def generate_ai_act_register(
-    organization_name: str,
-    responsible_person: str,
-    deployment_date: str,
-    ai_type: str,
-    ai_description: str,
-    risk_level: str,
-    risk_justification: str,
-    data_evaluation: str,
-    technical_docs: str,
-    human_oversight: str,
-    transparency_measures: str,
-    audit_frequency: str,
-    compliance_contact: str
-) -> str:
-    """
-    Génère un registre de conformité AI Act à partir des champs fournis.
-
-    Args:
-        organization_name: Le nom de l'organisation responsable du système IA.
-        responsible_person: Le nom de la personne référente.
-        deployment_date: La date de mise en service du système IA.
-        ai_type: Le type de système IA (ex: reconnaissance faciale, scoring).
-        ai_description: Une brève description du système IA.
-        risk_level: Le niveau de risque (ex: élevé, modéré).
-        risk_justification: La justification du niveau de risque déclaré.
-        data_evaluation: Les méthodes utilisées pour évaluer la qualité des données.
-        technical_docs: La documentation technique disponible.
-        human_oversight: Le niveau de supervision humaine prévu.
-        transparency_measures: Les mesures de transparence mises en œuvre.
-        audit_frequency: La fréquence prévue des audits du système.
-        compliance_contact: L'email ou contact de la personne en charge de la conformité.
-
-    Returns:
-        Un texte complet représentant le registre de conformité prêt à être exporté.
-    """
-    return f"""
-# Registre de Conformité – Système d’IA à Haut Risque
-
-## Informations générales
-
-- **Nom de l'organisation** : {organization_name}
-- **Responsable du système IA** : {responsible_person}
-- **Date de mise en service** : {deployment_date}
-- **Description du système IA** :  
-  {ai_description}
-
-## Catégorie de risque
-
-- **Type de système IA** : {ai_type}
-- **Risque identifié** : {risk_level}
-- **Justification de la classification** :  
-  {risk_justification}
-
-## Mesures de conformité appliquées
-
-- Évaluation des données utilisées :  
-  {data_evaluation}
-
-- Documentation technique produite :  
-  {technical_docs}
-
-- Contrôle humain prévu :  
-  {human_oversight}
-
-- Mécanismes de transparence intégrés :  
-  {transparency_measures}
-
-## Audit & révision
-
-- Fréquence de révision : {audit_frequency}
-- Responsable du suivi : {compliance_contact}
+#!/usr/bin/env python
+# coding=utf-8
+import mimetypes
+import os
+import re
+import shutil
+from typing import Optional
+
+from smolagents.agent_types import AgentAudio, AgentImage, AgentText, handle_agent_output_types
+from smolagents.agents import ActionStep, MultiStepAgent
+from smolagents.memory import MemoryStep
+from smolagents.utils import _is_package_available
+
+import gradio as gr
+from fpdf import FPDF
+from langdetect import detect
+
+# === PDF Export Function with Language Option ===
+def export_text_to_pdf(text, output_path="ai_act_register.pdf", language="fr"):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_auto_page_break(auto=True, margin=15)
+    pdf.set_font("Arial", size=12)
+
+    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.ln(10)
+    pdf.set_font("Arial", size=12)
+
+    for line in text.split('\n'):
+        pdf.multi_cell(0, 10, line)
+
+    pdf.output(output_path)
+    return output_path
+
+# === Sequential Questions ===
+QUESTIONS = [
+    ("organization_name", "What is the name of your organization?"),
+    ("responsible_person", "Who is responsible for this AI system?"),
+    ("deployment_date", "When is the AI system scheduled to be deployed?"),
+    ("ai_type", "What type of AI system is it?"),
+    ("ai_description", "Please briefly describe what the system does."),
+    ("risk_level", "What is the risk level of this system (e.g., high, medium)?"),
+    ("risk_justification", "Why do you consider it this risk level?"),
+    ("data_evaluation", "How have you evaluated the training data?"),
+    ("technical_docs", "What technical documentation is available?"),
+    ("human_oversight", "What kind of human oversight is planned?"),
+    ("transparency_measures", "What transparency mechanisms are in place?"),
+    ("audit_frequency", "How often will the system be audited?"),
+    ("compliance_contact", "Who is the contact person for compliance (email or name)?")
+]
+
+RESPONSES = {}
+
+# === Interactive Collection Flow ===
+def step_by_step_agent(user_input, state):
+    if state is None:
+        state = {"step": 0, "answers": {}}
+
+    step = state["step"]
+    answers = state["answers"]
+
+    # Save answer from previous step
+    if step > 0:
+        key, _ = QUESTIONS[step - 1]
+        answers[key] = user_input
+
+    # If more questions to ask
+    if step < len(QUESTIONS):
+        next_question = QUESTIONS[step][1]
+        state["step"] += 1
+        return next_question, state
+
+    # All questions answered, generate document
+    filled_template = f"""
+# AI Act Compliance Register
+
+## General Information
+- **Organization**: {answers['organization_name']}
+- **Responsible Person**: {answers['responsible_person']}
+- **Deployment Date**: {answers['deployment_date']}
+- **System Description**: {answers['ai_description']}
+
+## Risk Category
+- **Type**: {answers['ai_type']}
+- **Risk Level**: {answers['risk_level']}
+- **Justification**: {answers['risk_justification']}
+
+## Compliance Measures
+- **Data Evaluation**: {answers['data_evaluation']}
+- **Technical Docs**: {answers['technical_docs']}
+- **Human Oversight**: {answers['human_oversight']}
+- **Transparency Measures**: {answers['transparency_measures']}
+
+## Audit & Follow-up
+- **Audit Frequency**: {answers['audit_frequency']}
+- **Compliance Contact**: {answers['compliance_contact']}
 
 ---
+Generated by AI Act Assistant.
+"""
 
-> Document généré automatiquement selon le Règlement AI Act (UE).  
-> Ce modèle est à valider par un professionnel du droit selon le contexte d’usage.
-"""
+    pdf_path = export_text_to_pdf(filled_template, language="en")
+    return f"✅ All answers received. Download your AI Act compliance PDF below.", {"done": True, "pdf": pdf_path}
+
+# === UI for Step-by-Step Form ===
+def launch_step_by_step_ui():
+    with gr.Blocks(fill_height=True) as demo:
+        chatbot = gr.Chatbot()
+        msg = gr.Textbox(label="Your answer")
+        state = gr.State()
+        file_output = gr.File(visible=False)
+
+        def chat_logic(user_msg, state):
+            reply, state = step_by_step_agent(user_msg, state)
+            messages = [gr.ChatMessage(role="user", content=user_msg)]
+
+            if isinstance(reply, str):
+                messages.append(gr.ChatMessage(role="assistant", content=reply))
+
+            outputs = {"chatbot": messages, "state": state}
+            if isinstance(state, dict) and state.get("done"):
+                outputs["file_output"] = state["pdf"]
+            return outputs
+
+        msg.submit(chat_logic, [msg, state], [chatbot, state, file_output])
+
+    demo.launch()
+
+if __name__ == "__main__":
+    launch_step_by_step_ui()