import joblib
import pandas as pd
import streamlit as st

model = joblib.load('model (3).joblib')
unique_values = joblib.load('unique_values (3).joblib')

unique_CL =  unique_values["Color"]
unique_SC =  unique_values["Spectral_Class"]

def main():
    st.title("My star type")
    st.image("https://img.freepik.com/free-vector/different-types-stars-dark-space_1308-37762.jpg")
    st.text("example -> \nK = 27739, L = 849420 , R = 1252, A_M = -7.59 , Color = Blue ,Spectral_Class = B\n -> Type = Hyper Giants")
    with st.form("questionaire"):
         K = st.number_input('Temperature')
         L = st.number_input('Relative Luminosity')
         R = st.number_input('Relative Radius')
         AM = st.number_input('Absolute Magnitude')
         Color = st.selectbox("General Color of Spectrum",options=unique_CL)
         Spectral_Class = st.selectbox("Spectral_Class",options=unique_SC)
         clicked = st.form_submit_button("Predict type")
         if clicked:
            result = model.predict(pd.DataFrame({"Temperature(K)": [K],
                                                 "Relative Luminosity(Watts)": [L],
                                                 "Relative Radius(m)": [R],
                                                 "Absolute Magnitude": [AM],
                                                 "Color": [Color],
                                                 "Spectral_Class": [Spectral_Class]
                                                 })
            )
            if result[0] == 0:
                result = "Red Dwarf"
            elif result[0] == 1:
                result = "Brown Dwarf"
            elif result[0] == 2:
                result = "White Dwarf"
            elif result[0] == 3:
                result = "Main Sequence"
            elif result[0] == 4:
                result = "Super Giants"
            else:
                result = "Hyper Giants"
                
            st.write(f"star type... {result}")
           
if __name__=="__main__":
    main()