from typing import Any, List, Callable
import roop.globals
import insightface
import cv2
import numpy as np

from roop.typing import Face, Frame
from roop.utilities import resolve_relative_path



class FaceSwapInsightFace():
    model_swap_insightface = None


    processorname = 'faceswap'
    type = 'swap'


    def Initialize(self, devicename):
        if self.model_swap_insightface is None:
            model_path = resolve_relative_path('../models/inswapper_128.onnx')
            self.model_swap_insightface = insightface.model_zoo.get_model(model_path, providers=roop.globals.execution_providers)

    
    def Run(self, source_face: Face, target_face: Face, temp_frame: Frame) -> Frame:
        img_fake, M = self.model_swap_insightface.get(temp_frame, target_face, source_face, paste_back=False)
        target_face.matrix = M
        return img_fake 


    def Release(self):
        del self.model_swap_insightface
        self.model_swap_insightface = None