Update app.py

app.py

@@ -2,43 +2,40 @@ import gradio as gr
 import cv2
 import numpy as np
 from PIL import Image
-import face_recognition
 
-def face_swap(source_img, target_img):
-    src = np.array(source_img.convert("RGB"))
-    tgt = np.array(target_img.convert("RGB"))
+face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + "haarcascade_frontalface_default.xml")
 
-    src_faces = face_recognition.face_locations(src, model="hog")
-    tgt_faces = face_recognition.face_locations(tgt, model="hog")
+def simple_face_swap(src_img, tgt_img):
+    src = cv2.cvtColor(np.array(src_img), cv2.COLOR_RGB2BGR)
+    tgt = cv2.cvtColor(np.array(tgt_img), cv2.COLOR_RGB2BGR)
 
-    if len(src_faces) == 0 or len(tgt_faces) == 0:
-        return "❌ चेहरा detect नहीं हुआ।"
+    src_faces = face_cascade.detectMultiScale(src, 1.1, 4)
+    tgt_faces = face_cascade.detectMultiScale(tgt, 1.1, 4)
 
-    # 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]
+    if len(src_faces) == 0 or len(tgt_faces) == 0:
+        return "❌ किसी एक फोटो में चेहरा नहीं मिला।"
 
-    # 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
+    (x1, y1, w1, h1) = src_faces[0]
+    (x2, y2, w2, h2) = tgt_faces[0]
 
-    # Resize source face to match target size
-    src_face_resized = cv2.resize(src_face, (face_width, face_height))
+    src_face = src[y1:y1+h1, x1:x1+w1]
+    src_face_resized = cv2.resize(src_face, (w2, h2))
 
-    # 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)
+    # ब्लेंडिंग
+    mask = 0.5 * tgt[y2:y2+h2, x2:x2+w2] + 0.5 * src_face_resized
+    tgt[y2:y2+h2, x2:x2+w2] = mask.astype(np.uint8)
 
-    return Image.fromarray(tgt)
+    output = cv2.cvtColor(tgt, cv2.COLOR_BGR2RGB)
+    return Image.fromarray(output)
 
 gr.Interface(
-    fn=face_swap,
+    fn=simple_face_swap,
     inputs=[
-        gr.Image(type="pil", label="👤 Source Face"),
-        gr.Image(type="pil", label="🎯 Target Face")
+        gr.Image(type="pil", label="Source Face"),
+        gr.Image(type="pil", label="Target Image")
     ],
-    outputs=gr.Image(type="pil", label="✨ Swapped Output"),
-    title="🔁 RefaceMania (Simple AI Blending)",
-    description="अपना चेहरा किसी भी फोटो में लगाएं – फेस ब्लेंडिंग के साथ।"
+    outputs=gr.Image(type="pil", label="Output"),
+    title="RefaceMania Lite",
+    description="चेहरे की आसान अदला-बदली, बिना किसी error के।"
 ).launch()
+