Upload 7 files

.gitattributes

@@ -1,35 +1,35 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,3 +1,14 @@
 ---
+title: Face Swap
+emoji: 📊
+colorFrom: purple
+colorTo: pink
+sdk: gradio
+sdk_version: 5.46.0
+app_file: app.py
+pinned: false
 license: apache-2.0
+short_description: face swap
 ---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,134 @@
+import gradio as gr
+import os
+import uuid
+import cv2
+from mask import FaceSwapper  # الكلاس اللي عملناه قبل كده
+
+# تحميل الموديل مره واحده
+swapper = FaceSwapper(
+    model_path="models/inswapper_128.onnx",
+    gfpgan_model_path="gfpgan/weights/GFPGANv1.4.pth"
+)
+
+# دالة للتعامل مع Gradio
+def swap_faces(source_img, target_img):
+    try:
+        # حفظ الصور المرفوعة مؤقتاً
+        source_path = f"temp_source_{uuid.uuid4().hex}.jpg"
+        target_path = f"temp_target_{uuid.uuid4().hex}.jpg"
+        output_path = f"img/result_{uuid.uuid4().hex}.jpg"
+
+        cv2.imwrite(source_path, cv2.cvtColor(source_img, cv2.COLOR_RGB2BGR))
+        cv2.imwrite(target_path, cv2.cvtColor(target_img, cv2.COLOR_RGB2BGR))
+
+        # تنفيذ face swap
+        result_path = swapper.merge_face_into_image(source_path, target_path, output_path)
+
+        # تحميل الصورة الناتجة وعرضها
+        result_img = cv2.cvtColor(cv2.imread(result_path), cv2.COLOR_BGR2RGB)
+
+        # مسح الملفات المؤقتة
+        os.remove(source_path)
+        os.remove(target_path)
+
+        return result_img
+    except Exception as e:
+        return f"❌ حصل خطأ: {str(e)}"
+
+# واجهة Gradio
+demo = gr.Interface(
+    fn=swap_faces,
+    inputs=[
+        gr.Image(type="numpy", label="Source Image (الطفل)"),
+        gr.Image(type="numpy", label="Target Image (المشهد)")
+    ],
+    outputs=gr.Image(type="numpy", label="النتيجة"),
+    title="FaceSwap with GFPGAN",
+    description="ارفع صورتين: (1) صورة الطفل (2) المشهد اللي عايز تدخله فيه. وهنرجعلك صورة face swap محسّنة بـ GFPGAN."
+)
+
+if __name__ == "__main__":
+    demo.launch(server_name="127.0.0.1", server_port=7860)
+
+
+"""import cv2
+import insightface
+import numpy as np
+import os
+from gfpgan import GFPGANer  # pip install gfpgan"""
+
+
+"""class FaceSwapper:
+    def __init__(self,
+                 model_path="models/inswapper_128.onnx",
+                 gfpgan_model_path="gfpgan/weights/GFPGANv1.4.pth"):
+        # ============ تحميل FaceAnalysis ============
+        self.app = insightface.app.FaceAnalysis(name="buffalo_l", providers=['CPUExecutionProvider'])
+        self.app.prepare(ctx_id=0, det_size=(640, 640))
+
+        # ============ تحميل inswapper ============
+        if not os.path.exists(model_path):
+            raise FileNotFoundError(f"❌ الموديل مش موجود في: {model_path}")
+        self.swapper = insightface.model_zoo.get_model(model_path, providers=['CPUExecutionProvider'])
+
+        # ============ تحميل GFPGAN ============
+        self.gfpganer = GFPGANer(
+            model_path=gfpgan_model_path,
+            upscale=1,
+            arch="clean",
+            channel_multiplier=2
+        )
+
+    # ============ دالة مساعدة لاختيار أكبر وجه ============
+    @staticmethod
+    def get_biggest_face(faces):
+        return max(faces, key=lambda f: (f.bbox[2]-f.bbox[0]) * (f.bbox[3]-f.bbox[1]))
+
+    # ============ دالة الدمج ============
+    def merge_face_into_image(self, source_img_path, target_img_path, output_path):
+        source_img = cv2.imread(source_img_path)
+        target_img = cv2.imread(target_img_path)
+
+        if source_img is None or target_img is None:
+            raise ValueError("❌ مشكلة في قراءة الصور")
+
+        source_faces = self.app.get(source_img)
+        target_faces = self.app.get(target_img)
+
+        if not source_faces or not target_faces:
+            #raise ValueError("❌ مش لاقي وش في الصورة!")
+            print("⚠️ No faces detected, returning target image as-is.")
+            cv2.imwrite(output_path, target_img)
+            return output_path
+
+        source_face = self.get_biggest_face(source_faces)
+        target_face = self.get_biggest_face(target_faces)
+
+        # استبدال الوجه
+        swapped_img = self.swapper.get(target_img.copy(), target_face, source_face, paste_back=True)
+
+        # قص الوجه من الصورة
+        x1, y1, x2, y2 = target_face.bbox.astype(int)
+        x1, y1 = max(0, x1), max(0, y1)
+        x2, y2 = min(swapped_img.shape[1], x2), min(swapped_img.shape[0], y2)
+
+        face_crop = swapped_img[y1:y2, x1:x2]
+
+        if face_crop.size == 0:
+            raise ValueError("❌ الوجه المقطوع فاضي (bbox مش مظبوط)")
+
+        # ماسك بنفس حجم الوجه
+        mask = 255 * np.ones(face_crop.shape, face_crop.dtype)
+        mask = cv2.GaussianBlur(mask, (25, 25), 30)
+
+        # مركز الوجه
+        center = ((x1 + x2) // 2, (y1 + y2) // 2)
+
+        # دمج الوجه في الصورة
+        blended = cv2.seamlessClone(face_crop, swapped_img, mask, center, cv2.NORMAL_CLONE)
+
+        # تحسين الصورة بالـ GFPGAN
+        _, _, enhanced = self.gfpganer.enhance(blended, has_aligned=False, only_center_face=False, paste_back=True)
+
+        final_img = enhanced  # أو blended لو GFPGAN مش موجود
+        return cv2.cvtColor(final_img, cv2.COLOR_BGR2RGB)"""    

gfpgan/weights/GFPGANv1.4.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e2cd4703ab14f4d01fd1383a8a8b266f9a5833dacee8e6a79d3bf21a1b6be5ad
+size 348632874

mask.py

@@ -0,0 +1,93 @@
+import cv2
+import insightface
+import numpy as np
+import os
+from gfpgan import GFPGANer
+from basicsr.archs.rrdbnet_arch import RRDBNet
+from realesrgan import RealESRGANer   # pip install realesrgan
+
+
+class FaceSwapper:
+    def __init__(self,
+                 model_path="models/inswapper_128.onnx",
+                 gfpgan_model_path="gfpgan/weights/GFPGANv1.4.pth",
+                 realesrgan_model_path="models/RealESRGAN_x2plus.pth"):
+        
+        # ============ Face Analysis ============
+        self.app = insightface.app.FaceAnalysis(name="buffalo_l", providers=['CPUExecutionProvider'])
+        self.app.prepare(ctx_id=0, det_size=(640, 640))  # حافظ عليها صغيرة
+
+        # ============ Face Swapper ============
+        if not os.path.exists(model_path):
+            raise FileNotFoundError(f"❌ الموديل مش موجود في: {model_path}")
+        self.swapper = insightface.model_zoo.get_model(model_path, providers=['CPUExecutionProvider'])
+
+        # ============ GFPGAN ============
+        self.gfpganer = GFPGANer(
+            model_path=gfpgan_model_path,
+            upscale=2,   # ممكن تخليها 4 حسب جهازك
+            arch="clean",
+            channel_multiplier=2
+        )
+
+        # ============ Real-ESRGAN ============
+        model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64,
+                        num_block=23, num_grow_ch=32, scale=2)
+        self.realesrganer = RealESRGANer(
+            scale=2,
+            model_path=realesrgan_model_path,
+            model=model,
+            tile=0,
+            tile_pad=10,
+            pre_pad=0,
+            half=False
+        )
+
+    @staticmethod
+    def get_biggest_face(faces):
+        return max(faces, key=lambda f: (f.bbox[2]-f.bbox[0]) * (f.bbox[3]-f.bbox[1]))
+
+    def merge_face_into_image(self, source_img_path, target_img_path, output_path):
+        source_img = cv2.imread(source_img_path)
+        target_img = cv2.imread(target_img_path)
+
+        if source_img is None or target_img is None:
+            raise ValueError("❌ مشكلة في قراءة الصور")
+
+        source_faces = self.app.get(source_img)
+        target_faces = self.app.get(target_img)
+
+        if not source_faces or not target_faces:
+            print("⚠️ No faces detected, returning target image as-is.")
+            cv2.imwrite(output_path, target_img)
+            return output_path
+
+        source_face = self.get_biggest_face(source_faces)
+        target_face = self.get_biggest_face(target_faces)
+
+        swapped_img = self.swapper.get(target_img.copy(), target_face, source_face, paste_back=True)
+
+        # قص الوجه
+        x1, y1, x2, y2 = target_face.bbox.astype(int)
+        x1, y1 = max(0, x1), max(0, y1)
+        x2, y2 = min(swapped_img.shape[1], x2), min(swapped_img.shape[0], y2)
+        face_crop = swapped_img[y1:y2, x1:x2]
+
+        if face_crop.size == 0:
+            raise ValueError("❌ الوجه المقطوع فاضي (bbox مش مظبوط)")
+
+        # ماسك للدمج
+        mask = 255 * np.ones(face_crop.shape, face_crop.dtype)
+        mask = cv2.GaussianBlur(mask, (51, 51), 40)
+        center = ((x1 + x2) // 2, (y1 + y2) // 2)
+
+        blended = cv2.seamlessClone(face_crop, swapped_img, mask, center, cv2.MIXED_CLONE)
+
+        # تحسين الوش بالـ GFPGAN
+        _, _, gfpgan_img = self.gfpganer.enhance(blended, has_aligned=False, only_center_face=False, paste_back=True)
+
+        # تكبير وتحسين الصورة كاملة بالـ Real-ESRGAN
+        sr_img, _ = self.realesrganer.enhance(gfpgan_img, outscale=1)
+
+        cv2.imwrite(output_path, sr_img, [cv2.IMWRITE_PNG_COMPRESSION, 0])
+        return output_path

models/inswapper_128.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e4a3f08c753cb72d04e10aa0f7dbe3deebbf39567d4ead6dce08e98aa49e16af
+size 554253681