Update Gradio_UI.py

Gradio_UI.py

@@ -19,7 +19,7 @@ DOCUMENT_TO_MODULE = {
     "GDPR": {
         "Data Processing Record": "tools.gdpr_data_record",
         "DPIA (Data Protection Impact Assessment)": "tools.gdpr_dpia",
-         "Data Breach Notification Log": "tools.gdpr_breach_log",
+        "Data Breach Notification Log": "tools.gdpr_breach_log",
         "Subject Access Request Record": "tools.gdpr_sar_log",
         "Consent Management Log": "tools.gdpr_consent_log",
         "Data Processor Agreements": "tools.gdpr_dpa_log"
@@ -38,7 +38,7 @@ def export_pdf(title, answers, lang="en"):
     pdf.cell(0, 10, title, ln=True, align='C')
     pdf.ln(8)
 
-    # Metadata section
+    # Metadata
     company = answers.get("What is the name of your organization?", "N/A")
     name = answers.get("What is your name?", "N/A")
     role = answers.get("What is your role?", "N/A")
@@ -51,7 +51,7 @@ def export_pdf(title, answers, lang="en"):
     pdf.cell(0, 8, f"Timestamp: {timestamp}", ln=True, align='C')
     pdf.ln(10)
 
-    # Main content
+    # Answers
     pdf.set_font("Arial", '', 12)
     pdf.set_text_color(0, 0, 0)
     for label, value in answers.items():
@@ -95,17 +95,16 @@ def continue_conversation(user_input):
     step = conversation_state["step"]
 
     if step < len(conversation_state["questions"]):
-    key, question = conversation_state["questions"][step]
-    conversation_state["answers"][question] = user_input
-    conversation_state["step"] += 1
-
-    try:
-        if user_input and len(user_input.strip()) > 3:
-            conversation_state["lang"] = detect(user_input)
-    except:
-        conversation_state["lang"] = "en"
-
+        key, question = conversation_state["questions"][step]
+        conversation_state["answers"][question] = user_input
+        conversation_state["step"] += 1
 
+        # Language detection (safe fallback)
+        try:
+            if user_input and len(user_input.strip()) > 3:
+                conversation_state["lang"] = detect(user_input)
+        except:
+            conversation_state["lang"] = "en"
 
     if conversation_state["step"] < len(conversation_state["questions"]):
         next_question = conversation_state["questions"][conversation_state["step"]][1]
@@ -135,7 +134,6 @@ with gr.Blocks(
     }
     """
 ) as demo:
-    # GDPR Notice
     gr.Markdown("""
     ### 🔒 GDPR Notice  
     This assistant does not store any personal data.