Updated Module

app.py

@@ -4,7 +4,7 @@ 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"]
@@ -25,13 +25,16 @@ if selected_option == "Image Filters":
     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,filter,uploaded_file.name.split("."))
+            image_transformations(result,inp_img,filter,uploaded_file.name.split("."))
 
     elif img_operation_mode == "Take a snap from Webcam":
         result = webcam()

images.py

@@ -5,6 +5,7 @@ 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
 # from helper.utils import file_checker
 # from helper.descriptor import file_info
 # from helper.rcnn_utils import generate_rcnn_mask
@@ -18,7 +19,7 @@ def get_binary_file_downloader_html(bin_file,file_label='File'):
     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,filter,img_extension = "jpg"):
+def image_transformations(result,inp_img,filter,img_extension = "jpg"):
         result = result[:,:,:3]
         # filter_info(filter)
         if filter == "Basic Image Editing":
@@ -53,6 +54,11 @@ def image_transformations(result,filter,img_extension = "jpg"):
             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]

modules/ThugLife/main.py

@@ -0,0 +1,50 @@
+import cv2
+import numpy as np
+import imutils
+import streamlit as st
+import base64
+def detect_face(input_img):
+    face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
+    gray_image = cv2.cvtColor(input_img, cv2.COLOR_BGR2GRAY)
+    faces = face_cascade.detectMultiScale(gray_image, scaleFactor=1.1, minNeighbors=5)
+
+    if len(faces) == 1:
+        return faces[0]
+    else:
+        return None
+
+def thug_life(input_img,angle,x,y):
+    face_coords = detect_face(input_img)
+
+    if face_coords is None:
+        return input_img
+
+    sunglass_img = cv2.imread('media/glasses.png', cv2.IMREAD_UNCHANGED)
+    sunglass_img = imutils.rotate(sunglass_img, angle=angle)
+    face_width, face_height = face_coords[2], face_coords[3]
+    sunglass_img_resized = cv2.resize(sunglass_img, (face_width, face_height))
+
+    # Check if the image has an alpha channel
+    if sunglass_img_resized.shape[2] == 4:
+        alpha_channel_resized = cv2.resize(sunglass_img_resized[:, :, 3], (face_width, face_height))
+    else:
+        # If no alpha channel, create a default one
+        alpha_channel_resized = np.ones((face_height, face_width), dtype=np.uint8) * 255
+
+    thug_glass = alpha_channel_resized / 255.0
+    thug_glass = np.stack([thug_glass] * 3, axis=-1)
+    offset_y = int(face_height * y)  # Adjust the offset based on your preference
+    start_y = max(0, face_coords[1] - offset_y)
+    offset_x = int(face_width * x)  # Adjust the offset based on your preference
+    start_x = max(0, face_coords[1] - offset_x)
+    roi = input_img[start_y: start_y + face_height, start_x:start_x + face_width]
+
+    glass_bgr = sunglass_img_resized[:, :, :3]
+    overlay = (1 - thug_glass) * roi + thug_glass * glass_bgr
+    input_img[start_y: start_y + face_height, start_x:start_x + face_width] = overlay
+    music_file = 'media/ThugLife.mp3'
+    audio_bytes = open(music_file,'rb').read()
+    audio_bytes = base64.b64encode(audio_bytes).decode()
+    st.markdown(f'<audio autoplay controls><source src="data:audio/mp3;base64,{audio_bytes}" type="audio/mp3"></audio>', unsafe_allow_html=True)
+
+    return input_img

requiremments.txt → requirements.txt

@@ -59,3 +59,4 @@ validators==0.22.0
 watchdog==3.0.0
 webcam==1.34
 zipp==3.17.0
+