Update Gradio_UI.py

Gradio_UI.py

@@ -75,7 +75,6 @@ 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
 
@@ -85,16 +84,15 @@ with gr.Blocks(title="Regulatory Compliance Generator") as demo:
 
     with gr.Row():
         reg_dropdown = gr.Dropdown(choices=list(DOCUMENT_TO_MODULE.keys()), label="Select Regulation")
-        doc_dropdown = gr.Dropdown(label="Select Document Type")
+        doc_dropdown = gr.Dropdown(label="Select Document Type", choices=[])
 
     def update_doc_dropdown(selected_reg):
-    if selected_reg in DOCUMENT_TO_MODULE:
-        return gr.update(choices=list(DOCUMENT_TO_MODULE[selected_reg].keys()), value=None)
-    return gr.update(choices=[], value=None)
+        if selected_reg in DOCUMENT_TO_MODULE:
+            return gr.update(choices=list(DOCUMENT_TO_MODULE[selected_reg].keys()), value=None)
+        return gr.update(choices=[], value=None)
 
     reg_dropdown.change(update_doc_dropdown, inputs=reg_dropdown, outputs=doc_dropdown)
 
-
     start_btn = gr.Button("🚀 Start")
     chatbox = gr.Chatbot(label="🧑‍⚖️ Legalbot", value=[])
     user_input = gr.Textbox(label="Your answer", placeholder="Type your answer and press Enter")
@@ -102,6 +100,6 @@ with gr.Blocks(title="Regulatory Compliance Generator") as demo:
 
     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)  # clears input
+    user_input.submit(lambda: "", None, user_input)
 
 demo.launch()