|
import cv2 |
|
import os |
|
import base64 |
|
import numpy as np |
|
from PIL import ImageFont, ImageDraw, Image |
|
import streamlit as st |
|
from modules.Cartoon.main import cartoon |
|
from modules.ThugLife.main import thug_life |
|
|
|
|
|
|
|
|
|
img_extensions = ["jpg","png","jpeg"] |
|
|
|
def get_binary_file_downloader_html(bin_file,file_label='File'): |
|
with open(bin_file,'rb') as f: |
|
data = f.read() |
|
bin_str = base64.b64encode(data).decode() |
|
href = f'<h3><a href="data:application/octet-stream;base64,{bin_str}" download="{os.path.basename(bin_file)}">Download {file_label}</a></h3>' |
|
return href |
|
|
|
def image_transformations(result,inp_img,filter,img_extension = "jpg"): |
|
result = result[:,:,:3] |
|
|
|
if filter == "Basic Image Editing": |
|
gamma = st.slider("Gamma Correction",0.0,5.0,1.0,step = 0.1) |
|
saturation = st.slider("Saturation", 0.0, 2.0, 1.0, step=0.1) |
|
blurring = st.checkbox("Bluring", False) |
|
if blurring: |
|
col1,col2 = st.columns(2) |
|
blur_area = col1.slider("Blur Area",1,100,1,2) |
|
blur_intensity = col2.slider("Blur Intensity", 0, 500, 0, 1) |
|
result = cv2.GaussianBlur(result, (blur_area, 1), blur_intensity) |
|
|
|
apply_vignette = st.checkbox("Apply Vignette effect",False) |
|
if apply_vignette: |
|
vignette_effect = st.slider("Vignette Intensity",1,120,1) |
|
rows, cols = result.shape[:2] |
|
kernel_x = cv2.getGaussianKernel(cols, vignette_effect + 139) |
|
kernel_y = cv2.getGaussianKernel(rows, vignette_effect + 139) |
|
kernel = kernel_y*kernel_x.T |
|
filter = 255 * kernel / np.linalg.norm(kernel) |
|
vignette_img = np.copy(result) |
|
for i in range(3): |
|
vignette_img[:,:,i] = vignette_img[:,:,i]*filter |
|
result = vignette_img |
|
|
|
hsvImg = cv2.cvtColor(result, cv2.COLOR_BGR2HSV) |
|
hsvImg[..., 1] = np.clip(hsvImg[..., 1] * saturation, 0, 255) |
|
hsvImg[..., 2] = np.power((hsvImg[..., 2] / 255.0), 1 / (gamma + 0.1)) * 255.0 |
|
result = cv2.cvtColor(hsvImg, cv2.COLOR_HSV2BGR) |
|
elif filter == "Cartoonie": |
|
line_size = st.slider("Number of edges",3,101,7,2) |
|
blurVal = st.slider("Blurr effect",3,101,7,2) |
|
totalCols = st.slider("Total Color in images",2,100,11,1) |
|
result = cartoon(result,[line_size,blurVal,totalCols]) |
|
elif filter == "Thug Life": |
|
angle = st.slider("Angle",-360,360,0,1) |
|
x = st.slider("Shift X",-1.0,1.0,0.0,0.1) |
|
y = st.slider("Shift Y", -1.0, 1.0, 0.0, 0.1) |
|
result = thug_life(inp_img,angle,x,y) |
|
if np.all(result != None): |
|
st.image(result, use_column_width=True, clamp=True, channels="BGR") |
|
filename = img_extension[0] + "." + img_extension[-1] |
|
cv2.imwrite(filename, result) |
|
st.markdown(get_binary_file_downloader_html(filename, |
|
'From Here '), |
|
unsafe_allow_html=True) |
