import joblib
import pandas as pd
import streamlit as st 

model = joblib.load('model.joblib')
unique_values = joblib.load('unique_values.joblib')
    
unique_color =  unique_values["color"]

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

    with st.form("questionaire"):
        color = st.selectbox("Color", unique_color)
        
        clicked = st.form_submit_button("Predict Color")
        if clicked:
            result=model.predict(pd.DataFrame({"room": [room],
                                               "partitions": [partition],
                                               "Rows": [Row],
                                               "Columns": [Columns],
                                               "Color": [Color]}))
            if 'Color' = 'Red' : 
                print('Red')
            elif 'Color' = 'Blue':
                print('BLue')
            elif 'Color' = 'Yellow':
                print('Yellow')
            else 'Color' = 'Green':
                print('Green')
            #result = 'Red' = Red if result[0] == 1 else 'Blue'
            st.success('The predicted color is {}'.format(result))

if __name__=='__main__':
    main()