Update app.py

app.py

@@ -15,9 +15,7 @@ DESCRIPTION = '''
 
 LICENSE = """
 <p>
-This Health Assistant is designed to provide helpful healthcare information; however, it may make mistakes and is not designed to replace professional medical care. It is not intended to diagnose any condition or disease. Always consult with a qualified healthcare provider for any medical concerns.
-<br><br>
-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.
+This Health Assistant is designed to provide helpful healthcare information; however, it may make mistakes and is not designed to replace professional medical care. It is not intended to diagnose any condition or disease. Always consult with a qualified healthcare provider for any medical concerns.\n\nI 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.
 </p>
 """
 
@@ -53,13 +51,21 @@ 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
+                   ) -> str:
     """
     Generate a streaming response using the llama3-8b model.
+    Args:
+        message (str): The input message.
+        history (list): The conversation history used by ChatInterface.
+        temperature (float): The temperature for generating the response.
+        max_new_tokens (int): The maximum number of new tokens to generate.
+    Returns:
+        str: The generated response.
     """
+    
     conversation = []
     for user, assistant in history:
         conversation.extend([
@@ -117,30 +123,12 @@ def chat_llama3_8b(message: str,
     # 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
 
-# JavaScript snippet to conditionally show examples if ?examples=true is present in the URL.
-js_code = """
-<script>
-window.addEventListener("load", function(){
-    const urlParams = new URLSearchParams(window.location.search);
-    if(urlParams.get('examples') !== 'true'){
-         var elem = document.getElementById("examples-container");
-         if (elem) {
-             elem.style.display = "none";
-         }
-    }
-});
-</script>
-"""
-
 # Gradio block
 chatbot = gr.Chatbot(height=450, placeholder=PLACEHOLDER, label='Gradio ChatInterface')
 
 with gr.Blocks(fill_height=True, css=css) as demo:
-    gr.Markdown(DESCRIPTION)
-    
-    # Include the JavaScript so it runs on the client side.
-    gr.HTML(js_code)
     
+    gr.Markdown(DESCRIPTION)
     gr.ChatInterface(
         fn=chat_llama3_8b,
         chatbot=chatbot,
@@ -150,23 +138,17 @@ with gr.Blocks(fill_height=True, css=css) as demo:
             gr.Slider(minimum=0.6, maximum=0.6, step=0.1, value=0.6, label="Temperature", render=False),
             gr.Slider(minimum=1024, maximum=4096, step=128, value=2048, label="Max new tokens", render=False),
         ],
+        examples=[
+            ['What is PrEP, and do I need it?'],
+            ['What medications help manage being undetectable with HIV?'],
+            ['How do I know if an abortion is the right option?'],
+            ['How can I access birth-control in states where it is regulated?']
         cache_examples=False,
+        ]
+
     )
     
-    # Wrap your examples in a container with an id.
-    with gr.Column(elem_id="examples-container"):
-        gr.Markdown("### Examples")
-        gr.Examples(
-            examples=[
-                ['What is PrEP, and do I need it?', ''],
-                ['What medications help manage being undetectable with HIV?', ''],
-                ['How do I know if an abortion is the right option?', ''],
-                ['How can I access birth-control in states where it is regulated?', '']
-            ],
-            inputs=chatbot,
-        )
-    
     gr.Markdown(LICENSE)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()