Update Gradio_UI.py

Gradio_UI.py

@@ -4,6 +4,7 @@ from langdetect import detect
 import time
 import importlib
 
+# === Map document choices to modules ===
 DOCUMENT_TO_MODULE = {
     "AI Act": {
         "Compliance Register": "tools.ai_act_generator",
@@ -46,24 +47,40 @@ DOCUMENT_TO_MODULE = {
         "System Testing Record": "tools.nis2_system_testing",
         "Third-Party Access Review": "tools.nis2_third_party_access_review",
         "Vulnerability Management Log": "tools.nis2_vulnerability_log"
+    },
+    "DORA": {
+        "Third-Party ICT Risk Register": "tools.nis2_third_party_access_review",
+        "ICT Asset Inventory": "tools.nis2_asset_inventory",
+        "Incident Reporting Log": "tools.nis2_incident_log",
+        "Risk Register": "tools.nis2_risk_register",
+        "Backup & Recovery": "tools.nis2_backup_recovery_log",
+        "Penetration Testing Summary": "tools.nis2_penetration_test_summary",
+        "Disaster Recovery Plan": "tools.nis2_disaster_recovery_plan",
+        "Security Testing Strategy": "tools.nis2_security_testing_strategy",
+        "ICT Risk Management Policy": "tools.dora_risk_management_policy",
+        "ICT Change Management Log": "tools.dora_change_management_log",
+        "ICT Incident Communication Strategy": "tools.dora_incident_communication_strategy",
+        "ICT Service Continuity Plan": "tools.dora_service_continuity_plan",
+        "ICT Third-Party Contract Checklist": "tools.dora_third_party_contract_checklist",
+        "ICT Security Policy": "tools.dora_security_policy",
+        "DORA Self-Assessment Checklist": "tools.dora_self_assessment",
+        "ICT Training Record": "tools.dora_training_record",
+        "Data Localization & Sovereignty Log": "tools.dora_data_localization_log",
+        "ICT Business Impact Assessment": "tools.dora_bia"
     }
 }
 
-
-
 # === PDF Export Function ===
 def export_pdf(title, answers, lang="en"):
     pdf = FPDF()
     pdf.add_page()
     pdf.set_auto_page_break(auto=True, margin=15)
 
-    # Title
     pdf.set_font("Arial", 'B', 16)
     pdf.set_text_color(0, 51, 102)
     pdf.cell(0, 10, title, ln=True, align='C')
     pdf.ln(8)
 
-    # 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")
@@ -76,7 +93,6 @@ def export_pdf(title, answers, lang="en"):
     pdf.cell(0, 8, f"Timestamp: {timestamp}", ln=True, align='C')
     pdf.ln(10)
 
-    # Answers
     pdf.set_font("Arial", '', 12)
     pdf.set_text_color(0, 0, 0)
     for label, value in answers.items():
@@ -124,7 +140,6 @@ def continue_conversation(user_input):
         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)