Update api/disease.py

api/disease.py

@@ -1,280 +1,288 @@
-# app/api/disease.py
-from fastapi import APIRouter, HTTPException
-from pydantic import BaseModel
-import numpy as np
-import pandas as pd
-import pickle
-import ast
-import os
-
-router = APIRouter()
-
-BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-model_path = os.path.join(BASE_DIR, "models", "svc2.pkl")
-
-# Load models and data
-with open(model_path, "rb") as f:
-    svc = pickle.load(f)
-
-# Load data files
-DATA_DIR = os.path.join(BASE_DIR, "data")
-
-description = pd.read_csv(os.path.join(DATA_DIR, "description.csv"))
-precautions = pd.read_csv(os.path.join(DATA_DIR, "precautions_df.csv"))
-medications = pd.read_csv(os.path.join(DATA_DIR, "medications.csv"))
-diets = pd.read_csv(os.path.join(DATA_DIR, "diets.csv"))
-workout = pd.read_csv(os.path.join(DATA_DIR, "workout_df.csv"))
-
-# Load symptoms dict and disease list
-symptoms_dict = {
-    "itching": 0,
-    "skin_rash": 1,
-    "nodal_skin_eruptions": 2,
-    "continuous_sneezing": 3,
-    "shivering": 4,
-    "chills": 5,
-    "joint_pain": 6,
-    "stomach_pain": 7,
-    "acidity": 8,
-    "ulcers_on_tongue": 9,
-    "muscle_wasting": 10,
-    "vomiting": 11,
-    "burning_micturition": 12,
-    "spotting_ urination": 13,
-    "fatigue": 14,
-    "weight_gain": 15,
-    "anxiety": 16,
-    "cold_hands_and_feets": 17,
-    "mood_swings": 18,
-    "weight_loss": 19,
-    "restlessness": 20,
-    "lethargy": 21,
-    "patches_in_throat": 22,
-    "irregular_sugar_level": 23,
-    "cough": 24,
-    "high_fever": 25,
-    "sunken_eyes": 26,
-    "breathlessness": 27,
-    "sweating": 28,
-    "dehydration": 29,
-    "indigestion": 30,
-    "headache": 31,
-    "yellowish_skin": 32,
-    "dark_urine": 33,
-    "nausea": 34,
-    "loss_of_appetite": 35,
-    "pain_behind_the_eyes": 36,
-    "back_pain": 37,
-    "constipation": 38,
-    "abdominal_pain": 39,
-    "diarrhoea": 40,
-    "mild_fever": 41,
-    "yellow_urine": 42,
-    "yellowing_of_eyes": 43,
-    "acute_liver_failure": 44,
-    "fluid_overload": 45,
-    "swelling_of_stomach": 46,
-    "swelled_lymph_nodes": 47,
-    "malaise": 48,
-    "blurred_and_distorted_vision": 49,
-    "phlegm": 50,
-    "throat_irritation": 51,
-    "redness_of_eyes": 52,
-    "sinus_pressure": 53,
-    "runny_nose": 54,
-    "congestion": 55,
-    "chest_pain": 56,
-    "weakness_in_limbs": 57,
-    "fast_heart_rate": 58,
-    "pain_during_bowel_movements": 59,
-    "pain_in_anal_region": 60,
-    "bloody_stool": 61,
-    "irritation_in_anus": 62,
-    "neck_pain": 63,
-    "dizziness": 64,
-    "cramps": 65,
-    "bruising": 66,
-    "obesity": 67,
-    "swollen_legs": 68,
-    "swollen_blood_vessels": 69,
-    "puffy_face_and_eyes": 70,
-    "enlarged_thyroid": 71,
-    "brittle_nails": 72,
-    "swollen_extremeties": 73,
-    "excessive_hunger": 74,
-    "extra_marital_contacts": 75,
-    "drying_and_tingling_lips": 76,
-    "slurred_speech": 77,
-    "knee_pain": 78,
-    "hip_joint_pain": 79,
-    "muscle_weakness": 80,
-    "stiff_neck": 81,
-    "swelling_joints": 82,
-    "movement_stiffness": 83,
-    "spinning_movements": 84,
-    "loss_of_balance": 85,
-    "unsteadiness": 86,
-    "weakness_of_one_body_side": 87,
-    "loss_of_smell": 88,
-    "bladder_discomfort": 89,
-    "foul_smell_of urine": 90,
-    "continuous_feel_of_urine": 91,
-    "passage_of_gases": 92,
-    "internal_itching": 93,
-    "toxic_look_(typhos)": 94,
-    "depression": 95,
-    "irritability": 96,
-    "muscle_pain": 97,
-    "altered_sensorium": 98,
-    "red_spots_over_body": 99,
-    "belly_pain": 100,
-    "abnormal_menstruation": 101,
-    "dischromic _patches": 102,
-    "watering_from_eyes": 103,
-    "increased_appetite": 104,
-    "polyuria": 105,
-    "family_history": 106,
-    "mucoid_sputum": 107,
-    "rusty_sputum": 108,
-    "lack_of_concentration": 109,
-    "visual_disturbances": 110,
-    "receiving_blood_transfusion": 111,
-    "receiving_unsterile_injections": 112,
-    "coma": 113,
-    "stomach_bleeding": 114,
-    "distention_of_abdomen": 115,
-    "history_of_alcohol_consumption": 116,
-    "fluid_overload.1": 117,
-    "blood_in_sputum": 118,
-    "prominent_veins_on_calf": 119,
-    "palpitations": 120,
-    "painful_walking": 121,
-    "pus_filled_pimples": 122,
-    "blackheads": 123,
-    "scurring": 124,
-    "skin_peeling": 125,
-    "silver_like_dusting": 126,
-    "small_dents_in_nails": 127,
-    "inflammatory_nails": 128,
-    "blister": 129,
-    "red_sore_around_nose": 130,
-    "yellow_crust_ooze": 131,
-}  # Keep your full symptoms_dict here
-
-diseases_list = {
-    15: "Fungal infection",
-    4: "Allergy",
-    16: "GERD",
-    9: "Chronic cholestasis",
-    14: "Drug Reaction",
-    33: "Peptic ulcer diseae",
-    1: "AIDS",
-    12: "Diabetes ",
-    17: "Gastroenteritis",
-    6: "Bronchial Asthma",
-    23: "Hypertension ",
-    30: "Migraine",
-    7: "Cervical spondylosis",
-    32: "Paralysis (brain hemorrhage)",
-    28: "Jaundice",
-    29: "Malaria",
-    8: "Chicken pox",
-    11: "Dengue",
-    37: "Typhoid",
-    40: "hepatitis A",
-    19: "Hepatitis B",
-    20: "Hepatitis C",
-    21: "Hepatitis D",
-    22: "Hepatitis E",
-    3: "Alcoholic hepatitis",
-    36: "Tuberculosis",
-    10: "Common Cold",
-    34: "Pneumonia",
-    13: "Dimorphic hemmorhoids(piles)",
-    18: "Heart attack",
-    39: "Varicose veins",
-    26: "Hypothyroidism",
-    24: "Hyperthyroidism",
-    25: "Hypoglycemia",
-    31: "Osteoarthristis",
-    5: "Arthritis",
-    0: "(vertigo) Paroymsal  Positional Vertigo",
-    2: "Acne",
-    38: "Urinary tract infection",
-    35: "Psoriasis",
-    27: "Impetigo",
-}  # Keep your full diseases_list here
-
-
-display_to_actual = {sym.replace("_", " "): sym for sym in symptoms_dict.keys()}
-
-
-class SymptomsInput(BaseModel):
-    symptoms: list[str]
-
-
-def get_predicted_value(symptom_list):
-    input_vector = np.zeros(len(symptoms_dict))
-    for sym in symptom_list:
-        sym_clean = display_to_actual.get(sym)
-        if sym_clean:
-            input_vector[symptoms_dict[sym_clean]] = 1
-    return diseases_list[svc.predict([input_vector])[0]]
-
-
-def helper(disease):
-    desc = description[description["Disease"] == disease]["Description"]
-    desc = desc.values[0] if not desc.empty else "Description not available."
-
-    pre = precautions[precautions["Disease"] == disease].iloc[:, 1:].dropna(axis=1)
-    pre = (
-        pre.values.flatten().tolist()
-        if not pre.empty
-        else ["Precautions not available."]
-    )
-
-    med = medications[medications["Disease"] == disease]["Medication"]
-    med = (
-        ast.literal_eval(med.values[0])
-        if not med.empty
-        else ["Medications not available."]
-    )
-
-    die = diets[diets["Disease"] == disease]["Diet"]
-    die = (
-        ast.literal_eval(die.values[0])
-        if not die.empty
-        else ["Diet recommendations not available."]
-    )
-
-    wrkout = workout[workout["disease"] == disease]["workout"]
-    wrkout = (
-        wrkout.values.flatten().tolist()
-        if not wrkout.empty
-        else ["Workouts not available."]
-    )
-
-    return desc, pre, med, die, wrkout
-
-
-@router.post(
-    "/predict", name="rog_dristi", description="Predict disease based on symptoms"
-)
-def predict_disease(data: SymptomsInput):
-    if not data.symptoms:
-        raise HTTPException(status_code=400, detail="No symptoms provided.")
-
-    try:
-        predicted_disease = get_predicted_value(data.symptoms)
-        desc, pre, med, die, wrkout = helper(predicted_disease)
-
-        return {
-            "predicted_disease": predicted_disease,
-            "description": desc,
-            "precautions": pre,
-            "medications": med,
-            "diet": die,
-            "workouts": wrkout,
-        }
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+# app/api/disease.py
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
+import numpy as np
+import pandas as pd
+import pickle
+import ast
+import os
+from huggingface_hub import hf_hub_download
+
+router = APIRouter()
+
+# HF Repo
+REPO_ID = "uiuxarghya/test-store"
+REPO_TYPE = "dataset"
+
+# Download files from Hugging Face
+svc_path = hf_hub_download(repo_id=REPO_ID, filename="models/svc2.pkl", repo_type=REPO_TYPE)
+description_path = hf_hub_download(repo_id=REPO_ID, filename="data/description.csv", repo_type=REPO_TYPE)
+precautions_path = hf_hub_download(repo_id=REPO_ID, filename="data/precautions_df.csv", repo_type=REPO_TYPE)
+medications_path = hf_hub_download(repo_id=REPO_ID, filename="data/medications.csv", repo_type=REPO_TYPE)
+diets_path = hf_hub_download(repo_id=REPO_ID, filename="data/diets.csv", repo_type=REPO_TYPE)
+workout_path = hf_hub_download(repo_id=REPO_ID, filename="data/workout_df.csv", repo_type=REPO_TYPE)
+
+# Load model
+with open(svc_path, "rb") as f:
+    svc = pickle.load(f)
+
+# Load dataframes
+description = pd.read_csv(description_path)
+precautions = pd.read_csv(precautions_path)
+medications = pd.read_csv(medications_path)
+diets = pd.read_csv(diets_path)
+workout = pd.read_csv(workout_path)
+
+# Load symptoms dict and disease list
+symptoms_dict = {
+    "itching": 0,
+    "skin_rash": 1,
+    "nodal_skin_eruptions": 2,
+    "continuous_sneezing": 3,
+    "shivering": 4,
+    "chills": 5,
+    "joint_pain": 6,
+    "stomach_pain": 7,
+    "acidity": 8,
+    "ulcers_on_tongue": 9,
+    "muscle_wasting": 10,
+    "vomiting": 11,
+    "burning_micturition": 12,
+    "spotting_ urination": 13,
+    "fatigue": 14,
+    "weight_gain": 15,
+    "anxiety": 16,
+    "cold_hands_and_feets": 17,
+    "mood_swings": 18,
+    "weight_loss": 19,
+    "restlessness": 20,
+    "lethargy": 21,
+    "patches_in_throat": 22,
+    "irregular_sugar_level": 23,
+    "cough": 24,
+    "high_fever": 25,
+    "sunken_eyes": 26,
+    "breathlessness": 27,
+    "sweating": 28,
+    "dehydration": 29,
+    "indigestion": 30,
+    "headache": 31,
+    "yellowish_skin": 32,
+    "dark_urine": 33,
+    "nausea": 34,
+    "loss_of_appetite": 35,
+    "pain_behind_the_eyes": 36,
+    "back_pain": 37,
+    "constipation": 38,
+    "abdominal_pain": 39,
+    "diarrhoea": 40,
+    "mild_fever": 41,
+    "yellow_urine": 42,
+    "yellowing_of_eyes": 43,
+    "acute_liver_failure": 44,
+    "fluid_overload": 45,
+    "swelling_of_stomach": 46,
+    "swelled_lymph_nodes": 47,
+    "malaise": 48,
+    "blurred_and_distorted_vision": 49,
+    "phlegm": 50,
+    "throat_irritation": 51,
+    "redness_of_eyes": 52,
+    "sinus_pressure": 53,
+    "runny_nose": 54,
+    "congestion": 55,
+    "chest_pain": 56,
+    "weakness_in_limbs": 57,
+    "fast_heart_rate": 58,
+    "pain_during_bowel_movements": 59,
+    "pain_in_anal_region": 60,
+    "bloody_stool": 61,
+    "irritation_in_anus": 62,
+    "neck_pain": 63,
+    "dizziness": 64,
+    "cramps": 65,
+    "bruising": 66,
+    "obesity": 67,
+    "swollen_legs": 68,
+    "swollen_blood_vessels": 69,
+    "puffy_face_and_eyes": 70,
+    "enlarged_thyroid": 71,
+    "brittle_nails": 72,
+    "swollen_extremeties": 73,
+    "excessive_hunger": 74,
+    "extra_marital_contacts": 75,
+    "drying_and_tingling_lips": 76,
+    "slurred_speech": 77,
+    "knee_pain": 78,
+    "hip_joint_pain": 79,
+    "muscle_weakness": 80,
+    "stiff_neck": 81,
+    "swelling_joints": 82,
+    "movement_stiffness": 83,
+    "spinning_movements": 84,
+    "loss_of_balance": 85,
+    "unsteadiness": 86,
+    "weakness_of_one_body_side": 87,
+    "loss_of_smell": 88,
+    "bladder_discomfort": 89,
+    "foul_smell_of urine": 90,
+    "continuous_feel_of_urine": 91,
+    "passage_of_gases": 92,
+    "internal_itching": 93,
+    "toxic_look_(typhos)": 94,
+    "depression": 95,
+    "irritability": 96,
+    "muscle_pain": 97,
+    "altered_sensorium": 98,
+    "red_spots_over_body": 99,
+    "belly_pain": 100,
+    "abnormal_menstruation": 101,
+    "dischromic _patches": 102,
+    "watering_from_eyes": 103,
+    "increased_appetite": 104,
+    "polyuria": 105,
+    "family_history": 106,
+    "mucoid_sputum": 107,
+    "rusty_sputum": 108,
+    "lack_of_concentration": 109,
+    "visual_disturbances": 110,
+    "receiving_blood_transfusion": 111,
+    "receiving_unsterile_injections": 112,
+    "coma": 113,
+    "stomach_bleeding": 114,
+    "distention_of_abdomen": 115,
+    "history_of_alcohol_consumption": 116,
+    "fluid_overload.1": 117,
+    "blood_in_sputum": 118,
+    "prominent_veins_on_calf": 119,
+    "palpitations": 120,
+    "painful_walking": 121,
+    "pus_filled_pimples": 122,
+    "blackheads": 123,
+    "scurring": 124,
+    "skin_peeling": 125,
+    "silver_like_dusting": 126,
+    "small_dents_in_nails": 127,
+    "inflammatory_nails": 128,
+    "blister": 129,
+    "red_sore_around_nose": 130,
+    "yellow_crust_ooze": 131,
+}  # Keep your full symptoms_dict here
+
+diseases_list = {
+    15: "Fungal infection",
+    4: "Allergy",
+    16: "GERD",
+    9: "Chronic cholestasis",
+    14: "Drug Reaction",
+    33: "Peptic ulcer diseae",
+    1: "AIDS",
+    12: "Diabetes ",
+    17: "Gastroenteritis",
+    6: "Bronchial Asthma",
+    23: "Hypertension ",
+    30: "Migraine",
+    7: "Cervical spondylosis",
+    32: "Paralysis (brain hemorrhage)",
+    28: "Jaundice",
+    29: "Malaria",
+    8: "Chicken pox",
+    11: "Dengue",
+    37: "Typhoid",
+    40: "hepatitis A",
+    19: "Hepatitis B",
+    20: "Hepatitis C",
+    21: "Hepatitis D",
+    22: "Hepatitis E",
+    3: "Alcoholic hepatitis",
+    36: "Tuberculosis",
+    10: "Common Cold",
+    34: "Pneumonia",
+    13: "Dimorphic hemmorhoids(piles)",
+    18: "Heart attack",
+    39: "Varicose veins",
+    26: "Hypothyroidism",
+    24: "Hyperthyroidism",
+    25: "Hypoglycemia",
+    31: "Osteoarthristis",
+    5: "Arthritis",
+    0: "(vertigo) Paroymsal  Positional Vertigo",
+    2: "Acne",
+    38: "Urinary tract infection",
+    35: "Psoriasis",
+    27: "Impetigo",
+}  # Keep your full diseases_list here
+
+
+display_to_actual = {sym.replace("_", " "): sym for sym in symptoms_dict.keys()}
+
+
+class SymptomsInput(BaseModel):
+    symptoms: list[str]
+
+
+def get_predicted_value(symptom_list):
+    input_vector = np.zeros(len(symptoms_dict))
+    for sym in symptom_list:
+        sym_clean = display_to_actual.get(sym)
+        if sym_clean:
+            input_vector[symptoms_dict[sym_clean]] = 1
+    return diseases_list[svc.predict([input_vector])[0]]
+
+
+def helper(disease):
+    desc = description[description["Disease"] == disease]["Description"]
+    desc = desc.values[0] if not desc.empty else "Description not available."
+
+    pre = precautions[precautions["Disease"] == disease].iloc[:, 1:].dropna(axis=1)
+    pre = (
+        pre.values.flatten().tolist()
+        if not pre.empty
+        else ["Precautions not available."]
+    )
+
+    med = medications[medications["Disease"] == disease]["Medication"]
+    med = (
+        ast.literal_eval(med.values[0])
+        if not med.empty
+        else ["Medications not available."]
+    )
+
+    die = diets[diets["Disease"] == disease]["Diet"]
+    die = (
+        ast.literal_eval(die.values[0])
+        if not die.empty
+        else ["Diet recommendations not available."]
+    )
+
+    wrkout = workout[workout["disease"] == disease]["workout"]
+    wrkout = (
+        wrkout.values.flatten().tolist()
+        if not wrkout.empty
+        else ["Workouts not available."]
+    )
+
+    return desc, pre, med, die, wrkout
+
+
+@router.post(
+    "/predict", name="rog_dristi", description="Predict disease based on symptoms"
+)
+def predict_disease(data: SymptomsInput):
+    if not data.symptoms:
+        raise HTTPException(status_code=400, detail="No symptoms provided.")
+
+    try:
+        predicted_disease = get_predicted_value(data.symptoms)
+        desc, pre, med, die, wrkout = helper(predicted_disease)
+
+        return {
+            "predicted_disease": predicted_disease,
+            "description": desc,
+            "precautions": pre,
+            "medications": med,
+            "diet": die,
+            "workouts": wrkout,
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))