Update Gradio_UI.py

Gradio_UI.py

@@ -60,11 +60,14 @@ def start_conversation(reg, doc):
     conversation_state["questions"] = module.get_questions()
     conversation_state["tool"] = module
 
-    return f"📋 {conversation_state['questions'][0][1]}"
+    first_question = conversation_state["questions"][0][1]
+    return [{"role": "assistant", "content": f"📋 {first_question}"}]
 
 # === Continue Chat ===
 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
@@ -73,10 +76,12 @@ def continue_conversation(user_input):
 
     if conversation_state["step"] < len(conversation_state["questions"]):
         next_question = conversation_state["questions"][conversation_state["step"]][1]
-        return f"🧠 {next_question}", None
+        messages.append({"role": "assistant", "content": f"🧠 {next_question}"})
+        return messages, None
     else:
         pdf_path = export_pdf(conversation_state["title"], conversation_state["answers"], lang=conversation_state["lang"])
-        return "✅ Done! Your document is ready to download.", pdf_path
+        messages.append({"role": "assistant", "content": "✅ Done! Your document is ready to download."})
+        return messages, pdf_path
 
 # === Gradio Interface ===
 with gr.Blocks(title="Regulatory Compliance Generator") as demo:
@@ -94,12 +99,12 @@ with gr.Blocks(title="Regulatory Compliance Generator") as demo:
     reg_dropdown.change(update_doc_dropdown, inputs=reg_dropdown, outputs=doc_dropdown)
 
     start_btn = gr.Button("🚀 Start")
-    chatbox = gr.Chatbot(label="🧑‍⚖️ Legalbot", value=[])
+    chatbox = gr.Chatbot(label="🧑‍⚖️ Legalbot", value=[], type="messages")  # <-- Updated to use new format
     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)  # Clear input
 
 demo.launch()