import numpy as np
import streamlit as st
import tensorflow as tf
from PIL import Image
import io
from model_methods import predict, orig_img, plot_gradient_maps, gradCAM

# configuration of the page
st.set_page_config(
    layout='wide',
    page_icon='👁️',
    page_title='Eye Disease Classifier',
    initial_sidebar_state='auto'
)

st.title('👁️ Eye Disease classifier')
st.info('Only classifies **Cataract**, **Diabetic retinopathy**, **Glaucoma** or **Normal**. \n\n Model is restricted to giving **1** class at a time')

new_img = st.file_uploader('Please upload your retinal image in .png or .jpeg/.jpg format')

def predict_upload():
  result = predict(new_img) # result is a probabilities array 
  classes = ['cataract', 'diabetic retinopathy', 'glaucoma', 'normal']
  max_result = (np.max(result, axis=-1)) * 100 # max probability
  pred_prob = np.format_float_positional(max_result, precision=2) # format probability
  pred_class = classes[(np.argmax(result, axis=-1)[0])] # string
  st.write(f'### There is a')
  st.success(f'# {pred_prob}% probability')
  st.write(f'### that this retinal image shows')
  st.success(f'# {pred_class}')
  
# instantiate submit button
if st.button('Classify'):
   if new_img is not None:
        with st.sidebar: 
            predict_upload()   

   col1, col2, col3 = st.columns(3)
   with col1:
        st.image(new_img)
        st.caption('Original')
        
   with col2:
        input_im = orig_img(new_img) # output tensor
        plot_gradient_maps(input_im)
        st.caption('Saliency map')
        
   with col3:
        gradCAM(new_img, intensity=0.5, res=250)
        st.caption('Activation heatmap')
        
   if new_img is None:
        with st.sidebar: 
             st.warning('''
             Unable to detect image. 
             Please upload retinal image for classification. 
             \n\n Thank you 🙏
             ''')

st.write('##')   # create space         
st.write('DISCLAIMER: THIS WEBSITE DOES NOT REPLACE MEDICAL ADVICE \n\n The information, including but not limited to, text, graphics, images and other material contained on this website are for informational purposes only. No material on this site is intended to be a substitute for professional medical advice, diagnosis or treatment. Always seek the advice of a physician or other qualified health care provider with any questions you may have regarding a medical condition or treatment and before undertaking a new health care regimen, and never disregard professional medical advice or delay in seeking it because of something you have read on this website.')