Update Gradio_UI.py

Gradio_UI.py

@@ -1,39 +1,22 @@
-# gradio_UI.py
+# Gradio_UI.py
 import gradio as gr
 from fpdf import FPDF
 from langdetect import detect
 import time
+import importlib
 
-# === Define documents/questions structure ===
-DOCUMENTS = {
+# === Map document choices to modules ===
+DOCUMENT_TO_MODULE = {
     "AI Act": {
-        "Compliance Register": [
-            ("organization_name", "What is the name of your organization?"),
-            ("responsible_person", "Who is responsible for this AI system?"),
-            ("deployment_date", "When is the AI system scheduled to be deployed?"),
-            ("ai_type", "What type of AI system is it?"),
-            ("ai_description", "What does the system do?"),
-            ("risk_level", "What is the risk level?"),
-            ("risk_justification", "Why this level?"),
-        ],
-        "High-Risk System Record": [
-            ("organization_name", "Organization name?"),
-            ("system_purpose", "Purpose of the high-risk AI system?"),
-            ("annex_category", "Which Annex III category applies?"),
-            ("risk_controls", "Risk mitigation measures?"),
-        ]
+        "Compliance Register": "tools.ai_act_register",
+        "High-Risk System Record": "tools.ai_high_risk_system"
     },
     "GDPR": {
-        "Data Processing Record": [
-            ("data_controller", "Who is the data controller?"),
-            ("legal_basis", "Legal basis for processing?"),
-            ("data_subjects", "Categories of data subjects?"),
-            ("data_retention", "Retention period for data?"),
-        ]
+        "Data Processing Record": "tools.gdpr_data_record"
     }
 }
 
-# === PDF Export ===
+# === PDF Export Function ===
 def export_pdf(title, answers, lang="en"):
     pdf = FPDF()
     pdf.add_page()
@@ -56,7 +39,7 @@ def export_pdf(title, answers, lang="en"):
     pdf.output(filename)
     return filename
 
-# === App State ===
+# === State Memory ===
 conversation_state = {
     "step": 0,
     "questions": [],
@@ -65,13 +48,19 @@ conversation_state = {
     "lang": "en"
 }
 
+# === Load Tool Dynamically ===
 def start_conversation(reg, doc):
     conversation_state["step"] = 0
     conversation_state["answers"] = {}
-    conversation_state["questions"] = DOCUMENTS[reg][doc]
     conversation_state["title"] = f"{reg} - {doc}"
+
+    module_path = DOCUMENT_TO_MODULE[reg][doc]
+    module = importlib.import_module(module_path)
+    conversation_state["questions"] = module.get_questions()
+
     return f"📋 {conversation_state['questions'][0][1]}"
 
+# === Continue Chat ===
 def continue_conversation(user_input):
     step = conversation_state["step"]
     if step < len(conversation_state["questions"]):
@@ -87,24 +76,21 @@ def continue_conversation(user_input):
         pdf_path = export_pdf(conversation_state["title"], conversation_state["answers"], lang=conversation_state["lang"])
         return "✅ Done! Your document is ready to download.", pdf_path
 
-# === Gradio UI ===
+# === Gradio Interface ===
 with gr.Blocks(title="Regulatory Compliance Generator") as demo:
     gr.Markdown("## 📑 Regulation Document Generator")
 
     with gr.Row():
-        reg_dropdown = gr.Dropdown(choices=list(DOCUMENTS.keys()), label="Select Regulation")
+        reg_dropdown = gr.Dropdown(choices=list(DOCUMENT_TO_MODULE.keys()), label="Select Regulation")
         doc_dropdown = gr.Dropdown(label="Select Document Type")
 
-    reg_dropdown.change(lambda r: list(DOCUMENTS[r].keys()), inputs=reg_dropdown, outputs=doc_dropdown)
+    reg_dropdown.change(lambda r: list(DOCUMENT_TO_MODULE[r].keys()), inputs=reg_dropdown, outputs=doc_dropdown)
 
     start_btn = gr.Button("🚀 Start")
     chatbox = gr.Chatbot(label="🧑‍⚖️ Legalbot")
     user_input = gr.Textbox(label="Your answer")
     pdf_output = gr.File(label="Download PDF", visible=True)
 
-    def reset():
-        return "", "", "", [], None
-
     start_btn.click(start_conversation, inputs=[reg_dropdown, doc_dropdown], outputs=chatbox)
     user_input.submit(continue_conversation, inputs=user_input, outputs=[chatbox, pdf_output])
     user_input.submit(lambda: "", None, user_input)