Update app.py

app.py

@@ -15,10 +15,11 @@ if not os.path.exists(inswapper_path):
 swapper = insightface.model_zoo.get_model(inswapper_path, providers=['CPUExecutionProvider'])
 
 def preprocess_image(img):
-    img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-    img_eq = cv2.equalizeHist(img_gray)
-    img_rgb = cv2.cvtColor(img_eq, cv2.COLOR_GRAY2RGB)
-    return img_rgb
+    # Adjust brightness and contrast to improve quality
+    alpha = 1.2  # Contrast control
+    beta = 10    # Brightness control
+    adjusted = cv2.convertScaleAbs(img, alpha=alpha, beta=beta)
+    return adjusted
 
 def sharpen_image(img):
     kernel = np.array([[0, -1, 0], [-1, 5, -1], [0, -1, 0]])
@@ -27,11 +28,17 @@ def sharpen_image(img):
 
 def swap_faces(src_img, dst_img, blur_strength=3, sharpen=False):
     try:
+        # Preprocess while preserving color
         src = preprocess_image(src_img)
         dst = preprocess_image(dst_img)
 
-        src_faces = app.get(src)
-        dst_faces = app.get(dst)
+        # Convert to RGB for face detection
+        src_rgb = cv2.cvtColor(src, cv2.COLOR_BGR2RGB)
+        dst_rgb = cv2.cvtColor(dst, cv2.COLOR_BGR2RGB)
+
+        # Detect faces
+        src_faces = app.get(src_rgb)
+        dst_faces = app.get(dst_rgb)
 
         if not src_faces or not dst_faces:
             raise ValueError("No faces detected in one of the images.")
@@ -39,14 +46,18 @@ def swap_faces(src_img, dst_img, blur_strength=3, sharpen=False):
         src_face = src_faces[0]
         dst_face = dst_faces[0]
 
-        swapped_img = swapper.get(dst, dst_face, src_face, paste_back=True)
+        # Perform face swap using inswapper
+        swapped_img = swapper.get(dst_rgb, dst_face, src_face, paste_back=True)
 
+        # Post-processing: Apply slight blur for smoother blending
         if blur_strength > 0:
             swapped_img = cv2.GaussianBlur(swapped_img, (blur_strength, blur_strength), 0)
 
+        # Optional: Apply lightweight sharpening
         if sharpen:
             swapped_img = sharpen_image(swapped_img)
 
+        # Convert back to BGR for output
         result = cv2.cvtColor(swapped_img, cv2.COLOR_RGB2BGR)
         return result
     except Exception as e: