Update Gradio_UI.py

Gradio_UI.py

@@ -82,30 +82,31 @@ 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
 
-# === 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>
-    """
-
 # === Gradio Interface ===
-with gr.Blocks(title="Regulatory Compliance Generator") as demo:
+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;
+    }
+    """
+) as demo:
+    # GDPR Notice
     gr.Markdown("""
     ### 🔒 GDPR Notice  
     This assistant does not store any personal data.  
@@ -113,8 +114,6 @@ with gr.Blocks(title="Regulatory Compliance Generator") as demo:
     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()),
@@ -139,15 +138,11 @@ 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=[], type="messages")
+    chatbox = gr.Chatbot(label="🧑‍⚖️ Legalbot", value=[], type="messages", elem_classes=["chatbot"])
     user_input = gr.Textbox(label="Your answer", placeholder="Type your answer and press Enter")
     pdf_output = gr.File(label="Download PDF", visible=True)
 
-    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)
+    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)