import os
import cv2
import numpy as np
import streamlit as st
from webcam import webcam
from images import image_transformations
from PIL import Image
title = st.empty()
title.header("Image Filter")
img_extensions = ["jpg","png","jpeg"]

def show_info():
    st.subheader('''
    Creator Jainam Jain: 
    ''')
    st.markdown("### **Filters used:Cartoon")
    st.write("")
st.sidebar.subheader("Choose the mode of operation: ")
selected_option = st.sidebar.selectbox("",["Select from below","Image Filters","Video Filters"])

if selected_option == "Image Filters":
    title.header("Image Filters")
    img_operation_mode = st.selectbox("Upload Images from: ",["--  Select from below  --","Local Storage","Take a snap from Webcam"])

    if img_operation_mode == "Local Storage":
        uploaded_file = st.file_uploader("Upload images from local storage here",type = img_extensions)
        if uploaded_file is not None:

            image_byte = np.asarray(bytearray(uploaded_file.read()), dtype=np.uint8)
            uploaded_file.seek(0)
            img_bytes = uploaded_file.read()
            decoded_img = cv2.imdecode(np.frombuffer(img_bytes, np.uint8), -1)
            result = decoded_img
            inp_img = cv2.imdecode(image_byte, 1)
            filter = st.selectbox("Choose an Image filter: ",["Basic Image Editing","Thug Life","Green Screen","Moustaches",\
            "Devil-ie","Heart Eyes","John Cena XD","Cartoonie","Face Blur"],0)
            image_transformations(result,inp_img,filter,uploaded_file.name.split("."))

    elif img_operation_mode == "Take a snap from Webcam":
        result = webcam()
        if result is None:
            st.write("Waiting for capture...")
        else:
            st.write("Got an image from the webcam :P")
            result = cv2.cvtColor(np.asarray(result,np.uint8),cv2.COLOR_RGB2BGR)

            filter = st.selectbox("Choose an Image filter: ",["Basic Image Editing","Thug Life","Green Screen","Moustaches",\
            "Devil-ie","Heart Eyes","John Cena XD","Cartoonie","Face Blur"],0)
            image_transformations(result,filter)

    else:
        st.sidebar.markdown('''
        This section contains filters to be applied on images.

        Images can be uploaded either from local storage (or) from your webcamera.
        ''')
        st.sidebar.markdown("Choose the source of image, from the drop list on right :point_right:")
        st.sidebar.subheader("Tips for operating on Image Filters: ")
        st.sidebar.markdown('''
            * Un edited pictures provide the best results.
            * Link to download the edited pictures are at the bottom of the page
            ''')