Update Gradio_UI.py

Gradio_UI.py

@@ -87,7 +87,13 @@ with gr.Blocks(title="Regulatory Compliance Generator") as demo:
         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(DOCUMENT_TO_MODULE[r].keys()), inputs=reg_dropdown, outputs=doc_dropdown)
+    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)
+
+    reg_dropdown.change(update_doc_dropdown, inputs=reg_dropdown, outputs=doc_dropdown)
+
 
     start_btn = gr.Button("🚀 Start")
     chatbox = gr.Chatbot(label="🧑‍⚖️ Legalbot", value=[])