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_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("Hypertension", options=unique_hypertension)
        Heart_Disease = st.selectbox("Heart_Disease", options=unique_heart_disease)
        Work_type = st.selectbox("Work_type", options=unique_work_type)
        Residence = st.selectbox("Residence", options=unique_residence_type)
        Glucose_Level = st.number_input("Glucose_Level")
        BMI = st.number_input("BMI")
        Smoking = st.selectbox("Smoking", options=unique_smoking_status)

        # clicked==True only when the button is clicked
        clicked = st.form_submit_button("Predict Stroke")
        if clicked:
            result=model.predict(pd.DataFrame({"Gender": [Gender],
                                               "Age": [Age],
                                               "Hypertension": [Hypertension],
                                               "Heart_Disease": [Heart_Disease],
                                               "Work_type": [Work_type],
                                               "Residence": [Residence],
                                               "Avg_Glucose_Level": [Glucose_Level],
                                               "BMI": [BMI],
                                               "Smoking": [Smoking]}))
            # Show prediction
            result = 'Yes' if result[0] == 1 else 'No'
            st.success("Your predicted Stroke is ",result)
# Run main()
if __name__ == "__main__":
    main()