import streamlit as st 
import joblib
import pandas as pd

# st.markdown(
#     """
#     <style>
#     body {
#         background-color: #C9E4DE; 
#     }
#     </style>
#     """,
#     unsafe_allow_html=True
# )
# try:
#     model = joblib.load('model_campus')
#     st.success("Model loaded successfully!")
# except Exception as e:
#     st.error(f"Error loading model: {e}")
#     st.stop()
# model = joblib.load(open('/Users/nishthapandey/Desktop/PlacementPrediction/model_campus_placement_rf.joblib','rb'))

def predict_placement(data):
    # Preprocess the data
    new_data = pd.DataFrame(data, index=[0])

    # Make prediction
    prediction = model_campus.predict(new_data)[0]
    prob = model_campus.predict_proba(new_data)[0][1]

    return prediction, prob

def main():
    st.header('Placement Prediciton App')
    st.markdown('This app uses historical data to predict whether a student will be placed in a company based on their profile.')

    gender = st.radio('Gender', ['Male', 'Female'])
    ssc_p = st.number_input('Secondary School Percentage', min_value=0.0, max_value=100.0, value=50.0, step=0.1)
    ssc_b = st.radio('Board of Education (SSC)', ['Central', 'Others'])
    hsc_p = st.number_input('Higher Secondary Percentage', min_value=0.0, max_value=100.0, value=50.0, step=0.1)
    hsc_b = st.radio('Board of Education (HSC)', ['Central', 'Others'])
    degree_p = st.number_input('UG Percentage', min_value=0.0, max_value=100.0, value=50.0, step=0.1)
    branch = st.selectbox('Branch of Study', ['CSE', 'ECE/EN', 'Others'])
    workex = st.radio('Work Experience', ['Yes', 'No'])
    certifications = st.number_input('Number of Certifications', min_value=0, max_value=10, value=0)
    etest_p = st.number_input('Employability Test Score', min_value=0.0, max_value=100.0, value=50.0, step=0.1)
    backlogs = st.number_input('Number of Backlogs', min_value=0, max_value=10, value=0)
    
    if st.button('predict'):
        new_data = {
            'gender': 0 if gender == "Male" else 1,
            'ssc_p': ssc_p,
            'ssc_b': 1 if ssc_b == "Central" else 0,
            'hsc_p': hsc_p,
            'hsc_b': 1 if hsc_b == "Central" else 0,
            'degree_p': degree_p,
            'Branch': 2 if branch == "ECE/EN" else 1 if branch == "CSE" else 0,
            'Workex': 1 if workex == "Yes" else 0,
            'Certifications': certifications,
            'etest_p': etest_p,
            'Backlogs': backlogs,
    }

        prediction, probability = predict_placement(new_data)
        st.write(f'Percentage of getting placed: {probability*100:.2f}%')


if __name__=='__main__':
    main()