import joblib
import pandas as pd
import streamlit as st


EDU_DICT = {'Iris-setosa': 1,
            'Iris-versicolor': 2,
            'Iris-virginica': 3,
            }

model = joblib.load('model.joblib')
unique_values = joblib.load('unique_values.joblib')
    
def main():
    st.title("Iris's Class")

    with st.form("questionaire"):
        sepal length = st.slider("Sepal_length", min_value=0, max_value=6)
        sepal width = st.slider("Sepal_width", min_value=0, max_value=6)
        petal length = st.slider("Petal_length", min_value=0, max_value=6)
        petal width = st.slider("Petal_width", min_value=0, max_value=6)

        # clicked==True only when the button is clicked
        clicked = st.form_submit_button("Predict class")
        if clicked:
            result=model.predict(pd.DataFrame({"sepal length": [sepal length],
                                               "sepal width": [sepal width],
                                               "petal length": [petal length],
                                               "petal width": [petal width]}))
            # Show prediction
            result = "class" : [iris_DICT[class]]
            st.success("Your predicted class is"+result)
if __name__ == "__main__":
    main()