Update app.py

app.py

@@ -7,7 +7,6 @@ import os
 
 groq_api_key = os.environ["groq_api_key"] 
 
-
 # Initialize Groq client
 groq_client = Groq(api_key=groq_api_key)  # Replace with your Groq API key
 
@@ -15,23 +14,55 @@ groq_client = Groq(api_key=groq_api_key)  # Replace with your Groq API key
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-hi-en")
 classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
-# Medical configuration
+# Expanded Medical Conditions
 conditions = [
-    "Asthma", "COPD", "Pneumonia", "Heart Failure", "Diabetes", 
-    "Hypertension", "Migraine", "Gastroenteritis", "COVID-19", "Anemia"
+    # Respiratory Diseases
+    "Asthma", "COPD", "Pneumonia", "Tuberculosis", "COVID-19", "Bronchitis", "Pulmonary Fibrosis", "Cystic Fibrosis", "Emphysema", "Sleep Apnea",
+    # Cardiovascular Diseases
+    "Heart Failure", "Hypertension", "Hypotension", "Coronary Artery Disease", "Arrhythmia", "Atrial Fibrillation", "Myocardial Infarction", "Peripheral Artery Disease", "Stroke", "Aneurysm",
+    # Endocrine & Metabolic Diseases
+    "Diabetes Type 1", "Diabetes Type 2", "Hyperthyroidism", "Hypothyroidism", "Addison’s Disease", "Cushing’s Syndrome", "Polycystic Ovary Syndrome", "Metabolic Syndrome", "Hypoglycemia", "Hyperglycemia",
+    # Neurological Disorders
+    "Migraine", "Epilepsy", "Parkinson’s Disease", "Multiple Sclerosis", "Alzheimer’s Disease", "Huntington’s Disease", "ALS", "Meningitis", "Encephalitis", "Bell’s Palsy",
+    # Digestive System Diseases
+    "Gastroenteritis", "Acid Reflux", "Peptic Ulcer", "Crohn’s Disease", "Ulcerative Colitis", "Irritable Bowel Syndrome", "Hepatitis A", "Hepatitis B", "Hepatitis C", "Cirrhosis", "Gallstones", "Pancreatitis",
+    # Blood & Immune System Disorders
+    "Anemia", "Leukemia", "Lymphoma", "Sickle Cell Disease", "Hemophilia", "Thrombocytopenia", "Lupus", "Rheumatoid Arthritis", "HIV/AIDS", "Sepsis", "Aplastic Anemia",
+    # Musculoskeletal Disorders
+    "Arthritis", "Osteoporosis", "Gout", "Fibromyalgia", "Scoliosis", "Ankylosing Spondylitis", "Carpal Tunnel Syndrome", "Muscular Dystrophy", "Tendinitis", "Bone Fractures",
+    # Skin Conditions
+    "Eczema", "Psoriasis", "Acne", "Rosacea", "Fungal Infections", "Vitiligo", "Hives", "Melanoma", "Alopecia", "Sunburn",
+    # Infectious Diseases
+    "Influenza", "Malaria", "Dengue Fever", "Chickenpox", "Measles", "Mumps", "Typhoid Fever", "Tetanus", "Rabies", "Lyme Disease",
+    # Kidney & Urinary Tract Diseases
+    "Chronic Kidney Disease", "Kidney Stones", "UTI", "Bladder Cancer", "Polycystic Kidney Disease", "Nephritis", "Hydronephrosis", "Glomerulonephritis", "Interstitial Cystitis", "Prostatitis",
+    # Reproductive System Disorders
+    "Endometriosis", "Ovarian Cysts", "Erectile Dysfunction", "Prostate Cancer", "Uterine Fibroids", "Pelvic Inflammatory Disease", "Infertility", "Testicular Cancer", "Menstrual Disorders", "Premature Ovarian Failure",
+    # Mental Health Disorders
+    "Depression", "Anxiety Disorder", "Bipolar Disorder", "Schizophrenia", "PTSD", "OCD", "ADHD", "Autism Spectrum Disorder", "Sleep Disorders", "Substance Abuse Disorder"
 ]
 
+# Expanded Specialist Mapping
 specialist_mapping = {
     "Asthma": ("Pulmonologist", "Respiratory System"),
     "COPD": ("Pulmonologist", "Respiratory System"),
     "Pneumonia": ("Pulmonologist", "Respiratory System"),
+    "Tuberculosis": ("Infectious Disease Specialist", "Respiratory System"),
+    "COVID-19": ("Infectious Disease Specialist", "Immune System"),
     "Heart Failure": ("Cardiologist", "Cardiovascular System"),
-    "Diabetes": ("Endocrinologist", "Endocrine System"),
     "Hypertension": ("Cardiologist", "Cardiovascular System"),
+    "Diabetes Type 1": ("Endocrinologist", "Endocrine System"),
+    "Diabetes Type 2": ("Endocrinologist", "Endocrine System"),
     "Migraine": ("Neurologist", "Nervous System"),
     "Gastroenteritis": ("Gastroenterologist", "Digestive System"),
-    "COVID-19": ("Infectious Disease Specialist", "Immune System"),
-    "Anemia": ("Hematologist", "Blood Disorders")
+    "Anemia": ("Hematologist", "Blood Disorders"),
+    "Depression": ("Psychiatrist", "Mental Health"),
+    "Anxiety Disorder": ("Psychiatrist", "Mental Health"),
+    "Chronic Kidney Disease": ("Nephrologist", "Urinary System"),
+    "UTI": ("Urologist", "Urinary System"),
+    "Osteoporosis": ("Orthopedic Specialist", "Musculoskeletal System"),
+    "Psoriasis": ("Dermatologist", "Skin Disorders"),
+    "Epilepsy": ("Neurologist", "Nervous System")
 }
 
 def generate_expert_analysis(condition, symptoms):