Update app.py

app.py

@@ -149,19 +149,34 @@ def execute(image, prompt, debug=False):
     current_img = imgs[image]
     cv2.imwrite('base_image.jpg', current_img)
     cv2.imwrite("mask_person.jpg", maskPerson('base_image.jpg'))
-    #cv2.imwrite("mask.jpg", maskHead('base_image.jpg'))
+    cv2.imwrite("mask_face.jpg", maskHead('base_image.jpg'))
+      
     
     im = Image.open('base_image.jpg')
     np_arr = np.array(im)
     rgb_image = cv2.cvtColor(np_arr, cv2.COLOR_BGR2RGB)
 
     im =  Image.fromarray(rgb_image)
-    person = Image.open('mask_person.jpg')
+      
+    person = np.array(Image.open('mask_person.jpg'))
+    face = np.array(Image.open('mask_face.jpg'))
 
+    person_gray = cv2.cvtColor(person, cv2.COLOR_BGR2GRAY)
+    face_gray = cv2.cvtColor(face, cv2.COLOR_BGR2GRAY)
+
+    _, mask = cv2.threshold(face_gray, 1, 255, cv2.THRESH_BINARY_INV)
+    mask_inv = cv2.bitwise_not(mask)
+    person_masked = cv2.bitwise_and(person_gray, person_gray, mask=mask_inv)
+    face_masked = cv2.bitwise_and(face_gray, face_gray, mask=mask)
+    result = cv2.add(person_masked, face_masked)
+    cv2.imwrite('join.jpg', result)
+
+    fund_mask = (Image.open('join.jpg')
+    
     result0 = pipe(
             prompt=f"{prompt} {TRIGGER}",
             image=im,
-            mask_image=person,
+            mask_image=fund_mask,
             width=1024,
             height=1024,
             strength=0.85,
@@ -193,8 +208,10 @@ def execute(image, prompt, debug=False):
     if debug:
         response.append(im)
         response.append(person)
-        response.append(mask)
+        response.append(face)
+        response.append(fund_mask)
         response.append(result0)
+        response.append(mask)
 
     response.append(result)
   return response