import streamlit as st
import pandas as pd
from joblib import load

# Load the trained model and scaler
model = load('loandefaulter.joblib')
scaler = load('scaler.joblib')

# Define numerical features for scaling (only those that were used during training)
num_features = [
    'loan_amnt', 'int_rate', 'installment', 'annual_inc', 'dti',
    'revol_bal', 'revol_util', 'total_acc', 'mort_acc'
]

# Create the Streamlit app
st.set_page_config(page_title='Loan Default Prediction', layout='wide')

# App title and description
st.markdown("""
    <style>
    .title { font-size: 36px; font-weight: bold; color: #2E86C1; }
    .description { font-size: 20px; color: #34495E; }
    .input-container { margin-top: 20px; }
    .slider-container { margin: 10px 0; }
    </style>
    <div class="title">Loan Default Prediction</div>
    <div class="description">Enter the loan details below to get a prediction on whether the loan will be defaulted.</div>
""", unsafe_allow_html=True)

# Input fields with sliders
loan_amnt = st.slider('Loan Amount', min_value=0.0, max_value=1000000.0, step=1000.0, value=10000.0)
int_rate = st.slider('Interest Rate (%)', min_value=0.0, max_value=30.0, step=0.1, value=5.0)
installment = st.slider('EMI Amount', min_value=0.0, max_value=10000.0, step=10.0, value=200.0)
annual_inc = st.slider('Annual Income', min_value=0.0, max_value=1000000.0, step=1000.0, value=50000.0)

# CIBIL score input as a text field
cibil_score = st.text_input('CIBIL Score (Enter a number between 300 and 900)', value='700')

# Convert the CIBIL score input to a numeric value
try:
    cibil_score = int(cibil_score)
except ValueError:
    st.error("Please enter a valid number for CIBIL Score.")

# Set default values for the missing features
dti = 0.0  # Example default value for DTI
revol_bal = 0.0  # Example default value for Revolving Balance
revol_util = 0.0  # Example default value for Revolving Utilization
total_acc = 0  # Example default value for Total Accounts
mort_acc = 0  # Example default value for Mortgage Accounts

loan_amnt_by_income = loan_amnt / (annual_inc + 1)

# Create a DataFrame for the input
input_data = pd.DataFrame({
    'loan_amnt': [loan_amnt],
    'int_rate': [int_rate],
    'installment': [installment],
    'annual_inc': [annual_inc],
    'dti': [dti],
    'revol_bal': [revol_bal],
    'revol_util': [revol_util],
    'total_acc': [total_acc],
    'mort_acc': [mort_acc]
})

# Scale the numerical features that were used to fit the scaler
input_data[num_features] = scaler.transform(input_data[num_features])

# Add the additional feature (not part of scaling)
input_data['loan_amnt_by_income'] = [loan_amnt_by_income]


input_data['cibil_score'] = cibil_score  
input_data = input_data[num_features + ['loan_amnt_by_income']]  

# Predict using the model
if st.button('Predict'):
    if 300 <= cibil_score <= 900:  # Ensure CIBIL score is within the valid range
        prediction = model.predict(input_data)
        result = "Defaulted" if prediction[0] == 1 else "Not Defaulted"
        color = "red" if prediction[0] == 1 else "green"
        st.markdown(f"""
            <div style="font-size: 24px; color: {color}; font-weight: bold;">Prediction: {result}</div>
        """, unsafe_allow_html=True)
    else:
        st.error("Please enter a CIBIL Score between 300 and 900.")