Update Gradio_UI.py

Gradio_UI.py

@@ -12,6 +12,8 @@ from smolagents.memory import MemoryStep
 from smolagents.utils import _is_package_available
 
 import gradio as gr
+import csv
+import datetime
 from fpdf import FPDF
 from langdetect import detect
 
@@ -51,8 +53,6 @@ QUESTIONS = [
     ("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:
@@ -98,15 +98,13 @@ def step_by_step_agent(user_input, state):
 Generated by AI Act Assistant.
 """
 
-    import csv
-import datetime
-
     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"]
@@ -120,13 +118,14 @@ import datetime
         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}
+    return f"{flag} Language detected: {detected_lang.upper()}\n✅ 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:
+        reset_csv = gr.Button("🗑️ Reset CSV File")
         csv_download = gr.File(value="ai_act_registers.csv", label="📥 Download All Registers (.csv)", visible=os.path.exists("ai_act_registers.csv"))
+
         initial_message = gr.ChatMessage(role="assistant", content=(
             "👋 Welcome! I will guide you through the AI Act compliance form.\n"
             "Let's begin with a few questions to generate your compliance register.\n"
@@ -151,8 +150,14 @@ def launch_step_by_step_ui():
         def restart_conversation():
             return [initial_message], {"step": 0, "answers": {}}, None
 
+        def delete_csv():
+            if os.path.exists("ai_act_registers.csv"):
+                os.remove("ai_act_registers.csv")
+            return None
+
         msg.submit(chat_logic, [msg, state], [chatbot, state, file_output])
         restart_button.click(restart_conversation, outputs=[chatbot, state, file_output])
+        reset_csv.click(delete_csv, outputs=[])
 
     demo.launch()