import numpy as np
import streamlit as st
import pickle
import time
import base64

bg = "static/bg2.jpg"
bg_ext = "jpg"

st.markdown(
    f"""
    <style>
    .reportview-container {{
        background: url(data:image/{bg_ext};base64,{base64.b64encode(open(bg, "rb").read()).decode()})
    }}
    </style>
    """,
    unsafe_allow_html=True
)

st.title("Lung Cancer Prediction")
st.subheader("Enter Details")

arr = []

col1, col2 = st.columns(2)
with col1:
    x = st.selectbox('Gender',['Male','Female'])
    if x=='Male':
        arr.append(1)
    else:
        arr.append(0)

with col2:
    x = st.number_input('Age',0,100,1)
    arr.append(x)

col1, col2, col3 = st.columns(3)
with col1:
    x = st.selectbox('Smoking',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)
with col2:
    x = st.selectbox('Yellow Fingers',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)
with col3:
    x = st.selectbox('Anxiety',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)


col1, col2, col3 = st.columns(3)
with col1:
    x = st.selectbox('Peer Pressure',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)
with col2:
    x = st.selectbox('Chronic Disease',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)
with col3:
    x = st.selectbox('Fatigue',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)


col1, col2, col3 = st.columns(3)
with col1:
    x = st.selectbox('Allergy',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)
with col2:
    x = st.selectbox('Weezing',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)
with col3:
    x = st.selectbox('Alcohol Consumption',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)


col1, col2, col3, col4 = st.columns(4)
with col1:
    x = st.selectbox('Coughing',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)
with col2:
    x = st.selectbox('Shortness Of Breath',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)
with col3:
    x = st.selectbox('Swallowing Difficulty',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)
with col4:
    x = st.selectbox('Chest Pain',['Yes','No'])
    if x=='Yes':
        arr.append(1)
    else:
        arr.append(0)



if st.button('Predict'):
    lst = np.array(arr).reshape(1, 15)
    loaded_model = pickle.load(open("model/lung_cancer.pkl", "rb"))
    result = loaded_model.predict(lst)

    with st.spinner("Predicting.....Have Patience"):
        time.sleep(2)
    if result==1:
        st.error("Chance Of Disease")
    if result==0:
        st.success("No Chance Of Disease")