run.py

import gradio as gr
import cv2
import numpy as np
from insightface.app import FaceAnalysis
import tempfile
import os
import shutil

# Forcer InsightFace à utiliser un répertoire accessible
os.environ['INSIGHTFACE_ROOT'] = '/tmp/.insightface'

def swap_face(source_face, target_face, frame):
    src_emb = source_face.normed_embedding
    tgt_bbox = target_face.bbox.astype(int)
    resized_face = cv2.resize(source_face.img, (tgt_bbox[2]-tgt_bbox[0], tgt_bbox[3]-tgt_bbox[1]))
    mask = np.zeros_like(resized_face)
    center = (mask.shape[1]//2, mask.shape[0]//2)
    radius = int(min(mask.shape) * 0.45)
    cv2.circle(mask, center, radius, (255,255,255), -1)
    mask = cv2.GaussianBlur(mask, (15,15), 5)
    center = ((tgt_bbox[0]+tgt_bbox[2])//2, (tgt_bbox[1]+tgt_bbox[3])//2)
    result = cv2.seamlessClone(resized_face, frame, mask, center, cv2.NORMAL_CLONE)
    return result

def process_video(source_img, target_video):
    try:
        face_app = FaceAnalysis(name="buffalo_l", root="/tmp/.insightface")
        face_app.prepare(ctx_id=0, det_size=(640, 640))

        source_faces = face_app.get(source_img)
        if not source_faces:
            raise ValueError("Aucun visage trouvé dans l'image source.")
        source_face = source_faces[0]

        temp_output = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)

        cap = cv2.VideoCapture(target_video)
        fps = cap.get(cv2.CAP_PROP_FPS)
        frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
        frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
        out = cv2.VideoWriter(temp_output.name, fourcc, fps, (frame_width, frame_height))

        while True:
            ret, frame = cap.read()
            if not ret:
                break

            target_faces = face_app.get(frame)
            for face in target_faces:
                frame = swap_face(source_face, face, frame)

            out.write(frame)

        cap.release()
        out.release()

        final_path = tempfile.mktemp(suffix=".mp4")
        shutil.copy(temp_output.name, final_path)

        return final_path

    except Exception as e:
        print(f"Erreur lors du traitement : {str(e)}")
        return None

# Interface Gradio
demo = gr.Interface(
    fn=process_video,
    inputs=[
        gr.Image(label="Visage Source", type="numpy"),
        gr.Video(label="Vidéo Cible"),
    ],
    outputs=gr.Video(label="Vidéo Résultat"),
    title="🎬 FaceSwap Pro",
    description="Échangez des visages dans une vidéo.",
    allow_flagging="never"
)

if __name__ == "__main__":
    demo.launch()