Update tools/ai_act_generator.py

tools/ai_act_generator.py

@@ -1,8 +1,9 @@
 # NEW TOOL: Generate AI Act Compliance Register
 from smolagents import tool
-
-#!/usr/bin/env python
+c#!/usr/bin/env python
 # coding=utf-8
+import csv
+import datetime
 import mimetypes
 import os
 import re
@@ -64,18 +65,15 @@ def step_by_step_agent(user_input, state):
     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
+        return next_question, state, None  # Ensure return matches expected 3 outputs
 
-    # All questions answered, generate document
     filled_template = f"""
 # AI Act Compliance Register
 
@@ -104,30 +102,62 @@ def step_by_step_agent(user_input, state):
 Generated by AI Act Assistant.
 """
 
-    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}
+    
+
+    try:
+        detected_lang = detect(filled_template)
+    except:
+        detected_lang = "en"
+
+    flag = "🇬🇧" if detected_lang == "en" else "🇫🇷"
+    # Save to CSV
+    csv_file = "ai_act_registers.csv"
+    fieldnames = [key for key, _ in QUESTIONS] + ["timestamp"]
+    row_data = {**answers, "timestamp": datetime.datetime.now().isoformat()}
+
+    file_exists = os.path.isfile(csv_file)
+    with open(csv_file, mode="a", newline="", encoding="utf-8") as csvfile:
+        writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
+        if not file_exists:
+            writer.writeheader()
+        writer.writerow(row_data)
+
+    pdf_path = export_text_to_pdf(filled_template, language=detected_lang)
+    return f"{flag} Language detected: {detected_lang.upper()}
+✅ All answers received. Download your AI Act compliance PDF below.", {"done": True, "pdf": pdf_path}, 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()
+        initial_question = QUESTIONS[0][1]
+        initial_message = gr.ChatMessage(role="assistant", content="""
+👋 Welcome! I will guide you through the AI Act compliance form.
+Let's begin with a few questions to generate your compliance register.
+
+What is the name of your organization?
+""")
+        stored_messages = gr.State([initial_message])
+        chatbot = gr.Chatbot(type="messages", value=stored_messages)
         msg = gr.Textbox(label="Your answer")
         state = gr.State()
         file_output = gr.File(visible=False)
+        restart_button = gr.Button("🔁 Restart")
 
         def chat_logic(user_msg, state):
-            reply, state = step_by_step_agent(user_msg, state)
+            reply, updated_state, file_path = 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
+            file_path = file_path or ""
+            return messages, updated_state, file_path
+
+        def restart_conversation():
+            return [initial_message], {"step": 0, "answers": {}}, None
 
         msg.submit(chat_logic, [msg, state], [chatbot, state, file_output])
+        restart_button.click(restart_conversation, outputs=[chatbot, state, file_output])
 
     demo.launch()