Update Gradio_UI.py

Gradio_UI.py

@@ -90,14 +90,22 @@ def continue_conversation(user_input):
         return messages, pdf_path
 
 # === Gradio Interface ===
-with gr.Blocks(title="Regulatory Compliance Generator", css="""
+with gr.Blocks(
+    title="Regulatory Compliance Generator",
+    css="""
     footer, 
     a[href*="gradio.app"],
     a[href*="huggingface.co"],
-    button[aria-label="Share"] {
+    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:
+    """
+) as demo:
     # GDPR Notice
     gr.Markdown("""
     ### 🔒 GDPR Notice  
@@ -105,26 +113,6 @@ with gr.Blocks(title="Regulatory Compliance Generator", css="""
     All responses are processed temporarily to generate your document.  
     You are responsible for the accuracy and legality of the information submitted.
     """)
-    
-    # 🔒 Hide share button & links
-    gr.HTML("""
-    <style>
-        footer, 
-        a[href*="gradio.app"],
-        a[href*="huggingface.co"],
-        button[aria-label="Share"],
-        button[title="Share"],
-        button[data-testid="share-button"],
-        .svelte-1ipelgc[data-testid="block-settings-button"],
-        .svelte-drgfjz[data-testid="share-button"],
-        .svelte-drgfjz svg path[d^="M15 8a3"] {  /* Specific SVG icon path targeting */
-            display: none !important;
-        }
-    </style>
-    """)
-
-
-
 
     with gr.Row():
         reg_dropdown = gr.Dropdown(
@@ -150,7 +138,7 @@ with gr.Blocks(title="Regulatory Compliance Generator", css="""
     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)