Update app.py

app.py

@@ -1,17 +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", "medical", "disease", "symptom", "treatment", "doctor", "patient", "medicine"]
-    message = message.lower()
-    for keyword in health_keywords:
-        if keyword in message:
-            return True
-    return False
 
 def respond(
     message,
@@ -21,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]})
 
@@ -43,12 +34,13 @@ def respond(
         temperature=temperature,
         top_p=top_p,
     ):
-        token = message.choices.delta.content
+        token = message.choices[0].delta.content
 
         response += token
         yield response
 
-# Custom CSS to make it look like ChatGPT
+
+# CSS for styling the interface
 css = """
 body {
     font-family: 'Inter', sans-serif;
@@ -115,60 +107,26 @@ body {
 }
 """
 
-# Create a custom chat interface using gr.Blocks
-with gr.Blocks(css=css) as demo:
-    gr.Markdown("# Health Assistant Chatbot")
-    gr.Markdown("### Ask me any health-related questions.")
-    chatbot = gr.Chatbot(
-        value=[],
-        show_user_avatar=False,
-        show_bot_avatar=False,
-        width="100%",
-    )
-    input_box = gr.Textbox(
-        label="Type your message here",
-        placeholder="Type your message here",
-        show_label=False,
-    )
-    system_message = gr.Textbox(
-        value="You are a virtual Doctor Assistant. Your role is to assist healthcare professionals by providing accurate, evidence-based medical information, offering treatment options, and supporting patient care. Always prioritize patient safety, provide concise answers, and clearly state that your advice does not replace a doctor's judgment. Do not diagnose or prescribe treatments without human oversight.",
-        label="System message",
-        visible=False,
-    )
-    max_tokens = gr.Slider(
-        minimum=1,
-        maximum=2048,
-        value=512,
-        step=1,
-        label="Max new tokens",
-        visible=False,
-    )
-    temperature = gr.Slider(
-        minimum=0.1,
-        maximum=4.0,
-        value=0.7,
-        step=0.1,
-        label="Temperature",
-        visible=False,
-    )
-    top_p = gr.Slider(
-        minimum=0.1,
-        maximum=1.0,
-        value=0.95,
-        step=0.05,
-        label="Top-p (nucleus sampling)",
-        visible=False,
-    )
-
-    def update_chat(message, history, system_message, max_tokens, temperature, top_p):
-        response = respond(message, history, system_message, max_tokens, temperature, top_p)
-        return chatbot.update(value=history + [(message, response)])
-
-    input_box.submit(
-        update_chat,
-        inputs=[input_box, chatbot, system_message, max_tokens, temperature, top_p],
-        outputs=chatbot,
-    )
+"""
+For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
+"""
+demo = gr.ChatInterface(
+    respond,
+    additional_inputs=[
+        gr.Textbox(value="You are a virtual Doctor Assistant. Your role is to assist healthcare professionals by providing accurate, evidence-based medical information, offering treatment options, and supporting patient care. Always prioritize patient safety, provide concise answers, and clearly state that your advice does not replace a doctor's judgment. Do not diagnose or prescribe treatments without human oversight.", label="System message", visible=False),
+        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens", visible=False),
+        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature", visible=False),
+        gr.Slider(
+            minimum=0.1,
+            maximum=1.0,
+            value=0.95,
+            step=0.05,
+            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)