Update app.py

app.py

@@ -286,24 +286,21 @@ class HealthcareChatbot:
         return self._get_rule_based_response(user_input, language)
     
     def _get_groq_response(self, user_input, language):
-        system_prompt = """You are Dr. SehatAI, a professional medical AI assistant for Pakistani patients. 
-        Provide accurate, professional medical advice in a compassionate doctor-patient manner.
+        system_prompt = """You are SehatAI, a professional healthcare assistant designed for Pakistani patients. 
+        Provide accurate, culturally appropriate medical advice in a compassionate manner.
+        Focus on preventive care, symptom explanation, and when to seek medical attention.
+        Always emphasize that you are an AI assistant and recommend consulting healthcare professionals for serious conditions.
+        Be specific, practical, and provide actionable advice.
         
-        CRITICAL GUIDELINES:
-        1. NEVER start with generic welcome messages or lists of services
-        2. ALWAYS provide specific, direct answers to the user's exact question
-        3. Respond like a professional doctor would to a patient's specific concern
-        4. Provide detailed medical explanations, practical advice, and when to seek help
-        5. Be empathetic but professional and authoritative
-        6. Focus on the specific medical issue the user is asking about
-        7. If it's a greeting, respond professionally and ask how you can help with their health concerns
-        
-        Format your response as a medical professional would speak to a patient."""
+        IMPORTANT: Provide detailed, specific answers directly related to the user's query. 
+        Do not give generic responses - tailor your answer specifically to what the user is asking.
+        If they ask about specific symptoms, conditions, or treatments, provide comprehensive information about that topic.
+        Use your medical knowledge to give practical, actionable advice."""
         
         if language == 'Urdu':
-            system_prompt += " Respond in Urdu with professional medical terminology. Be direct and specific to the user's question."
+            system_prompt += " Respond in Urdu with proper medical terminology and cultural sensitivity. Provide detailed, specific answers in Urdu."
         else:
-            system_prompt += " Respond in English with professional medical advice. Be direct and specific to the user's question."
+            system_prompt += " Respond in English with clear, professional medical advice. Provide detailed, specific answers in English."
         
         messages = [
             {"role": "system", "content": system_prompt},
@@ -330,7 +327,10 @@ class HealthcareChatbot:
     def _get_rule_based_response(self, user_input, language):
         user_input = user_input.lower().strip()
         
-        # Remove generic greeting response - always provide specific answers
+        # Greeting patterns
+        if any(word in user_input for word in ['hello', 'hi', 'hey', 'salam', 'السلام علیکم']):
+            return self._format_greeting_response(language)
+        
         # Heart disease related queries
         if any(word in user_input for word in ['heart', 'cardiac', 'chest pain', 'cholesterol', 'دل', 'سینے میں درد']):
             return self._format_heart_response(language)
@@ -367,41 +367,42 @@ class HealthcareChatbot:
         elif any(word in user_input for word in ['health', 'wellness', 'prevention', 'صحت', 'تندرستی']):
             return self._format_general_health_response(language)
         
-        # For greetings or unknown queries, provide professional medical assistance response
+        # Default response for unknown queries
         else:
-            return self._format_professional_response(user_input, language)
+            return self._format_default_response(language)
     
-    def _format_professional_response(self, user_input, language):
-        """Professional response for greetings and general queries"""
+    def _format_greeting_response(self, language):
         if language == 'English':
-            return f"""
-**Dr. SehatAI:** Thank you for your message. I understand you're asking about: "{user_input}"
-
-As your medical AI assistant, I'm here to provide specific, professional healthcare guidance. Could you please tell me more about your specific health concern so I can give you the most accurate and helpful medical advice?
+            return """
+**🤖 Welcome to SehatAI Health Assistant!**
 
-I specialize in:
-• Cardiovascular health and heart conditions
-• Diabetes management and prevention
-• Blood pressure control and hypertension
-• General wellness and preventive care
+I'm here to help you with:
+• Heart health and cardiovascular diseases
+• Diabetes and blood sugar management  
+• Blood pressure and hypertension
+• Nutrition and diet advice
+• Exercise and fitness guidance
+• Symptom assessment
+• General health and wellness
 
-Please share more details about your specific symptoms or health questions, and I'll provide personalized medical guidance.
+Please ask me specific health-related questions, and I'll provide detailed, helpful information.
 
 *Note: I'm an AI assistant. For serious medical concerns, always consult a qualified healthcare professional.*
 """
         else:
-            return f"""
-**ڈاکٹر سیحت اے آئی:** آپ کے پیغام کا شکریہ۔ میں سمجھتا ہوں کہ آپ پوچھ رہے ہیں: "{user_input}"
-
-آپ کے میڈیکل AI اسسٹنٹ کے طور پر، میں مخصوص، پیشہ ورانہ ہیلتھ کیئر رہنمائی فراہم کرنے کے لیے یہاں ہوں۔ براہ کرم مجھے اپنے مخصوص صحت کے مسئلے کے بارے میں مزید بتائیں تاکہ میں آپ کو سب سے درست اور مددگار طبی مشورہ دے سکوں۔
+            return """
+**🤖 سیحت اے آئی ہیلتھ اسسٹنٹ میں خوش آمدید!**
 
-میں اس میں مہارت رکھتا ہوں:
-• قلبی صحت اور دل کے حالات
-• ذیابیطس کا انتظام اور روک تھام
-• بلڈ پریشر کنٹرول اور ہائی بلڈ پریشر
-• عام تندرستی اور احتیاطی دیکھ بھال
+میں آپ کی مدد کے لیے یہاں ہوں:
+• دل کی صحت اور قلبی امراض
+• ذیابیطس اور بلڈ شوگر مینجمنٹ
+• بلڈ پریشر اور ہائی بلڈ پریشر
+• غذائیت اور غذا کے مشورے
+• ورزش اور فٹنس گائیڈنس
+• علامات کی تشخیص
+• عام صحت اور تندرستی
 
-براہ کرم اپنی مخصوص علامات یا صحت کے سوالات کے بارے میں مزید تفصیلات شیئر کریں، اور میں ذاتی نوعیت کی طبی رہنمائی فراہم کروں گا۔
+براہ کرم مجھ سے مخصوص صحت سے متعلق سوالات پوچھیں، اور میں تفصیلی، مددگار معلومات فراہم کروں گا۔
 
 *نوٹ: میں ایک AI اسسٹنٹ ہوں۔ سنگین طبی مسائل کے لیے، ہمیشہ کوالیفائیڈ ہیلتھ کیئر پروفیشنل سے مشورہ کریں۔*
 """
@@ -410,501 +411,1275 @@ Please share more details about your specific symptoms or health questions, and
         knowledge = self.medical_knowledge_base['heart_disease']
         if language == 'English':
             return f"""
-**Dr. SehatAI - Cardiac Specialist:**
+**❤️ Comprehensive Heart Health Information**
 
-Based on your question about heart health, here's my professional assessment:
-
-**Cardiac Symptoms to Monitor:**
+**🚨 Common Symptoms:**
 {chr(10).join(['• ' + symptom for symptom in knowledge['symptoms']])}
 
-**Key Risk Factors:**
+**⚠️ Risk Factors:**
 {chr(10).join(['• ' + factor for factor in knowledge['risk_factors']])}
 
-**Preventive Cardiology Recommendations:**
+**🛡️ Prevention Strategies:**
 {chr(10).join(['• ' + tip for tip in knowledge['prevention']])}
 
-**🚨 Cardiac Emergency Indicators:**
+**🚑 Emergency Warning Signs (Seek Immediate Care):**
 {chr(10).join(['• ' + sign for sign in knowledge['emergency_signs']])}
 
-**Clinical Monitoring Protocol:**
-• Regular BP monitoring (weekly if hypertensive)
-• Lipid profile every 6-12 months
-• ECG if symptomatic
-• Stress testing as clinically indicated
+**📊 Recommended Monitoring:**
+• Regular blood pressure checks
+• Cholesterol levels annually if high risk
+• ECG if symptoms present
+• Stress test if recommended by doctor
 
-*Medical Note: Cardiovascular disease remains the leading cause of mortality. Early intervention significantly improves outcomes.*
+*🔬 Clinical Note: Early detection saves lives. Regular check-ups are crucial for heart health.*
 """
         else:
             return f"""
-**ڈاکٹر سیحت اے آئی - کارڈیاک اسپیشلسٹ:**
-
-دل کی صحت کے بارے میں آپ کے سوال کی بنیاد پر، یہ میرا پیشہ ورانہ جائزہ ہے:
+**❤️ دل کی صحت کی جامع معلومات**
 
-**نگرانی کرنے والی قلبی علامات:**
+**🚨 عام علامات:**
 {chr(10).join(['• ' + symptom for symptom in ['سینے میں درد یا بے چینی', 'سانس لینے میں دشواری', 'بازوؤں، گردن، جبڑے میں درد', 'متلی', 'تھکاوٹ', 'چکر آنا']])}
 
-**اہم خطرے کے عوامل:**
+**⚠️ خطرے کے عوامل:**
 {chr(10).join(['• ' + factor for factor in ['ہائی بلڈ پریشر', 'ہائی کولیسٹرول', 'سگریٹ نوشی', 'ذیابیطس', 'موٹاپا', 'خاندانی تاریخ', 'جسمانی غیر فعالیت']])}
 
-**احتیاطی کارڈیالوجی سفارشات:**
+**🛡️ بچاؤ کی حکمت عملی:**
 {chr(10).join(['• ' + tip for tip in ['سیر شدہ چکنائی سے پاک صحت مند غذا', 'روزانہ 30 منٹ باقاعدہ ورزش', 'سگریٹ نوشی سے پرہیز', 'وزن کا انتظام', 'تنا�� کا انتظام', 'باقاعدہ چیک اپ']])}
 
-**🚨 قلبی ہنگامی اشارے:**
+**🚑 ہنگامی انتباہی علامات (فوری علاج حاصل کریں):**
 {chr(10).join(['• ' + sign for sign in ['شدید سینے میں درد', 'سانس لینے میں شدید دشواری', 'بیہوش ہونا', 'تیز دل کی دھڑکن', 'بازو/کندھوں میں پھیلنے والا درد']])}
 
-*طبی نوٹ: قلبی بیماری اموات کی سب سے بڑی وجہ بنی ہوئی ہے۔ ابتدائی مداخلت نتائج کو نمایاں طور پر بہتر بناتی ہے۔*
+*🔬 کلینیکل نوٹ: ابتدائی تشخیص جان بچاتی ہے۔ دل کی صحت کے لیے باقاعدہ چیک اپ بہت ضروری ہیں۔*
 """
     
     def _format_diabetes_response(self, language):
         knowledge = self.medical_knowledge_base['diabetes']
         if language == 'English':
             return f"""
-**Dr. SehatAI - Endocrinology Specialist:**
-
-Regarding your diabetes inquiry, here's my clinical perspective:
+**🩺 Comprehensive Diabetes Information**
 
-**Common Diabetic Symptoms:**
+**🚨 Common Symptoms:**
 {chr(10).join(['• ' + symptom for symptom in knowledge['symptoms']])}
 
-**Diabetes Risk Profile:**
+**⚠️ Risk Factors:**
 {chr(10).join(['• ' + factor for factor in knowledge['risk_factors']])}
 
-**Comprehensive Diabetes Management:**
+**🛡️ Prevention & Management:**
 {chr(10).join(['• ' + tip for tip in knowledge['prevention']])}
 
-**🚨 Diabetic Emergency Situations:**
+**🚑 Emergency Signs (Seek Immediate Care):**
 {chr(10).join(['• ' + sign for sign in knowledge['emergency_signs']])}
 
-**Glycemic Control Targets:**
-• Fasting glucose: 80-130 mg/dL
-• Postprandial glucose: <180 mg/dL  
-• HbA1c: <7.0% (individualized)
-• Regular ophthalmologic exams
-• Annual foot examination
+**📊 Monitoring Guidelines:**
+• Fasting blood sugar: 80-130 mg/dL
+• Post-meal blood sugar: <180 mg/dL
+• HbA1c: <7.0% (quarterly if uncontrolled)
+• Regular foot and eye examinations
 
-*Clinical Insight: Consistent glucose monitoring and lifestyle modification are cornerstone to diabetes management.*
+*🔬 Clinical Note: Consistent monitoring and lifestyle management are key to diabetes control.*
 """
         else:
             return f"""
-**ڈاکٹر سیحت اے آئی - اینڈوکرینالوجی اسپیشلسٹ:**
+**🩺 ذیابیطس کی جامع معلومات**
 
-ذیابیطس کے بارے میں آپ کی استفسار کے حوالے سے، یہ میرا کلینیکل نقطہ نظر ہے:
-
-**عام ذیابیطس کی علامات:**
+**🚨 عام علامات:**
 {chr(10).join(['• ' + symptom for symptom in ['زیادہ پیاس اور بھوک لگنا', 'بار بار پیشاب آنا', 'تھکاوٹ', 'دھندلا نظر آنا', 'زخموں کا دیر سے بھرنا', 'بے وجہ وزن میں کمی']])}
 
-**ذیابیطس کا خطرہ پروفائل:**
+**⚠️ خطرے کے عوامل:**
 {chr(10).join(['• ' + factor for factor in ['خاندانی تاریخ', 'وزن زیادہ ہونا (BMI > 25)', 'ہائی بلڈ پریشر', 'جسمانی غیر فعالیت', '45 سال سے زیادہ عمر', 'حمل کی ذیابیطس کی تاریخ']])}
 
-**جامع ذیابیطس مینجمنٹ:**
+**🛡️ بچاؤ اور انتظام:**
 {chr(10).join(['• ' + tip for tip in ['کنٹرول کاربوہائیڈریٹ کے ساتھ متوازن غذا', 'باقاعدہ جسمانی سرگرمی', 'وزن کا انتظام', 'بلڈ شوگر مانیٹرنگ', 'باقاعدہ صحت کی اسکریننگ']])}
 
-**🚨 ذیابیطس کی ہنگامی صورتحال:**
+**🚑 ہنگامی علامات (فوری علاج حاصل کریں):**
 {chr(10).join(['• ' + sign for sign in ['بلڈ شوگر >300 mg/dL یا <70 mg/dL', 'الجھن یا بے ترتیبی', 'پھل کی سی بو والی سانس', 'تیز سانس لینا', 'ہوش کا ضیاع']])}
 
-*کلینیکل بصیرت: مستقل گلوکوز مانیٹرنگ اور طرز زندگی میں ترمیم ذیابیطس کے انتظام کی بنیاد ہیں۔*
+*🔬 کلینیکل نوٹ: مستقل نگرانی اور طرز زندگی کا انتظام ذیابیطس کے کنٹرول کی کلید ہے۔*
 """
     
     def _format_hypertension_response(self, language):
         knowledge = self.medical_knowledge_base['hypertension']
         if language == 'English':
             return f"""
-**Dr. SehatAI - Hypertension Specialist:**
-
-Regarding your blood pressure concerns, here's my medical assessment:
+**💓 Comprehensive Blood Pressure Information**
 
-**Hypertension Symptoms:**
+**🚨 Common Symptoms:**
 {chr(10).join(['• ' + symptom for symptom in knowledge['symptoms']])}
 
-**Hypertension Risk Factors:**
+**⚠️ Risk Factors:**
 {chr(10).join(['• ' + factor for factor in knowledge['risk_factors']])}
 
-**Blood Pressure Management Protocol:**
+**🛡️ Prevention & Control:**
 {chr(10).join(['• ' + tip for tip in knowledge['prevention']])}
 
-**🚨 Hypertensive Crisis Indicators:**
+**🚑 Hypertensive Emergency Signs:**
 {chr(10).join(['• ' + sign for sign in knowledge['emergency_signs']])}
 
-**Blood Pressure Classification:**
-• Normal: <120/<80 mmHg
+**📊 Blood Pressure Classification:**
+• Normal: <120/80 mmHg
 • Elevated: 120-129/<80 mmHg  
-• Stage 1 HTN: 130-139/80-89 mmHg
-• Stage 2 HTN: ≥140/90 mmHg
-• Hypertensive Crisis: >180/>120 mmHg
+• Stage 1 Hypertension: 130-139/80-89 mmHg
+• Stage 2 Hypertension: ≥140/90 mmHg
 
-*Clinical Note: Hypertension is often asymptomatic until complications develop. Regular monitoring is essential.*
+*🔬 Clinical Note: Silent killer - often no symptoms. Regular monitoring is essential.*
 """
         else:
             return f"""
-**ڈاکٹر سیحت اے آئی - ہائی بلڈ پریشر اسپیشلسٹ:**
-
-بلڈ پریشر کے بارے میں آپ کی تشویش کے حوالے سے، یہ میری طبی تشخیص ہے:
+**💓 بلڈ پریشر کی جامع معلومات**
 
-**ہائی بلڈ پریشر کی علامات:**
+**🚨 عام علامات:**
 {chr(10).join(['• ' + symptom for symptom in ['سر درد', 'چکر آنا', 'دھندلا نظر آنا', 'سانس لینے میں دشواری', 'سینے میں درد', 'ناک سے خون آنا']])}
 
-**ہائی بلڈ پریشر کے خطرے کے عوامل:**
+**⚠️ خطرے کے عوامل:**
 {chr(10).join(['• ' + factor for factor in ['خاندانی تاریخ', '65 سال سے زیادہ عمر', 'موٹاپا', 'زیادہ نمک کا استعمال', 'تناؤ', 'شراب کا استعمال', 'سگریٹ نوشی']])}
 
-**بلڈ پریشر مینجمنٹ پروٹوکول:**
+**🛡️ بچاؤ اور کنٹرول:**
 {chr(10).join(['• ' + tip for tip in ['ڈیش ڈائٹ (کم سوڈیم)', 'باقاعدہ ایروبک ورزش', 'وزن کا انتظام', 'تناؤ میں کمی', 'شراب کو محدود کریں', 'سگریٹ نوشی سے پرہیز']])}
 
-**🚨 ہائی بلڈ پریشر کے بحران کے اشارے:**
+**🚑 ہائی بلڈ پریشر کی ہنگامی علامات:**
 {chr(10).join(['• ' + sign for sign in ['شدید سر درد', 'سینے میں درد', 'سانس لینے میں دشواری', 'بینائی میں تبدیلی', 'الجھن', 'دورے']])}
 
-*کلینیکل نوٹ: ہائی بلڈ پریشر اکثر پیچیدگیاں پیدا ہونے تک بے علامت ہوتا ہے۔ باقاعدہ نگرانی ضروری ہے۔*
+*🔬 کلینیکل نوٹ: خاموش قاتل - اکثر کوئی علامات نہیں ہوتیں۔ باقاعدہ نگرانی ضروری ہے۔*
 """
     
     def _format_nutrition_response(self, language):
         if language == 'English':
             return """
-**Dr. SehatAI - Clinical Nutrition Specialist:**
+**🍎 Comprehensive Nutrition Guide**
 
-Based on your nutritional inquiry, here are evidence-based dietary recommendations:
+**🥦 Healthy Eating Principles:**
+• Eat 5-9 servings of fruits/vegetables daily
+• Choose whole grains over refined
+• Include lean proteins
+• Stay hydrated (8-10 glasses water)
+• Limit processed foods and sugar
 
-**Core Nutritional Principles:**
-• Consume 5-9 servings of fruits/vegetables daily
-• Prioritize whole grains over refined carbohydrates
-• Include lean protein sources (poultry, fish, legumes)
-• Maintain hydration (2-3 liters daily)
-• Minimize processed foods and added sugars
+**🎯 Specific Recommendations:**
+• **Heart Health**: Mediterranean diet, omega-3 rich foods
+• **Diabetes**: Controlled carb intake, high fiber
+• **Hypertension**: Low sodium (<2300mg/day), potassium-rich foods
 
-**Condition-Specific Dietary Modifications:**
-• **Cardiac Health**: Mediterranean diet, omega-3 fatty acids
-• **Diabetes**: Controlled carbohydrate intake, high fiber
-• **Hypertension**: DASH diet (<2300mg sodium/day), potassium-rich foods
-
-**Cultural Dietary Adaptations for Pakistan:**
+**💡 Pakistani Dietary Adaptations:**
 • Choose whole wheat chapati over refined flour
-• Incorporate lentils (daal) and legumes regularly
-• Utilize healthy cooking oils (canola, olive, mustard)
-• Limit ghee and saturated fat consumption
+• Include lentils (daal) and legumes regularly
+• Use healthy cooking oils (canola, olive)
+• Limit ghee and butter usage
 
-*Nutritional Note: Balanced nutrition forms the foundation of preventive healthcare and chronic disease management.*
+*Remember: Balanced nutrition is foundation of good health.*
 """
         else:
             return """
-**ڈاکٹر سیحت اے آئی - کلینیکل نیوٹریشن اسپیشلسٹ:**
-
-آپ کی غذائی استفسار کی بنیاد پر، یہ ثبوت پر مبنی غذائی سفارشات ہیں:
+**🍎 غذائیت کی جامع گائیڈ**
 
-**بنیادی غذائی اصول:**
+**🥦 صحت مند کھانے کے اصول:**
 • روزانہ 5-9 سرونگ پھل/سبزیاں کھائیں
-• ریفائنڈ کاربوہائیڈریٹس پر سارا اناج ترجیح دیں
-• لیان پروٹین ذرائع (پولٹری، مچھلی، پھلیاں) شامل کریں
-• ہائیڈریشن برقرار رکھیں (روزانہ 2-3 لیٹر)
-• پراسیسڈ فوڈز اور اضافی چینی کو کم سے کم کریں
+• ریفائنڈ کی بجائے سارا اناج منتخب کریں
+• لیان پروٹین شامل کریں
+• ہائیڈریٹ رہیں (8-10 گلاس پانی)
+• پراسیسڈ فوڈز اور چینی کو محدود کریں
 
-**حالت کی مخصوص غذائی ترامیم:**
-• **قلبی صحت**: بحیرہ روم کی غذا، اومیگا 3 فیٹی ایسڈ
+**🎯 مخصوص سفارشات:**
+• **دل کی صحت**: بحیرہ روم کی غذا، اومیگا 3 سے بھرپور غذائیں
 • **ذیابیطس**: کنٹرول کاربوہائیڈریٹ انٹیک، زیادہ فائبر
-• **ہائی بلڈ پریشر**: ڈیش ڈائٹ (<2300mg سوڈیم/دن)، پوٹاشیم سے بھرپور غذائیں
+• **ہائی بلڈ پریشر**: کم سوڈیم (<2300mg/day)، پوٹاشیم سے بھرپور غذائیں
 
-**پاکستان کے لیے ثقافتی غذائی موافقت:**
-• ریفائنڈ آٹے پر whole wheat چپاتی کو ترجیح دیں
+**💡 پاکستانی غذائی موافقت:**
+• ریفائنڈ آٹے کی بجائے whole wheat چپاتی منتخب کریں
 • دالوں اور پھلیوں کو باقاعدہ شامل کریں
-• صحت مند ککنگ آئلز (کینولا، زیتون، سرسوں) استعمال کریں
-• گھی اور سیر شدہ چکنائی کے استعمال کو محدود کریں
+• صحت مند ککنگ آئلز (کینولا، زیتون) استعمال کریں
+• گھی اور مکھن کے استعمال کو محدود کریں
 
-*غذائی نوٹ: متوازن غذائیت احتیاطی ہیلتھ کیئر اور دائمی بیماری کے انتظام کی بنیاد بنتی ہے۔*
+*یاد رکھیں: متوازن غذائیت اچھی صحت کی بنیاد ہے۔*
 """
     
     def _format_exercise_response(self, language):
         if language == 'English':
             return """
-**Dr. SehatAI - Exercise Physiology Specialist:**
-
-Regarding physical activity, here's my exercise prescription:
-
-**General Exercise Guidelines:**
-• 150 minutes moderate-intensity aerobic exercise weekly
-• Strength training 2-3 times weekly
-• Daily physical activity integration
-• Flexibility and balance exercises
-• Adequate recovery between sessions
-
-**Disease-Specific Exercise Prescriptions:**
-• **Cardiac Patients**: 30 minutes moderate cardio daily, supervised if high-risk
-• **Diabetic Patients**: Combination of aerobic and resistance training
-• **Hypertensive Patients**: Regular aerobic activity, avoid isometric exercises
-
-**Climate Considerations for Pakistan:**
-• Exercise during cooler morning/evening hours
-• Maintain hydration with electrolyte balance
-• Wear lightweight, breathable clothing
-• Monitor for heat-related illness symptoms
-
-*Exercise Physiology Note: Regular physical activity is potent medicine for chronic disease prevention and management.*
+**💪 Comprehensive Exercise Guide**
+
+**🏃‍♂️ Exercise Recommendations:**
+• 150 mins moderate exercise weekly
+• Strength training 2x weekly
+• Stay active throughout day
+• Include flexibility exercises
+• Get adequate rest between workouts
+
+**🎯 Disease-Specific Exercise:**
+• **Heart Health**: 30 mins moderate cardio daily
+• **Diabetes**: Combination of cardio and strength training
+• **Hypertension**: Regular aerobic exercise
+
+**🌞 Pakistani Climate Considerations:**
+• Exercise in cooler morning/evening hours
+• Stay hydrated with water and electrolytes
+• Wear light, breathable clothing
+
+*Important: Consult doctor before starting new exercise program.*
 """
         else:
             return """
-**ڈاکٹر سیحت اے آئی - ورزش فزیالوجی اسپیشلسٹ:**
-
-جسمانی سرگرمی کے بارے میں، یہ میری ورزش کی نسخہ ہے:
+**💪 ورزش کی جامع گائیڈ**
 
-**عام ورزش کے رہنما اصول:**
-• ہفتے میں 150 منٹ اعتدال پسند شدت کی ایروبک ورزش
-• ہفتے میں 2-3 بار طاقت کی تربیت
-• روزانہ جسمانی سرگرمی کا انضمام
-• لچک اور توازن کی ورزشیں
-• سیشن کے درمیان مناسب بحالی
+**🏃‍♂️ ورزش کی سفارشات:**
+• ہفتے میں 150 منٹ اعتدال پسند ورزش
+• ہفتے میں 2 بار طاقت کی تربیت
+• پورے دن متحرک رہیں
+• لچک کی ورزشیں شامل کریں
+• ورک آؤٹس کے درمیان مناسب آرام حاصل کریں
 
-**بیماری کی مخصوص ورزش کے نسخے:**
-• **قلبی مریض**: روز��نہ 30 منٹ اعتدال پسند کارڈیو، اگر اعلی خطرہ ہو تو نگرانی میں
-• **ذیابیطس کے مریض**: ایروبک اور مزاحمتی تربیت کا مجموعہ
-• **ہائی بلڈ پریشر کے مریض**: باقاعدہ ایروبک سرگرمی، آئسومیٹرک ورزشوں سے پرہیز
+**🎯 بیماری کی مخصوص ورزش:**
+• **دل کی صحت**: روزانہ 30 منٹ اعتدال پسند کارڈیو
+• **ذیابیطس**: کارڈیو اور طاقت کی تربیت کا مجموعہ
+• **ہائی بلڈ پریشر**: باقاعدہ ایروبک ورزش
 
-**پاکستان کے لیے آب و ہیا کے تحفظات:**
+**🌞 پاکستانی آب و ہوا کے تحفظات:**
 • ٹھنڈے صبح/شام کے اوقات میں ورزش کریں
-• الیکٹرولائٹ توازن کے ساتھ ہائیڈریشن برقرار رکھیں
-• ہلکے وزن، ہوا دار کپڑے پہنیں
-• گرمی سے متعلق بیماری کی علامات کی نگرانی کریں
+• پانی اور الیکٹرولائٹس کے ساتھ ہائیڈریٹ رہیں
+• ہلکے، ہوا دار کپڑے پہنیں
 
-*ورزش فزیالوجی نوٹ: باقاعدہ جسمانی سرگرمی دائمی بیماری کی روک تھام اور انتظام کے لیے طاقتور دوا ہے۔*
+*اہم: نئی ورزش پروگرام شروع کرنے سے پہلے ڈاکٹر سے مشورہ کریں۔*
 """
     
     def _format_symptoms_response(self, language):
         if language == 'English':
             return """
-**Dr. SehatAI - Clinical Assessment Specialist:**
-
-Regarding symptom evaluation, here's my clinical guidance:
-
-**When to Seek Medical Evaluation:**
-• Persistent fever >103°F (39.4°C) unresponsive to antipyretics
-• Respiratory distress or chest pain
-• Severe abdominal pain or trauma
-• Acute neurological changes (weakness, confusion, vision changes)
-• Unexplained weight loss >10% body weight in 6 months
-• Symptoms persisting beyond 2 weeks without improvement
-
-**🚨 Emergency Department Indicators:**
-• Acute respiratory distress or stridor
-• Chest pain with radiation, diaphoresis, or nausea
-• Uncontrolled hemorrhage
-• Sudden severe pain anywhere
-• Altered mental status or syncope
-• Seizure activity
-
-**Clinical Assessment Protocol:**
-• Comprehensive history taking
-• Vital signs assessment
-• Systematic physical examination
-• Appropriate diagnostic testing
-• Specialist referral when indicated
-
-*Medical Note: Symptom assessment requires professional medical evaluation for accurate diagnosis and treatment.*
+**🔍 Symptom Assessment Guidance**
+
+**🩺 When to Seek Medical Care:**
+• Persistent fever above 103°F (39.4°C)
+• Difficulty breathing or chest pain
+• Severe abdominal pain
+• Sudden dizziness, weakness, or confusion
+• Unexplained weight loss (>10% body weight)
+• Symptoms lasting more than 2 weeks
+
+**🚨 Emergency Warning Signs (Go to ER):**
+• Difficulty breathing or shortness of breath
+• Chest pain or pressure lasting >5 minutes
+• Severe bleeding that doesn't stop
+• Sudden severe pain anywhere in body
+• Loss of consciousness or fainting
+
+*⚠️ Important: This is general guidance. Always consult healthcare professional for proper diagnosis.*
 """
         else:
             return """
-**ڈاکٹر سیحت اے آئی - کلینیکل تشخیص اسپیشلسٹ:**
-
-علامات کی تشخیص کے بارے میں، یہ میری کلینیکل رہنمائی ہے:
-
-**طبی تشخیص کب حاصل کریں:**
-• اینٹیپائریٹکس کے لیے غیر receptive مستقل بخار >103°F (39.4°C)
-• سانس کی تکلیف یا سینے میں درد
-• شدید پیٹ کا درد یا چوٹ
-• شدید اعصابی تبدیلیاں (کمزوری، الجھن، بینائی میں تبدیلی)
-• 6 ماہ میں جسمانی وزن کا >10% بے وجہ وزن میں کمی
-• بہتری کے بغیر 2 ہفتے سے زیادہ علامات برقرار رہنا
-
-**🚨 ایمرجنسی ڈیپارٹمنٹ کے اشارے:**
-• شدید سانس کی تکلیف یا سٹرائیڈر
-• تابکاری، پسینہ آنا، یا متلی کے ساتھ سینے میں درد
-• غیر کنٹرول خون بہنا
-• کہیں بھی اچانک شدید درد
-• تبدیل ذہنی حیثیت یا سنکوپ
-• دورے کی سرگرمی
-
-**کلینیکل تشخیص پروٹوکول:**
-• جامع تاریخ لینا
-• اہم علامات کی تشخیص
-• نظامی جسمانی معائنہ
-• مناسب تشخیصی ٹیسٹنگ
-• اشارہ کردہ ہو تو ماہر کی رجوع
-
-*طبی نوٹ: علامات کی تشخیص کے لیے درست تشخیص اور علاج کے لیے پیشہ ورانہ طبی تشخیص کی ضرورت ہوتی ہے۔*
+**🔍 علامات کی تشخیص کی رہنمائی**
+
+**🩺 طبی دیکھ بھال کب حاصل کریں:**
+• 103°F (39.4°C) سے اوپر مستقل بخار
+• سانس لینے میں دشواری یا سینے میں درد
+• شدید پیٹ کا درد
+• اچانک چکر آنا، کمزوری، یا الجھن
+• بے وجہ وزن میں کمی (>10% جسمانی وزن)
+• 2 ہفتے سے زیادہ علامات برقرار رہنا
+
+**🚨 ہنگامی انتباہی علامات (ایمرجنسی روم جائیں):**
+• سانس لینے میں دشواری یا سانس کی قلت
+• 5 منٹ سے زیادہ سینے میں درد یا دباؤ
+• شدید خون بہنا جو رک نہیں رہا
+• جسم میں کہیں بھی اچانک شدید درد
+• ہوش کا ضیاع یا بیہوشی
+
+*⚠️ اہم: یہ عام رہنمائی ہے۔ مناسب تشخیص کے لیے ہمیشہ ہیلتھ کیئر پروفیشنل سے مشورہ کریں۔*
 """
     
     def _format_medication_response(self, language):
         if language == 'English':
             return """
-**Dr. SehatAI - Clinical Pharmacology Specialist:**
-
-Regarding medication management, here's my pharmacological guidance:
-
-**Medication Safety Protocol:**
-• Adhere strictly to prescribed dosing schedules
-• Never share prescription medications
-• Maintain current medication list including OTC products
-• Disclose all medications to all healthcare providers
-• Understand therapeutic goals and potential adverse effects
-
-**🚨 Medication Adverse Reaction Indicators:**
-• Anaphylaxis (urticaria, angioedema, respiratory distress)
-• Bronchospasm or wheezing post-medication
-• Severe gastrointestinal symptoms
-• Unexplained bleeding or bruising
-• Neurological changes (confusion, hallucinations)
-
-**Medication Management Principles:**
-• Regular medication reconciliation
-• Adherence assessment at each visit
-• Monitoring for drug interactions
-• Appropriate storage and disposal
-• Understanding administration techniques
-
-*Pharmacology Note: Medication management requires ongoing professional supervision for optimal therapeutic outcomes.*
+**💊 Medication Safety & Information**
+
+**📋 General Medication Guidelines:**
+• Take medications exactly as prescribed
+• Never share prescriptions with others
+• Keep updated medication list with you
+• Inform all doctors about all medications
+• Understand purpose and side effects of each medicine
+
+**🚨 Medication Warning Signs:**
+• Severe allergic reactions (rash, swelling)
+• Difficulty breathing after taking medicine
+• Severe stomach pain or vomiting
+• Unusual bleeding or bruising
+• Confusion or hallucinations
+
+*Note: Always follow your healthcare provider's specific instructions.*
 """
         else:
             return """
-**ڈاکٹر سیحت اے آئی - کلینیکل فارماکالوجی اسپیشلسٹ:**
-
-دوائیوں کے انتظام کے بارے میں، یہ میری فارماکولوجیکل رہنمائی ہے:
-
-**دوائیوں کی حفاظت کا پروٹوکول:**
-• تجویز کردہ خوراک کے شیڈول پر سختی سے عمل کریں
-• کبھی بھی نسخہ کی دوائیں شیئر نہ کریں
-• OTC مصنوعات سمیت موجودہ دوائیوں کی فہرست برقرار رکھیں
-• تمام ہیلتھ کیئر فراہم کرنے والوں کو تمام دوائیں بتائیں
-• علاج کے مقاصد اور ممکنہ مضر اثرات کو سمجھیں
-
-**🚨 دوائیوں کے مضر رد عمل کے اشارے:**
-• انافیلیکسس (urticaria، angioedema، سانس کی تکلیف)
-• دوائی کے بعد برونکوسپازم یا wheezing
-• شدید gastrointestinal علامات
-• بے وجہ خون بہنا یا چوٹ لگنا
-• اعصابی تبدیلیاں (الجھن، توہمات)
-
-**دوائیوں کے انتظام کے اصول:**
-• باقاعدہ دوائیوں کی مطابقت پذیری
-• ہر وزٹ پر پابندی کی تشخیص
-• دوائیوں کے تعاملات کی نگرانی
-• مناسب اسٹوریج اور ضائع
-• انتظامیہ کی تکنیکوں کو سمجھنا
-
-*فارماکالوجی نوٹ: دوائیوں کے انتظام کے لیے بہترین علاج کے نتائج کے لیے مسلسل پیشہ ورانہ نگرانی کی ضرورت ہوتی ہے۔*
+**💊 دوائیوں کی حفاظت اور معلومات**
+
+**📋 عام دوائیوں کے رہنما اصول:**
+• دوائیں بالکل تجویز کردہ طریقے سے لیں
+• کبھی بھی دوسروں کے ساتھ نسخے شیئر نہ کریں
+• اپنے ساتھ اپ ڈیٹ شدہ دوائیوں کی فہرست رکھیں
+• تمام ڈاکٹروں کو تمام دوائیوں کے بارے میں بتائیں
+• ہر دوا کے مقصد اور مضر اثرات کو سمجھیں
+
+**🚨 دوائیوں کی انتباہی علامات:**
+• شدید الرجک رد عمل (خارش، سوجن)
+• دوا لینے کے بعد سانس لینے میں دشواری
+• شدید پیٹ میں درد یا الٹی
+• غیر معمولی خون بہنا یا چوٹ لگنا
+• الجھن یا توہمات
+
+*نوٹ: ہمیشہ اپنے ہیلتھ کیئر فراہم کرنے والے کی مخصوص ہدایات پر عمل کریں۔*
 """
     
     def _format_emergency_response(self, language):
         if language == 'English':
             return """
-**Dr. SehatAI - Emergency Medicine Protocol:**
-
-**🚨 IMMEDIATE MEDICAL ATTENTION REQUIRED**
-
-**Emergency Response Protocol:**
-• Activate emergency medical services (1122 in Pakistan)
-• Do not attempt self-transport to hospital
-• Maintain patient in position of comfort
-• Prepare relevant medical information
-• Withhold oral intake
-
-**Emergency Department Presentation Indicators:**
-• Acute chest pain or pressure
-• Respiratory distress or hypoxia
-• Uncontrolled hemorrhage
-• Acute neurological deficit
-• Major trauma or burns
-• Suspected poisoning or overdose
-• Seizure activity
-
-**Pre-Hospital Management:**
-• Ensure scene safety
-• Basic life support if trained
-• Continuous monitoring of vital signs
-• Preparation for EMS arrival
-• Documentation of event timeline
-
-**⚠️ CRITICAL: This constitutes emergency medical guidance only. Seek immediate professional medical evaluation.**
+**🚨 EMERGENCY MEDICAL GUIDANCE**
+
+**📞 Immediate Actions Required:**
+• Call emergency services (1122 in Pakistan)
+• Don't drive yourself to hospital
+• Keep patient calm and comfortable
+• Have medical information ready
+• Don't give food or drink
+
+**🏥 When to Go to Emergency Room:**
+• Chest pain or pressure
+• Difficulty breathing
+• Severe bleeding
+• Sudden weakness or paralysis
+• Severe burns
+• Poisoning or overdose
+• Seizures
+
+*⚠️ THIS IS EMERGENCY GUIDANCE ONLY. SEEK PROFESSIONAL MEDICAL HELP IMMEDIATELY.*
 """
         else:
             return """
-**ڈاکٹر سیحت اے آئی - ایمرجنسی میڈیسن پروٹوکول:**
-
-**🚨 فوری طبی توجہ درکار**
-
-**ہنگامی ردعمل پروٹوکول:**
-• ہنگامی طبی خدمات کو چالو کریں (پاکستان میں 1122)
-• ہسپتال خود نقل و حمل کی کوشش نہ کریں
-• مریض کو آرام کی پوزیشن میں رکھیں
-• متعلقہ طبی معلومات تیار کریں
-• زبانی انٹیک کو روکیں
-
-**ایمرجنسی ڈیپارٹمنٹ پیشکش کے اشارے:**
-• شدید سینے میں درد یا دباؤ
-• سانس کی تکلیف یا hypoxia
-• غیر کنٹرول خون بہنا
-• شدید اعصابی خسارہ
-• بڑی چوٹ یا جلنے
-• مشتبہ زہر یا اوور ڈوز
-• دورے کی سرگرمی
-
-**ہسپتال سے پہلے کے انتظام:**
-• منظر کی حفاظت کو یقینی بنائیں
-• اگر تربیت یافتہ ہوں تو بنیادی زندگی کی حمایت
-• اہم علامات کی مسلسل نگرانی
-• EMS کی آمد کی تیاری
-• واقعہ کی ٹائم لائن کی دستاویزات
-
-**⚠️ اہم: یہ صرف ہنگامی طبی رہنمائی پر مشتمل ہے۔ فوری پیشہ ورانہ طبی تشخیص حاصل کریں۔**
+**🚨 ہنگامی طبی رہنمائی**
+
+**📞 فوری کارروائی کی ضرورت ہے:**
+• ہنگامی خدمات کو کال کریں (پاکستان میں 1122)
+• خود ہسپتال نہ جائیں
+• مریض کو پرسکون اور آرام دہ رکھیں
+• طبی معلومات تیار رکھیں
+• کھانا یا پانی نہ دیں
+
+**🏥 ایمرجنسی روم کب جائیں:**
+• سینے میں درد یا دباؤ
+• سانس لینے میں دشواری
+• شدید خون بہنا
+• اچانک کمزوری یا فالج
+• شدید جلنے
+• زہر یا اوور ڈوز
+• دورے
+
+*⚠️ یہ صرف ہنگامی رہنمائی ہے۔ فوری طور پر پیشہ ورانہ طبی مدد حاصل کریں۔*
 """
     
     def _format_general_health_response(self, language):
         if language == 'English':
             return """
-**Dr. SehatAI - Preventive Medicine Specialist:**
-
-Regarding general health and wellness, here's my comprehensive preventive guidance:
-
-**Lifestyle Medicine Recommendations:**
-• Sleep hygiene: 7-9 hours quality sleep nightly
-• Stress management: Mindfulness, meditation, coping strategies
-• Tobacco cessation: Complete avoidance
-• Alcohol moderation: ≤1 drink/day women, ≤2 drinks/day men
-• Regular preventive health screenings
-
-**Preventive Healthcare Protocol:**
-• Annual comprehensive physical examination
-• Age-appropriate vaccination schedule
-• Hand hygiene and infection control
-• Sun protection and skin surveillance
-• Mental health screening and support
-
-**Public Health Considerations for Pakistan:**
-• Access to clean drinking water
-• Consumption of fresh, seasonal produce
-• Food safety and hygiene practices
-• Adequate vitamin D synthesis through sensible sun exposure
-
-*Preventive Medicine Insight: Sustainable health improvements result from consistent lifestyle modifications and regular preventive care.*
+**🌿 Comprehensive Health & Wellness Guide**
+
+**💚 Lifestyle Recommendations:**
+• 7-9 hours quality sleep nightly
+• Stress management techniques
+• No smoking
+• Limit alcohol to 1-2 drinks/day
+• Regular health check-ups
+
+**🛡️ Preventive Healthcare:**
+• Annual physical exams
+• Vaccinations up to date
+• Regular hand washing
+• Sun protection
+• Mental health care
+
+**🌟 Healthy Habits for Pakistan:**
+• Drink filtered/boiled water
+• Eat fresh, locally available fruits and vegetables
+• Practice good food hygiene
+• Get adequate sunlight exposure (vitamin D)
+
+*Remember: Small consistent changes lead to lasting health improvements!*
 """
         else:
             return """
-**ڈاکٹر سیحت اے آئی - احتیاطی میڈیسن اسپیشلسٹ:**
-
-عام صحت اور تندرستی کے بارے میں، یہ میری جامع احتیاطی رہنمائی ہے:
-
-**طرز زندگی کی میڈیسن سفارشات:**
-• نیند کی حفظان صحت: رات میں 7-9 گھنٹے معیاری نیند
-• تناؤ کا انتظام: مائنڈفلنیس، مراقبہ، نمٹنے کی حکمت عملی
-• تمباکو نوشی بند کرنا: مکمل پرہیز
-• شراب اعتدال: ≤1 ڈرنک/دن خواتین، ≤2 ڈرنکس/دن مرد
-• باقاعدہ احتیاطی صحت کی اسکریننگ
-
-**احتیاطی ہیلتھ کیئر پروٹوکول:**
-• سالانہ جامع جسمانی معائنہ
-• عمر کے مناسب ویکسینیشن شیڈول
-• ہاتھ کی حفظان صحت اور انفیکشن کنٹرول
-• سورج سے تحفظ اور جلد کی نگرانی
-• ذہنی صحت کی اسکریننگ اور حمایت
-
-**پاکستان کے لیے عوامی صحت کے تحفظات:**
-• صاف پینے کے پانی تک رسائی
-• تازہ، موسمی پیداوار کی کھپت
-• غذائی تحفظ اور حفظان صحت کے طریقے
-• سمجھدار سورج کی نمائش کے ذریعے مناسب وٹامن ڈی ترکیب
-
-*احتیاطی میڈیسن بصیرت: پائیدار صحت کی بہتری مسلسل طرز زندگی میں ترمیم اور باقاعدہ احتیاطی دیکھ بھال کا نتیجہ ہے۔*
-"""
+**🌿 صحت اور تندرستی کی جامع گائیڈ**
+
+**💚 طرز زندگی کی سفارشات:**
+• رات میں 7-9 گھنٹے معیاری نیند
+• تناؤ کے انتظام کی تکنیکیں
+• سگریٹ نوشی سے پرہیز
+• شراب کو 1-2 ڈرنکس/دن تک محدود کریں
+• باقاعدہ صحت کی جانچ
+
+**🛡️ احتیاطی ہیلتھ کیئر:**
+• سالانہ جسمانی امتحانات
+• ویکسینیشن اپ ڈیٹ
+• باقاعدہ ہاتھ دھونا
+• سورج سے تحفظ
+• ذہنی صحت کی دیکھ بھال
+
+**🌟 پاکستان کے لیے صحت مند عادات:**
+• فلٹر/ابلا ہوا پانی پیئیں
+• تازہ، مقامی طور پر دستیاب پھل اور سبزیاں کھائیں
+• اچھی غذا کی حفظان صحت کی مشق کریں
+• مناسب سورج کی روشنی (وٹامن ڈی) حاصل کریں
+
+*یاد رکھیں: چھوٹی مسلسل تبدیلیاں پائیدار صحت کی بہتری کا باعث بنتی ہیں!*
+"""
+    
+    def _format_default_response(self, language):
+        if language == 'English':
+            return """
+**🤖 SehatAI Health Assistant**
+
+I understand you're asking about health topics. I'm specially trained to help with:
+
+**💚 Common Health Areas:**
+• Heart disease and cardiovascular health
+• Diabetes management and prevention  
+• Blood pressure control
+• Nutrition and dietary guidance
+• Exercise and physical activity
+• Symptom assessment
+• Medication information
+• Emergency situations
+• General wellness and prevention
+
+Please ask specific questions for detailed medical information.
+
+*⚠️ Remember: I'm an AI assistant. For personal medical advice, always consult qualified healthcare professionals.*
+"""
+        else:
+            return """
+**🤖 سیحت اے آئی ہیلتھ اسسٹنٹ**
+
+میں سمجھتا ہوں کہ آپ صحت کے موضوعات کے بارے میں پوچھ رہے ہیں۔ میں خاص طور پر مدد کے لیے تربیت یافتہ ہوں:
+
+**💚 عام صحت کے شعبے:**
+• دل کی بیماری اور قلبی صحت
+• ذیابیطس کا انتظام اور روک تھام
+• بلڈ پریشر کنٹرول
+• غذائیت اور غذائی رہنمائی
+• ورزش اور جسمانی سرگرمی
+• علامات کی تشخیص
+• دوائ��وں کی معلومات
+• ہنگامی حالات
+• عام تندرستی اور روک تھام
+
+براہ کرم تفصیلی طبی معلومات کے لیے مخصوص سوالات پوچھیں۔
+
+*⚠️ یاد رکھیں: میں ایک AI اسسٹنٹ ہوں۔ ذاتی طبی مشورے کے لیے، ہمیشہ کوالیفائیڈ ہیلتھ کیئر پروفیشنلز سے مشورہ کریں۔*
+"""
+
+class PDFReportGenerator:
+    def __init__(self):
+        self.pdf = FPDF()
+        self.pdf.set_auto_page_break(auto=True, margin=15)
+        
+    def generate_report(self, patient_data, risk_scores, language='English'):
+        self.pdf.add_page()
+        
+        # Title
+        self.pdf.set_font('Arial', 'B', 16)
+        if language == 'English':
+            self.pdf.cell(0, 10, 'SEHATAI MEDICAL ASSESSMENT REPORT', 0, 1, 'C')
+        else:
+            self.pdf.cell(0, 10, 'SEHATAI MEDICAL ASSESSMENT REPORT', 0, 1, 'C')
+        
+        self.pdf.ln(10)
+        
+        # Patient Information
+        self.pdf.set_font('Arial', 'B', 12)
+        self.pdf.cell(0, 10, 'Patient Information:', 0, 1)
+        
+        self.pdf.set_font('Arial', '', 10)
+        patient_info = [
+            f"Name: {self._safe_encode(patient_data.get('name', 'Not provided'))}",
+            f"Age: {patient_data.get('age', 'Not provided')} years",
+            f"Gender: {self._safe_encode(patient_data.get('gender', 'Not provided'))}",
+            f"Contact: {self._safe_encode(patient_data.get('contact', 'Not provided'))}"
+        ]
+        
+        for info in patient_info:
+            self.pdf.cell(0, 8, info, 0, 1)
+        
+        self.pdf.ln(5)
+        
+        # Clinical Parameters
+        self.pdf.set_font('Arial', 'B', 12)
+        self.pdf.cell(0, 10, 'Clinical Parameters:', 0, 1)
+        
+        self.pdf.set_font('Arial', '', 10)
+        clinical_info = [
+            f"Blood Pressure: {patient_data.get('bp_systolic', 'N/A')}/{patient_data.get('bp_diastolic', 'N/A')} mmHg",
+            f"Heart Rate: {patient_data.get('heart_rate', 'N/A')} bpm",
+            f"Cholesterol: {patient_data.get('cholesterol', 'N/A')} mg/dL",
+            f"Glucose: {patient_data.get('glucose', 'N/A')} mg/dL",
+            f"BMI: {patient_data.get('bmi', 'N/A')}"
+        ]
+        
+        for info in clinical_info:
+            self.pdf.cell(0, 8, info, 0, 1)
+        
+        self.pdf.ln(5)
+        
+        # Symptoms
+        self.pdf.set_font('Arial', 'B', 12)
+        self.pdf.cell(0, 10, 'Reported Symptoms:', 0, 1)
+        
+        self.pdf.set_font('Arial', '', 10)
+        symptoms = patient_data.get('symptoms', {})
+        symptom_list = [
+            f"Chest Pain: {'Yes' if symptoms.get('chest_pain') else 'No'}",
+            f"Shortness of Breath: {'Yes' if symptoms.get('shortness_breath') else 'No'}",
+            f"Palpitations: {'Yes' if symptoms.get('palpitations') else 'No'}",
+            f"Fatigue: {'Yes' if symptoms.get('fatigue') else 'No'}",
+            f"Dizziness: {'Yes' if symptoms.get('dizziness') else 'No'}",
+            f"Blurred Vision: {'Yes' if symptoms.get('blurred_vision') else 'No'}"
+        ]
+        
+        for symptom in symptom_list:
+            self.pdf.cell(0, 8, symptom, 0, 1)
+        
+        self.pdf.ln(5)
+        
+        # Risk Assessment
+        self.pdf.set_font('Arial', 'B', 12)
+        self.pdf.cell(0, 10, 'Risk Assessment Results:', 0, 1)
+        
+        self.pdf.set_font('Arial', '', 10)
+        risk_info = [
+            f"Heart Disease Risk: {risk_scores.get('heart', 0):.1%}",
+            f"Diabetes Risk: {risk_scores.get('diabetes', 0):.1%}",
+            f"Hypertension Risk: {risk_scores.get('hypertension', 0):.1%}",
+            f"Overall Priority Score: {risk_scores.get('priority', 0):.1%}"
+        ]
+        
+        for info in risk_info:
+            self.pdf.cell(0, 8, info, 0, 1)
+        
+        self.pdf.ln(5)
+        
+        # Recommendation
+        self.pdf.set_font('Arial', 'B', 12)
+        self.pdf.cell(0, 10, 'Clinical Recommendation:', 0, 1)
+        
+        self.pdf.set_font('Arial', '', 10)
+        recommendation = self._safe_encode(risk_scores.get('recommendation', 'No recommendation available'))
+        self.pdf.multi_cell(0, 8, recommendation)
+        
+        self.pdf.ln(10)
+        
+        # Footer
+        self.pdf.set_font('Arial', 'I', 8)
+        self.pdf.cell(0, 10, f'Generated by SehatAI on: {datetime.now().strftime("%Y-%m-%d %H:%M")}', 0, 1, 'C')
+        
+        # Return the PDF as bytes
+        return self.pdf.output()
+    
+    def _safe_encode(self, text):
+        """Safely encode text for PDF generation"""
+        if text is None:
+            return ""
+        try:
+            return str(text)
+        except:
+            return "Encoding error"
+
+# Load ML models with enhanced error handling
+@st.cache_resource(show_spinner=False)
+def load_models():
+    models = {}
+    
+    # Try multiple approaches to load models
+    model_files = {
+        'heart': ['heart_model.pkl', 'heart_disease_model.pkl', 'models/heart_model.pkl'],
+        'diabetes': ['diabetes_model.pkl', 'diabetes_model.pkl', 'models/diabetes_model.pkl'],
+        'hypertension': ['hypertension_model.pkl', 'models/hypertension_model.pkl']
+    }
+    
+    def try_load_model(model_name, file_list):
+        for file_path in file_list:
+            if os.path.exists(file_path):
+                try:
+                    model = joblib.load(file_path)
+                    # Success message removed from interface as requested
+                    return model
+                except Exception as e:
+                    continue
+        return None
+    
+    models['heart'] = try_load_model("Heart Disease", model_files['heart'])
+    models['diabetes'] = try_load_model("Diabetes", model_files['diabetes'])
+    models['hypertension'] = try_load_model("Hypertension", model_files['hypertension'])
+    
+    # If no models found, use rule-based
+    if not any(models.values()):
+        st.info("🔍 Using advanced rule-based assessment (no ML models found)")
+    
+    return models
+
+def calculate_priority_score(heart_risk, diabetes_risk, hypertension_risk):
+    """Calculate integrated priority score with clinical weighting"""
+    # Clinical severity weighting
+    priority = (
+        heart_risk * 0.45 +      # Highest weight for cardiac issues
+        diabetes_risk * 0.25 +   # Medium weight for diabetes
+        hypertension_risk * 0.30 # Medium weight for hypertension
+    )
+    
+    return min(1.0, priority)  # Cap at 1.0
+
+def get_priority_recommendation(priority_score, language='English'):
+    """Get priority-based recommendation with clinical thresholds"""
+    if priority_score >= 0.75:
+        if language == 'Urdu':
+            return "EMERGENCY_CARE", "اعلی ترجیح - فوری ہنگامی علاج کی ضرورت", "risk-high"
+        else:
+            return "EMERGENCY_CARE", "High Priority - Immediate Emergency Care Required", "risk-high"
+    elif priority_score >= 0.55:
+        if language == 'Urdu':
+            return "SAME_DAY_CONSULT", "درمیانی ترجیح - اسی دن مشورہ ضروری", "risk-medium"
+        else:
+            return "SAME_DAY_CONSULT", "Medium Priority - Same Day Consultation Required", "risk-medium"
+    else:
+        if language == 'Urdu':
+            return "ROUTINE_APPOINTMENT", "کم ترجیح - روٹین اپائنٹمنٹ", "risk-low"
+        else:
+            return "ROUTINE_APPOINTMENT", "Low Priority - Routine Appointment", "risk-low"
+
+def validate_patient_data(age, bp_systolic, bp_diastolic, heart_rate):
+    """Validate patient data for realistic clinical values"""
+    errors = []
+    
+    if age < 1 or age > 120:
+        errors.append("Age must be between 1 and 120 years")
+    if bp_systolic < 70 or bp_systolic > 250:
+        errors.append("Systolic BP must be between 70 and 250 mmHg")
+    if bp_diastolic < 40 or bp_diastolic > 150:
+        errors.append("Diastolic BP must be between 40 and 150 mmHg")
+    if heart_rate < 30 or heart_rate > 200:
+        errors.append("Heart rate must be between 30 and 200 bpm")
+    
+    return errors
+
+def predict_with_models(age, bp_systolic, bp_diastolic, heart_rate, cholesterol, glucose, bmi, symptoms, models):
+    """Predict risks using loaded ML models with fallback to rule-based"""
+    heart_risk, diabetes_risk, hypertension_risk = 0.0, 0.0, 0.0
+    
+    # Heart disease prediction
+    if models.get('heart') is not None:
+        try:
+            heart_features = np.array([[
+                age, bp_systolic, cholesterol, heart_rate,
+                symptoms['chest_pain'], symptoms['shortness_breath'], 
+                symptoms['palpitations'], bmi, glucose
+            ]])
+            heart_risk = models['heart'].predict_proba(heart_features)[0][1]
+        except Exception as e:
+            heart_risk = predict_risk_rule_based_heart(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms)
+    else:
+        heart_risk = predict_risk_rule_based_heart(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms)
+    
+    # Diabetes prediction
+    if models.get('diabetes') is not None:
+        try:
+            diabetes_features = np.array([[
+                age, glucose, bmi, cholesterol,
+                symptoms['fatigue'], symptoms['blurred_vision'],
+                bp_systolic, heart_rate
+            ]])
+            diabetes_risk = models['diabetes'].predict_proba(diabetes_features)[0][1]
+        except Exception as e:
+            diabetes_risk = predict_risk_rule_based_diabetes(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms)
+    else:
+        diabetes_risk = predict_risk_rule_based_diabetes(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms)
+    
+    # Hypertension prediction
+    if models.get('hypertension') is not None:
+        try:
+            hypertension_features = np.array([[
+                age, bp_systolic, bp_diastolic, bmi,
+                symptoms['dizziness'], symptoms['palpitations'],
+                heart_rate, cholesterol
+            ]])
+            hypertension_risk = models['hypertension'].predict_proba(hypertension_features)[0][1]
+        except Exception as e:
+            hypertension_risk = predict_risk_rule_based_hypertension(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms)
+    else:
+        hypertension_risk = predict_risk_rule_based_hypertension(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms)
+    
+    return heart_risk, diabetes_risk, hypertension_risk
+
+def predict_risk_rule_based_heart(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms):
+    """Advanced rule-based heart risk prediction"""
+    risk = 0.0
+    
+    # Age factor
+    if age > 60: risk += 0.25
+    elif age > 45: risk += 0.15
+    elif age > 30: risk += 0.05
+    
+    # Blood pressure factor
+    if bp_systolic > 180 or bp_diastolic > 120: risk += 0.30
+    elif bp_systolic > 140 or bp_diastolic > 90: risk += 0.20
+    elif bp_systolic > 130 or bp_diastolic > 85: risk += 0.10
+    
+    # Cholesterol factor
+    if cholesterol > 240: risk += 0.20
+    elif cholesterol > 200: risk += 0.10
+    
+    # BMI factor
+    if bmi > 35: risk += 0.15
+    elif bmi > 30: risk += 0.10
+    elif bmi > 25: risk += 0.05
+    
+    # Glucose factor
+    if glucose > 200: risk += 0.10
+    elif glucose > 126: risk += 0.05
+    
+    # Symptom factors
+    if symptoms['chest_pain']: risk += 0.25
+    if symptoms['shortness_breath']: risk += 0.15
+    if symptoms['palpitations']: risk += 0.10
+    
+    return min(0.95, max(0.05, risk))
+
+def predict_risk_rule_based_diabetes(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms):
+    """Advanced rule-based diabetes risk prediction"""
+    risk = 0.0
+    
+    # Age factor
+    if age > 50: risk += 0.20
+    elif age > 40: risk += 0.10
+    elif age > 30: risk += 0.05
+    
+    # Glucose factor (most important)
+    if glucose > 200: risk += 0.40
+    elif glucose > 126: risk += 0.30
+    elif glucose > 100: risk += 0.15
+    
+    # BMI factor
+    if bmi > 35: risk += 0.25
+    elif bmi > 30: risk += 0.15
+    elif bmi > 25: risk += 0.08
+    
+    # Blood pressure factor
+    if bp_systolic > 140 or bp_diastolic > 90: risk += 0.10
+    
+    # Symptom factors
+    if symptoms['fatigue']: risk += 0.10
+    if symptoms['blurred_vision']: risk += 0.08
+    
+    return min(0.95, max(0.05, risk))
+
+def predict_risk_rule_based_hypertension(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms):
+    """Advanced rule-based hypertension risk prediction"""
+    risk = 0.0
+    
+    # Blood pressure factor (most important)
+    if bp_systolic > 180 or bp_diastolic > 120: risk += 0.50
+    elif bp_systolic > 140 or bp_diastolic > 90: risk += 0.35
+    elif bp_systolic > 130 or bp_diastolic > 85: risk += 0.20
+    
+    # Age factor
+    if age > 60: risk += 0.15
+    elif age > 45: risk += 0.08
+    
+    # BMI factor
+    if bmi > 30: risk += 0.15
+    elif bmi > 25: risk += 0.08
+    
+    # Symptom factors
+    if symptoms['dizziness']: risk += 0.10
+    if symptoms['palpitations']: risk += 0.08
+    
+    return min(0.95, max(0.05, risk))
+
+def create_download_link(pdf_bytes, filename, text):
+    """Create a download link for PDF"""
+    b64 = base64.b64encode(pdf_bytes).decode()
+    href = f'<a href="data:application/octet-stream;base64,{b64}" download="{filename}">{text}</a>'
+    return href
+
+def main():
+    # Load custom CSS
+    local_css()
+    init_session_state()
+    
+    # Load ML models in background without showing messages
+    if not st.session_state.models_loaded:
+        with st.spinner("🔄 Loading AI models..."):
+            models = load_models()
+            st.session_state.models = models
+            st.session_state.models_loaded = True
+    
+    models = st.session_state.models
+    
+    # Language selector at top right with Pakistan flag colors
+    col_lang, col_space = st.columns([1, 5])
+    with col_lang:
+        st.markdown('<div class="pakistan-flag">🇵🇰</div>', unsafe_allow_html=True)
+        language = st.selectbox("Language", ["English", "Urdu"], key="language_selector", label_visibility="collapsed")
+        st.session_state.current_language = language
+    
+    # Main header with new name
+    st.markdown('<h1 class="main-header">🏥 SehatAI</h1>', unsafe_allow_html=True)
+    if language == "English":
+        st.markdown('<h3 style="text-align: center; color: #00612A; margin-bottom: 20px;">Smart Healthcare Triage & Risk Assessment System</h3>', unsafe_allow_html=True)
+    else:
+        st.markdown('<h3 style="text-align: center; color: #00612A; margin-bottom: 20px;">ذہین ہیلتھ کیئر ٹریج اور خطرے کی تشخیص کا نظام</h3>', unsafe_allow_html=True)
+    
+    # Display random healthcare quote
+    import random
+    quote = random.choice(HEALTHCARE_QUOTES)
+    
+    st.markdown(f'''
+    <div class="quote-section">
+        <div class="quote-text">"{quote['english']}"</div>
+        <div class="quote-text">"{quote['urdu']}"</div>
+        <div class="quote-author">- {quote['author']}</div>
+    </div>
+    ''', unsafe_allow_html=True)
+    
+    # Create tabs
+    if language == "English":
+        tab_names = ["Patient Assessment", "Health Assistant", "Medical Reports"]
+    else:
+        tab_names = ["مریض تشخیص", "ہیلتھ اسسٹنٹ", "طبی رپورٹس"]
+    
+    tab1, tab2, tab3 = st.tabs(tab_names)
+    
+    with tab1:
+        # Patient Assessment Form
+        st.markdown('<div class="section-container">', unsafe_allow_html=True)
+        
+        if language == "English":
+            st.markdown('<h2 style="color: #00401A; border-bottom: 3px solid #00401A; padding-bottom: 10px;">👨‍⚕️ Comprehensive Patient Assessment</h2>', unsafe_allow_html=True)
+            st.write("Complete the following form for comprehensive patient risk assessment and priority scoring")
+        else:
+            st.markdown('<h2 style="color: #00401A; border-bottom: 3px solid #00401A; padding-bottom: 10px;">👨‍⚕️ جامع مریض تشخیص</h2>', unsafe_allow_html=True)
+            st.write("جامع مریض کے خطرے کی تشخیص اور ترجیحی اسکورنگ کے لیے درج ذیل فارم کو مکمل کریں")
+        
+        with st.form("patient_assessment_form"):
+            col1, col2 = st.columns(2)
+            
+            with col1:
+                # Basic Information
+                if language == "English":
+                    st.subheader("Personal Information")
+                    name = st.text_input("Full Name", placeholder="Enter patient's full name")
+                    age = st.number_input("Age", min_value=1, max_value=120, value=45, 
+                                        help="Patient's age in years")
+                    gender = st.selectbox("Gender", ["Male", "Female", "Other"])
+                    contact = st.text_input("Contact Number", placeholder="03XX-XXXXXXX")
+                else:
+                    st.subheader("ذاتی معلومات")
+                    name = st.text_input("مکمل نام", placeholder="مریض کا مکمل نام درج کریں")
+                    age = st.number_input("عمر", min_value=1, max_value=120, value=45,
+                                        help="مریض کی عمر سالوں میں")
+                    gender = st.selectbox("جنس", ["مرد", "عورت", "دیگر"])
+                    contact = st.text_input("رابطہ نمبر", placeholder="03XX-XXXXXXX")
+            
+            with col2:
+                # Vital Signs
+                if language == "English":
+                    st.subheader("Clinical Parameters")
+                    bp_systolic = st.slider("Blood Pressure (systolic)", 
+                                          min_value=70, max_value=250, value=120,
+                                          help="Systolic blood pressure in mmHg")
+                    bp_diastolic = st.slider("Blood Pressure (diastolic)", 
+                                           min_value=40, max_value=150, value=80,
+                                          help="Diastolic blood pressure in mmHg")
+                    heart_rate = st.slider("Heart Rate (bpm)", 
+                                         min_value=30, max_value=200, value=72,
+                                         help="Heart beats per minute")
+                    cholesterol = st.slider("Cholesterol Level (mg/dL)", 
+                                          min_value=100, max_value=400, value=180)
+                    glucose = st.slider("Blood Glucose (mg/dL)", 
+                                      min_value=50, max_value=500, value=95)
+                    bmi = st.slider("BMI", min_value=15.0, max_value=40.0, value=23.5, step=0.1)
+                else:
+                    st.subheader("کلینیکل پیرامیٹرز")
+                    bp_systolic = st.slider("بلڈ پریشر (سسٹولک)", 
+                                          min_value=70, max_value=250, value=120,
+                                          help="سسٹولک بلڈ پریشر mmHg میں")
+                    bp_diastolic = st.slider("بلڈ پریشر (ڈائیسٹولک)", 
+                                           min_value=40, max_value=150, value=80,
+                                          help="ڈائیسٹولک بلڈ پریشر mmHg میں")
+                    heart_rate = st.slider("دل کی دھڑکن (bpm)", 
+                                         min_value=30, max_value=200, value=72,
+                                         help="دل کی دھڑکن فی منٹ")
+                    cholesterol = st.slider("کولیسٹرول کی سطح (mg/dL)", 
+                                          min_value=100, max_value=400, value=180)
+                    glucose = st.slider("خون میں گلوکوز (mg/dL)", 
+                                      min_value=50, max_value=500, value=95)
+                    bmi = st.slider("باڈی ماس انڈیکس", min_value=15.0, max_value=40.0, value=23.5, step=0.1)
+            
+            # Symptoms Section
+            if language == "English":
+                st.subheader("Reported Symptoms")
+                col3, col4 = st.columns(2)
+                with col3:
+                    chest_pain = st.checkbox("Chest Pain or Discomfort")
+                    shortness_breath = st.checkbox("Shortness of Breath")
+                    palpitations = st.checkbox("Heart Palpitations")
+                with col4:
+                    fatigue = st.checkbox("Persistent Fatigue")
+                    dizziness = st.checkbox("Dizziness or Lightheadedness")
+                    blurred_vision = st.checkbox("Blurred Vision")
+            else:
+                st.subheader("رپورٹ کردہ علامات")
+                col3, col4 = st.columns(2)
+                with col3:
+                    chest_pain = st.checkbox("سینے میں درد یا بے چینی")
+                    shortness_breath = st.checkbox("سانس لینے میں دشواری")
+                    palpitations = st.checkbox("دل کی دھڑکن میں اضافہ")
+                with col4:
+                    fatigue = st.checkbox("مسلسل تھکاوٹ")
+                    dizziness = st.checkbox("چکر آنا یا سر ہلکا محسوس ہونا")
+                    blurred_vision = st.checkbox("دھندلا نظر آنا")
+            
+            # Assessment Button
+            if language == "English":
+                assess_button = st.form_submit_button("🚀 Calculate Risk Score & Priority", 
+                                                    use_container_width=True)
+            else:
+                assess_button = st.form_submit_button("🚀 خطرے کا اسکور اور ترجیح معلوم کریں", 
+                                                    use_container_width=True)
+            
+            if assess_button:
+                # Validate inputs
+                validation_errors = validate_patient_data(age, bp_systolic, bp_diastolic, heart_rate)
+                
+                if validation_errors:
+                    for error in validation_errors:
+                        st.error(f"❌ {error}")
+                else:
+                    try:
+                        with st.spinner("🔍 Analyzing patient data and calculating risks..."):
+                            # Prepare symptoms dictionary
+                            symptoms_dict = {
+                                'chest_pain': chest_pain,
+                                'shortness_breath': shortness_breath,
+                                'palpitations': palpitations,
+                                'fatigue': fatigue,
+                                'dizziness': dizziness,
+                                'blurred_vision': blurred_vision
+                            }
+                            
+                            # Store patient data
+                            st.session_state.patient_data = {
+                                'name': name,
+                                'age': age,
+                                'gender': gender,
+                                'contact': contact,
+                                'bp_systolic': bp_systolic,
+                                'bp_diastolic': bp_diastolic,
+                                'heart_rate': heart_rate,
+                                'cholesterol': cholesterol,
+                                'glucose': glucose,
+                                'bmi': bmi,
+                                'symptoms': symptoms_dict
+                            }
+                            
+                            # Predict risks using ML models
+                            heart_risk, diabetes_risk, hypertension_risk = predict_with_models(
+                                age, bp_systolic, bp_diastolic, heart_rate, 
+                                cholesterol, glucose, bmi, symptoms_dict, models
+                            )
+                            
+                            # Calculate integrated priority score
+                            priority_score = calculate_priority_score(
+                                heart_risk, diabetes_risk, hypertension_risk
+                            )
+                            
+                            priority_level, recommendation, risk_class = get_priority_recommendation(
+                                priority_score, language
+                            )
+                            
+                            # Store results
+                            st.session_state.risk_scores = {
+                                'heart': heart_risk,
+                                'diabetes': diabetes_risk,
+                                'hypertension': hypertension_risk,
+                                'priority': priority_score,
+                                'recommendation': recommendation,
+                                'level': priority_level
+                            }
+                            
+                            # Add to assessment history
+                            st.session_state.assessment_history.append({
+                                'timestamp': datetime.now(),
+                                'patient_data': st.session_state.patient_data.copy(),
+                                'risk_scores': st.session_state.risk_scores.copy()
+                            })
+                        
+                        # Display results
+                        st.markdown("---")
+                        st.success("✅ Risk assessment completed successfully!")
+                        
+                        # Risk Scores Visualization
+                        if language == "English":
+                            st.subheader("📊 Disease Risk Assessment Dashboard")
+                        else:
+                            st.subheader("📊 بیماری کے خطرے کی تشخیص ڈیش بورڈ")
+                        
+                        col5, col6, col7, col8 = st.columns(4)
+                        
+                        risk_metrics = [
+                            (heart_risk, "Heart Disease", "❤️", "#FF6B6B"),
+                            (diabetes_risk, "Diabetes", "🩺", "#4ECDC4"),
+                            (hypertension_risk, "Hypertension", "💓", "#45B7D1"),
+                            (priority_score, "Priority Score", "🎯", "#96CEB4")
+                        ]
+                        
+                        for (value, title, emoji, color), col in zip(risk_metrics, [col5, col6, col7, col8]):
+                            with col:
+                                fig = go.Figure(go.Indicator(
+                                    mode = "gauge+number+delta",
+                                    value = value,
+                                    domain = {'x': [0, 1], 'y': [0, 1]},
+                                    title = {'text': f"{emoji} {title}", 'font': {'size': 14}},
+                                    gauge = {
+                                        'axis': {'range': [0, 1], 'tickwidth': 1},
+                                        'bar': {'color': color},
+                                        'steps': [
+                                            {'range': [0, 0.3], 'color': "lightgreen"},
+                                            {'range': [0.3, 0.7], 'color': "yellow"},
+                                            {'range': [0.7, 1], 'color': "red"}
+                                        ],
+                                        'threshold': {
+                                            'line': {'color': "black", 'width': 4},
+                                            'thickness': 0.75,
+                                            'value': 0.7
+                                        }
+                                    }
+                                ))
+                                fig.update_layout(height=250, margin=dict(l=10, r=10, t=50, b=10))
+                                st.plotly_chart(fig, use_container_width=True)
+                        
+                        # Priority Recommendation
+                        st.markdown(f'<div class="priority-box {risk_class}">', unsafe_allow_html=True)
+                        if language == "English":
+                            st.markdown(f"## 🎯 Clinical Priority Recommendation")
+                            st.markdown(f"### {recommendation}")
+                            st.markdown(f"**Overall Risk Score:** `{priority_score:.3f}`")
+                            st.markdown(f"**Recommended Action:** `{priority_level.replace('_', ' ').title()}`")
+                            
+                            # Additional clinical guidance
+                            if priority_level == "EMERGENCY_CARE":
+                                st.warning("🚨 **Immediate Action Required:** Patient should be directed to emergency department without delay.")
+                            elif priority_level == "SAME_DAY_CONSULT":
+                                st.info("ℹ️ **Urgent Consultation:** Schedule appointment within 24 hours.")
+                            else:
+                                st.success("✅ **Routine Care:** Schedule within regular appointment system.")
+                                
+                        else:
+                            st.markdown(f"## 🎯 کلینیکل ترجیحی سفارش")
+                            st.markdown(f"### {recommendation}")
+                            st.markdown(f"**کل خطرے کا اسکور:** `{priority_score:.3f}`")
+                            st.markdown(f"**سفارش کردہ عمل:** `{priority_level.replace('_', ' ').title()}`")
+                            
+                            if priority_level == "EMERGENCY_CARE":
+                                st.warning("🚨 **فوری کارروائی ضروری:** مریض کو بغیر کسی تاخیر کے ایمرجنسی ڈیپارٹمنٹ بھیجا جائے۔")
+                            elif priority_level == "SAME_DAY_CONSULT":
+                                st.info("ℹ️ **فوری مشاورت:** 24 گھنٹے کے اندر اپائنٹمنٹ شیڈول کریں۔")
+                            else:
+                                st.success("✅ **روٹین کیئر:** معمول کی اپائنٹمنٹ سسٹم کے اندر شیڈول کریں۔")
+                                
+                        st.markdown('</div>', unsafe_allow_html=True)
+                            
+                    except Exception as e:
+                        st.error(f"❌ Error in risk assessment: {str(e)}")
+        
+        st.markdown('</div>', unsafe_allow_html=True)
+    
+    with tab2:
+        # Healthcare Chatbot
+        st.markdown('<div class="section-container">', unsafe_allow_html=True)
+        
+        if language == "English":
+            st.markdown('<h2 style="color: #00401A; border-bottom: 3px solid #00401A; padding-bottom: 10px;">💬 SehatAI Health Assistant</h2>', unsafe_allow_html=True)
+            st.write("Ask health-related questions and get instant, reliable medical information")
+        else:
+            st.markdown('<h2 style="color: #00401A; border-bottom: 3px solid #00401A; padding-bottom: 10px;">💬 سیحت اے آئی ہیلتھ اسسٹنٹ</h2>', unsafe_allow_html=True)
+            st.write("صحت سے متعلق سوالات پوچھیں اور فوری، قابل اعتماد طبی معلومات حاصل کریں")
+        
+        # Groq API Configuration - Hidden from interface as requested
+        # Check if API key is available in environment or secrets
+        groq_api_key = st.secrets.get("GROQ_API_KEY", "") if hasattr(st, 'secrets') else ""
+        
+        if groq_api_key:
+            st.session_state.groq_api_key = groq_api_key
+            try:
+                groq_client = GroqClient(groq_api_key)
+                chatbot = HealthcareChatbot(groq_client)
+            except Exception as e:
+                chatbot = HealthcareChatbot()
+        else:
+            chatbot = HealthcareChatbot()
+        
+        # Display chat history
+        for message in st.session_state.chat_history:
+            if message["role"] == "user":
+                st.markdown(f'<div class="chat-message user-message"><strong>You:</strong> {message["content"]}</div>', unsafe_allow_html=True)
+            else:
+                st.markdown(f'<div class="chat-message bot-message"><strong>🤖 Health Assistant:</strong> {message["content"]}</div>', unsafe_allow_html=True)
+        
+        # Chat input
+        if prompt := st.chat_input(
+            "Type your health question here..." if language == "English" 
+            else "اپنا صحت کا سوال یہاں ٹائپ کریں..."
+        ):
+            # Add user message to chat history
+            st.session_state.chat_history.append({"role": "user", "content": prompt})
+            
+            # Generate bot response
+            with st.spinner("💭 Analyzing your question..." if language == "English" else "💭 آپ کا سوال تجزیہ ہو رہا ہے..."):
+                response = chatbot.get_response(prompt, language)
+                st.session_state.chat_history.append({"role": "assistant", "content": response})
+            
+            # Limit chat history to last 10 messages
+            if len(st.session_state.chat_history) > 10:
+                st.session_state.chat_history = st.session_state.chat_history[-10:]
+            
+            st.rerun()
+        
+        # Quick action buttons
+        if language == "English":
+            st.subheader("Quick Health Topics")
+        else:
+            st.subheader("فوری صحت کے موضوعات")
+        
+        col_qa1, col_qa2, col_qa3, col_qa4 = st.columns(4)
+        
+        with col_qa1:
+            if st.button("❤️ Heart Health", use_container_width=True):
+                if language == "English":
+                    query = "Tell me about heart disease symptoms and prevention"
+                else:
+                    query = "دل کی بیماری کی علامات اور بچاؤ کے بارے میں بتائیں"
+                st.session_state.chat_history.append({"role": "user", "content": query})
+                st.rerun()
+        
+        with col_qa2:
+            if st.button("🩺 Diabetes", use_container_width=True):
+                if language == "English":
+                    query = "What are diabetes symptoms and management strategies?"
+                else:
+                    query = "ذیابیطس کی علامات اور انتظام کی حکمت عملی کیا ہیں؟"
+                st.session_state.chat_history.append({"role": "user", "content": query})
+                st.rerun()
+        
+        with col_qa3:
+            if st.button("💓 Blood Pressure", use_container_width=True):
+                if language == "English":
+                    query = "How to control high blood pressure naturally?"
+                else:
+                    query = "قدرتی طور پر ہائی ��لڈ پریشر کو کیسے کنٹرول کریں؟"
+                st.session_state.chat_history.append({"role": "user", "content": query})
+                st.rerun()
+        
+        with col_qa4:
+            if st.button("🍎 Nutrition", use_container_width=True):
+                if language == "English":
+                    query = "What is a healthy diet for heart health?"
+                else:
+                    query = "دل کی صحت کے لیے صحت مند غذا کیا ہے؟"
+                st.session_state.chat_history.append({"role": "user", "content": query})
+                st.rerun()
+        
+        st.markdown('</div>', unsafe_allow_html=True)
+    
+    with tab3:
+        # Medical Reports
+        st.markdown('<div class="section-container">', unsafe_allow_html=True)
+        
+        if language == "English":
+            st.markdown('<h2 style="color: #00401A; border-bottom: 3px solid #00401A; padding-bottom: 10px;">📋 Medical Reports & Analytics</h2>', unsafe_allow_html=True)
+            st.write("Download comprehensive medical reports and view assessment analytics")
+        else:
+            st.markdown('<h2 style="color: #00401A; border-bottom: 3px solid #00401A; padding-bottom: 10px;">📋 طبی رپورٹس اور تجزیات</h2>', unsafe_allow_html=True)
+            st.write("جامع طبی رپورٹس ڈاؤن لوڈ کریں اور تشخیص کے تجزیات دیکھیں")
+        
+        if st.session_state.assessment_history:
+            # Latest assessment
+            latest_assessment = st.session_state.assessment_history[-1]
+            
+            # Download PDF report
+            if language == "English":
+                st.subheader("Download Patient Report")
+            else:
+                st.subheader("مریض رپورٹ ڈاؤن لوڈ کریں")
+            
+            pdf_generator = PDFReportGenerator()
+            pdf_bytes = pdf_generator.generate_report(
+                latest_assessment['patient_data'], 
+                latest_assessment['risk_scores'], 
+                language
+            )
+            
+            if language == "English":
+                st.markdown(create_download_link(pdf_bytes, "sehatai_medical_report.pdf", "📥 Download Medical Report (PDF)"), unsafe_allow_html=True)
+            else:
+                st.markdown(create_download_link(pdf_bytes, "sehatai_medical_report.pdf", "📥 طبی رپورٹ ڈاؤن لوڈ کریں (PDF)"), unsafe_allow_html=True)
+            
+            # Analytics Section
+            if language == "English":
+                st.subheader("Assessment Analytics")
+            else:
+                st.subheader("تشخیص کے تجزیات")
+            
+            col_anal1, col_anal2, col_anal3 = st.columns(3)
+            
+            with col_anal1:
+                st.metric(
+                    "Total Assessments" if language == "English" else "کل تشخیص",
+                    len(st.session_state.assessment_history)
+                )
+            
+            with col_anal2:
+                latest_priority = latest_assessment['risk_scores']['priority']
+                st.metric(
+                    "Current Priority Score" if language == "English" else "موجودہ ترجیحی اسکور",
+                    f"{latest_priority:.1%}"
+                )
+            
+            with col_anal3:
+                high_risk_count = sum(1 for assess in st.session_state.assessment_history 
+                                    if assess['risk_scores']['priority'] > 0.7)
+                st.metric(
+                    "High Risk Cases" if language == "English" else "اعلی خطرہ والے معاملات",
+                    high_risk_count
+                )
+            
+            # Risk distribution chart
+            if len(st.session_state.assessment_history) > 1:
+                risk_data = pd.DataFrame([
+                    {
+                        'Heart Risk': assess['risk_scores']['heart'],
+                        'Diabetes Risk': assess['risk_scores']['diabetes'],
+                        'Hypertension Risk': assess['risk_scores']['hypertension'],
+                        'Assessment': i+1
+                    } for i, assess in enumerate(st.session_state.assessment_history[-5:])
+                ])
+                
+                fig = px.line(risk_data, x='Assessment', y=['Heart Risk', 'Diabetes Risk', 'Hypertension Risk'],
+                            title="Risk Trend Analysis" if language == "English" else "خطرے کے رجحان کا تجزیہ",
+                            color_discrete_map={
+                                'Heart Risk': '#FF6B6B',
+                                'Diabetes Risk': '#4ECDC4', 
+                                'Hypertension Risk': '#45B7D1'
+                            })
+                st.plotly_chart(fig, use_container_width=True)
+        
+        else:
+            if language == "English":
+                st.info("👆 Complete a patient assessment to generate reports and view analytics")
+            else:
+                st.info("👆 رپورٹس تیار کرنے اور تجزیات دیکھنے کے لیے مریض کی تشخیص مکمل کریں")
+        
+        st.markdown('</div>', unsafe_allow_html=True)
+
+if __name__ == "__main__":
+    main()