app.py

import os
import json
import numpy as np
import streamlit as st
import onnxruntime as rt
from sklearn.preprocessing import StandardScaler
from sklearn.ensemble import RandomForestClassifier

scalerSession = rt.InferenceSession("standard_scaler.onnx")
modelSession = rt.InferenceSession("random_forest_model.onnx")

with open("random_forest_model.json", "r") as f:
    params = json.load(f)
    model = RandomForestClassifier(**params)
    del(params)

def softmax(x):
    """Compute softmax values for each sets of scores in x."""
    e_x = np.exp(x - np.max(x))
    return e_x / e_x.sum()

with open("scaler.json", "r") as f:
    params = json.load(f)
    scaler = StandardScaler(**params)
    del(params)

def calculate_bmi(weight, height):
    # Calculate BMI
    bmi = weight / (height ** 2)
    
    return bmi

def is_number(text):
    try:
        # Try to convert the text to a float
        float(text)
        return True
    except ValueError:
        # If conversion fails, it's not a number
        return False
    
def diabetic_pedigree_function(mother, father, siblings):
    """
    Calculate a scaled Diabetic Pedigree Function (DPF) for an individual,
    aiming for an output range of approximately (0.078, 2.42).

    Parameters:
    mother (int): 1 if the mother has diabetes, 0 otherwise.
    father (int): 1 if the father has diabetes, 0 otherwise.
    siblings (list): A list of 0s and 1s representing siblings' diabetes status.

    Returns:
    float: The scaled diabetic pedigree function score.
    """
    # Assign weights to each family member
    mother_weight = 0.5
    father_weight = 0.5
    sibling_weight = 0.25
    
    # Calculate the weighted contributions
    family_history = (mother * mother_weight) + (father * father_weight) + (siblings) * sibling_weight
    
    # Add a scaling factor to shift the range
    scaling_factor = 1.2
    bias = 0.078  # Minimum value in the desired range
    
    # Final scaled DPF score
    dpf_score = family_history * scaling_factor + bias
    
    return round(dpf_score, 3)  # Rounded for clarity



if "question_no" not in st.session_state:
    st.session_state.question_no = 1

st.title(f"Know your diabetic status")
if st.session_state.question_no == 1:

    progress = st.progress((st.session_state.question_no) / 10)
    name = st.text_input(f"What is your name?", key=st.session_state.question_no,placeholder="Type your answer here")

    if st.button("Next->"):
        if name:
            st.session_state.name = name
        else:
            st.session_state.name = "Anonymous user"
        st.session_state.question_no += 1

elif st.session_state.question_no == 2:
    progress = st.progress((st.session_state.question_no) / 10)
    age = st.text_input(f"How old are you?", key=st.session_state.question_no, placeholder="Type your answer")
    if st.button("Next->"):
        if not is_number(age):
            st.warning("Please enter a valid age")
        else:
            st.session_state.age = int(age)
            st.session_state.question_no += 1

elif st.session_state.question_no == 3:
    progress = st.progress((st.session_state.question_no) / 10)
    gender = st.selectbox(f"What is your gender?", ["Male", "Female", "Other"], key=st.session_state.question_no)
    if st.button("Next->"):
        st.session_state.gender = gender
        if gender == "Male":
            st.session_state.pregnancies = 0
            st.session_state.question_no += 2
        else:
            st.session_state.question_no += 1

elif st.session_state.question_no == 4:
    progress = st.progress((st.session_state.question_no) / 10)
    pregnancies = st.text_input(f"How many times have you been pregnant?", key=st.session_state.question_no)
    if st.button("Next->"):
        if not is_number(pregnancies):
            st.warning("Please enter a vaild Input!!!!")
        else:
            st.session_state.pregnancies = int(pregnancies)
            st.session_state.question_no += 1

elif st.session_state.question_no == 5:
    progress = st.progress((st.session_state.question_no) / 10)
    glucose = st.text_input(f"Enter your glucose level", key=st.session_state.question_no)
    if st.button("Next->"):
        if not is_number(glucose):
            st.warning("Please enter a valid input!!!!")
        else:
            st.session_state.glucose = int(glucose)
            st.session_state.question_no += 1

elif st.session_state.question_no == 6:
    progress = st.progress((st.session_state.question_no) / 10)
    bp = st.text_input(f"Enter your blood pressure", key=st.session_state.question_no)
    if st.button("Next->"):
        if not is_number(bp):
            st.warning("Please enter valid input!!!")
        else:
            st.session_state.bp = int(bp)
            st.session_state.question_no += 1

elif st.session_state.question_no == 7:
    progress = st.progress((st.session_state.question_no) / 10)
    height = st.text_input(f"Enter your height in cm:")
    if st.button("Next->"):
        if not is_number(height):
            st.warning("Please enter valid input!!!!")
        else:
            st.session_state.height = float(height)/100
            st.session_state.question_no += 1

elif st.session_state.question_no == 8:
    progress = st.progress((st.session_state.question_no) / 10)
    weight = st.text_input(f"Enter your weight in KG")
    if st.button("Next->"):
        if not is_number(weight):
            st.warning("Please enter valid input!!!")
        else:
            st.session_state.weight = float(weight)
            st.session_state.bmi = calculate_bmi(st.session_state.weight, st.session_state.height)
            st.session_state.question_no += 1

elif st.session_state.question_no == 9:
    progress = st.progress((st.session_state.question_no) / 10)
    st.write("Select the members with diabetes in your family")
    diabeticMother = st.checkbox("Mother")
    diabeticFather = st.checkbox("Father")
    diabeticSibling = st.text_input("Enter the number of diabetic siblings in your family", key=st.session_state.question_no)
    if st.button("Next->"):
        if not is_number(diabeticSibling):
            st.warning("Please enter valid number for diabetic siblings!!!!")
        else:
            st.session_state.diabeticMother = 1 if diabeticMother else 0
            st.session_state.diabeticFather = 1 if diabeticFather else 0
            st.session_state.diabeticSibling = int(diabeticSibling)
            st.session_state.dpf = diabetic_pedigree_function(st.session_state.diabeticMother, st.session_state.diabeticFather, st.session_state.diabeticSibling)
            st.session_state.question_no += 1

elif st.session_state.question_no == 10:

    progress = st.progress((st.session_state.question_no) / 10)
    input_name = scalerSession.get_inputs()[0].name

    # Transform the new data using the ONNX scaler
    transformed_data = scalerSession.run(None, {input_name: [[st.session_state.pregnancies, st.session_state.glucose, st.session_state.bp, st.session_state.bmi, st.session_state.dpf, st.session_state.age]]})[0]
    # Prepare input data (convert to numpy array)
    input_name = modelSession.get_inputs()[0].name
    test_data = np.array(transformed_data, dtype=np.float32)
    
    # Run the model
    out = modelSession.run(None, {input_name: test_data})[1]
    # out = model.predict([st.session_state.pregnancies, st.session_state.glucose, st.session_state.bp, st.session_state.bmi, st.session_state.dpf])
    out = softmax(list(out[0].values()))[1]
    risk = "high" if out <= 0.3 else "medium" if out <= 0.6 else "high" if out <= 0.9 else "very high"

    st.write(f"Risk of type 2 diabetes is {risk}")

    st.session_state.risk = risk

    if st.button("Reset"):
        for key in st.session_state.keys():
            del st.session_state[key]
        st.rerun()  # Reloads the app to reflect the changes

    # if st.button("AI Consult"):