Update Gradio_UI.py

Gradio_UI.py

@@ -4,7 +4,6 @@ from langdetect import detect
 import time
 import importlib
 
-# === Map document choices to modules ===
 DOCUMENT_TO_MODULE = {
     "AI Act": {
         "Compliance Register": "tools.ai_act_generator",
@@ -25,29 +24,33 @@ DOCUMENT_TO_MODULE = {
         "Data Processor Agreements": "tools.gdpr_dpa_log"
     },
     "NIS2": {
-        "Incident Response Log": "tools.nis2_incident_response_log",
-        "Asset Inventory": "tools.nis2_asset_inventory",
-        "Business Continuity Plan": "tools.nis2_business_continuity_plan",
         "Access Control Log": "tools.nis2_access_control_log",
-        "System Testing Record": "tools.nis2_system_testing_record",
-        "Supply Chain Risk Record": "tools.nis2_supply_chain_risk_record",
-        "Incident Communication Plan": "tools.nis2_incident_communication_plan",
-        "Vulnerability Management Log": "tools.nis2_vulnerability_management_log",
-        "Security Audit Record": "tools.nis2_security_audit_record",
-        "Disaster Recovery Plan": "tools.nis2_disaster_recovery_plan",
+        "Asset Inventory": "tools.nis2_asset_inventory",
+        "Backup & Recovery Log": "tools.nis2_backup_recovery_log",
+        "Business Continuity Plan": "tools.nis2_business_continuity",
         "Compliance Audit Log": "tools.nis2_compliance_audit_log",
-        "Penetration Test Summary": "tools.nis2_penetration_test_summary",
         "Data Encryption Inventory": "tools.nis2_data_encryption_inventory",
+        "Disaster Recovery Plan": "tools.nis2_disaster_recovery_plan",
+        "Incident Communication Plan": "tools.nis2_incident_communication",
+        "Incident Response Log": "tools.nis2_incident_log",
         "Logging & Monitoring Policy": "tools.nis2_logging_monitoring_policy",
         "Network Architecture Diagram Log": "tools.nis2_network_architecture_log",
-        "Security Awareness Training Record": "tools.nis2_security_training_record",
-        "Backup & Recovery Log": "tools.nis2_backup_recovery_log",
+        "Penetration Test Summary": "tools.nis2_penetration_test_summary",
+        "Physical Security Checklist": "tools.nis2_physical_security_checklist",
         "Risk Assessment Matrix": "tools.nis2_risk_assessment_matrix",
-        "Third-Party Access Review": "tools.nis2_third_party_access_review"
+        "Risk Register": "tools.nis2_risk_register",
+        "Security Audit Record": "tools.nis2_security_audit",
+        "Security Testing Strategy": "tools.nis2_security_testing_strategy",
+        "Security Awareness Training Record": "tools.nis2_security_training_record",
+        "Supply Chain Risk Record": "tools.nis2_supply_chain_risk",
+        "System Testing Record": "tools.nis2_system_testing",
+        "Third-Party Access Review": "tools.nis2_third_party_access_review",
+        "Vulnerability Management Log": "tools.nis2_vulnerability_log"
     }
 }
 
 
+
 # === PDF Export Function ===
 def export_pdf(title, answers, lang="en"):
     pdf = FPDF()