import streamlit as st
from tensorflow import keras
from PIL import Image
import io

import model


def configure():
    st.set_page_config(page_title="Low-light image enhancement")
    if "model" not in st.session_state:
        st.session_state["model"]: keras.Model = model.create_model()


def describe_service():
    st.title("Low-light image enhancement")
    st.subheader("Just upload your low-light image and get the processed one!")


def process_image():
    uploaded_file: io.BytesIO = st.file_uploader(
        label="Choose a file (you can upload new files without refreshing the page)",
        type=["png", "jpg", "jpeg"],
    )
    if uploaded_file:
        placeholder: st.delta_generator.DeltaGenerator = st.empty()
        placeholder.info("The image is being processed. It may take some time. Wait, please...")
        image: Image.Image = model.run_model(uploaded_file, st.session_state["model"])
        placeholder.empty()
        placeholder.image(image)


def main():
    describe_service()
    process_image()


if __name__ == "__main__":
    configure()
    main()