Upload 16 files

.env.example

@@ -0,0 +1,2 @@
+GROQ_API_KEY="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"
+HUGGING_FACE_TOKEN="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"

.gitattributes

@@ -1,35 +1,5 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+# Auto detect text files and perform LF normalization
+* text=auto
+agent/chroma_db/b75a830d-8736-407d-ac23-12c31b1e0ede/data_level0.bin filter=lfs diff=lfs merge=lfs -text
+agent/chroma_db/b75a830d-8736-407d-ac23-12c31b1e0ede/index_metadata.pickle filter=lfs diff=lfs merge=lfs -text
+agent/chroma_db/chroma.sqlite3 filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,2 @@
+agent/chroma_db/chroma.sqlite3
+/agent/chroma_db

agent/chroma_db/b75a830d-8736-407d-ac23-12c31b1e0ede/data_level0.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c7e2a5c66a30e0d9228b85d06681048e2d25425ad5b7f8f10b672c87ac37e001
+size 321200

agent/chroma_db/b75a830d-8736-407d-ac23-12c31b1e0ede/index_metadata.pickle

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8ccf314efa9aa64c9e7790bcb8ca39796ba96ba08ffbab69ea2d128df6078db8
+size 1155658

agent/chroma_db/chroma.sqlite3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4074cbf4b190145e083f50a61bfe8d6a0c9d7a68a54f61d64a329e8f093b16e4
+size 395534336

agent/image_label_config.json

@@ -0,0 +1,63 @@
+{
+  "id2label": {
+    "0": "atopic_dermatitis",
+    "1": "basal_cell_carcinoma",
+    "2": "eczema",
+    "3": "fungal_infections",
+    "4": "keratosis",
+    "5": "melanocytic_nevus",
+    "6": "melanoma",
+    "7": "psoriasis",
+    "8": "viral_warts"
+  },
+  "label2id": {
+    "atopic_dermatitis": 0,
+    "basal_cell_carcinoma": 1,
+    "eczema": 2,
+    "fungal_infections": 3,
+    "keratosis": 4,
+    "melanocytic_nevus": 5,
+    "melanoma": 6,
+    "psoriasis": 7,
+    "viral_warts": 8
+  },
+  "class_names": [
+    "atopic_dermatitis",
+    "basal_cell_carcinoma",
+    "eczema",
+    "fungal_infections",
+    "keratosis",
+    "melanocytic_nevus",
+    "melanoma",
+    "psoriasis",
+    "viral_warts"
+  ],
+  "num_labels": 9,
+  "class_weights": {
+    "0": 1.0,
+    "1": 1.0,
+    "2": 1.0,
+    "3": 1.0,
+    "4": 1.0,
+    "5": 1.0,
+    "6": 1.0,
+    "7": 1.0,
+    "8": 1.0
+  },
+  "version": "2.0-kaggle",
+  "model_name": "microsoft/beit-base-patch16-224-pt22k-ft22k",
+  "image_size": {
+    "height": 224,
+    "width": 224
+  },
+  "platform": "kaggle",
+  "improvements": [
+    "Medical-safe augmentation",
+    "Class weights for imbalance",
+    "Reduced batch size (16)",
+    "Optimized epochs (20)",
+    "Dropout (0.1)",
+    "Early stopping",
+    "Test-time augmentation support"
+  ]
+}

agent/image_symptom_tool.py

@@ -0,0 +1,56 @@
+from langchain_core.tools import tool
+import requests
+import base64
+from PIL import Image
+import io
+
+@tool
+def analyze_symptom_image(image_url: str, image_description: str = "") -> str:
+    """
+    Analyzes symptom images using vision classification model via FastAPI backend.
+    Input: Image URL and optional image description
+    Output: Image symptom analysis with confidence scores 
+    """
+    try:
+        # Download image from URL
+        print(f"📥 Downloading image from: {image_url}")
+        image_response = requests.get(image_url, timeout=15)
+        image_response.raise_for_status()
+        image = Image.open(io.BytesIO(image_response.content)).convert('RGB')
+        
+        # Convert to base64 for FastAPI transmission
+        buffered = io.BytesIO()
+        image.save(buffered, format="PNG")
+        img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
+        
+        # Call your HF FastAPI Space
+        api_url = "https://datdevsteve-nivra-vision-diagnosis.hf.space/run/predict"
+        payload = {
+            "data": [img_base64],
+            "description": image_description or ""
+        }
+        
+        print("🔬 Calling Nivra Vision FastAPI backend...")
+        response = requests.post(api_url, json=payload, timeout=60)
+        response.raise_for_status()
+        
+        result = response.json()
+        
+        # Extract diagnosis from HF Space response format
+        if "data" in result and len(result["data"]) > 0:
+            diagnosis = result["data"][0]
+            return f"""
+[SYMPTOM IMAGE ANALYSIS - SUCCESS]:
+✅ FastAPI Backend Response: {diagnosis}
+
+🔍 **Image Description**: {image_description or "Not provided"}
+📡 **Backend**: nivra-vision-diagnosis HF Space"""
+        else:
+            return "[SYMPTOM IMAGE ANALYSIS - WARNING]: No diagnosis returned from backend"
+            
+    except requests.exceptions.Timeout:
+        return "[SYMPTOM IMAGE ANALYSIS - ERROR]: Backend timeout. Please try again."
+    except requests.exceptions.RequestException as e:
+        return f"[SYMPTOM IMAGE ANALYSIS - ERROR]: Network error: {str(e)}"
+    except Exception as e:
+        return f"[SYMPTOM IMAGE ANALYSIS - ERROR]: Unexpected error: {str(e)}"

agent/rag_retriever.py

@@ -0,0 +1,22 @@
+from langchain_community.vectorstores import Chroma
+from langchain_huggingface import HuggingFaceEmbeddings
+from langchain_text_splitters import RecursiveCharacterTextSplitter
+import os
+
+class NivraRAGRetriever:
+    def __init__(self, chroma_path: str = "./chroma_db"):
+        self.embeddings = HuggingFaceEmbeddings(
+            model_name="sentence-transformers/all-MiniLM-L6-v2"
+        )
+        self.vectorstore = Chroma(
+            persist_directory=chroma_path,
+            embedding_function=self.embeddings
+        )
+        self.retriever = self.vectorstore.as_retriever(
+            search_type="similarity_score_threshold",
+            search_kwargs={"score_threshold": 0.7, "k":5}
+        )
+    
+    def getRelevantDocs(self, query:str):
+        return self.retriever.invoke(query)
+    

agent/text_label_config.json

@@ -0,0 +1,1516 @@
+{
+  "id2label": {
+    "0": "(vertigo) Paroymsal  Positional Vertigo",
+    "1": "AIDS",
+    "2": "Acne",
+    "3": "Alcoholic hepatitis",
+    "4": "Allergy",
+    "5": "Arthritis",
+    "6": "Bronchial Asthma",
+    "7": "Cervical spondylosis",
+    "8": "Chicken pox",
+    "9": "Chronic cholestasis",
+    "10": "Common Cold",
+    "11": "Dengue",
+    "12": "Diabetes",
+    "13": "Dimorphic hemmorhoids(piles)",
+    "14": "Drug Reaction",
+    "15": "Fungal infection",
+    "16": "GERD",
+    "17": "Gastroenteritis",
+    "18": "Heart attack",
+    "19": "Hepatitis B",
+    "20": "Hepatitis C",
+    "21": "Hepatitis D",
+    "22": "Hepatitis E",
+    "23": "Hypertension",
+    "24": "Hyperthyroidism",
+    "25": "Hypoglycemia",
+    "26": "Hypothyroidism",
+    "27": "Impetigo",
+    "28": "Jaundice",
+    "29": "Malaria",
+    "30": "Migraine",
+    "31": "Osteoarthristis",
+    "32": "Paralysis (brain hemorrhage)",
+    "33": "Peptic ulcer diseae",
+    "34": "Pneumonia",
+    "35": "Psoriasis",
+    "36": "Tuberculosis",
+    "37": "Typhoid",
+    "38": "Urinary tract infection",
+    "39": "Varicose veins",
+    "40": "abdominal aortic aneurysm",
+    "41": "abdominal hernia",
+    "42": "abscess of nose",
+    "43": "abscess of the lung",
+    "44": "abscess of the pharynx",
+    "45": "acanthosis nigricans",
+    "46": "acariasis",
+    "47": "achalasia",
+    "48": "actinic keratosis",
+    "49": "acute bronchiolitis",
+    "50": "acute bronchitis",
+    "51": "acute bronchospasm",
+    "52": "acute glaucoma",
+    "53": "acute kidney injury",
+    "54": "acute otitis media",
+    "55": "acute pancreatitis",
+    "56": "acute respiratory distress syndrome (ards)",
+    "57": "acute sinusitis",
+    "58": "acute stress reaction",
+    "59": "adhesive capsulitis of the shoulder",
+    "60": "adjustment reaction",
+    "61": "adrenal adenoma",
+    "62": "alcohol abuse",
+    "63": "alcohol intoxication",
+    "64": "alcohol withdrawal",
+    "65": "alcoholic liver disease",
+    "66": "allergy to animals",
+    "67": "alopecia",
+    "68": "alzheimer disease",
+    "69": "amblyopia",
+    "70": "amyloidosis",
+    "71": "amyotrophic lateral sclerosis (als)",
+    "72": "anal fissure",
+    "73": "anal fistula",
+    "74": "anemia",
+    "75": "anemia due to chronic kidney disease",
+    "76": "anemia due to malignancy",
+    "77": "anemia of chronic disease",
+    "78": "angina",
+    "79": "ankylosing spondylitis",
+    "80": "anxiety",
+    "81": "aortic valve disease",
+    "82": "aphakia",
+    "83": "aphthous ulcer",
+    "84": "aplastic anemia",
+    "85": "appendicitis",
+    "86": "arrhythmia",
+    "87": "arthritis of the hip",
+    "88": "ascending cholangitis",
+    "89": "asperger syndrome",
+    "90": "asthma",
+    "91": "astigmatism",
+    "92": "atelectasis",
+    "93": "athlete's foot",
+    "94": "atonic bladder",
+    "95": "atrial fibrillation",
+    "96": "atrial flutter",
+    "97": "atrophic skin condition",
+    "98": "atrophic vaginitis",
+    "99": "atrophy of the corpus cavernosum",
+    "100": "attention deficit hyperactivity disorder (adhd)",
+    "101": "autism",
+    "102": "autonomic nervous system disorder",
+    "103": "avascular necrosis",
+    "104": "balanitis",
+    "105": "bell palsy",
+    "106": "benign kidney cyst",
+    "107": "benign paroxysmal positional vertical (bppv)",
+    "108": "benign prostatic hyperplasia (bph)",
+    "109": "benign vaginal discharge (leukorrhea)",
+    "110": "bipolar disorder",
+    "111": "birth trauma",
+    "112": "bladder cancer",
+    "113": "bladder disorder",
+    "114": "bladder obstruction",
+    "115": "blepharitis",
+    "116": "blepharospasm",
+    "117": "bone cancer",
+    "118": "bone disorder",
+    "119": "bone spur of the calcaneous",
+    "120": "brachial neuritis",
+    "121": "brain cancer",
+    "122": "breast cancer",
+    "123": "breast cyst",
+    "124": "breast infection (mastitis)",
+    "125": "broken tooth",
+    "126": "bunion",
+    "127": "burn",
+    "128": "bursitis",
+    "129": "callus",
+    "130": "carbon monoxide poisoning",
+    "131": "cardiac arrest",
+    "132": "cardiomyopathy",
+    "133": "carpal tunnel syndrome",
+    "134": "cat scratch disease",
+    "135": "cataract",
+    "136": "celiac disease",
+    "137": "cellulitis or abscess of mouth",
+    "138": "central atherosclerosis",
+    "139": "central retinal artery or vein occlusion",
+    "140": "cerebral edema",
+    "141": "cerebral palsy",
+    "142": "cervical cancer",
+    "143": "cervical disorder",
+    "144": "cervicitis",
+    "145": "chalazion",
+    "146": "chickenpox",
+    "147": "chlamydia",
+    "148": "cholecystitis",
+    "149": "choledocholithiasis",
+    "150": "cholesteatoma",
+    "151": "chondromalacia of the patella",
+    "152": "chorioretinitis",
+    "153": "chronic back pain",
+    "154": "chronic constipation",
+    "155": "chronic glaucoma",
+    "156": "chronic inflammatory demyelinating polyneuropathy (cidp)",
+    "157": "chronic kidney disease",
+    "158": "chronic knee pain",
+    "159": "chronic obstructive pulmonary disease (copd)",
+    "160": "chronic otitis media",
+    "161": "chronic pain disorder",
+    "162": "chronic pancreatitis",
+    "163": "chronic rheumatic fever",
+    "164": "chronic sinusitis",
+    "165": "cirrhosis",
+    "166": "coagulation (bleeding) disorder",
+    "167": "cold sore",
+    "168": "colonic polyp",
+    "169": "colorectal cancer",
+    "170": "complex regional pain syndrome",
+    "171": "concussion",
+    "172": "conduct disorder",
+    "173": "conductive hearing loss",
+    "174": "congenital heart defect",
+    "175": "congenital malformation syndrome",
+    "176": "conjunctivitis",
+    "177": "conjunctivitis due to allergy",
+    "178": "conjunctivitis due to bacteria",
+    "179": "conjunctivitis due to virus",
+    "180": "connective tissue disorder",
+    "181": "contact dermatitis",
+    "182": "conversion disorder",
+    "183": "cornea infection",
+    "184": "corneal abrasion",
+    "185": "corneal disorder",
+    "186": "coronary atherosclerosis",
+    "187": "cranial nerve palsy",
+    "188": "crohn disease",
+    "189": "croup",
+    "190": "crushing injury",
+    "191": "cryptorchidism",
+    "192": "cyst of the eyelid",
+    "193": "cystic fibrosis",
+    "194": "cysticercosis",
+    "195": "cystitis",
+    "196": "de quervain disease",
+    "197": "decubitus ulcer",
+    "198": "deep vein thrombosis (dvt)",
+    "199": "degenerative disc disease",
+    "200": "delirium",
+    "201": "dementia",
+    "202": "dental caries",
+    "203": "depression",
+    "204": "dermatitis due to sun exposure",
+    "205": "developmental disability",
+    "206": "deviated nasal septum",
+    "207": "diabetic ketoacidosis",
+    "208": "diabetic peripheral neuropathy",
+    "209": "diabetic retinopathy",
+    "210": "diaper rash",
+    "211": "dislocation of the ankle",
+    "212": "dislocation of the elbow",
+    "213": "dislocation of the finger",
+    "214": "dislocation of the foot",
+    "215": "dislocation of the hip",
+    "216": "dislocation of the knee",
+    "217": "dislocation of the patella",
+    "218": "dislocation of the shoulder",
+    "219": "dislocation of the vertebra",
+    "220": "dislocation of the wrist",
+    "221": "dissociative disorder",
+    "222": "diverticulitis",
+    "223": "diverticulosis",
+    "224": "down syndrome",
+    "225": "drug abuse",
+    "226": "drug abuse (barbiturates)",
+    "227": "drug abuse (cocaine)",
+    "228": "drug abuse (methamphetamine)",
+    "229": "drug abuse (opioids)",
+    "230": "drug poisoning due to medication",
+    "231": "drug withdrawal",
+    "232": "dry eye of unknown cause",
+    "233": "dumping syndrome",
+    "234": "dyshidrosis",
+    "235": "dysthymic disorder",
+    "236": "ear drum damage",
+    "237": "ear wax impaction",
+    "238": "eating disorder",
+    "239": "ectopic pregnancy",
+    "240": "ectropion",
+    "241": "eczema",
+    "242": "emphysema",
+    "243": "empyema",
+    "244": "encephalitis",
+    "245": "endocarditis",
+    "246": "endometrial cancer",
+    "247": "endometrial hyperplasia",
+    "248": "endometriosis",
+    "249": "endophthalmitis",
+    "250": "envenomation from spider or animal bite",
+    "251": "ependymoma",
+    "252": "epididymitis",
+    "253": "epidural hemorrhage",
+    "254": "epilepsy",
+    "255": "erectile dysfunction",
+    "256": "erythema multiforme",
+    "257": "esophageal cancer",
+    "258": "esophagitis",
+    "259": "essential tremor",
+    "260": "eustachian tube dysfunction (ear disorder)",
+    "261": "extrapyramidal effect of drugs",
+    "262": "eye alignment disorder",
+    "263": "factitious disorder",
+    "264": "female genitalia infection",
+    "265": "female infertility of unknown cause",
+    "266": "fetal alcohol syndrome",
+    "267": "fibroadenoma",
+    "268": "fibrocystic breast disease",
+    "269": "fibromyalgia",
+    "270": "flat feet",
+    "271": "floaters",
+    "272": "flu",
+    "273": "fluid overload",
+    "274": "folate deficiency",
+    "275": "food allergy",
+    "276": "foreign body in the ear",
+    "277": "foreign body in the eye",
+    "278": "foreign body in the gastrointestinal tract",
+    "279": "foreign body in the throat",
+    "280": "foreign body in the vagina",
+    "281": "fracture of the ankle",
+    "282": "fracture of the arm",
+    "283": "fracture of the facial bones",
+    "284": "fracture of the finger",
+    "285": "fracture of the foot",
+    "286": "fracture of the hand",
+    "287": "fracture of the jaw",
+    "288": "fracture of the leg",
+    "289": "fracture of the neck",
+    "290": "fracture of the patella",
+    "291": "fracture of the pelvis",
+    "292": "fracture of the rib",
+    "293": "fracture of the shoulder",
+    "294": "fracture of the skull",
+    "295": "fracture of the vertebra",
+    "296": "friedrich ataxia",
+    "297": "frostbite",
+    "298": "fungal infection of the hair",
+    "299": "fungal infection of the skin",
+    "300": "galactorrhea of unknown cause",
+    "301": "gallstone",
+    "302": "ganglion cyst",
+    "303": "gastritis",
+    "304": "gastroduodenal ulcer",
+    "305": "gastroesophageal reflux disease (gerd)",
+    "306": "gastrointestinal hemorrhage",
+    "307": "gastroparesis",
+    "308": "genital herpes",
+    "309": "gestational diabetes",
+    "310": "glaucoma",
+    "311": "glucocorticoid deficiency",
+    "312": "goiter",
+    "313": "gonorrhea",
+    "314": "gout",
+    "315": "graves disease",
+    "316": "guillain barre syndrome",
+    "317": "gum disease",
+    "318": "gynecomastia",
+    "319": "hammer toe",
+    "320": "head and neck cancer",
+    "321": "head injury",
+    "322": "headache after lumbar puncture",
+    "323": "heart block",
+    "324": "heart failure",
+    "325": "heat exhaustion",
+    "326": "hemangioma",
+    "327": "hemarthrosis",
+    "328": "hematoma",
+    "329": "hemiplegia",
+    "330": "hemochromatosis",
+    "331": "hemolytic anemia",
+    "332": "hemophilia",
+    "333": "hemorrhoids",
+    "334": "hepatic encephalopathy",
+    "335": "hepatitis A",
+    "336": "hepatitis due to a toxin",
+    "337": "herniated disk",
+    "338": "herpangina",
+    "339": "hiatal hernia",
+    "340": "hidradenitis suppurativa",
+    "341": "hirschsprung disease",
+    "342": "hirsutism",
+    "343": "histoplasmosis",
+    "344": "hormone disorder",
+    "345": "hpv",
+    "346": "human immunodeficiency virus infection (hiv)",
+    "347": "hydatidiform mole",
+    "348": "hydrocele of the testicle",
+    "349": "hydrocephalus",
+    "350": "hydronephrosis",
+    "351": "hypercalcemia",
+    "352": "hypercholesterolemia",
+    "353": "hyperemesis gravidarum",
+    "354": "hyperhidrosis",
+    "355": "hyperkalemia",
+    "356": "hyperlipidemia",
+    "357": "hypernatremia",
+    "358": "hyperopia",
+    "359": "hyperosmotic hyperketotic state",
+    "360": "hypertension of pregnancy",
+    "361": "hypertensive heart disease",
+    "362": "hypertrophic obstructive cardiomyopathy (hocm)",
+    "363": "hypocalcemia",
+    "364": "hypokalemia",
+    "365": "hyponatremia",
+    "366": "hypothermia",
+    "367": "hypovolemia",
+    "368": "idiopathic absence of menstruation",
+    "369": "idiopathic excessive menstruation",
+    "370": "idiopathic infrequent menstruation",
+    "371": "idiopathic irregular menstrual cycle",
+    "372": "idiopathic nonmenstrual bleeding",
+    "373": "idiopathic painful menstruation",
+    "374": "ileus",
+    "375": "impulse control disorder",
+    "376": "indigestion",
+    "377": "induced abortion",
+    "378": "infection of open wound",
+    "379": "infectious gastroenteritis",
+    "380": "ingrown toe nail",
+    "381": "inguinal hernia",
+    "382": "injury of the ankle",
+    "383": "injury to internal organ",
+    "384": "injury to the abdomen",
+    "385": "injury to the arm",
+    "386": "injury to the face",
+    "387": "injury to the finger",
+    "388": "injury to the hand",
+    "389": "injury to the hip",
+    "390": "injury to the knee",
+    "391": "injury to the leg",
+    "392": "injury to the shoulder",
+    "393": "injury to the spinal cord",
+    "394": "injury to the trunk",
+    "395": "insect bite",
+    "396": "interstitial lung disease",
+    "397": "intertrigo (skin condition)",
+    "398": "intestinal cancer",
+    "399": "intestinal disease",
+    "400": "intestinal malabsorption",
+    "401": "intestinal obstruction",
+    "402": "intracerebral hemorrhage",
+    "403": "intracranial abscess",
+    "404": "intracranial hemorrhage",
+    "405": "intussusception",
+    "406": "iridocyclitis",
+    "407": "iron deficiency anemia",
+    "408": "irritable bowel syndrome",
+    "409": "ischemia of the bowel",
+    "410": "ischemic heart disease",
+    "411": "itching of unknown cause",
+    "412": "jaw disorder",
+    "413": "joint effusion",
+    "414": "juvenile rheumatoid arthritis",
+    "415": "kidney cancer",
+    "416": "kidney disease due to longstanding hypertension",
+    "417": "kidney failure",
+    "418": "kidney stone",
+    "419": "knee ligament or meniscus tear",
+    "420": "labyrinthitis",
+    "421": "lactose intolerance",
+    "422": "laryngitis",
+    "423": "lateral epicondylitis (tennis elbow)",
+    "424": "leukemia",
+    "425": "lewy body dementia",
+    "426": "lice",
+    "427": "lichen planus",
+    "428": "lichen simplex",
+    "429": "lipoma",
+    "430": "liver cancer",
+    "431": "liver disease",
+    "432": "lumbago",
+    "433": "lung cancer",
+    "434": "lung contusion",
+    "435": "lyme disease",
+    "436": "lymphadenitis",
+    "437": "lymphedema",
+    "438": "lymphogranuloma venereum",
+    "439": "lymphoma",
+    "440": "macular degeneration",
+    "441": "magnesium deficiency",
+    "442": "male genitalia infection",
+    "443": "malignant hypertension",
+    "444": "marijuana abuse",
+    "445": "mastectomy",
+    "446": "mastoiditis",
+    "447": "melanoma",
+    "448": "meniere disease",
+    "449": "meningioma",
+    "450": "meningitis",
+    "451": "menopause",
+    "452": "metabolic disorder",
+    "453": "metastatic cancer",
+    "454": "missed abortion",
+    "455": "mitral valve disease",
+    "456": "mittelschmerz",
+    "457": "molluscum contagiosum",
+    "458": "mononeuritis",
+    "459": "mononucleosis",
+    "460": "mucositis",
+    "461": "multiple myeloma",
+    "462": "multiple sclerosis",
+    "463": "mumps",
+    "464": "muscle spasm",
+    "465": "muscular dystrophy",
+    "466": "myasthenia gravis",
+    "467": "myelodysplastic syndrome",
+    "468": "myocarditis",
+    "469": "myoclonus",
+    "470": "myopia",
+    "471": "myositis",
+    "472": "narcolepsy",
+    "473": "nasal polyp",
+    "474": "necrotizing fasciitis",
+    "475": "neonatal jaundice",
+    "476": "nerve impingement near the shoulder",
+    "477": "neuralgia",
+    "478": "neurofibromatosis",
+    "479": "neuropathy due to drugs",
+    "480": "neurosis",
+    "481": "nonalcoholic liver disease (nash)",
+    "482": "noninfectious gastroenteritis",
+    "483": "normal pressure hydrocephalus",
+    "484": "nose disorder",
+    "485": "obesity",
+    "486": "obsessive compulsive disorder (ocd)",
+    "487": "obstructive sleep apnea (osa)",
+    "488": "omphalitis",
+    "489": "onychomycosis",
+    "490": "open wound from surgical incision",
+    "491": "open wound of the abdomen",
+    "492": "open wound of the arm",
+    "493": "open wound of the back",
+    "494": "open wound of the ear",
+    "495": "open wound of the eye",
+    "496": "open wound of the finger",
+    "497": "open wound of the foot",
+    "498": "open wound of the hand",
+    "499": "open wound of the lip",
+    "500": "open wound of the mouth",
+    "501": "open wound of the neck",
+    "502": "open wound of the nose",
+    "503": "open wound of the shoulder",
+    "504": "oppositional disorder",
+    "505": "optic neuritis",
+    "506": "oral leukoplakia",
+    "507": "oral mucosal lesion",
+    "508": "oral thrush (yeast infection)",
+    "509": "orbital cellulitis",
+    "510": "orthostatic hypotension",
+    "511": "osteoarthritis",
+    "512": "osteochondroma",
+    "513": "osteochondrosis",
+    "514": "osteomyelitis",
+    "515": "osteoporosis",
+    "516": "otitis externa (swimmer's ear)",
+    "517": "otitis media",
+    "518": "otosclerosis",
+    "519": "ovarian cancer",
+    "520": "ovarian cyst",
+    "521": "ovarian torsion",
+    "522": "overflow incontinence",
+    "523": "pain after an operation",
+    "524": "pain disorder affecting the neck",
+    "525": "pancreatic cancer",
+    "526": "panic attack",
+    "527": "panic disorder",
+    "528": "parasitic disease",
+    "529": "parathyroid adenoma",
+    "530": "parkinson disease",
+    "531": "paronychia",
+    "532": "paroxysmal supraventricular tachycardia",
+    "533": "paroxysmal ventricular tachycardia",
+    "534": "pelvic fistula",
+    "535": "pelvic inflammatory disease",
+    "536": "pelvic organ prolapse",
+    "537": "pemphigus",
+    "538": "pericarditis",
+    "539": "peripheral arterial disease",
+    "540": "peripheral arterial embolism",
+    "541": "peripheral nerve disorder",
+    "542": "perirectal infection",
+    "543": "peritonitis",
+    "544": "peritonsillar abscess",
+    "545": "persistent vomiting of unknown cause",
+    "546": "personality disorder",
+    "547": "peyronie disease",
+    "548": "pharyngitis",
+    "549": "phimosis",
+    "550": "pilonidal cyst",
+    "551": "pinguecula",
+    "552": "pinworm infection",
+    "553": "pituitary adenoma",
+    "554": "pituitary disorder",
+    "555": "pityriasis rosea",
+    "556": "placenta previa",
+    "557": "placental abruption",
+    "558": "plantar fasciitis",
+    "559": "pleural effusion",
+    "560": "pneumoconiosis",
+    "561": "pneumothorax",
+    "562": "poisoning due to analgesics",
+    "563": "poisoning due to anticonvulsants",
+    "564": "poisoning due to antidepressants",
+    "565": "poisoning due to antihypertensives",
+    "566": "poisoning due to antimicrobial drugs",
+    "567": "poisoning due to antipsychotics",
+    "568": "poisoning due to ethylene glycol",
+    "569": "poisoning due to gas",
+    "570": "poisoning due to opioids",
+    "571": "poisoning due to sedatives",
+    "572": "polycystic kidney disease",
+    "573": "polycystic ovarian syndrome (pcos)",
+    "574": "polycythemia vera",
+    "575": "polymyalgia rheumatica",
+    "576": "post-traumatic stress disorder (ptsd)",
+    "577": "postoperative infection",
+    "578": "postpartum depression",
+    "579": "preeclampsia",
+    "580": "pregnancy",
+    "581": "premature atrial contractions (pacs)",
+    "582": "premature ovarian failure",
+    "583": "premature rupture of amniotic membrane",
+    "584": "premature ventricular contractions (pvcs)",
+    "585": "premenstrual tension syndrome",
+    "586": "presbyacusis",
+    "587": "presbyopia",
+    "588": "priapism",
+    "589": "primary immunodeficiency",
+    "590": "primary insomnia",
+    "591": "primary kidney disease",
+    "592": "primary thrombocythemia",
+    "593": "problem during pregnancy",
+    "594": "prostate cancer",
+    "595": "prostatitis",
+    "596": "protein deficiency",
+    "597": "pseudohypoparathyroidism",
+    "598": "pseudotumor cerebri",
+    "599": "psychosexual disorder",
+    "600": "psychotic disorder",
+    "601": "pterygium",
+    "602": "pulmonary congestion",
+    "603": "pulmonary embolism",
+    "604": "pulmonary eosinophilia",
+    "605": "pulmonary fibrosis",
+    "606": "pulmonary hypertension",
+    "607": "pyelonephritis",
+    "608": "pyloric stenosis",
+    "609": "pyogenic skin infection",
+    "610": "raynaud disease",
+    "611": "rectal disorder",
+    "612": "restless leg syndrome",
+    "613": "retinal detachment",
+    "614": "retinopathy due to high blood pressure",
+    "615": "rhabdomyolysis",
+    "616": "rheumatoid arthritis",
+    "617": "rosacea",
+    "618": "rotator cuff injury",
+    "619": "salivary gland disorder",
+    "620": "sarcoidosis",
+    "621": "scabies",
+    "622": "scar",
+    "623": "scarlet fever",
+    "624": "schizophrenia",
+    "625": "sciatica",
+    "626": "scleritis",
+    "627": "scleroderma",
+    "628": "scoliosis",
+    "629": "seasonal allergies (hay fever)",
+    "630": "sebaceous cyst",
+    "631": "seborrheic dermatitis",
+    "632": "seborrheic keratosis",
+    "633": "sensorineural hearing loss",
+    "634": "sepsis",
+    "635": "septic arthritis",
+    "636": "shingles (herpes zoster)",
+    "637": "sialoadenitis",
+    "638": "sick sinus syndrome",
+    "639": "sickle cell anemia",
+    "640": "sickle cell crisis",
+    "641": "sinus bradycardia",
+    "642": "sjogren syndrome",
+    "643": "skin cancer",
+    "644": "skin disorder",
+    "645": "skin pigmentation disorder",
+    "646": "skin polyp",
+    "647": "smoking or tobacco addiction",
+    "648": "social phobia",
+    "649": "soft tissue sarcoma",
+    "650": "somatization disorder",
+    "651": "spermatocele",
+    "652": "spherocytosis",
+    "653": "spina bifida",
+    "654": "spinal stenosis",
+    "655": "spinocerebellar ataxia",
+    "656": "spondylitis",
+    "657": "spondylolisthesis",
+    "658": "spondylosis",
+    "659": "spontaneous abortion",
+    "660": "sprain or strain",
+    "661": "stenosis of the tear duct",
+    "662": "stomach cancer",
+    "663": "strep throat",
+    "664": "stress incontinence",
+    "665": "stricture of the esophagus",
+    "666": "stroke",
+    "667": "stye",
+    "668": "subacute thyroiditis",
+    "669": "subarachnoid hemorrhage",
+    "670": "subconjunctival hemorrhage",
+    "671": "subdural hemorrhage",
+    "672": "substance-related mental disorder",
+    "673": "syndrome of inappropriate secretion of adh (siadh)",
+    "674": "syphilis",
+    "675": "syringomyelia",
+    "676": "systemic lupus erythematosis (sle)",
+    "677": "teething syndrome",
+    "678": "temporary or benign blood in urine",
+    "679": "temporomandibular joint disorder",
+    "680": "tendinitis",
+    "681": "tension headache",
+    "682": "testicular cancer",
+    "683": "testicular disorder",
+    "684": "testicular torsion",
+    "685": "thoracic aortic aneurysm",
+    "686": "thoracic outlet syndrome",
+    "687": "threatened pregnancy",
+    "688": "thrombocytopenia",
+    "689": "thrombophlebitis",
+    "690": "thyroid cancer",
+    "691": "thyroid disease",
+    "692": "thyroid nodule",
+    "693": "tic (movement) disorder",
+    "694": "tietze syndrome",
+    "695": "tinnitus of unknown cause",
+    "696": "tonsillar hypertrophy",
+    "697": "tonsillitis",
+    "698": "tooth abscess",
+    "699": "tooth disorder",
+    "700": "torticollis",
+    "701": "tourette syndrome",
+    "702": "toxic multinodular goiter",
+    "703": "tracheitis",
+    "704": "transient ischemic attack",
+    "705": "trichiasis",
+    "706": "trichomonas infection",
+    "707": "trigeminal neuralgia",
+    "708": "trigger finger (finger disorder)",
+    "709": "tuberous sclerosis",
+    "710": "ulcerative colitis",
+    "711": "urethral disorder",
+    "712": "urethral stricture",
+    "713": "urethral valves",
+    "714": "urethritis",
+    "715": "urge incontinence",
+    "716": "urinary tract obstruction",
+    "717": "uterine atony",
+    "718": "uterine cancer",
+    "719": "uterine fibroids",
+    "720": "uveitis",
+    "721": "vaginal cyst",
+    "722": "vaginal yeast infection",
+    "723": "vaginismus",
+    "724": "vaginitis",
+    "725": "valley fever",
+    "726": "varicocele of the testicles",
+    "727": "vasculitis",
+    "728": "venous insufficiency",
+    "729": "vertebrobasilar insufficiency",
+    "730": "vesicoureteral reflux",
+    "731": "viral exanthem",
+    "732": "viral hepatitis",
+    "733": "viral warts",
+    "734": "vitamin b12 deficiency",
+    "735": "vitamin d deficiency",
+    "736": "vitreous degeneration",
+    "737": "vitreous hemorrhage",
+    "738": "vocal cord polyp",
+    "739": "volvulus",
+    "740": "von willebrand disease",
+    "741": "vulvar cancer",
+    "742": "vulvar disorder",
+    "743": "vulvodynia",
+    "744": "wernicke korsakoff syndrome",
+    "745": "white blood cell disease",
+    "746": "whooping cough",
+    "747": "wilson disease",
+    "748": "yeast infection",
+    "749": "zenker diverticulum"
+  },
+  "label2id": {
+    "(vertigo) Paroymsal  Positional Vertigo": 0,
+    "AIDS": 1,
+    "Acne": 2,
+    "Alcoholic hepatitis": 3,
+    "Allergy": 4,
+    "Arthritis": 5,
+    "Bronchial Asthma": 6,
+    "Cervical spondylosis": 7,
+    "Chicken pox": 8,
+    "Chronic cholestasis": 9,
+    "Common Cold": 10,
+    "Dengue": 11,
+    "Diabetes": 12,
+    "Dimorphic hemmorhoids(piles)": 13,
+    "Drug Reaction": 14,
+    "Fungal infection": 15,
+    "GERD": 16,
+    "Gastroenteritis": 17,
+    "Heart attack": 18,
+    "Hepatitis B": 19,
+    "Hepatitis C": 20,
+    "Hepatitis D": 21,
+    "Hepatitis E": 22,
+    "Hypertension": 23,
+    "Hyperthyroidism": 24,
+    "Hypoglycemia": 25,
+    "Hypothyroidism": 26,
+    "Impetigo": 27,
+    "Jaundice": 28,
+    "Malaria": 29,
+    "Migraine": 30,
+    "Osteoarthristis": 31,
+    "Paralysis (brain hemorrhage)": 32,
+    "Peptic ulcer diseae": 33,
+    "Pneumonia": 34,
+    "Psoriasis": 35,
+    "Tuberculosis": 36,
+    "Typhoid": 37,
+    "Urinary tract infection": 38,
+    "Varicose veins": 39,
+    "abdominal aortic aneurysm": 40,
+    "abdominal hernia": 41,
+    "abscess of nose": 42,
+    "abscess of the lung": 43,
+    "abscess of the pharynx": 44,
+    "acanthosis nigricans": 45,
+    "acariasis": 46,
+    "achalasia": 47,
+    "actinic keratosis": 48,
+    "acute bronchiolitis": 49,
+    "acute bronchitis": 50,
+    "acute bronchospasm": 51,
+    "acute glaucoma": 52,
+    "acute kidney injury": 53,
+    "acute otitis media": 54,
+    "acute pancreatitis": 55,
+    "acute respiratory distress syndrome (ards)": 56,
+    "acute sinusitis": 57,
+    "acute stress reaction": 58,
+    "adhesive capsulitis of the shoulder": 59,
+    "adjustment reaction": 60,
+    "adrenal adenoma": 61,
+    "alcohol abuse": 62,
+    "alcohol intoxication": 63,
+    "alcohol withdrawal": 64,
+    "alcoholic liver disease": 65,
+    "allergy to animals": 66,
+    "alopecia": 67,
+    "alzheimer disease": 68,
+    "amblyopia": 69,
+    "amyloidosis": 70,
+    "amyotrophic lateral sclerosis (als)": 71,
+    "anal fissure": 72,
+    "anal fistula": 73,
+    "anemia": 74,
+    "anemia due to chronic kidney disease": 75,
+    "anemia due to malignancy": 76,
+    "anemia of chronic disease": 77,
+    "angina": 78,
+    "ankylosing spondylitis": 79,
+    "anxiety": 80,
+    "aortic valve disease": 81,
+    "aphakia": 82,
+    "aphthous ulcer": 83,
+    "aplastic anemia": 84,
+    "appendicitis": 85,
+    "arrhythmia": 86,
+    "arthritis of the hip": 87,
+    "ascending cholangitis": 88,
+    "asperger syndrome": 89,
+    "asthma": 90,
+    "astigmatism": 91,
+    "atelectasis": 92,
+    "athlete's foot": 93,
+    "atonic bladder": 94,
+    "atrial fibrillation": 95,
+    "atrial flutter": 96,
+    "atrophic skin condition": 97,
+    "atrophic vaginitis": 98,
+    "atrophy of the corpus cavernosum": 99,
+    "attention deficit hyperactivity disorder (adhd)": 100,
+    "autism": 101,
+    "autonomic nervous system disorder": 102,
+    "avascular necrosis": 103,
+    "balanitis": 104,
+    "bell palsy": 105,
+    "benign kidney cyst": 106,
+    "benign paroxysmal positional vertical (bppv)": 107,
+    "benign prostatic hyperplasia (bph)": 108,
+    "benign vaginal discharge (leukorrhea)": 109,
+    "bipolar disorder": 110,
+    "birth trauma": 111,
+    "bladder cancer": 112,
+    "bladder disorder": 113,
+    "bladder obstruction": 114,
+    "blepharitis": 115,
+    "blepharospasm": 116,
+    "bone cancer": 117,
+    "bone disorder": 118,
+    "bone spur of the calcaneous": 119,
+    "brachial neuritis": 120,
+    "brain cancer": 121,
+    "breast cancer": 122,
+    "breast cyst": 123,
+    "breast infection (mastitis)": 124,
+    "broken tooth": 125,
+    "bunion": 126,
+    "burn": 127,
+    "bursitis": 128,
+    "callus": 129,
+    "carbon monoxide poisoning": 130,
+    "cardiac arrest": 131,
+    "cardiomyopathy": 132,
+    "carpal tunnel syndrome": 133,
+    "cat scratch disease": 134,
+    "cataract": 135,
+    "celiac disease": 136,
+    "cellulitis or abscess of mouth": 137,
+    "central atherosclerosis": 138,
+    "central retinal artery or vein occlusion": 139,
+    "cerebral edema": 140,
+    "cerebral palsy": 141,
+    "cervical cancer": 142,
+    "cervical disorder": 143,
+    "cervicitis": 144,
+    "chalazion": 145,
+    "chickenpox": 146,
+    "chlamydia": 147,
+    "cholecystitis": 148,
+    "choledocholithiasis": 149,
+    "cholesteatoma": 150,
+    "chondromalacia of the patella": 151,
+    "chorioretinitis": 152,
+    "chronic back pain": 153,
+    "chronic constipation": 154,
+    "chronic glaucoma": 155,
+    "chronic inflammatory demyelinating polyneuropathy (cidp)": 156,
+    "chronic kidney disease": 157,
+    "chronic knee pain": 158,
+    "chronic obstructive pulmonary disease (copd)": 159,
+    "chronic otitis media": 160,
+    "chronic pain disorder": 161,
+    "chronic pancreatitis": 162,
+    "chronic rheumatic fever": 163,
+    "chronic sinusitis": 164,
+    "cirrhosis": 165,
+    "coagulation (bleeding) disorder": 166,
+    "cold sore": 167,
+    "colonic polyp": 168,
+    "colorectal cancer": 169,
+    "complex regional pain syndrome": 170,
+    "concussion": 171,
+    "conduct disorder": 172,
+    "conductive hearing loss": 173,
+    "congenital heart defect": 174,
+    "congenital malformation syndrome": 175,
+    "conjunctivitis": 176,
+    "conjunctivitis due to allergy": 177,
+    "conjunctivitis due to bacteria": 178,
+    "conjunctivitis due to virus": 179,
+    "connective tissue disorder": 180,
+    "contact dermatitis": 181,
+    "conversion disorder": 182,
+    "cornea infection": 183,
+    "corneal abrasion": 184,
+    "corneal disorder": 185,
+    "coronary atherosclerosis": 186,
+    "cranial nerve palsy": 187,
+    "crohn disease": 188,
+    "croup": 189,
+    "crushing injury": 190,
+    "cryptorchidism": 191,
+    "cyst of the eyelid": 192,
+    "cystic fibrosis": 193,
+    "cysticercosis": 194,
+    "cystitis": 195,
+    "de quervain disease": 196,
+    "decubitus ulcer": 197,
+    "deep vein thrombosis (dvt)": 198,
+    "degenerative disc disease": 199,
+    "delirium": 200,
+    "dementia": 201,
+    "dental caries": 202,
+    "depression": 203,
+    "dermatitis due to sun exposure": 204,
+    "developmental disability": 205,
+    "deviated nasal septum": 206,
+    "diabetic ketoacidosis": 207,
+    "diabetic peripheral neuropathy": 208,
+    "diabetic retinopathy": 209,
+    "diaper rash": 210,
+    "dislocation of the ankle": 211,
+    "dislocation of the elbow": 212,
+    "dislocation of the finger": 213,
+    "dislocation of the foot": 214,
+    "dislocation of the hip": 215,
+    "dislocation of the knee": 216,
+    "dislocation of the patella": 217,
+    "dislocation of the shoulder": 218,
+    "dislocation of the vertebra": 219,
+    "dislocation of the wrist": 220,
+    "dissociative disorder": 221,
+    "diverticulitis": 222,
+    "diverticulosis": 223,
+    "down syndrome": 224,
+    "drug abuse": 225,
+    "drug abuse (barbiturates)": 226,
+    "drug abuse (cocaine)": 227,
+    "drug abuse (methamphetamine)": 228,
+    "drug abuse (opioids)": 229,
+    "drug poisoning due to medication": 230,
+    "drug withdrawal": 231,
+    "dry eye of unknown cause": 232,
+    "dumping syndrome": 233,
+    "dyshidrosis": 234,
+    "dysthymic disorder": 235,
+    "ear drum damage": 236,
+    "ear wax impaction": 237,
+    "eating disorder": 238,
+    "ectopic pregnancy": 239,
+    "ectropion": 240,
+    "eczema": 241,
+    "emphysema": 242,
+    "empyema": 243,
+    "encephalitis": 244,
+    "endocarditis": 245,
+    "endometrial cancer": 246,
+    "endometrial hyperplasia": 247,
+    "endometriosis": 248,
+    "endophthalmitis": 249,
+    "envenomation from spider or animal bite": 250,
+    "ependymoma": 251,
+    "epididymitis": 252,
+    "epidural hemorrhage": 253,
+    "epilepsy": 254,
+    "erectile dysfunction": 255,
+    "erythema multiforme": 256,
+    "esophageal cancer": 257,
+    "esophagitis": 258,
+    "essential tremor": 259,
+    "eustachian tube dysfunction (ear disorder)": 260,
+    "extrapyramidal effect of drugs": 261,
+    "eye alignment disorder": 262,
+    "factitious disorder": 263,
+    "female genitalia infection": 264,
+    "female infertility of unknown cause": 265,
+    "fetal alcohol syndrome": 266,
+    "fibroadenoma": 267,
+    "fibrocystic breast disease": 268,
+    "fibromyalgia": 269,
+    "flat feet": 270,
+    "floaters": 271,
+    "flu": 272,
+    "fluid overload": 273,
+    "folate deficiency": 274,
+    "food allergy": 275,
+    "foreign body in the ear": 276,
+    "foreign body in the eye": 277,
+    "foreign body in the gastrointestinal tract": 278,
+    "foreign body in the throat": 279,
+    "foreign body in the vagina": 280,
+    "fracture of the ankle": 281,
+    "fracture of the arm": 282,
+    "fracture of the facial bones": 283,
+    "fracture of the finger": 284,
+    "fracture of the foot": 285,
+    "fracture of the hand": 286,
+    "fracture of the jaw": 287,
+    "fracture of the leg": 288,
+    "fracture of the neck": 289,
+    "fracture of the patella": 290,
+    "fracture of the pelvis": 291,
+    "fracture of the rib": 292,
+    "fracture of the shoulder": 293,
+    "fracture of the skull": 294,
+    "fracture of the vertebra": 295,
+    "friedrich ataxia": 296,
+    "frostbite": 297,
+    "fungal infection of the hair": 298,
+    "fungal infection of the skin": 299,
+    "galactorrhea of unknown cause": 300,
+    "gallstone": 301,
+    "ganglion cyst": 302,
+    "gastritis": 303,
+    "gastroduodenal ulcer": 304,
+    "gastroesophageal reflux disease (gerd)": 305,
+    "gastrointestinal hemorrhage": 306,
+    "gastroparesis": 307,
+    "genital herpes": 308,
+    "gestational diabetes": 309,
+    "glaucoma": 310,
+    "glucocorticoid deficiency": 311,
+    "goiter": 312,
+    "gonorrhea": 313,
+    "gout": 314,
+    "graves disease": 315,
+    "guillain barre syndrome": 316,
+    "gum disease": 317,
+    "gynecomastia": 318,
+    "hammer toe": 319,
+    "head and neck cancer": 320,
+    "head injury": 321,
+    "headache after lumbar puncture": 322,
+    "heart block": 323,
+    "heart failure": 324,
+    "heat exhaustion": 325,
+    "hemangioma": 326,
+    "hemarthrosis": 327,
+    "hematoma": 328,
+    "hemiplegia": 329,
+    "hemochromatosis": 330,
+    "hemolytic anemia": 331,
+    "hemophilia": 332,
+    "hemorrhoids": 333,
+    "hepatic encephalopathy": 334,
+    "hepatitis A": 335,
+    "hepatitis due to a toxin": 336,
+    "herniated disk": 337,
+    "herpangina": 338,
+    "hiatal hernia": 339,
+    "hidradenitis suppurativa": 340,
+    "hirschsprung disease": 341,
+    "hirsutism": 342,
+    "histoplasmosis": 343,
+    "hormone disorder": 344,
+    "hpv": 345,
+    "human immunodeficiency virus infection (hiv)": 346,
+    "hydatidiform mole": 347,
+    "hydrocele of the testicle": 348,
+    "hydrocephalus": 349,
+    "hydronephrosis": 350,
+    "hypercalcemia": 351,
+    "hypercholesterolemia": 352,
+    "hyperemesis gravidarum": 353,
+    "hyperhidrosis": 354,
+    "hyperkalemia": 355,
+    "hyperlipidemia": 356,
+    "hypernatremia": 357,
+    "hyperopia": 358,
+    "hyperosmotic hyperketotic state": 359,
+    "hypertension of pregnancy": 360,
+    "hypertensive heart disease": 361,
+    "hypertrophic obstructive cardiomyopathy (hocm)": 362,
+    "hypocalcemia": 363,
+    "hypokalemia": 364,
+    "hyponatremia": 365,
+    "hypothermia": 366,
+    "hypovolemia": 367,
+    "idiopathic absence of menstruation": 368,
+    "idiopathic excessive menstruation": 369,
+    "idiopathic infrequent menstruation": 370,
+    "idiopathic irregular menstrual cycle": 371,
+    "idiopathic nonmenstrual bleeding": 372,
+    "idiopathic painful menstruation": 373,
+    "ileus": 374,
+    "impulse control disorder": 375,
+    "indigestion": 376,
+    "induced abortion": 377,
+    "infection of open wound": 378,
+    "infectious gastroenteritis": 379,
+    "ingrown toe nail": 380,
+    "inguinal hernia": 381,
+    "injury of the ankle": 382,
+    "injury to internal organ": 383,
+    "injury to the abdomen": 384,
+    "injury to the arm": 385,
+    "injury to the face": 386,
+    "injury to the finger": 387,
+    "injury to the hand": 388,
+    "injury to the hip": 389,
+    "injury to the knee": 390,
+    "injury to the leg": 391,
+    "injury to the shoulder": 392,
+    "injury to the spinal cord": 393,
+    "injury to the trunk": 394,
+    "insect bite": 395,
+    "interstitial lung disease": 396,
+    "intertrigo (skin condition)": 397,
+    "intestinal cancer": 398,
+    "intestinal disease": 399,
+    "intestinal malabsorption": 400,
+    "intestinal obstruction": 401,
+    "intracerebral hemorrhage": 402,
+    "intracranial abscess": 403,
+    "intracranial hemorrhage": 404,
+    "intussusception": 405,
+    "iridocyclitis": 406,
+    "iron deficiency anemia": 407,
+    "irritable bowel syndrome": 408,
+    "ischemia of the bowel": 409,
+    "ischemic heart disease": 410,
+    "itching of unknown cause": 411,
+    "jaw disorder": 412,
+    "joint effusion": 413,
+    "juvenile rheumatoid arthritis": 414,
+    "kidney cancer": 415,
+    "kidney disease due to longstanding hypertension": 416,
+    "kidney failure": 417,
+    "kidney stone": 418,
+    "knee ligament or meniscus tear": 419,
+    "labyrinthitis": 420,
+    "lactose intolerance": 421,
+    "laryngitis": 422,
+    "lateral epicondylitis (tennis elbow)": 423,
+    "leukemia": 424,
+    "lewy body dementia": 425,
+    "lice": 426,
+    "lichen planus": 427,
+    "lichen simplex": 428,
+    "lipoma": 429,
+    "liver cancer": 430,
+    "liver disease": 431,
+    "lumbago": 432,
+    "lung cancer": 433,
+    "lung contusion": 434,
+    "lyme disease": 435,
+    "lymphadenitis": 436,
+    "lymphedema": 437,
+    "lymphogranuloma venereum": 438,
+    "lymphoma": 439,
+    "macular degeneration": 440,
+    "magnesium deficiency": 441,
+    "male genitalia infection": 442,
+    "malignant hypertension": 443,
+    "marijuana abuse": 444,
+    "mastectomy": 445,
+    "mastoiditis": 446,
+    "melanoma": 447,
+    "meniere disease": 448,
+    "meningioma": 449,
+    "meningitis": 450,
+    "menopause": 451,
+    "metabolic disorder": 452,
+    "metastatic cancer": 453,
+    "missed abortion": 454,
+    "mitral valve disease": 455,
+    "mittelschmerz": 456,
+    "molluscum contagiosum": 457,
+    "mononeuritis": 458,
+    "mononucleosis": 459,
+    "mucositis": 460,
+    "multiple myeloma": 461,
+    "multiple sclerosis": 462,
+    "mumps": 463,
+    "muscle spasm": 464,
+    "muscular dystrophy": 465,
+    "myasthenia gravis": 466,
+    "myelodysplastic syndrome": 467,
+    "myocarditis": 468,
+    "myoclonus": 469,
+    "myopia": 470,
+    "myositis": 471,
+    "narcolepsy": 472,
+    "nasal polyp": 473,
+    "necrotizing fasciitis": 474,
+    "neonatal jaundice": 475,
+    "nerve impingement near the shoulder": 476,
+    "neuralgia": 477,
+    "neurofibromatosis": 478,
+    "neuropathy due to drugs": 479,
+    "neurosis": 480,
+    "nonalcoholic liver disease (nash)": 481,
+    "noninfectious gastroenteritis": 482,
+    "normal pressure hydrocephalus": 483,
+    "nose disorder": 484,
+    "obesity": 485,
+    "obsessive compulsive disorder (ocd)": 486,
+    "obstructive sleep apnea (osa)": 487,
+    "omphalitis": 488,
+    "onychomycosis": 489,
+    "open wound from surgical incision": 490,
+    "open wound of the abdomen": 491,
+    "open wound of the arm": 492,
+    "open wound of the back": 493,
+    "open wound of the ear": 494,
+    "open wound of the eye": 495,
+    "open wound of the finger": 496,
+    "open wound of the foot": 497,
+    "open wound of the hand": 498,
+    "open wound of the lip": 499,
+    "open wound of the mouth": 500,
+    "open wound of the neck": 501,
+    "open wound of the nose": 502,
+    "open wound of the shoulder": 503,
+    "oppositional disorder": 504,
+    "optic neuritis": 505,
+    "oral leukoplakia": 506,
+    "oral mucosal lesion": 507,
+    "oral thrush (yeast infection)": 508,
+    "orbital cellulitis": 509,
+    "orthostatic hypotension": 510,
+    "osteoarthritis": 511,
+    "osteochondroma": 512,
+    "osteochondrosis": 513,
+    "osteomyelitis": 514,
+    "osteoporosis": 515,
+    "otitis externa (swimmer's ear)": 516,
+    "otitis media": 517,
+    "otosclerosis": 518,
+    "ovarian cancer": 519,
+    "ovarian cyst": 520,
+    "ovarian torsion": 521,
+    "overflow incontinence": 522,
+    "pain after an operation": 523,
+    "pain disorder affecting the neck": 524,
+    "pancreatic cancer": 525,
+    "panic attack": 526,
+    "panic disorder": 527,
+    "parasitic disease": 528,
+    "parathyroid adenoma": 529,
+    "parkinson disease": 530,
+    "paronychia": 531,
+    "paroxysmal supraventricular tachycardia": 532,
+    "paroxysmal ventricular tachycardia": 533,
+    "pelvic fistula": 534,
+    "pelvic inflammatory disease": 535,
+    "pelvic organ prolapse": 536,
+    "pemphigus": 537,
+    "pericarditis": 538,
+    "peripheral arterial disease": 539,
+    "peripheral arterial embolism": 540,
+    "peripheral nerve disorder": 541,
+    "perirectal infection": 542,
+    "peritonitis": 543,
+    "peritonsillar abscess": 544,
+    "persistent vomiting of unknown cause": 545,
+    "personality disorder": 546,
+    "peyronie disease": 547,
+    "pharyngitis": 548,
+    "phimosis": 549,
+    "pilonidal cyst": 550,
+    "pinguecula": 551,
+    "pinworm infection": 552,
+    "pituitary adenoma": 553,
+    "pituitary disorder": 554,
+    "pityriasis rosea": 555,
+    "placenta previa": 556,
+    "placental abruption": 557,
+    "plantar fasciitis": 558,
+    "pleural effusion": 559,
+    "pneumoconiosis": 560,
+    "pneumothorax": 561,
+    "poisoning due to analgesics": 562,
+    "poisoning due to anticonvulsants": 563,
+    "poisoning due to antidepressants": 564,
+    "poisoning due to antihypertensives": 565,
+    "poisoning due to antimicrobial drugs": 566,
+    "poisoning due to antipsychotics": 567,
+    "poisoning due to ethylene glycol": 568,
+    "poisoning due to gas": 569,
+    "poisoning due to opioids": 570,
+    "poisoning due to sedatives": 571,
+    "polycystic kidney disease": 572,
+    "polycystic ovarian syndrome (pcos)": 573,
+    "polycythemia vera": 574,
+    "polymyalgia rheumatica": 575,
+    "post-traumatic stress disorder (ptsd)": 576,
+    "postoperative infection": 577,
+    "postpartum depression": 578,
+    "preeclampsia": 579,
+    "pregnancy": 580,
+    "premature atrial contractions (pacs)": 581,
+    "premature ovarian failure": 582,
+    "premature rupture of amniotic membrane": 583,
+    "premature ventricular contractions (pvcs)": 584,
+    "premenstrual tension syndrome": 585,
+    "presbyacusis": 586,
+    "presbyopia": 587,
+    "priapism": 588,
+    "primary immunodeficiency": 589,
+    "primary insomnia": 590,
+    "primary kidney disease": 591,
+    "primary thrombocythemia": 592,
+    "problem during pregnancy": 593,
+    "prostate cancer": 594,
+    "prostatitis": 595,
+    "protein deficiency": 596,
+    "pseudohypoparathyroidism": 597,
+    "pseudotumor cerebri": 598,
+    "psychosexual disorder": 599,
+    "psychotic disorder": 600,
+    "pterygium": 601,
+    "pulmonary congestion": 602,
+    "pulmonary embolism": 603,
+    "pulmonary eosinophilia": 604,
+    "pulmonary fibrosis": 605,
+    "pulmonary hypertension": 606,
+    "pyelonephritis": 607,
+    "pyloric stenosis": 608,
+    "pyogenic skin infection": 609,
+    "raynaud disease": 610,
+    "rectal disorder": 611,
+    "restless leg syndrome": 612,
+    "retinal detachment": 613,
+    "retinopathy due to high blood pressure": 614,
+    "rhabdomyolysis": 615,
+    "rheumatoid arthritis": 616,
+    "rosacea": 617,
+    "rotator cuff injury": 618,
+    "salivary gland disorder": 619,
+    "sarcoidosis": 620,
+    "scabies": 621,
+    "scar": 622,
+    "scarlet fever": 623,
+    "schizophrenia": 624,
+    "sciatica": 625,
+    "scleritis": 626,
+    "scleroderma": 627,
+    "scoliosis": 628,
+    "seasonal allergies (hay fever)": 629,
+    "sebaceous cyst": 630,
+    "seborrheic dermatitis": 631,
+    "seborrheic keratosis": 632,
+    "sensorineural hearing loss": 633,
+    "sepsis": 634,
+    "septic arthritis": 635,
+    "shingles (herpes zoster)": 636,
+    "sialoadenitis": 637,
+    "sick sinus syndrome": 638,
+    "sickle cell anemia": 639,
+    "sickle cell crisis": 640,
+    "sinus bradycardia": 641,
+    "sjogren syndrome": 642,
+    "skin cancer": 643,
+    "skin disorder": 644,
+    "skin pigmentation disorder": 645,
+    "skin polyp": 646,
+    "smoking or tobacco addiction": 647,
+    "social phobia": 648,
+    "soft tissue sarcoma": 649,
+    "somatization disorder": 650,
+    "spermatocele": 651,
+    "spherocytosis": 652,
+    "spina bifida": 653,
+    "spinal stenosis": 654,
+    "spinocerebellar ataxia": 655,
+    "spondylitis": 656,
+    "spondylolisthesis": 657,
+    "spondylosis": 658,
+    "spontaneous abortion": 659,
+    "sprain or strain": 660,
+    "stenosis of the tear duct": 661,
+    "stomach cancer": 662,
+    "strep throat": 663,
+    "stress incontinence": 664,
+    "stricture of the esophagus": 665,
+    "stroke": 666,
+    "stye": 667,
+    "subacute thyroiditis": 668,
+    "subarachnoid hemorrhage": 669,
+    "subconjunctival hemorrhage": 670,
+    "subdural hemorrhage": 671,
+    "substance-related mental disorder": 672,
+    "syndrome of inappropriate secretion of adh (siadh)": 673,
+    "syphilis": 674,
+    "syringomyelia": 675,
+    "systemic lupus erythematosis (sle)": 676,
+    "teething syndrome": 677,
+    "temporary or benign blood in urine": 678,
+    "temporomandibular joint disorder": 679,
+    "tendinitis": 680,
+    "tension headache": 681,
+    "testicular cancer": 682,
+    "testicular disorder": 683,
+    "testicular torsion": 684,
+    "thoracic aortic aneurysm": 685,
+    "thoracic outlet syndrome": 686,
+    "threatened pregnancy": 687,
+    "thrombocytopenia": 688,
+    "thrombophlebitis": 689,
+    "thyroid cancer": 690,
+    "thyroid disease": 691,
+    "thyroid nodule": 692,
+    "tic (movement) disorder": 693,
+    "tietze syndrome": 694,
+    "tinnitus of unknown cause": 695,
+    "tonsillar hypertrophy": 696,
+    "tonsillitis": 697,
+    "tooth abscess": 698,
+    "tooth disorder": 699,
+    "torticollis": 700,
+    "tourette syndrome": 701,
+    "toxic multinodular goiter": 702,
+    "tracheitis": 703,
+    "transient ischemic attack": 704,
+    "trichiasis": 705,
+    "trichomonas infection": 706,
+    "trigeminal neuralgia": 707,
+    "trigger finger (finger disorder)": 708,
+    "tuberous sclerosis": 709,
+    "ulcerative colitis": 710,
+    "urethral disorder": 711,
+    "urethral stricture": 712,
+    "urethral valves": 713,
+    "urethritis": 714,
+    "urge incontinence": 715,
+    "urinary tract obstruction": 716,
+    "uterine atony": 717,
+    "uterine cancer": 718,
+    "uterine fibroids": 719,
+    "uveitis": 720,
+    "vaginal cyst": 721,
+    "vaginal yeast infection": 722,
+    "vaginismus": 723,
+    "vaginitis": 724,
+    "valley fever": 725,
+    "varicocele of the testicles": 726,
+    "vasculitis": 727,
+    "venous insufficiency": 728,
+    "vertebrobasilar insufficiency": 729,
+    "vesicoureteral reflux": 730,
+    "viral exanthem": 731,
+    "viral hepatitis": 732,
+    "viral warts": 733,
+    "vitamin b12 deficiency": 734,
+    "vitamin d deficiency": 735,
+    "vitreous degeneration": 736,
+    "vitreous hemorrhage": 737,
+    "vocal cord polyp": 738,
+    "volvulus": 739,
+    "von willebrand disease": 740,
+    "vulvar cancer": 741,
+    "vulvar disorder": 742,
+    "vulvodynia": 743,
+    "wernicke korsakoff syndrome": 744,
+    "white blood cell disease": 745,
+    "whooping cough": 746,
+    "wilson disease": 747,
+    "yeast infection": 748,
+    "zenker diverticulum": 749
+  },
+  "num_labels": 750,
+  "max_length": 512,
+  "version": "3.0",
+  "improvements": [
+    "Increased diseases covered",
+    "Better generalization",
+    "Consistent 512 token length",
+    "Improved regularization",
+    "Better data augmentation"
+  ]
+}

agent/text_symptom_tool.py

@@ -0,0 +1,50 @@
+import json
+import requests
+from langchain_core.tools import tool
+
+@tool
+def analyze_symptom_text(symptoms: str) -> str:
+    """
+    Analyzes text-based symptoms using classification model via api backend.
+    Input: Patient's symptom description text.
+    Output: Predicted conditions with confidence scores.
+    """
+    try:
+        print(f"🩺 Analyzing symptoms: {symptoms[:50]}...")
+        
+        # Call your HF ClinicalBERT FastAPI Space
+        api_url = "https://datdevsteve-nivra-text-diagnosis.hf.space/run/predict"
+        payload = {
+            "data": [symptoms],
+            "fn_index": 0  # Default prediction function
+        }
+        
+        print("🔬 Calling ClinicalBERT FastAPI backend...")
+        response = requests.post(api_url, json=payload, timeout=30)
+        response.raise_for_status()
+        
+        result = response.json()
+        
+        # Extract diagnosis from HF Space response format
+        if "data" in result and len(result["data"]) > 0:
+            diagnosis = result["data"][0]
+            
+            # Parse confidence if available, else default format
+            if isinstance(diagnosis, list) and len(diagnosis) > 0:
+                diagnosis = diagnosis[0]
+                
+            return f"""
+[TEXT SYMPTOM ANALYSIS - SUCCESS]:
+✅ FastAPI Backend Response: {diagnosis}
+
+📡 **Backend**: nivra-text-diagnosis HF Space"""
+        else:
+            # Fallback with generic advice
+            return "[TEXT SYMPTOM ANALYSIS - WARNING]: No diagnosis returned from backend. Please consult a doctor."
+            
+    except requests.exceptions.Timeout:
+        return "[TEXT SYMPTOM ANALYSIS - ERROR]: Analysis timeout. Please try again or consult a doctor."
+    except requests.exceptions.RequestException as e:
+        return f"[TEXT SYMPTOM ANALYSIS - ERROR]: Network error: {str(e)}. Please consult a doctor."
+    except Exception as e:
+        return f"[TEXT SYMPTOM ANALYSIS - ERROR]: Unexpected error: {str(e)}. Please consult a doctor."

app.py

@@ -0,0 +1,31 @@
+import gradio as gr
+from nivra_agent import nivra_chat
+import os
+
+# Gradio interface for HF Spaces
+def chat_interface(message, history, image=None, audio=None):
+    # Build multimodal input
+    input_text = message
+    if image:
+        input_text += f"\n[IMAGE: {image}]"
+    if audio:
+        input_text += f"\n[AUDIO: {audio}]"
+    
+    response = nivra_chat(input_text)
+    history.append((message, response))
+    return history, ""
+
+demo = gr.ChatInterface(
+    chat_interface,
+    title="🩺 Nivra AI Healthcare Assistant",
+    description="India-first symptom diagnosis: Text + Image + Voice",
+    examples=[
+        ["I have fever and chills"],
+        ["Skin rash", gr.Image(type="filepath")],
+        ["Stomach pain, vomiting"]
+    ],
+    multimodal=True
+)
+
+if __name__ == "__main__":
+    demo.launch()

nivra_agent.py

@@ -0,0 +1,120 @@
+#=========================================
+#|| NIVRA AI HEALTHCARE ASSISTANT AGENT ||
+#=========================================
+
+from langchain_groq import ChatGroq
+from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
+from langchain_core.messages import HumanMessage, AIMessage
+from langchain.agents import create_tool_calling_agent, AgentExecutor
+from langchain_core.tools import tool
+from agent.rag_retriever import NivraRAGRetriever
+from agent.text_symptom_tool import analyze_symptom_text
+from agent.image_symptom_tool import analyze_symptom_image
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+rag_tool = NivraRAGRetriever.getRelevantDocs
+#instantiate
+llm = ChatGroq(
+    temperature=0.1,
+    model_name="llama-3.1-70b-versatile",
+    api_key= os.getenv("GROQ_API_KEY")
+)
+tools = [analyze_symptom_image, analyze_symptom_text, rag_tool]
+system_prompt = """You are Nivra, a smart and helpful AI Healthcare Assistant with multimodal capabilities. Your role is to only provide medical attention to the user in a structured format of text or voice.
+
+You are equipped with these tools:
+- analyze_symptom_image: Performs symptom analysis on provided image input to diagnose skin conditions
+- analyze_symptom_text: Performs symptom analysis on provided text input to diagnose diseases  
+- rag_tool: Refers to medical knowledgebase for additional context
+
+Always remember these rules:
+- Output text should be structured for easy TTS narration when [VOICE MODE ONLY] is detected
+- You ARE NOT A DOCTOR - provide disease information ONLY, NO treatment advice  
+- Always provide doctor referral for Cancer, Dengue, Malaria, or high-risk diagnoses
+- ALWAYS correlate with rag_tool for medical knowledge
+- Keep descriptions short/crisp to limit token usage
+
+Basic Output Structure:
+[TOOLS USED]
+[SYMPTOMS] 
+[PRIMARY DIAGNOSIS]
+[DIAGNOSIS DESCRIPTION with RAG Knowledgebase]
+[FIRST AID] 
+[EMERGENCY CONSULTATION REQUIRED]
+
+FEW-SHOT EXAMPLES (FOLLOW EXACT FORMAT):
+
+(EXAMPLE 1- Voice Input)
+Input: [VOICE MODE] "I have fever, chills and severe headache."
+---
+[TOOLS USED] analyze_symptom_text, rag_tool [/TOOLS USED]
+[SYMPTOMS] Fever, Chills, Headache [/SYMPTOMS]
+[PRIMARY DIAGNOSIS] Malaria (78% confidence) [/PRIMARY DIAGNOSIS] 
+[DIAGNOSIS DESCRIPTION]
+Malaria is caused by Plasmodium parasite spread by Anopheles mosquitoes. 
+It multiplies in red blood cells causing fever, chills, headache cycles.
+Common in India during monsoon season.
+[/DIAGNOSIS DESCRIPTION]
+[FIRST AID]
+Rest completely and drink plenty of fluids. Seek immediate medical attention for malaria test and treatment.
+[/FIRST AID]
+[EMERGENCY] Yes [/EMERGENCY]
+
+(EXAMPLE 2- Image Input)
+Input: "I have skin rash" + rash.jpg
+---
+[TOOLS USED] analyze_symptom_image, rag_tool [/TOOLS USED]
+[SYMPTOMS] Red scaly patches, itching [/SYMPTOMS]
+[PRIMARY DIAGNOSIS] Psoriasis (82% confidence) [/PRIMARY DIAGNOSIS]
+[DIAGNOSIS DESCRIPTION]
+Psoriasis is a chronic autoimmune skin disorder causing rapid skin cell growth. 
+Results in thick, dry, scaly patches. Non-contagious, affects 2-3% population.
+[/DIAGNOSIS DESCRIPTION]
+[FIRST AID]
+Keep skin clean and moisturized. Avoid scratching. Consult dermatologist for management.
+[/FIRST AID]
+[EMERGENCY] No [/EMERGENCY]
+
+(EXAMPLE 3- Low Confidence)
+Input: "Stomach pain, vomiting frequently"
+---
+[TOOLS USED] analyze_symptom_text, rag_tool [/TOOLS USED]
+[SYMPTOMS] Stomach pain, vomiting [/SYMPTOMS]
+[PRIMARY DIAGNOSIS] Gastritis or Gastroenteritis [/PRIMARY DIAGNOSIS]
+[DIAGNOSIS DESCRIPTION]
+Multiple causes possible: acidity, infection, food poisoning. Clinical evaluation required.
+[/DIAGNOSIS DESCRIPTION]
+[FIRST AID]
+Seek medical consultation immediately. Ultrasound may be needed.
+[/FIRST AID]
+[EMERGENCY] Yes [/EMERGENCY]
+
+**CRITICAL**: High-risk triggers (EMERGENCY=Yes): melanoma, basal_cell_carcinoma, dengue, malaria, typhoid, cancer
+"""
+prompt = ChatPromptTemplate.from_messages([
+    ("system", system_prompt),
+    MessagesPlaceholder(variable_name="chat_history", optional=True),
+    ("user", "{input}"),
+    MessagesPlaceholder(variable_name="agent_scratchpad")
+])
+
+#create the agent
+agent = create_tool_calling_agent(llm, tools, prompt)
+agent_executor = AgentExecutor(
+    agent = agent,
+    tools = tools,
+    verbose = True,
+    handle_parsing_errors = True,
+    max_iterations = 5
+)
+
+def nivra_chat(user_input: str, chat_history: list = []):
+    """Main chat function to be invoked via mobile app"""
+    response = agent_executor.invoke({
+        "input": user_input,
+        "chat_history": chat_history
+    })
+    return response["output"]
+

requirements.txt

@@ -1,2 +1,29 @@
-gradio
-llama-cpp-python
+# LangChain Core (Agent + Tools)
+langchain>=0.3.1
+langchain-groq>=0.1.0
+langchain-community>=0.0.20
+langchain-core>=0.1.23
+langchain-huggingface>=0.1.0
+
+# RAG (Your chroma_db)
+chromadb>=0.5.11
+sentence-transformers>=2.2.2
+faiss-cpu>=1.7.4
+
+# Groq LLM
+groq>=0.4.0
+
+# HF Spaces UI
+gradio>=5.4.0
+
+# API Clients (Text + Vision Spaces)
+requests>=2.32.0
+
+# Image processing (for API calls)
+pillow>=12.1.0
+
+# Utilities
+python-dotenv>=1.0.0
+numpy>=1.24.0
+pandas>=2.0.0
+pydantic>=2.5.0