Update app.py

app.py

@@ -62,16 +62,47 @@ def load_data():
 
     # Encode diseases
     disease_dict = {
-        'Fungal infection': 0, 'Allergy': 1, 'GERD': 2, 'Chronic cholestasis': 3, 'Drug Reaction': 4,
-        'Peptic ulcer disease': 5, 'AIDS': 6, 'Diabetes': 7, 'Gastroenteritis': 8, 'Bronchial Asthma': 9,
-        'Hypertension': 10, 'Migraine': 11, 'Cervical spondylosis': 12, 'Paralysis (brain hemorrhage)': 13,
-        'Jaundice': 14, 'Malaria': 15, 'Chicken pox': 16, 'Dengue': 17, 'Typhoid': 18, 'Hepatitis A': 19,
-        'Hepatitis B': 20, 'Hepatitis C': 21, 'Hepatitis D': 22, 'Hepatitis E': 23, 'Alcoholic hepatitis': 24,
-        'Tuberculosis': 25, 'Common Cold': 26, 'Pneumonia': 27, 'Dimorphic hemorrhoids(piles)': 28,
-        'Heart attack': 29, 'Varicose veins': 30, 'Hypothyroidism': 31, 'Hyperthyroidism': 32,
-        'Hypoglycemia': 33, 'Osteoarthritis': 34, 'Arthritis': 35,
-        '(vertigo) Paroxysmal Positional Vertigo': 36, 'Acne': 37, 'Urinary tract infection': 38,
-        'Psoriasis': 39, 'Impetigo': 40
+        'Fungal infection': 0, 
+        'Allergy': 1, 
+        'GERD': 2, 
+        'Chronic cholestasis': 3, 
+        'Drug Reaction': 4,
+        'Peptic ulcer disease': 5,  # corrected spelling
+        'AIDS': 6, 
+        'Diabetes': 7,  # corrected spelling (removed extra space)
+        'Gastroenteritis': 8, 
+        'Bronchial Asthma': 9,
+        'Hypertension': 10,  # corrected spelling (removed extra space)
+        'Migraine': 11, 
+        'Cervical spondylosis': 12, 
+        'Paralysis (brain hemorrhage)': 13,
+        'Jaundice': 14, 
+        'Malaria': 15, 
+        'Chicken pox': 16, 
+        'Dengue': 17, 
+        'Typhoid': 18, 
+        'Hepatitis A': 19,  # corrected spelling (capitalized)
+        'Hepatitis B': 20, 
+        'Hepatitis C': 21, 
+        'Hepatitis D': 22, 
+        'Hepatitis E': 23, 
+        'Alcoholic hepatitis': 24,
+        'Tuberculosis': 25, 
+        'Common Cold': 26, 
+        'Pneumonia': 27, 
+        'Dimorphic hemorrhoids (piles)': 28,  # corrected spelling
+        'Heart attack': 29, 
+        'Varicose veins': 30, 
+        'Hypothyroidism': 31, 
+        'Hyperthyroidism': 32,
+        'Hypoglycemia': 33, 
+        'Osteoarthritis': 34,  # corrected spelling
+        'Arthritis': 35,
+        '(vertigo) Paroxysmal Positional Vertigo': 36,  # corrected spelling
+        'Acne': 37, 
+        'Urinary tract infection': 38,
+        'Psoriasis': 39, 
+        'Impetigo': 40
     }
 
     # Replace prognosis values with numerical categories