Update app.py

app.py

@@ -12,6 +12,8 @@ from datetime import datetime
 import base64
 from fpdf import FPDF
 import tempfile
+import requests
+import json
 
 # Set page config first
 st.set_page_config(
@@ -163,6 +165,13 @@ def local_css():
         font-weight: bold;
         margin: 0 10px;
     }
+    .api-key-section {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        padding: 15px;
+        border-radius: 10px;
+        margin: 10px 0;
+        color: white;
+    }
     </style>
     """, unsafe_allow_html=True)
 
@@ -178,6 +187,8 @@ def init_session_state():
         st.session_state.chat_history = []
     if 'current_language' not in st.session_state:
         st.session_state.current_language = 'English'
+    if 'groq_api_key' not in st.session_state:
+        st.session_state.groq_api_key = ''
 
 # Healthcare quotes database
 HEALTHCARE_QUOTES = [
@@ -208,8 +219,38 @@ HEALTHCARE_QUOTES = [
     }
 ]
 
+class GroqClient:
+    def __init__(self, api_key):
+        self.api_key = api_key
+        self.base_url = "https://api.groq.com/openai/v1"
+        self.headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json"
+        }
+    
+    def chat_completions_create(self, messages, model="llama3-8b-8192", temperature=0.3, max_tokens=1024):
+        try:
+            url = f"{self.base_url}/chat/completions"
+            data = {
+                "messages": messages,
+                "model": model,
+                "temperature": temperature,
+                "max_tokens": max_tokens,
+                "top_p": 0.9
+            }
+            
+            response = requests.post(url, headers=self.headers, json=data, timeout=30)
+            response.raise_for_status()
+            
+            result = response.json()
+            return result
+            
+        except Exception as e:
+            raise Exception(f"Groq API error: {str(e)}")
+
 class HealthcareChatbot:
-    def __init__(self):
+    def __init__(self, groq_client=None):
+        self.groq_client = groq_client
         self.medical_knowledge_base = {
             'heart_disease': {
                 'symptoms': ["Chest pain or discomfort", "Shortness of breath", "Pain in arms, neck, jaw", "Nausea", "Fatigue", "Dizziness"],
@@ -228,16 +269,55 @@ class HealthcareChatbot:
                 'risk_factors': ["Family history", "Age over 65", "Obesity", "High salt intake", "Stress", "Alcohol consumption", "Smoking"],
                 'prevention': ["DASH diet (low sodium)", "Regular aerobic exercise", "Weight management", "Stress reduction", "Limit alcohol", "No smoking"],
                 'emergency_signs': ["Severe headache", "Chest pain", "Difficulty breathing", "Vision changes", "Confusion", "Seizures"]
-            },
-            'general_health': {
-                'nutrition': ["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"],
-                'exercise': ["150 mins moderate exercise weekly", "Strength training 2x weekly", "Stay active throughout day", "Include flexibility exercises", "Get adequate rest between workouts"],
-                'lifestyle': ["7-9 hours quality sleep nightly", "Stress management techniques", "No smoking", "Limit alcohol to 1-2 drinks/day", "Regular health check-ups"],
-                'prevention': ["Annual physical exams", "Vaccinations up to date", "Regular hand washing", "Sun protection", "Mental health care"]
             }
         }
     
     def get_response(self, user_input, language='English'):
+        # If Groq client is available, use it for more intelligent responses
+        if self.groq_client:
+            try:
+                return self._get_groq_response(user_input, language)
+            except Exception as e:
+                st.error(f"Groq API error: {str(e)}. Using rule-based response.")
+        
+        # Fallback to rule-based responses
+        return self._get_rule_based_response(user_input, language)
+    
+    def _get_groq_response(self, user_input, language):
+        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."""
+        
+        if language == 'Urdu':
+            system_prompt += " Respond in Urdu with proper medical terminology and cultural sensitivity."
+        else:
+            system_prompt += " Respond in English with clear, professional medical advice."
+        
+        messages = [
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_input}
+        ]
+        
+        response = self.groq_client.chat_completions_create(
+            messages=messages,
+            model="llama3-8b-8192",
+            temperature=0.3,
+            max_tokens=1024
+        )
+        
+        bot_response = response['choices'][0]['message']['content']
+        
+        # Add disclaimer
+        if language == 'Urdu':
+            bot_response += "\n\n⚠️ براہ کرم نوٹ کریں: یہ ایک AI اسسٹنٹ ہے۔ سنگین طبی حالات کے لیے ہمیشہ کوالیفائیڈ ڈاکٹر سے مشورہ کریں۔"
+        else:
+            bot_response += "\n\n⚠️ Please note: This is an AI assistant. Always consult qualified doctors for serious medical conditions."
+        
+        return bot_response
+    
+    def _get_rule_based_response(self, user_input, language):
         user_input = user_input.lower().strip()
         
         # Greeting patterns
@@ -362,12 +442,6 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 **🚑 ہنگامی انتباہی علامات (فوری علاج حاصل کریں):**
 {chr(10).join(['• ' + sign for sign in ['شدید سینے میں درد', 'سانس لینے میں شدید دشواری', 'بیہوش ہونا', 'تیز دل کی دھڑکن', 'بازو/کندھوں میں پھیلنے والا درد']])}
 
-**📊 تجویز کردہ نگرانی:**
-• بلڈ پریشر کی باقاعدہ جانچ
-• اگر خطرہ زیادہ ہو تو سالانہ کولیسٹرول کی سطح
-• اگر علامات موجود ہوں تو ECG
-• اگر ڈاکٹر تجویز کرے تو اسٹریس ٹیسٹ
-
 *🔬 کلینیکل نوٹ: ابتدائی تشخیص جان بچاتی ہے۔ دل کی صحت کے لیے باقاعدہ چیک اپ بہت ضروری ہیں۔*
 """
     
@@ -413,12 +487,6 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 **🚑 ہنگامی علامات (فوری علاج حاصل کریں):**
 {chr(10).join(['• ' + sign for sign in ['بلڈ شوگر >300 mg/dL یا <70 mg/dL', 'الجھن یا بے ترتیبی', 'پھل کی سی بو والی سانس', 'تیز سانس لینا', 'ہوش کا ضیاع']])}
 
-**📊 نگرانی کے رہنما اصول:**
-• فاسٹنگ بلڈ شوگر: 80-130 mg/dL
-• کھانے کے بعد بلڈ شوگر: <180 mg/dL
-• HbA1c: <7.0% (اگر کنٹرول نہ ہو تو سہ ماہی)
-• پاؤں اور آنکھوں کا باقاعدہ معائنہ
-
 *🔬 کلینیکل نوٹ: مستقل نگرانی اور طرز زندگی کا انتظام ذیابیطس کے کنٹرول کی کلید ہے۔*
 """
     
@@ -464,35 +532,25 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 **🚑 ہائی بلڈ پریشر کی ہنگامی علامات:**
 {chr(10).join(['• ' + sign for sign in ['شدید سر درد', 'سینے میں درد', 'سانس لینے میں دشواری', 'بینائی میں تبدیلی', 'الجھن', 'دورے']])}
 
-**📊 بلڈ پریشر کی درجہ بندی:**
-• نارمل: <120/80 mmHg
-• بلند: 120-129/<80 mmHg
-• اسٹیج 1 ہائی بلڈ پریشر: 130-139/80-89 mmHg
-• اسٹیج 2 ہائی بلڈ پریشر: ≥140/90 mmHg
-
 *🔬 کلینیکل نوٹ: خاموش قاتل - اکثر کوئی علامات نہیں ہوتیں۔ باقاعدہ نگرانی ضروری ہے۔*
 """
     
     def _format_nutrition_response(self, language):
-        knowledge = self.medical_knowledge_base['general_health']
         if language == 'English':
-            return f"""
+            return """
 **🍎 Comprehensive Nutrition Guide**
 
 **🥦 Healthy Eating Principles:**
-{chr(10).join(['• ' + tip for tip in knowledge['nutrition']])}
+• 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
 
 **🎯 Specific Recommendations:**
 • **Heart Health**: Mediterranean diet, omega-3 rich foods
 • **Diabetes**: Controlled carb intake, high fiber
 • **Hypertension**: Low sodium (<2300mg/day), potassium-rich foods
-• **Weight Management**: Calorie control, portion management
-
-**🚫 Foods to Limit:**
-• Processed and packaged foods
-• Sugary drinks and sweets
-• High sodium snacks
-• Saturated and trans fats
 
 **💡 Pakistani Dietary Adaptations:**
 • Choose whole wheat chapati over refined flour
@@ -503,23 +561,20 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 *Remember: Balanced nutrition is foundation of good health.*
 """
         else:
-            return f"""
+            return """
 **🍎 غذائیت کی جامع گائیڈ**
 
 **🥦 صحت مند کھانے کے اصول:**
-{chr(10).join(['• ' + tip for tip in ['روزانہ 5-9 سرونگ پھل/سبزیاں کھائیں', 'ریفائنڈ کی بجائے سارا اناج منتخب کریں', 'لیان پروٹین شامل کریں', 'ہائیڈریٹ رہیں (8-10 گلاس پانی)', 'پراسیسڈ فوڈز اور چینی کو محدود کریں']])}
+• روزانہ 5-9 سرونگ پھل/سبزیاں کھائیں
+• ریفائنڈ کی بجائے سارا اناج منتخب کریں
+• لیان پروٹین شامل کریں
+• ہائیڈریٹ رہیں (8-10 گلاس پانی)
+• پراسیسڈ فوڈز اور چینی کو محدود کریں
 
 **🎯 مخصوص سفارشات:**
 • **دل کی صحت**: بحیرہ روم کی غذا، اومیگا 3 سے بھرپور غذائیں
 • **ذیابیطس**: کنٹرول کاربوہائیڈریٹ انٹیک، زیادہ فائبر
 • **ہائی بلڈ پریشر**: کم سوڈیم (<2300mg/day)، پوٹاشیم سے بھرپور غذائیں
-• **وزن کا انتظام**: کیلوری کنٹرول، حصے کا انتظام
-
-**🚫 محدود کرنے والی غذائیں:**
-• پراسیسڈ اور پیکجڈ فوڈز
-• میٹھے مشروبات اور مٹھائیاں
-• ہائی سوڈیم اسنیکس
-• سیر شدہ اور ٹرانس فیٹس
 
 **💡 پاکستانی غذائی موافقت:**
 • ریفائنڈ آٹے کی بجائے whole wheat چپاتی منتخب کریں
@@ -531,58 +586,49 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 """
     
     def _format_exercise_response(self, language):
-        knowledge = self.medical_knowledge_base['general_health']
         if language == 'English':
-            return f"""
+            return """
 **💪 Comprehensive Exercise Guide**
 
 **🏃‍♂️ Exercise Recommendations:**
-{chr(10).join(['• ' + tip for tip in knowledge['exercise']])}
+• 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
-• **Weight Management**: 300+ mins moderate exercise weekly
 
 **🌞 Pakistani Climate Considerations:**
 • Exercise in cooler morning/evening hours
 • Stay hydrated with water and electrolytes
 • Wear light, breathable clothing
-• Listen to your body in hot weather
-
-**🔄 Exercise Types:**
-• **Cardio**: Walking, jogging, cycling, swimming
-• **Strength**: Weight training, resistance bands
-• **Flexibility**: Yoga, stretching
-• **Balance**: Tai chi, balance exercises
 
 *Important: Consult doctor before starting new exercise program.*
 """
         else:
-            return f"""
+            return """
 **💪 ورزش کی جامع گائیڈ**
 
 **🏃‍♂️ ورزش کی سفارشات:**
-{chr(10).join(['• ' + tip for tip in ['ہفتے میں 150 منٹ اعتدال پسند ورزش', 'ہفتے میں 2 بار طاقت کی تربیت', 'پورے دن متحرک رہیں', 'لچک کی ورزشیں شامل کریں', 'ورک آؤٹس کے درمیان مناسب آرام حاصل کریں']])}
+• ہفتے میں 150 منٹ اعتدال پسند ورزش
+• ہفتے میں 2 بار طاقت کی تربیت
+• پورے دن متحرک رہیں
+• لچک کی ورزشیں شامل کریں
+• ورک آؤٹس کے درمیان مناسب آرام حاصل کریں
 
 **🎯 بیماری کی مخصوص ورزش:**
 • **دل کی صحت**: روزانہ 30 منٹ اعتدال پسند کارڈیو
 • **ذیابیطس**: کارڈیو اور طاقت کی تربیت کا مجموعہ
 • **ہائی بلڈ پریشر**: باقاعدہ ایروبک ورزش
-• **وزن کا انتظام**: ہفتے میں 300+ منٹ اعتدال پسند ورزش
 
 **🌞 پاکستانی آب و ہوا کے تحفظات:**
 • ٹھنڈے صبح/شام کے اوقات میں ورزش کریں
 • پانی اور الیکٹرولائٹس کے ساتھ ہائیڈریٹ رہیں
 • ہلکے، ہوا دار کپڑے پہنیں
-• گرم موسم میں اپنے جسم کی بات سنیں
-
-**🔄 ورزش کی اقسام:**
-• **کارڈیو**: چہل قدمی، جاگنگ، سائیکل چلانا، تیراکی
-• **طاقت**: وزن کی تربیت، مزاحمتی بینڈ
-• **لچک**: یوگا، اسٹریچنگ
-• **توازن**: تائی چی، توازن کی ورزشیں
 
 *اہم: نئی ورزش پروگرام شروع کرنے سے پہلے ڈاکٹر سے مشورہ کریں۔*
 """
@@ -599,14 +645,6 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 • Sudden dizziness, weakness, or confusion
 • Unexplained weight loss (>10% body weight)
 • Symptoms lasting more than 2 weeks
-• Worsening of chronic conditions
-
-**🏠 Self-Care Tips:**
-• Rest and adequate hydration
-• Monitor symptoms and temperature regularly
-• Use over-the-counter medications as directed
-• Apply cold/heat packs as appropriate
-• Maintain proper nutrition
 
 **🚨 Emergency Warning Signs (Go to ER):**
 • Difficulty breathing or shortness of breath
@@ -614,8 +652,6 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 • Severe bleeding that doesn't stop
 • Sudden severe pain anywhere in body
 • Loss of consciousness or fainting
-• Sudden vision changes
-• Difficulty speaking or confusion
 
 *⚠️ Important: This is general guidance. Always consult healthcare professional for proper diagnosis.*
 """
@@ -630,14 +666,6 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 • اچانک چکر آنا، کمزوری، یا الجھن
 • بے وجہ وزن میں کمی (>10% جسمانی وزن)
 • 2 ہفتے سے زیادہ علامات برقرار رہنا
-• دائمی حالات کا بگڑنا
-
-**🏠 خود کی دیکھ بھال کے نکات:**
-• آرام اور مناسب ہائیڈریشن
-• علامات اور درجہ حرارت کی باقاعدہ نگرانی
-• ہدایت کے مطابق اوور دی کاؤنٹر ادویات استعمال کریں
-• حسب مناسب کولڈ/ہیٹ پیکس لگائیں
-• مناسب غذائیت برقرار رکھیں
 
 **🚨 ہنگامی انتباہی علامات (ایمرجنسی روم جائیں):**
 • سانس لینے میں دشواری یا سانس کی قلت
@@ -645,8 +673,6 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 • شدید خون بہنا جو رک نہیں رہا
 • جسم میں کہیں بھی اچانک شدید درد
 • ہوش کا ضیاع یا بیہوشی
-• اچانک بینائی میں تبدیلی
-• بولنے میں دشواری یا الجھن
 
 *⚠️ اہم: یہ عام رہنمائی ہے۔ مناسب تشخیص کے لیے ہمیشہ ہیلتھ کیئر پروفیشنل سے مشورہ کریں۔*
 """
@@ -663,12 +689,6 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 • Inform all doctors about all medications
 • Understand purpose and side effects of each medicine
 
-**⚠️ Common Medication Categories:**
-• **Heart**: Beta-blockers, ACE inhibitors, statins
-• **Diabetes**: Metformin, insulin, SGLT2 inhibitors  
-• **Hypertension**: Diuretics, calcium channel blockers
-• **Pain**: NSAIDs, acetaminophen (use cautiously)
-
 **🚨 Medication Warning Signs:**
 • Severe allergic reactions (rash, swelling)
 • Difficulty breathing after taking medicine
@@ -676,12 +696,6 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 • Unusual bleeding or bruising
 • Confusion or hallucinations
 
-**💡 Important Reminders:**
-• Don't stop medications without doctor's advice
-• Report side effects to your doctor
-• Keep medications in original containers
-• Check expiration dates regularly
-
 *Note: Always follow your healthcare provider's specific instructions.*
 """
         else:
@@ -695,12 +709,6 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 • تمام ڈاکٹروں کو تمام دوائیوں کے بارے میں بتائیں
 • ہر دوا کے مقصد اور مضر اثرات کو سمجھیں
 
-**⚠️ عام دوائیوں کی اقسام:**
-• **دل**: بیٹا بلاکرز، ACE inhibitors، statins
-• **ذیابیطس**: میٹفارمن، انسولین، SGLT2 inhibitors
-• **ہائی بلڈ پریشر**: ڈائیوریٹکس، کیلشیم چینل بلاکرز
-• **درد**: NSAIDs، acetaminophen (احتیاط سے استعمال کریں)
-
 **🚨 دوائیوں کی انتباہی علامات:**
 • شدید الرجک رد عمل (خارش، سوجن)
 • دوا لینے کے بعد سانس لینے میں دشواری
@@ -708,12 +716,6 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 • غیر معمولی خون بہنا یا چوٹ لگنا
 • الجھن یا توہمات
 
-**💡 اہم یاد دہانیاں:**
-• ڈاکٹر کے مشورے کے بغیر دوائیں بند نہ کریں
-• اپنے ڈاکٹر کو مضر اثرات کی رپورٹ کریں
-• دوائیں اصل کنٹینرز میں رکھیں
-• میعاد ختم ہونے کی تاریخوں کو باقاعدگی سے چیک کریں
-
 *نوٹ: ہمیشہ اپنے ہیلتھ کیئر فراہم کرنے والے کی مخصوص ہدایات پر عمل کریں۔*
 """
     
@@ -737,20 +739,6 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 • Severe burns
 • Poisoning or overdose
 • Seizures
-• Severe allergic reactions
-
-**🆘 First Aid Basics:**
-• **CPR**: Start if no breathing/pulse (30 compressions: 2 breaths)
-• **Bleeding**: Apply direct pressure
-• **Choking**: Perform Heimlich maneuver
-• **Burns**: Cool with running water
-• **Fainting**: Lay flat, elevate legs
-
-**📋 Emergency Preparedness:**
-• Keep emergency numbers handy
-• Know location of nearest hospital
-• Have medical history documented
-• Keep medications list updated
 
 *⚠️ THIS IS EMERGENCY GUIDANCE ONLY. SEEK PROFESSIONAL MEDICAL HELP IMMEDIATELY.*
 """
@@ -773,75 +761,60 @@ Please ask me specific health-related questions, and I'll provide detailed, help
 • شدید جلنے
 • زہر یا اوور ڈوز
 • دورے
-• شدید الرجک رد عمل
-
-**🆘 فرسٹ ایڈ کی بنیادی باتیں:**
-• **CPR**: اگر سانس/پلس نہ ہو تو شروع کریں (30 کمپریشن: 2 سانس)
-• **خون بہنا**: براہ راست دباؤ ڈالیں
-• **گلا گھٹنا**: ہیملک مانور انجام دیں
-• **جلنے**: پانی سے ٹھنڈا کریں
-• **بیہوشی**: سیدھا لیٹائیں، ٹانگیں اونچی کریں
-
-**📋 ہنگامی تیاری:**
-• ہنگامی نمبرز ہاتھ میں رکھیں
-• قریبی ہسپتال کا مقام جانیں
-• طبی تاریخ دستاویزی شکل میں رکھیں
-• دوائیوں کی فہرست اپ ڈیٹ رکھیں
 
 *⚠️ یہ صرف ہنگامی رہنمائی ہے۔ فوری طور پر پیشہ ورانہ طبی مدد حاصل کریں۔*
 """
     
     def _format_general_health_response(self, language):
-        knowledge = self.medical_knowledge_base['general_health']
         if language == 'English':
-            return f"""
+            return """
 **🌿 Comprehensive Health & Wellness Guide**
 
 **💚 Lifestyle Recommendations:**
-{chr(10).join(['• ' + tip for tip in knowledge['lifestyle']])}
+• 7-9 hours quality sleep nightly
+• Stress management techniques
+• No smoking
+• Limit alcohol to 1-2 drinks/day
+• Regular health check-ups
 
 **🛡️ Preventive Healthcare:**
-{chr(10).join(['• ' + tip for tip in knowledge['prevention']])}
-
-**🎯 Health Monitoring Schedule:**
-• **Daily**: Blood pressure (if hypertensive), blood sugar (if diabetic)
-• **Weekly**: Weight, exercise minutes
-• **Monthly**: Self-examinations (skin, breasts, testicles)
-• **Annual**: Complete physical, blood tests, dental checkup
-• **As recommended**: Cancer screenings, vision/hearing tests
+• 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)
-• Maintain social connections and mental wellbeing
 
 *Remember: Small consistent changes lead to lasting health improvements!*
 """
         else:
-            return f"""
+            return """
 **🌿 صحت اور تندرستی کی جامع گائیڈ**
 
 **💚 طرز زندگی کی سفارشات:**
-{chr(10).join(['• ' + tip for tip in ['رات میں 7-9 گھنٹے معیاری نیند', 'تناؤ کے انتظام کی تکنیکیں', 'سگریٹ نوشی سے پرہیز', 'شراب کو 1-2 ڈرنکس/دن تک محدود کریں', 'باقاعدہ صحت کی جانچ']])}
+• رات میں 7-9 گھنٹے معیاری نیند
+• تناؤ کے انتظام کی تکنیکیں
+• سگریٹ نوشی سے پرہیز
+• شراب کو 1-2 ڈرنکس/دن تک محدود کریں
+• باقاعدہ صحت کی جانچ
 
 **🛡️ احتیاطی ہیلتھ کیئر:**
-{chr(10).join(['• ' + tip for tip in ['سالانہ جسمانی امتحانات', 'ویکسینیشن اپ ڈیٹ', 'باقاعدہ ہاتھ دھونا', 'سورج سے تحفظ', 'ذہنی صحت کی دیکھ بھال']])}
-
-**🎯 صحت کی نگرانی کا شیڈول:**
-• **روزانہ**: بلڈ پریشر (اگر ہائی بلڈ پریشر ہو)، بلڈ شوگر (اگر ذیابیطس ہو)
-• **ہفتہ وار**: وزن، ورزش کے منٹ
-• **ماہانہ**: خود معائنہ (جلد، چھاتی، خصیے)
-• **سالانہ**: مکمل جسمانی، خون کے ٹیسٹ، دانتوں کا چیک اپ
-• **جیسا کہ تجویز کیا گیا ہے**: کینسر کی اسکریننگ، بینائی/سماعت کے ٹیسٹ
+• سالانہ جسمانی امتحانات
+• ویکسینیشن اپ ڈیٹ
+• باقاعدہ ہاتھ دھونا
+• سورج سے تحفظ
+• ذہنی صحت کی دیکھ بھال
 
 **🌟 پاکستان کے لیے صحت مند عادات:**
 • فلٹر/ابلا ہوا پانی پیئیں
 • تازہ، مقامی طور پر دستیاب پھل اور سبزیاں کھائیں
 • اچھی غذا کی حفظان صحت کی مشق کریں
 • مناسب سورج کی روشنی (وٹامن ڈی) حاصل کریں
-• سماجی رابطے اور ذہنی تندرستی برقرار رکھیں
 
 *یاد رکھیں: چھوٹی مسلسل تبدیلیاں پائیدار صحت کی بہتری کا باعث بنتی ہیں!*
 """
@@ -864,15 +837,7 @@ I understand you're asking about health topics. I'm specially trained to help wi
 • Emergency situations
 • General wellness and prevention
 
-**💡 How to Get Better Answers:**
-Please ask specific questions like:
-• "What are the symptoms of heart disease?"
-• "How to manage diabetes?"
-• "What foods help lower blood pressure?"
-• "Exercise recommendations for beginners"
-• "When to go to emergency room?"
-
-I'll provide detailed, accurate medical information to help you make informed health decisions.
+Please ask specific questions for detailed medical information.
 
 *⚠️ Remember: I'm an AI assistant. For personal medical advice, always consult qualified healthcare professionals.*
 """
@@ -893,15 +858,7 @@ I'll provide detailed, accurate medical information to help you make informed he
 • ہنگامی حالات
 • عام تندرستی اور روک تھام
 
-**💡 بہتر جوابات کیسے حاصل کریں:**
-براہ کرم مخصوص سوالات پوچھیں جیسے:
-• "دل کی بیماری کی علامات کیا ہیں؟"
-• "ذیابیطس کا انتظام کیسے کریں؟"
-• "کون سی غذائیں بلڈ پریشر کم کرنے میں مدد کرتی ہیں؟"
-• "ابتدائی افراد کے لیے ورزش کی سفارشات"
-• "ایمرجنسی روم کب جائیں؟"
-
-میں آپ کو معلوماتی صحت کے فیصلے کرنے میں مدد کے لیے تفصیلی، درست طبی معلومات فراہم کروں گا۔
+براہ کرم تفصیلی طبی معلومات کے لیے مخصوص سوالات پوچھیں۔
 
 *⚠️ یاد رکھیں: میں ایک AI اسسٹنٹ ہوں۔ ذاتی طبی مشورے کے لیے، ہمیشہ کوالیفائیڈ ہیلتھ کیئر پروفیشنلز سے مشورہ کریں۔*
 """
@@ -919,7 +876,6 @@ class PDFReportGenerator:
         if language == 'English':
             self.pdf.cell(0, 10, 'SEHATAI MEDICAL ASSESSMENT REPORT', 0, 1, 'C')
         else:
-            # For Urdu, we'll use English to avoid encoding issues
             self.pdf.cell(0, 10, 'SEHATAI MEDICAL ASSESSMENT REPORT', 0, 1, 'C')
         
         self.pdf.ln(10)
@@ -1010,7 +966,8 @@ class PDFReportGenerator:
         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 self.pdf.output(dest='S').encode('latin1', 'replace')
+        # Return the PDF as bytes
+        return self.pdf.output()
     
     def _safe_encode(self, text):
         """Safely encode text for PDF generation"""
@@ -1021,58 +978,39 @@ class PDFReportGenerator:
         except:
             return "Encoding error"
 
-# Load ML models with error handling
+# Load ML models with enhanced error handling
 @st.cache_resource(show_spinner=False)
 def load_models():
     models = {}
-    scalers = {}
     
-    try:
-        # Load heart disease model
-        if os.path.exists('heart_model.pkl'):
-            models['heart'] = joblib.load('heart_model.pkl')
-            st.success("✅ Heart disease model loaded successfully")
-        else:
-            st.warning("⚠️ Heart model file not found, using rule-based assessment")
-            models['heart'] = None
-    except Exception as e:
-        st.warning(f"⚠️ Error loading heart model: {str(e)}")
-        models['heart'] = None
+    # 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']
+    }
     
-    try:
-        # Load diabetes model
-        if os.path.exists('diabeties_model.pkl'):
-            models['diabetes'] = joblib.load('diabeties_model.pkl')
-            st.success("✅ Diabetes model loaded successfully")
-        else:
-            st.warning("⚠️ Diabetes model file not found, using rule-based assessment")
-            models['diabetes'] = None
-    except Exception as e:
-        st.warning(f"⚠️ Error loading diabetes model: {str(e)}")
-        models['diabetes'] = None
+    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)
+                    st.success(f"✅ {model_name} model loaded successfully from {file_path}")
+                    return model
+                except Exception as e:
+                    st.warning(f"⚠️ Error loading {model_name} from {file_path}: {str(e)}")
+                    continue
+        return None
     
-    try:
-        # Load hypertension model
-        if os.path.exists('hypertension_model.pkl'):
-            models['hypertension'] = joblib.load('hypertension_model.pkl')
-            st.success("✅ Hypertension model loaded successfully")
-        else:
-            st.warning("⚠️ Hypertension model file not found, using rule-based assessment")
-            models['hypertension'] = None
-    except Exception as e:
-        st.warning(f"⚠️ Error loading hypertension model: {str(e)}")
-        models['hypertension'] = 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'])
     
-    # Load scalers
-    try:
-        if os.path.exists('scaler.pkl'):
-            scalers['default'] = joblib.load('scaler.pkl')
-            st.success("✅ Scaler loaded successfully")
-    except:
-        st.warning("⚠️ Scaler not available, using standard scaling")
-        scalers['default'] = None
+    # 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, scalers
+    return models
 
 def calculate_priority_score(heart_risk, diabetes_risk, hypertension_risk):
     """Calculate integrated priority score with clinical weighting"""
@@ -1118,71 +1056,56 @@ def validate_patient_data(age, bp_systolic, bp_diastolic, heart_rate):
     
     return errors
 
-def predict_with_models(age, bp_systolic, bp_diastolic, heart_rate, cholesterol, glucose, bmi, symptoms, models, scalers):
-    """Predict risks using loaded ML models"""
+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
     
-    try:
-        # Prepare features for heart model
-        if models['heart'] is not None:
+    # 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
             ]])
-            
-            if scalers.get('default') is not None:
-                heart_features = scalers['default'].transform(heart_features)
-            
             heart_risk = models['heart'].predict_proba(heart_features)[0][1]
-        else:
+        except Exception as e:
             heart_risk = predict_risk_rule_based_heart(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms)
-    
-    except Exception as e:
+    else:
         heart_risk = predict_risk_rule_based_heart(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms)
     
-    try:
-        # Prepare features for diabetes model
-        if models['diabetes'] is not None:
+    # 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
             ]])
-            
-            if scalers.get('default') is not None:
-                diabetes_features = scalers['default'].transform(diabetes_features)
-            
             diabetes_risk = models['diabetes'].predict_proba(diabetes_features)[0][1]
-        else:
+        except Exception as e:
             diabetes_risk = predict_risk_rule_based_diabetes(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms)
-    
-    except Exception as e:
+    else:
         diabetes_risk = predict_risk_rule_based_diabetes(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms)
     
-    try:
-        # Prepare features for hypertension model
-        if models['hypertension'] is not None:
+    # 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
             ]])
-            
-            if scalers.get('default') is not None:
-                hypertension_features = scalers['default'].transform(hypertension_features)
-            
             hypertension_risk = models['hypertension'].predict_proba(hypertension_features)[0][1]
-        else:
+        except Exception as e:
             hypertension_risk = predict_risk_rule_based_hypertension(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms)
-    
-    except Exception as e:
+    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):
-    """Rule-based heart risk prediction"""
+    """Advanced rule-based heart risk prediction"""
     risk = 0.0
     
     # Age factor
@@ -1213,10 +1136,10 @@ def predict_risk_rule_based_heart(age, bp_systolic, bp_diastolic, cholesterol, g
     if symptoms['shortness_breath']: risk += 0.15
     if symptoms['palpitations']: risk += 0.10
     
-    return min(0.95, risk)
+    return min(0.95, max(0.05, risk))
 
 def predict_risk_rule_based_diabetes(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms):
-    """Rule-based diabetes risk prediction"""
+    """Advanced rule-based diabetes risk prediction"""
     risk = 0.0
     
     # Age factor
@@ -1241,10 +1164,10 @@ def predict_risk_rule_based_diabetes(age, bp_systolic, bp_diastolic, cholesterol
     if symptoms['fatigue']: risk += 0.10
     if symptoms['blurred_vision']: risk += 0.08
     
-    return min(0.95, risk)
+    return min(0.95, max(0.05, risk))
 
 def predict_risk_rule_based_hypertension(age, bp_systolic, bp_diastolic, cholesterol, glucose, bmi, symptoms):
-    """Rule-based hypertension risk prediction"""
+    """Advanced rule-based hypertension risk prediction"""
     risk = 0.0
     
     # Blood pressure factor (most important)
@@ -1264,11 +1187,11 @@ def predict_risk_rule_based_hypertension(age, bp_systolic, bp_diastolic, cholest
     if symptoms['dizziness']: risk += 0.10
     if symptoms['palpitations']: risk += 0.08
     
-    return min(0.95, risk)
+    return min(0.95, max(0.05, risk))
 
-def create_download_link(pdf_data, filename, text):
+def create_download_link(pdf_bytes, filename, text):
     """Create a download link for PDF"""
-    b64 = base64.b64encode(pdf_data).decode()
+    b64 = base64.b64encode(pdf_bytes).decode()
     href = f'<a href="data:application/octet-stream;base64,{b64}" download="{filename}">{text}</a>'
     return href
 
@@ -1279,10 +1202,7 @@ def main():
     
     # Load ML models
     with st.spinner("🔄 Loading AI models..."):
-        models, scalers = load_models()
-    
-    # Initialize chatbot
-    chatbot = HealthcareChatbot()
+        models = load_models()
     
     # Language selector at top right with Pakistan flag colors
     col_lang, col_space = st.columns([1, 5])
@@ -1454,7 +1374,7 @@ def main():
                             # 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, scalers
+                                cholesterol, glucose, bmi, symptoms_dict, models
                             )
                             
                             # Calculate integrated priority score
@@ -1574,6 +1494,34 @@ def main():
             st.markdown('<h2 style="color: #00401A; border-bottom: 3px solid #00401A; padding-bottom: 10px;">💬 سیحت اے آئی ہیلتھ اسسٹنٹ</h2>', unsafe_allow_html=True)
             st.write("صحت سے متعلق سوالات پوچھیں اور فوری، قابل اعتماد طبی معلومات حاصل کریں")
         
+        # Groq API Configuration
+        with st.expander("🔑 AI Assistant Configuration", expanded=False):
+            st.markdown('<div class="api-key-section">', unsafe_allow_html=True)
+            if language == "English":
+                groq_api_key = st.text_input("Enter Groq API Key:", type="password", 
+                                           value=st.session_state.groq_api_key,
+                                           help="Get your API key from https://console.groq.com")
+                st.caption("💡 Using Groq API enhances response quality with advanced AI. Leave empty for basic rule-based responses.")
+            else:
+                groq_api_key = st.text_input("Groq API کلید درج کریں:", type="password", 
+                                           value=st.session_state.groq_api_key,
+                                           help="API کلید حاصل کریں https://console.groq.com سے")
+                st.caption("💡 Groq API کا استعمال جدید AI کے ساتھ جوابات کی معیار کو بہتر بناتا ہے۔ بنیادی رول بیسڈ جوابات کے لیے خالی چھوڑ دیں۔")
+            st.markdown('</div>', unsafe_allow_html=True)
+            
+            if groq_api_key:
+                st.session_state.groq_api_key = groq_api_key
+                try:
+                    groq_client = GroqClient(groq_api_key)
+                    chatbot = HealthcareChatbot(groq_client)
+                    st.success("✅ Groq API configured successfully! Using advanced AI responses.")
+                except Exception as e:
+                    st.error(f"❌ Error configuring Groq API: {str(e)}")
+                    chatbot = HealthcareChatbot()
+            else:
+                chatbot = HealthcareChatbot()
+                st.info("ℹ️ Using rule-based health assistant. For enhanced responses, add Groq API key.")
+        
         # Display chat history
         for message in st.session_state.chat_history:
             if message["role"] == "user":
@@ -1610,69 +1558,38 @@ def main():
         
         with col_qa1:
             if st.button("❤️ Heart Health", use_container_width=True):
-                st.session_state.chat_history.append({
-                    "role": "user", 
-                    "content": "Tell me about heart disease symptoms and prevention"
-                })
+                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):
-                st.session_state.chat_history.append({
-                    "role": "user", 
-                    "content": "What are diabetes symptoms and management strategies?"
-                })
+                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):
-                st.session_state.chat_history.append({
-                    "role": "user", 
-                    "content": "How to control high blood pressure naturally?"
-                })
+                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):
-                st.session_state.chat_history.append({
-                    "role": "user", 
-                    "content": "What is a healthy diet for heart health?"
-                })
-                st.rerun()
-        
-        # Additional quick actions
-        col_qa5, col_qa6, col_qa7, col_qa8 = st.columns(4)
-        
-        with col_qa5:
-            if st.button("💪 Exercise", use_container_width=True):
-                st.session_state.chat_history.append({
-                    "role": "user", 
-                    "content": "Exercise recommendations for beginners"
-                })
-                st.rerun()
-        
-        with col_qa6:
-            if st.button("🔍 Symptoms", use_container_width=True):
-                st.session_state.chat_history.append({
-                    "role": "user", 
-                    "content": "When should I seek emergency medical care?"
-                })
-                st.rerun()
-        
-        with col_qa7:
-            if st.button("💊 Medications", use_container_width=True):
-                st.session_state.chat_history.append({
-                    "role": "user", 
-                    "content": "General medication safety guidelines"
-                })
-                st.rerun()
-        
-        with col_qa8:
-            if st.button("🚨 Emergency", use_container_width=True):
-                st.session_state.chat_history.append({
-                    "role": "user", 
-                    "content": "What are emergency warning signs?"
-                })
+                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)
@@ -1699,16 +1616,16 @@ def main():
                 st.subheader("مریض رپورٹ ڈاؤن لوڈ کریں")
             
             pdf_generator = PDFReportGenerator()
-            pdf_data = pdf_generator.generate_report(
+            pdf_bytes = pdf_generator.generate_report(
                 latest_assessment['patient_data'], 
                 latest_assessment['risk_scores'], 
                 language
             )
             
             if language == "English":
-                st.markdown(create_download_link(pdf_data, "sehatai_medical_report.pdf", "📥 Download Medical Report (PDF)"), unsafe_allow_html=True)
+                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_data, "sehatai_medical_report.pdf", "📥 طبی رپورٹ ڈاؤن لوڈ کریں (PDF)"), unsafe_allow_html=True)
+                st.markdown(create_download_link(pdf_bytes, "sehatai_medical_report.pdf", "📥 طبی رپورٹ ڈاؤن لوڈ کریں (PDF)"), unsafe_allow_html=True)
             
             # Analytics Section
             if language == "English":