Update app.py

app.py

@@ -26,8 +26,9 @@ print("Model loaded successfully!")
 print(f"Device map: {model.hf_device_map}")
 print(f"Model device: {next(model.parameters()).device}")
 
+
 # =======================================================
-# Generate Doctor Response (Stateless)
+# Generate Doctor Response (Stateless + Clean Replies)
 # =======================================================
 def generate_doctor_response(history):
     user_message = history[-1]["content"]
@@ -37,25 +38,23 @@ def generate_doctor_response(history):
         yield history
         return
 
-    # Single-turn prompt (no past context)
-    prompt = """# 🩺 You are a highly knowledgeable Medical Expert
-
-**Roles:**
-- **Doctor:** Diagnose symptoms, ask relevant follow-up questions, and provide medical advice.
-- **Nutritionist:** Give diet and lifestyle recommendations.
-- **Medical Teacher:** Explain complex medical terms and conditions in simple, understandable language.
+    # 🩺 New Prompt (no 'Patient:' or 'Doctor:' lines)
+    prompt = f"""
+You are a compassionate and professional medical expert.
+Your role is to help users by providing clear, empathetic, and accurate medical information.
 
-**Guidelines:**
-1. Ask **only one follow-up question at a time** if necessary.
-2. Provide advice after collecting enough information.
-3. Be **empathetic, professional, and conversational**.
-4. Include disclaimers when needed:  
-   ⚕️ *This is AI-generated information and not a substitute for professional medical advice. Please consult a healthcare provider for proper diagnosis and treatment.*
-5. Keep answers concise, clear, and actionable.
+Guidelines:
+1. Do NOT include words like 'Doctor:' or 'Patient:' in your replies.
+2. Respond naturally and directly to the user's concern.
+3. Keep answers short, clear, and medically sound.
+4. Add a disclaimer when appropriate: 
+   ⚕️ *This is AI-generated information and not a substitute for professional medical advice.*
 
-Respond naturally to the following patient message:
+Now, please respond to the user's message below:
 
-Patient: """ + user_message + "\nDoctor:"
+User: {user_message}
+Assistant:
+"""
 
     # Tokenize input
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
@@ -80,8 +79,8 @@ Patient: """ + user_message + "\nDoctor:"
 
     # Keep concise output
     response = ". ".join(response.split(". ")[:3]).strip()
-    if response.lower().startswith("doctor:"):
-        response = response[7:].strip()
+    if response.lower().startswith("assistant:"):
+        response = response[10:].strip()
     if len(response) < 10:
         response = "I understand your concern. Could you please provide more details about your symptoms?"