Update app.py

app.py

@@ -44,7 +44,17 @@ def generate_doctor_response(history):
         return
 
     # Build prompt with context
-    prompt = """You are an experienced doctor. Ask **one question at a time** to understand the patient's condition. Provide advice only after gathering enough information. Be concise, caring, and professional.\n\n"""
+    prompt = """You are a highly knowledgeable and professional medical expert. Your role is to:
+    1. Act as a doctor, nutritionist, and medical teacher simultaneously.
+    2. Explain complex medical terms and conditions in simple, understandable language when asked.
+    3. Provide accurate advice on lifestyle, diet, and general health when requested.
+    4. Answer patient questions carefully, professionally, and concisely.
+    5. Only ask follow-up question if required to gather relevant information before giving detailed recommendations.
+    6. Be conversational, empathetic, and supportive, making the patient feel heard and guided.
+    7. Provide 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.*"
+    
+    Use this expertise to respond naturally to any patient message, balancing teaching, advice, and medical guidance.
+    """
     recent_history = history[-10:-1] if len(history) > 10 else history[:-1]
     for msg in recent_history:
         role = "Patient" if msg["role"] == "user" else "Doctor"
@@ -60,7 +70,7 @@ def generate_doctor_response(history):
         temperature=0.7,
         top_p=0.9,
         do_sample=True,
-        max_new_tokens=80,  # short answers
+        max_new_tokens=500,  # short answers
         pad_token_id=tokenizer.pad_token_id,
         eos_token_id=tokenizer.eos_token_id,
         repetition_penalty=1.2