Update Gradio_UI.py

Gradio_UI.py

@@ -82,31 +82,30 @@ def continue_conversation(user_input):
 
     if conversation_state["step"] < len(conversation_state["questions"]):
         next_question = conversation_state["questions"][conversation_state["step"]][1]
-        messages.append({"role": "assistant", "content": f"🧠 {next_question}"})
+        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"])
         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",
-    css="""
-    footer, 
-    a[href*="gradio.app"],
-    a[href*="huggingface.co"],
-    button[aria-label="Share"],
-    button[title="Share"],
-    button[data-testid="share-button"],
-    .svelte-drgfjz[data-testid="share-button"],
-    .svelte-drgfjz svg path[d^="M15 8a3"],
-    .chatbot button[data-testid="share-button"] {
-        display: none !important;
-    }
+# === Custom CSS for Cleanup ===
+def get_custom_css():
+    return """
+    <style>
+        footer, 
+        a[href*='gradio.app'],
+        a[href*='huggingface.co'],
+        button[aria-label='Share'],
+        button[data-testid='share-button'],
+        button[title='Share'] {
+            display: none !important;
+        }
+    </style>
     """
-) as demo:
-    # GDPR Notice
+
+# === Gradio Interface ===
+with gr.Blocks(title="Regulatory Compliance Generator") as demo:
     gr.Markdown("""
     ### 🔒 GDPR Notice  
     This assistant does not store any personal data.  
@@ -114,6 +113,8 @@ with gr.Blocks(
     You are responsible for the accuracy and legality of the information submitted.
     """)
 
+    css_injector = gr.HTML(get_custom_css())
+
     with gr.Row():
         reg_dropdown = gr.Dropdown(
             choices=list(DOCUMENT_TO_MODULE.keys()),
@@ -138,11 +139,15 @@ with gr.Blocks(
     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", elem_classes=["chatbot"])
+    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)
+    def start_conversation_and_inject_css(reg, doc):
+        css_injector.update(get_custom_css())
+        return start_conversation(reg, doc)
+
+    start_btn.click(start_conversation_and_inject_css, 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)