Update app.py

app.py

@@ -31,6 +31,17 @@ model = LlamaForCausalLM.from_pretrained(
 generator = model.generate
 print("✅ ChatDoctor model loaded successfully!\n")
 
+# =============================
+# System Prompt for Medical Focus
+# =============================
+SYSTEM_PROMPT = """You are ChatDoctor, a professional medical AI assistant. Your role is to:
+- Provide helpful medical information and guidance
+- Ask relevant follow-up questions about symptoms
+- Offer general health advice
+- Remind patients to consult healthcare professionals for serious concerns
+
+You should ONLY engage in health and medical-related conversations. If someone greets you casually or asks non-medical questions, politely redirect them to ask about their health concerns."""
+
 # =============================
 # Stopping Criteria
 # =============================
@@ -54,16 +65,41 @@ class StopOnTokens(StoppingCriteria):
 # =============================
 conversation_history = []
 
+# =============================
+# Check if message is medical-related
+# =============================
+def is_greeting_or_casual(message):
+    """Check if message is just a casual greeting"""
+    casual_patterns = [
+        "hey", "hi", "hello", "sup", "what's up", "whats up",
+        "how are you", "good morning", "good evening", "good afternoon"
+    ]
+    
+    message_lower = message.lower().strip()
+    
+    # Check if it's ONLY a greeting (not combined with medical query)
+    if len(message_lower.split()) <= 3:
+        for pattern in casual_patterns:
+            if message_lower == pattern or message_lower.startswith(pattern):
+                return True
+    
+    return False
+
 # =============================
 # Get Response Function
 # =============================
 def get_response(user_input, history_context):
     """Generate response from ChatDoctor model"""
+    
+    # Handle casual greetings by redirecting to medical topics
+    if is_greeting_or_casual(user_input):
+        return "Hello! I'm ChatDoctor, your AI medical assistant. I'm here to help with your health concerns and medical questions. What health-related issue can I assist you with today?"
+    
     human_invitation = "Patient: "
     doctor_invitation = "ChatDoctor: "
 
     # Build conversation from history
-    history_text = []
+    history_text = [SYSTEM_PROMPT]
     for human, assistant in history_context:
         if human:
             history_text.append(human_invitation + human)
@@ -202,14 +238,14 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     # Chatbot Interface
     chatbot = gr.Chatbot(
         height=500,
-        placeholder="<div style='text-align: center; padding: 40px;'><h3>👋 Welcome to ChatDoctor!</h3><p>I'm here to discuss your health concerns. How can I assist you today?</p></div>",
+        placeholder="<div style='text-align: center; padding: 40px;'><h3>👋 Welcome to ChatDoctor!</h3><p>I'm here to discuss your health concerns. Please describe your symptoms or health question.</p></div>",
         show_label=False,
         avatar_images=(None, "🤖"),
     )
     
     with gr.Row():
         msg = gr.Textbox(
-            placeholder="Type your message here... (e.g., 'I have a headache')",
+            placeholder="Describe your symptoms or health concern here...",
             show_label=False,
             scale=9,
             container=False