Update app.py

app.py

@@ -39,19 +39,7 @@ def check_for_adult_content(image):
 
     return False  # No explicit content detected
 
-# Validate ROI to prevent out-of-bound errors
-def validate_roi(roi, image_shape):
-    x, y, w, h = roi
-    img_h, img_w = image_shape[:2]
-
-    x = max(0, x)
-    y = max(0, y)
-    w = min(w, img_w - x)
-    h = min(h, img_h - y)
-
-    return x, y, w, h
-
-# Face swapping function with sharpening
+# Face swapping function with added sharpening
 def swap_faces(destination_image, source_image):
     # Load the destination and source images from Streamlit inputs
     img = cv2.cvtColor(np.array(destination_image), cv2.COLOR_RGB2BGR)
@@ -70,13 +58,7 @@ def swap_faces(destination_image, source_image):
     res = img.copy()
     try:
         for face in faces:
-            bbox = face['bbox'].astype(int)
-            x, y, w, h = validate_roi(bbox, img.shape)  # Validate ROI
-
-            # Resize source face to match destination face size
-            test_face_resized = cv2.resize(test, (w, h))
-
-            res = swapper.get(res, face, test_face_resized, paste_back=True)
+            res = swapper.get(res, face, test_face, paste_back=True)
     except MemoryError:
         return "Memory error: Face swapping operation failed due to memory overload."
 
@@ -86,6 +68,9 @@ def swap_faces(destination_image, source_image):
                        [0, -1, 0]])
     sharpened_res = cv2.filter2D(res, -1, kernel)
 
+    # Commented smoothing (blurring) for testing purposes
+    smoothed_res = cv2.GaussianBlur(res, (5, 5), 0)
+
     # Convert the result to RGB for display in Streamlit
     sharpened_res_rgb = cv2.cvtColor(sharpened_res, cv2.COLOR_BGR2RGB)
 
@@ -161,4 +146,4 @@ def main():
 
 # Run the app
 if __name__ == '__main__':
-    main()
+    main()