Update Gradio_UI.py

Gradio_UI.py

@@ -8,7 +8,7 @@ import importlib
 # === Map document choices to modules ===
 DOCUMENT_TO_MODULE = {
     "AI Act": {
-        "Compliance Register": "tools.ai_act_register",
+        "Compliance Register": "tools.ai_act_generator",
         "High-Risk System Record": "tools.ai_high_risk_system"
     },
     "GDPR": {
@@ -45,7 +45,8 @@ conversation_state = {
     "questions": [],
     "answers": {},
     "title": "",
-    "lang": "en"
+    "lang": "en",
+    "tool": None
 }
 
 # === Load Tool Dynamically ===
@@ -57,6 +58,7 @@ def start_conversation(reg, doc):
     module_path = DOCUMENT_TO_MODULE[reg][doc]
     module = importlib.import_module(module_path)
     conversation_state["questions"] = module.get_questions()
+    conversation_state["tool"] = module
 
     return f"📋 {conversation_state['questions'][0][1]}"
 
@@ -73,6 +75,7 @@ def continue_conversation(user_input):
         next_question = conversation_state["questions"][conversation_state["step"]][1]
         return f"🧠 {next_question}", None
     else:
+        # Optionally: use conversation_state["tool"].run_tool() here if needed
         pdf_path = export_pdf(conversation_state["title"], conversation_state["answers"], lang=conversation_state["lang"])
         return "✅ Done! Your document is ready to download.", pdf_path
 
@@ -88,11 +91,11 @@ with gr.Blocks(title="Regulatory Compliance Generator") as demo:
 
     start_btn = gr.Button("🚀 Start")
     chatbox = gr.Chatbot(label="🧑‍⚖️ Legalbot")
-    user_input = gr.Textbox(label="Your answer")
+    user_input = gr.Textbox(label="Your answer", placeholder="Type your answer and press Enter")
     pdf_output = gr.File(label="Download PDF", visible=True)
 
     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)
+    user_input.submit(lambda: "", None, user_input)  # clears input
 
 demo.launch()