import joblib
import pandas as pd
import streamlit as st

model = joblib.load('model.joblib')
unique_values = joblib.load('unique_values.joblib')    
unique_gender =  unique_values["gender"]
unique_hypertension =  unique_values["hypertension"]
unique_heart_disease =  unique_values["heart_disease"]
unique_ever_married =  unique_values["ever_married"]
unique_work_type =  unique_values["work_type"]
unique_Residence_type = unique_values["Residence_type"]
unique_smoking_status = unique_values["smoking_status"]

def main():
    st.title("Stroke Prediction")

    with st.form("questionaire"):
        gender = st.selectbox("Gender" , options=unique_gender)
        age = st.slider("Age" , min_value=10 , max_value=100)
        hypertension= st.selectbox("Do you have Hypertension ? " , options=unique_hypertension)
        heart_disease= st.selectbox("Do you have any Heart Desease ? " , options=unique_heart_disease)
        ever_married = st.selectbox("Have you ever married ? " , options=unique_ever_married)
        work_type= st.selectbox("What kind of job do you do ? " , options=unique_work_type)
        Residence_type= st.selectbox("Which kind of place do you live in ? " , options=unique_Residence_type)
        smoking_status= st.selectbox("Do you smoke ? " , options=unique_smoking_status )
        bmi  = st.number_input(label="Input your BMI",step=1,format="%.2f")
        avg_glucose_level= st.number_input(label="Input your Average glucose level",step=1,format="%.2f")

        # clicked==True only when the button is clicked
        clicked = st.form_submit_button("Predict income")
        if clicked:
            result=model.predict(pd.DataFrame({"gender": [gender],
                                               "age": [age],
                                               "hypertension": [hypertension],
                                               "heart_disease": [heart_disease],
                                               "ever_married": [ever_married],
                                               "work_type": [work_type],
                                               "Residence_type": [Residence_type],
                                               "avg_glucose_level":[avg_glucose_level],
                                               "bmi":[bmi],
                                               "smoking_status":[smoking_status]}))
            # Show prediction
            result = 'You may not have a stroke' if result[0] == 1 else 'You may have a stroke'
            st.success("Your prediction :"+result)
            
# Run main()
if __name__ == "__main__" :
    main()