| 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 |
| ''') |
|
|
|
|
