Update Gradio_UI.py

Gradio_UI.py

@@ -16,13 +16,15 @@ DOCUMENT_TO_MODULE = {
         "Data Governance Record": "tools.data_governance_record",
         "Human Oversight Strategy (Art. 14)": "tools.human_oversight_strategy",
         "Corrective Action Log": "tools.corrective_action_log"
+        
+        
     },
     "GDPR": {
         "Data Processing Record": "tools.gdpr_data_record"
     }
 }
 
-# === Fallback PDF Export (used only if tool doesn't have its own export method) ===
+# === PDF Export Function ===
 def export_pdf(title, answers, lang="en"):
     pdf = FPDF()
     pdf.add_page()
@@ -85,18 +87,7 @@ def continue_conversation(user_input):
         messages.append({"role": "assistant", "content": f"🧠 {next_question}"})
         return messages, None
     else:
-        tool = conversation_state["tool"]
-        title = conversation_state["title"]
-        lang = conversation_state["lang"]
-        answers = conversation_state["answers"]
-
-        # === Prefer tool's export function if available ===
-        if hasattr(tool, "export_text_to_pdf"):
-            content = "\n".join([f"- **{label}**: {answers.get(key, '')}" for key, label in conversation_state["questions"]])
-            pdf_path = tool.export_text_to_pdf(content, language=lang)
-        else:
-            pdf_path = export_pdf(title, answers, lang=lang)
-
+        pdf_path = export_pdf(conversation_state["title"], conversation_state["answers"], lang=conversation_state["lang"])
         messages.append({"role": "assistant", "content": "✅ Done! Your document is ready to download."})
         return messages, pdf_path
 
@@ -136,4 +127,4 @@ with gr.Blocks(title="Regulatory Compliance Generator") as demo:
     user_input.submit(continue_conversation, inputs=user_input, outputs=[chatbox, pdf_output])
     user_input.submit(lambda: "", None, user_input)
 
-demo.launch()
+demo.launch()