Update Gradio_UI.py

Gradio_UI.py

@@ -67,7 +67,7 @@ def start_conversation(reg, doc):
 def continue_conversation(user_input):
     messages = [{"role": "user", "content": user_input}]
     step = conversation_state["step"]
-    
+
     if step < len(conversation_state["questions"]):
         key, question = conversation_state["questions"][step]
         conversation_state["answers"][question] = user_input
@@ -87,22 +87,21 @@ def continue_conversation(user_input):
 with gr.Blocks(title="Regulatory Compliance Generator") as demo:
     gr.Markdown("## 📑 Regulation Document Generator")
 
-with gr.Row():
-    reg_dropdown = gr.Dropdown(
-        choices=list(DOCUMENT_TO_MODULE.keys()),
-        label="Select Regulation",
-        value=None,
-        interactive=True,
-        info="-- Please choose a regulation --"
-    )
-    doc_dropdown = gr.Dropdown(
-        label="Select Document Type",
-        choices=[],
-        value=None,
-        interactive=True,
-        info="-- Select a document after choosing a regulation --"
-    )
-
+    with gr.Row():
+        reg_dropdown = gr.Dropdown(
+            choices=list(DOCUMENT_TO_MODULE.keys()),
+            label="Select Regulation",
+            value=None,
+            interactive=True,
+            info="-- Please choose a regulation --"
+        )
+        doc_dropdown = gr.Dropdown(
+            label="Select Document Type",
+            choices=[],
+            value=None,
+            interactive=True,
+            info="-- Select a document after choosing a regulation --"
+        )
 
     def update_doc_dropdown(selected_reg):
         if selected_reg in DOCUMENT_TO_MODULE:
@@ -112,12 +111,12 @@ with gr.Row():
     reg_dropdown.change(update_doc_dropdown, inputs=reg_dropdown, outputs=doc_dropdown)
 
     start_btn = gr.Button("🚀 Start")
-    chatbox = gr.Chatbot(label="🧑‍⚖️ Legalbot", value=[], type="messages")  # <-- Updated to use new format
+    chatbox = gr.Chatbot(label="🧑‍⚖️ Legalbot", value=[], type="messages")
     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)  # Clear input
+    user_input.submit(lambda: "", None, user_input)
 
 demo.launch()