Update app.py

app.py

@@ -9,7 +9,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN", None)
 
 DESCRIPTION = '''
 <div>
-<h1 style="text-align: center;">A.I. Healthcare</h1>
+  <h1 style="text-align: center;">A.I. Healthcare</h1>
 </div>
 '''
 
@@ -51,20 +51,26 @@ terminators = [
 
 @spaces.GPU(duration=60)
 def chat_llama3_8b(message: str, 
-                    history: list, 
-                    temperature: float, 
-                    max_new_tokens: int
-                   ) -> str:
+                   history: list, 
+                   temperature: float, 
+                   max_new_tokens: int,
+                   confirm: bool) -> str:
     """
     Generate a streaming response using the llama3-8b model.
+    
     Args:
         message (str): The input message.
-        history (list): The conversation history used by ChatInterface.
+        history (list): The conversation history.
         temperature (float): The temperature for generating the response.
         max_new_tokens (int): The maximum number of new tokens to generate.
+        confirm (bool): Whether the user has confirmed the age/disclaimer.
+        
     Returns:
         str: The generated response.
     """
+    # If the confirmation checkbox is not checked, return a short message immediately.
+    if not confirm:
+        return "⚠️ You must confirm that you meet the usage requirements before sending a message."
     
     conversation = []
     for user, assistant in history:
@@ -120,23 +126,53 @@ def chat_llama3_8b(message: str,
             outputs.append(text)
             yield "".join(outputs)
 
-    # Store the full response (including <think>) in history, but only show the user the cleaned response
-    history.append((message, full_response))  # Full assistant response saved for context
-
-# Gradio block
-chatbot = gr.Chatbot(height=450, placeholder=PLACEHOLDER, label='Gradio ChatInterface')
+    # Store the full response (including <think>) in history for context
+    history.append((message, full_response))
+
+# Custom JavaScript to disable the send button until confirmation is given.
+# (The JS waits for the checkbox with a label containing the specified text and then monitors its state.)
+CUSTOM_JS = """
+<script>
+document.addEventListener("DOMContentLoaded", function() {
+    // Poll for the confirmation checkbox and the send button inside the ChatInterface.
+    const interval = setInterval(() => {
+        // The checkbox is rendered as an <input type="checkbox"> with an associated label.
+        const checkbox = document.querySelector('input[type="checkbox"][aria-label*="I hereby confirm that I am at least 18 years of age"]');
+        // The send button might be a <button> element with a title or specific text. Adjust the selector as needed.
+        const sendButton = document.querySelector('button[title="Send"]');
+        if (checkbox && sendButton) {
+            sendButton.disabled = !checkbox.checked;
+            checkbox.addEventListener('change', function() {
+                sendButton.disabled = !checkbox.checked;
+            });
+            clearInterval(interval);
+        }
+    }, 500);
+});
+</script>
+"""
 
-with gr.Blocks(fill_height=True, css=css) as demo:
-    
+with gr.Blocks(css=css, title="A.I. Healthcare") as demo:
     gr.Markdown(DESCRIPTION)
-    gr.ChatInterface(
+    # Inject the custom JavaScript.
+    gr.HTML(CUSTOM_JS)
+    
+    # The ChatInterface below now includes additional inputs: the confirmation checkbox and the parameter sliders.
+    chat_interface = gr.ChatInterface(
         fn=chat_llama3_8b,
-        chatbot=chatbot,
-        fill_height=True,
-        additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
+        title="A.I. Healthcare Chat",
+        chatbot=gr.Chatbot(height=450, placeholder=PLACEHOLDER, label='Conversation'),
         additional_inputs=[
-            gr.Slider(minimum=0.6, maximum=0.6, step=0.1, value=0.6, label="Temperature", render=False),
-            gr.Slider(minimum=128, maximum=4096, step=64, value=1024, label="Max new tokens", render=False),
+            gr.Checkbox(
+                value=False,
+                label=("I hereby confirm that I am at least 18 years of age (or accompanied by a legal guardian "
+                       "who is at least 18 years old), understand that the information provided by this service "
+                       "is for informational purposes only and is not intended to diagnose or treat any medical condition, "
+                       "and acknowledge that I am solely responsible for verifying any information provided."),
+                elem_id="age_confirm_checkbox"
+            ),
+            gr.Slider(minimum=0.6, maximum=0.6, step=0.1, value=0.6, label="Temperature", visible=False),
+            gr.Slider(minimum=128, maximum=4096, step=64, value=1024, label="Max new tokens", visible=False),
         ],
         examples=[
             ['What are the common symptoms of diabetes?'],
@@ -146,6 +182,7 @@ with gr.Blocks(fill_height=True, css=css) as demo:
             ['What should I know about the side effects of common medications?']
         ],
         cache_examples=False,
+        allow_screenshot=False,
     )
     
     gr.Markdown(LICENSE)