Update app.py

app.py

@@ -1,37 +1,44 @@
 import gradio as gr
 import cv2
-from PIL import Image
 import numpy as np
+from PIL import Image
+import face_recognition
 
-face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + "haarcascade_frontalface_default.xml")
-
-def swap_faces(src_img, tgt_img):
-    src = np.array(src_img.convert("RGB"))
-    tgt = np.array(tgt_img.convert("RGB"))
-
-    src_gray = cv2.cvtColor(src, cv2.COLOR_RGB2GRAY)
-    tgt_gray = cv2.cvtColor(tgt, cv2.COLOR_RGB2GRAY)
+def face_swap(source_img, target_img):
+    src = np.array(source_img.convert("RGB"))
+    tgt = np.array(target_img.convert("RGB"))
 
-    src_faces = face_cascade.detectMultiScale(src_gray, 1.3, 5)
-    tgt_faces = face_cascade.detectMultiScale(tgt_gray, 1.3, 5)
+    src_faces = face_recognition.face_locations(src, model="hog")
+    tgt_faces = face_recognition.face_locations(tgt, model="hog")
 
     if len(src_faces) == 0 or len(tgt_faces) == 0:
-        return "❌ चेहरे नहीं मिले।"
+        return "❌ चेहरा detect नहीं हुआ।"
+
+    # Source face crop
+    top_s, right_s, bottom_s, left_s = src_faces[0]
+    src_face = src[top_s:bottom_s, left_s:right_s]
 
-    (x, y, w, h) = src_faces[0]
-    src_face = src[y:y+h, x:x+w]
+    # Target face replacement
+    top_t, right_t, bottom_t, left_t = tgt_faces[0]
+    face_height = bottom_t - top_t
+    face_width = right_t - left_t
 
-    (x2, y2, w2, h2) = tgt_faces[0]
-    src_face_resized = cv2.resize(src_face, (w2, h2))
+    # Resize source face to match target size
+    src_face_resized = cv2.resize(src_face, (face_width, face_height))
 
-    tgt[y2:y2+h2, x2:x2+w2] = src_face_resized
+    # Blend with target face area
+    blended = 0.5 * tgt[top_t:bottom_t, left_t:right_t] + 0.5 * src_face_resized
+    tgt[top_t:bottom_t, left_t:right_t] = blended.astype(np.uint8)
 
     return Image.fromarray(tgt)
 
 gr.Interface(
-    fn=swap_faces,
-    inputs=[gr.Image(type="pil", label="Source Face"), gr.Image(type="pil", label="Target Image")],
-    outputs=gr.Image(type="pil", label="Swapped Output"),
-    title="🤖 RefaceMania Lite",
-    description="Face Swapping tool (Fast Build, No Errors)"
+    fn=face_swap,
+    inputs=[
+        gr.Image(type="pil", label="👤 Source Face"),
+        gr.Image(type="pil", label="🎯 Target Face")
+    ],
+    outputs=gr.Image(type="pil", label="✨ Swapped Output"),
+    title="🔁 RefaceMania (Simple AI Blending)",
+    description="अपना चेहरा किसी भी फोटो में लगाएं – फेस ब्लेंडिंग के साथ।"
 ).launch()