Update app.py

app.py

@@ -1,52 +1,11 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Initialize the InferenceClient
+"""
+For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
+"""
 client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct")
 
-def is_health_related(message):
-    # Simple heuristic to check if the message is health-related
-    health_keywords = [
-  "health", "cancer", "diabetes", "hypertension", "asthma",
-  "arthritis", "cardiology", "neurology", "depression",
-  "anxiety", "cholesterol", "obesity", "allergies", "autism",
-  "osteoporosis", "stroke", "HIV", "AIDS", "hepatitis",
-  "tuberculosis", "malaria", "COVID-19", "Parkinson's",
-  "Alzheimer's", "dementia", "eczema", "psoriasis", "migraine",
-  "epilepsy", "insomnia", "fibromyalgia", "sclerosis",
-  "pneumonia", "influenza", "endometriosis", "infertility",
-  "glaucoma", "cataracts", "anemia", "leukemia", "lymphoma",
-  "melanoma", "dermatitis", "IBS", "Crohn's", "ulcers",
-  "GERD", "kidney disease", "liver disease", "cirrhosis",
-  "thyroid", "menopause", "prostate", "cholera",
-  "dysentery", "polio", "tetanus", "diphtheria",
-  "Zika", "Ebola", "mental health", "physical health",
-  "immunology", "orthopedics", "pediatrics", "geriatrics",
-  "surgery", "rehabilitation", "nutrition", "fitness",
-  "wellness", "vaccination", "therapy", "radiology",
-  "oncology", "pathology", "psychology", "psychiatry",
-  "urology", "gastroenterology", "dermatology", "hematology",
-  "toxicology", "pharmacology", "trauma", "injury",
-  "burns", "sepsis", "addiction", "substance abuse",
-  "symptoms", "diagnosis", "treatment", "medication",
-  "medicine", "patient", "healthcare", "hospital",
-  "clinic", "recovery", "remedy", "pain management",
-  "first aid", "disease prevention", "immunization",
-  "primary care", "specialist", "prescription", "chronic illness",
-  "acute care", "emergency care", "surgical procedures",
-  "therapy sessions", "mental therapy", "physical therapy",
-  "health monitoring", "health insurance", "medical advice",
-  "health tips", "diagnostic tests", "lab results",
-  "consultation", "second opinion", "side effects", "health checkup",
-  "disease management", "caregiver support", "home care", 
-  "alternative medicine", "herbal treatment", "integrative medicine",
-  "personalized care", "health records", "online consultation"
-]
-    message = message.lower()
-    for keyword in health_keywords:
-        if keyword in message:
-            return True
-    return False
 
 def respond(
     message,
@@ -56,14 +15,11 @@ def respond(
     temperature,
     top_p,
 ):
-    if not is_health_related(message):
-        return "Sorry, I can't help you with that because I am just a bot who can help with health-related queries."
-
     messages = [{"role": "system", "content": system_message}]
 
     for val in history:
-        if val:
-            messages.append({"role": "user", "content": val})
+        if val[0]:
+            messages.append({"role": "user", "content": val[0]})
         if val[1]:
             messages.append({"role": "assistant", "content": val[1]})
 
@@ -78,17 +34,19 @@ def respond(
         temperature=temperature,
         top_p=top_p,
     ):
-        token = message.choices.delta.content
+        token = message.choices[0].delta.content
 
         response += token
         yield response
 
+
 # CSS for styling the interface
 css = """
 body {
     background-color: #06688E; /* Dark background */
     color: white; /* Text color for better visibility */
 }
+
 .gr-button {
     background-color: #42B3CE !important; /* White button color */
     color: black !important; /* Black text for contrast */
@@ -96,14 +54,19 @@ body {
     padding: 8px 16px !important;
     border-radius: 5px !important;
 }
+
 .gr-button:hover {
     background-color: #e0e0e0 !important; /* Slightly lighter button on hover */
 }
+
 .gr-slider-container {
     color: white !important; /* Slider labels in white */
 }
 """
 
+"""
+For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
+"""
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
@@ -115,12 +78,12 @@ demo = gr.ChatInterface(
             maximum=1.0,
             value=0.95,
             step=0.05,
-            label="Top-p (nucleus sampling)",
-            visible=False
+            label="Top-p (nucleus sampling)",visible=False
         ),
     ],
     css=css,  # Pass the custom CSS here
 )
 
+
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)