import joblib
import pandas as pd
import streamlit as st

model = joblib.load('model.joblib')
unique_values = joblib.load('unique_values.joblib')
    
unique_sex =  unique_values["sex"]
unique_country =  unique_values["country"]
unique_returning =  unique_values["returning"]
unique_GImg1 =  unique_values["GImg1"]
unique_GImg2 =  unique_values["GImg2"]
unique_GImg3 =  unique_values["GImg3"]
unique_PImg1 =  unique_values["PImg1"]
unique_PImg2 =  unique_values["PImg2"]
unique_PImg3 =  unique_values["PImg3"]
unique_PImg4=  unique_values["PImg4"]
unique_PImg5 =  unique_values["PImg5"]
unique_TAudio1  =  unique_values["TAudio1"]
unique_TAudio2  =  unique_values["TAudio2"]
unique_TAudio3  =  unique_values["TAudio3"]
unique_QAudio1  =  unique_values["QAudio1"]
unique_QAudio2  =  unique_values["QAudio2"]
unique_QAudio3  =  unique_values["QAudio3"]
unique_Proxemics  =  unique_values["Proxemics"]

def main():
    st.title("Non verbal tourists data")
    with st.form("questionaire"):
        sex = st.selectbox("Sex", options = unique_sex)
        age = st.slider("Age", min_value=20, max_values=90)
        country = st.selectbox("Country of the client United Nations admitted countries", options = unique_country)
        GImg1 = st.selectbox("Handshake Indifferent", options = unique_GImg1)
        GImg2 = st.selectbox("Hug Indifferent", options = unique_GImg2)
        GImg3 = st.selectbox("Kiss Indifferent", options = unique_GImg3)
        PImg1 = st.selectbox("Consent posture Indifferent", options = unique_PImg1)
        PImg2 = st.selectbox("Interest posture Indifferent", options = unique_PImg2)
        PImg3 = st.selectbox("Neutral posture Indifferent", options = unique_PImg3)
        PImg4 = st.selectbox("Reflexive posture Indifferent", options = unique_PImg4)
        PImg5 = st.selectbox("Negative posture Indifferent", options = unique_PImg5)
        Tense= st.slider("Observed emotional clime", min_value=1, max_values=10)
        Hostile = st.slider("friendly Observed emotional clime", min_value=1, max_values=10)
        TAudio1 = st.selectbox("Authoritative Indifferent", options = unique_TAudio1)
        TAudio2 = st.selectbox("Sarcastic Indifferent", options = unique_TAudio2)   
        TAudio3 = st.selectbox("Friendly Indifferent", options = unique_TAudio3)   
        QAudio1 = st.selectbox("Spitting Indifferent", options = unique_QAudio1)   
        QAudio2 = st.selectbox("Hum Indifferent", options = unique_QAudio1)   
        QAudio3 = st.selectbox("Sigh Indifferent", options = unique_QAudio1)   
        Proxemics = st.selectbox("Physical distance preferred for the client", options = unique_Proxemics)
        # clicked==True only when the button is clicked
        clicked = st.form_submit_button("Predict Type of Client")
        if clicked:
            result=model.predict(pd.DataFrame({"sex":[sex],
                                               "age":[age],
                                               "country":[country],
                                               "GImg1":[GImg1],
                                               "GImg2":[GImg2],
                                               "GImg3":[GImg3],
                                               "PImg1":[PImg1],
                                               "PImg2":[PImg2],
                                               "PImg3":[PImg3],
                                               "PImg4":[PImg4],
                                               "PImg5":[PImg5],
                                               "Tense -relaxed":[Tense],
                                               "Hostile -friendly":[Hostile],
                                               "TAudio1":[TAudio1],
                                               "TAudio2":[TAudio2],
                                               "TAudio3":[TAudio3],
                                               "QAudio1":[QAudio1],  
                                               "QAudio2":[QAudio1],  
                                               "QAudio3":[QAudio1],
                                               "Proxemics":[Proxemics]}))
            # Show prediction
            result = 'low' if result[0] == 1 else 'high'
            st.success("Predict Type of Client is "+result) #แสดงผล
            
# Run main()
if __name__ == "__main__":
    main()