Update app.py

app.py

@@ -332,49 +332,39 @@ class HealthcareChatbot:
     def _get_rule_based_response(self, user_input, language):
         user_input_lower = user_input.lower().strip()
         
-        # For greetings, provide a professional medical response instead of generic welcome
+        # Improved response mapping
+        response_map = {
+            'heart': (['heart', 'cardiac', 'chest pain', 'cholesterol', 'دل', 'سینے میں درد', 'heart attack'], 
+                     self._format_heart_response),
+            'diabetes': (['diabetes', 'sugar', 'glucose', 'insulin', 'ذیابیطس', 'شوگر'], 
+                        self._format_diabetes_response),
+            'hypertension': (['blood pressure', 'hypertension', 'bp', 'بلڈ پریشر', 'ہائی بلڈ پریشر'], 
+                           self._format_hypertension_response),
+            'nutrition': (['diet', 'nutrition', 'food', 'eating', 'غذا', 'کھانا', 'diet'], 
+                         self._format_nutrition_response),
+            'exercise': (['exercise', 'workout', 'fitness', 'gym', 'ورزش', 'جم', 'physical'], 
+                        self._format_exercise_response),
+            'symptoms': (['symptom', 'pain', 'fever', 'headache', 'علامت', 'درد', 'بخار'], 
+                        self._format_symptoms_response),
+            'medicine': (['medicine', 'medication', 'drug', 'treatment', 'دوا', 'علاج'], 
+                        self._format_medication_response),
+            'emergency': (['emergency', 'urgent', 'help', 'ہنگامی', 'فوری'], 
+                         self._format_emergency_response),
+            'general': (['health', 'wellness', 'prevention', 'صحت', 'تندرستی'], 
+                       self._format_general_health_response)
+        }
+        
+        # Check for greetings first
         if any(word in user_input_lower for word in ['hello', 'hi', 'hey', 'salam', 'السلام علیکم']):
             return self._format_professional_greeting(user_input, language)
         
-        # Heart disease related queries
-        if any(word in user_input_lower for word in ['heart', 'cardiac', 'chest pain', 'cholesterol', 'دل', 'سینے میں درد']):
-            return self._format_heart_response(language)
-        
-        # Diabetes related queries
-        elif any(word in user_input_lower for word in ['diabetes', 'sugar', 'glucose', 'insulin', 'ذیابیطس', 'شوگر']):
-            return self._format_diabetes_response(language)
-        
-        # Hypertension related queries
-        elif any(word in user_input_lower for word in ['blood pressure', 'hypertension', 'bp', 'بلڈ پریشر', 'ہائی بلڈ پریشر']):
-            return self._format_hypertension_response(language)
-        
-        # Nutrition and diet
-        elif any(word in user_input_lower for word in ['diet', 'nutrition', 'food', 'eating', 'غذا', 'کھانا']):
-            return self._format_nutrition_response(language)
-        
-        # Exercise and fitness
-        elif any(word in user_input_lower for word in ['exercise', 'workout', 'fitness', 'gym', 'ورزش', 'جم']):
-            return self._format_exercise_response(language)
+        # Check for matching categories
+        for category, (keywords, response_func) in response_map.items():
+            if any(keyword in user_input_lower for keyword in keywords):
+                return response_func(language)
         
-        # Symptoms assessment
-        elif any(word in user_input_lower for word in ['symptom', 'pain', 'fever', 'headache', 'علامت', 'درد']):
-            return self._format_symptoms_response(language)
-        
-        # Medication queries
-        elif any(word in user_input_lower for word in ['medicine', 'medication', 'drug', 'treatment', 'دوا', 'علاج']):
-            return self._format_medication_response(language)
-        
-        # Emergency situations
-        elif any(word in user_input_lower for word in ['emergency', 'urgent', 'help', 'ہنگامی', 'فوری']):
-            return self._format_emergency_response(language)
-        
-        # General health queries
-        elif any(word in user_input_lower for word in ['health', 'wellness', 'prevention', 'صحت', 'تندرستی']):
-            return self._format_general_health_response(language)
-        
-        # For unknown queries, provide professional medical assistance response
-        else:
-            return self._format_professional_response(user_input, language)
+        # Default professional response for unknown queries
+        return self._format_professional_response(user_input, language)
     
     def _format_professional_greeting(self, user_input, language):
         """Professional response for greetings"""
@@ -404,38 +394,76 @@ What specific health issue would you like to discuss today?
 """
     
     def _format_professional_response(self, user_input, language):
-        """Professional response for unknown queries"""
+        """Improved professional response for unknown queries"""
         if language == 'English':
             return f"""
 **Dr. SehatAI:** I understand you're asking about: "{user_input}"
 
-As your medical AI assistant, I'm here to provide professional healthcare guidance. Could you please provide more details about your specific health concern so I can give you the most accurate medical advice?
+I'm here to provide professional medical guidance. Based on your query, here are the key areas I can help you with:
 
-I specialize in:
-• Cardiovascular health and heart conditions
-• Diabetes management and prevention
-• Blood pressure control and hypertension
-• General wellness and preventive care
+🔍 **Medical Assessment & Risk Prediction**
+• Heart disease risk evaluation
+• Diabetes screening and management  
+• Blood pressure monitoring
+• Symptom analysis and triage
 
-Please share more specific details about your symptoms or health questions.
+💊 **Treatment & Medication Guidance**
+• Medication safety information
+• Treatment protocols for common conditions
+• Drug interaction checks
 
-*Note: I'm an AI assistant. For serious medical concerns, always consult a qualified healthcare professional.*
+🩺 **Clinical Consultation**
+• Interpretation of medical reports
+• Preventive healthcare strategies
+• Lifestyle modification advice
+
+🚨 **Emergency Guidance**
+• Identifying critical symptoms
+• First aid recommendations
+• When to seek immediate care
+
+Please ask me specific questions about:
+• Your symptoms or health concerns
+• Medical test results interpretation  
+• Medication queries
+• Preventive health measures
+
+*Note: For medical emergencies, please call 1122 immediately or visit your nearest hospital.*
 """
         else:
             return f"""
 **ڈاکٹر سیحت اے آئی:** میں سمجھتا ہوں کہ آپ پوچھ رہے ہیں: "{user_input}"
 
-آپ کے میڈیکل AI اسسٹنٹ کے طور پر، میں پیشہ ورانہ ہیلتھ کیئر رہنمائی فراہم کرنے کے لیے یہاں ہوں۔ براہ کرم اپنے مخصوص صحت کے مسئلے کے بارے میں مزید تفصیلات فراہم کریں تاکہ میں آپ کو سب سے درست طبی مشورہ دے سکوں۔
+میں پیشہ ورانہ طبی رہنمائی فراہم کرنے کے لیے یہاں ہوں۔ آپ کے سوال کی بنیاد پر، میں آپ کی ان اہم شعبوں میں مدد کر سکتا ہوں:
 
-میں اس میں مہارت رکھتا ہوں:
-• قلبی صحت اور دل کے حالات
-• ذیابیطس کا انتظام اور روک تھام
-• بلڈ پریشر کنٹرول اور ہائی بلڈ پریشر
-• عام تندرستی اور احتیاطی دیکھ بھال
+🔍 **طبی تشخیص اور خطرے کی پیشگوئی**
+• دل کی بیماری کے خطرے کا جائزہ
+• ذیابیطس کی اسکریننگ اور انتظام
+• بلڈ پریشر کی نگرانی
+• علامات کا تجزیہ اور ترجیح
 
-براہ کرم اپنی علامات یا صحت کے سوالات کے بارے میں مزید مخصوص تفصیلات شیئر کریں۔
+💊 **علاج اور دوائیوں کی رہنمائی**
+• دوائیوں کی حفاظت کی معلومات
+• عام حالات کے لیے علاج کے طریقہ کار
+• دوائیوں کے باہمی تعامل کی چیکنگ
 
-*نوٹ: میں ایک AI اسسٹنٹ ہوں۔ سنگین طبی مسائل کے لیے، ہمیشہ کوالیفائیڈ ہیلتھ کیئر پروفیشنل سے مشورہ کریں۔*
+🩺 **کلینیکل مشاورت**
+• طبی رپورٹس کی تشریح
+• احتیاطی ہیلتھ کیئر کی حکمت عملی
+• طرز زندگی میں تبدیلی کے مشورے
+
+🚨 **ہنگامی رہنمائی**
+• critical علامات کی شناخت
+• فرسٹ ایڈ کی سفارشات
+• فوری دیکھ بھال کی ضرورت کا تعین
+
+براہ کرم مجھ سے ان کے بارے میں مخصوص سوالات پوچھیں:
+• آپ کی علامات یا صحت کے خدشات
+• طبی ٹیسٹ کے نتائج کی تشریح
+• دوائیوں کے سوالات
+• احتیاطی صحت کے اقدامات
+
+*نوٹ: طبی ایمرجنسیز کے لیے، براہ کرم فوری طور پر 1122 پر کال کریں یا اپنے قریبی ہسپتال جائیں۔*
 """
     
     def _format_heart_response(self, language):
@@ -443,7 +471,6 @@ Please share more specific details about your symptoms or health questions.
         if language == 'English':
             return f"""
 **Dr. SehatAI - Cardiac Specialist:**
-
 Based on your question about heart health, here's my professional assessment:
 
 **Cardiac Symptoms to Monitor:**
@@ -469,7 +496,6 @@ Based on your question about heart health, here's my professional assessment:
         else:
             return f"""
 **ڈاکٹر سیحت اے آئی - کارڈیاک اسپیشلسٹ:**
-
 دل کی صحت کے بارے میں آپ ک�� سوال کی بنیاد پر، یہ میرا پیشہ ورانہ جائزہ ہے:
 
 **نگرانی کرنے والی قلبی علامات:**
@@ -492,7 +518,6 @@ Based on your question about heart health, here's my professional assessment:
         if language == 'English':
             return f"""
 **Dr. SehatAI - Endocrinology Specialist:**
-
 Regarding your diabetes inquiry, here's my clinical perspective:
 
 **Common Diabetic Symptoms:**
@@ -519,7 +544,6 @@ Regarding your diabetes inquiry, here's my clinical perspective:
         else:
             return f"""
 **ڈاکٹر سیحت اے آئی - اینڈوکرینالوجی اسپیشلسٹ:**
-
 ذیابیطس کے بارے میں آپ کی استفسار کے حوالے سے، یہ میرا کلینیکل نقطہ نظر ہے:
 
 **عام ذیابیطس کی علامات:**
@@ -542,7 +566,6 @@ Regarding your diabetes inquiry, here's my clinical perspective:
         if language == 'English':
             return f"""
 **Dr. SehatAI - Hypertension Specialist:**
-
 Regarding your blood pressure concerns, here's my medical assessment:
 
 **Hypertension Symptoms:**
@@ -569,7 +592,6 @@ Regarding your blood pressure concerns, here's my medical assessment:
         else:
             return f"""
 **ڈاکٹر سیحت اے آئی - ہائی بلڈ پریشر اسپیشلسٹ:**
-
 بلڈ پریشر کے بارے میں آپ کی تشویش کے حوالے سے، یہ میری طبی تشخیص ہے:
 
 **ہائی بلڈ پریشر کی علامات:**
@@ -591,7 +613,6 @@ Regarding your blood pressure concerns, here's my medical assessment:
         if language == 'English':
             return """
 **Dr. SehatAI - Clinical Nutrition Specialist:**
-
 Based on your nutritional inquiry, here are evidence-based dietary recommendations:
 
 **Core Nutritional Principles:**
@@ -617,7 +638,6 @@ Based on your nutritional inquiry, here are evidence-based dietary recommendatio
         else:
             return """
 **ڈاکٹر سیحت اے آئی - کلینیکل نیوٹریشن اسپیشلسٹ:**
-
 آپ کی غذائی استفسار کی بنیاد پر، یہ ثبوت پر مبنی غذائی سفارشات ہیں:
 
 **بنیادی غذائی اصول:**
@@ -645,7 +665,6 @@ Based on your nutritional inquiry, here are evidence-based dietary recommendatio
         if language == 'English':
             return """
 **Dr. SehatAI - Exercise Physiology Specialist:**
-
 Regarding physical activity, here's my exercise prescription:
 
 **General Exercise Guidelines:**
@@ -671,7 +690,6 @@ Regarding physical activity, here's my exercise prescription:
         else:
             return """
 **ڈاکٹر سیحت اے آئی - ورزش فزیالوجی اسپیشلسٹ:**
-
 جسمانی سرگرمی کے بارے میں، یہ میری ورزش کی نسخہ ہے:
 
 **عام ورزش کے رہنما اصول:**
@@ -699,7 +717,6 @@ Regarding physical activity, here's my exercise prescription:
         if language == 'English':
             return """
 **Dr. SehatAI - Clinical Assessment Specialist:**
-
 Regarding symptom evaluation, here's my clinical guidance:
 
 **When to Seek Medical Evaluation:**
@@ -730,7 +747,6 @@ Regarding symptom evaluation, here's my clinical guidance:
         else:
             return """
 **ڈاکٹر سیحت اے آئی - کلینیکل تشخیص اسپیشلسٹ:**
-
 علامات کی تشخیص کے بارے میں، یہ میری کلینیکل رہنمائی ہے:
 
 **طبی تشخیص کب حاصل کریں:**
@@ -763,7 +779,6 @@ Regarding symptom evaluation, here's my clinical guidance:
         if language == 'English':
             return """
 **Dr. SehatAI - Clinical Pharmacology Specialist:**
-
 Regarding medication management, here's my pharmacological guidance:
 
 **Medication Safety Protocol:**
@@ -792,7 +807,6 @@ Regarding medication management, here's my pharmacological guidance:
         else:
             return """
 **ڈاکٹر سیحت اے آئی - کلینیکل فارماکالوجی اسپیشلسٹ:**
-
 دوائیوں کے انتظام کے بارے میں، یہ میری فارماکولوجیکل رہنمائی ہے:
 
 **دوائیوں کی حفاظت کا پروٹوکول:**
@@ -887,7 +901,6 @@ Regarding medication management, here's my pharmacological guidance:
         if language == 'English':
             return """
 **Dr. SehatAI - Preventive Medicine Specialist:**
-
 Regarding general health and wellness, here's my comprehensive preventive guidance:
 
 **Lifestyle Medicine Recommendations:**
@@ -915,7 +928,6 @@ Regarding general health and wellness, here's my comprehensive preventive guidan
         else:
             return """
 **ڈاکٹر سیحت اے آئی - احتیاطی میڈیسن اسپیشلسٹ:**
-
 عام صحت اور تندرستی کے بارے میں، یہ میری جامع احتیاطی رہنمائی ہے:
 
 **طرز زندگی کی میڈیسن سفارشات:**
@@ -1073,7 +1085,6 @@ def load_models():
             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
@@ -1083,10 +1094,6 @@ def load_models():
     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):
@@ -1659,6 +1666,34 @@ def main():
                     query = "دل کی صحت برقرار رکھنے کے لیے بہترین غذا کیا ہے؟"
                 st.session_state.chat_history.append({"role": "user", "content": query})
                 st.rerun()
+
+        # Test Medical Queries Section
+        if language == "English":
+            st.subheader("Test Medical Queries")
+            st.write("Quickly test the chatbot with these sample medical questions:")
+        else:
+            st.subheader("طبی سوالات کی جانچ")
+            st.write("ان نمونہ طبی سوالات کے ساتھ چیٹ بوٹ کو فوری طور پر آزمائیں:")
+        
+        test_col1, test_col2 = st.columns(2)
+        
+        with test_col1:
+            if st.button("Test Heart Query", use_container_width=True):
+                if language == "English":
+                    test_query = "I have chest pain and high cholesterol, what should I do?"
+                else:
+                    test_query = "مجھے سینے میں درد اور ہائی کولیسٹرول ہے، میں کیا کروں؟"
+                st.session_state.chat_history.append({"role": "user", "content": test_query})
+                st.rerun()
+        
+        with test_col2:
+            if st.button("Test Diabetes Query", use_container_width=True):
+                if language == "English":
+                    test_query = "What are the early symptoms of diabetes?"
+                else:
+                    test_query = "ذیابیطس کی ابتدائی علامات کیا ہیں؟"
+                st.session_state.chat_history.append({"role": "user", "content": test_query})
+                st.rerun()
         
         st.markdown('</div>', unsafe_allow_html=True)