fixing a small bug with add_face and the swap endpoint

Author: w4hns1nn

apipod-deploy/Dockerfile

@@ -0,0 +1,35 @@
+FROM python:3.8-slim
+
+SHELL ["/bin/bash", "-c"]
+
+# Install system dependencies together with optional extras.
+# Always install gcc for reliability; some ML packages still need compilation from source.
+# Others we detect automatically based on the deployment flags.
+RUN apt-get update && apt-get install -y \
+    ffmpeg \
+    gcc g++ \
+    libcudnn9-cuda-12 \
+    libcudnn9-dev-cuda-12 \
+    && apt-get clean
+# Ensure CUDA and cuDNN libraries are in the library path
+ENV LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}"
+
+# Set the working directory
+WORKDIR /
+
+# Install Python dependencies
+RUN pip install --upgrade pip
+RUN pip install runpod>=1.7.7 && \
+    pip install . --no-cache
+
+# Configure APIPod for RunPod Serverless
+ENV APIPOD_BACKEND="runpod"
+ENV APIPOD_DEPLOYMENT="serverless"
+
+ARG port=8080
+ENV APIPOD_PORT=$port
+EXPOSE $port
+ENV APIPOD_HOST="0.0.0.0"
+
+# Set the entrypoint
+CMD ["uvicorn", "face2face\server:app", "--host", "0.0.0.0", "--port", "8000"]

apipod-deploy/apipod.json

@@ -0,0 +1,27 @@
+{
+    "entrypoint": "face2face\\server.py",
+    "title": "Face2Face",
+    "python_version": "3.8",
+    "pytorch": false,
+    "tensorflow": false,
+    "onnx": "1.23.0",
+    "transformers": false,
+    "diffusers": false,
+    "cuda": false,
+    "system_packages": [],
+    "model_files": [
+        "face2face\\models\\face_occluder.onnx",
+        "face2face\\models\\face_enhancer\\gfpgan_1.4.onnx",
+        "face2face\\models\\face_enhancer\\gpen_bfr_1024.onnx",
+        "face2face\\models\\face_enhancer\\gpen_bfr_2048.onnx",
+        "face2face\\models\\face_enhancer\\gpen_bfr_256.onnx",
+        "face2face\\models\\face_enhancer\\gpen_bfr_512.onnx",
+        "face2face\\models\\insightface\\inswapper_128.onnx",
+        "face2face\\models\\insightface\\buffalo_l\\1k3d68.onnx",
+        "face2face\\models\\insightface\\buffalo_l\\2d106det.onnx",
+        "face2face\\models\\insightface\\buffalo_l\\det_10g.onnx",
+        "face2face\\models\\insightface\\buffalo_l\\genderage.onnx",
+        "face2face\\models\\insightface\\buffalo_l\\w600k_r50.onnx"
+    ],
+    "has_env_file": false
+}

apipod.json

@@ -0,0 +1,27 @@
+{
+    "entrypoint": "face2face\\server.py",
+    "title": "Face2Face",
+    "python_version": "3.8",
+    "pytorch": false,
+    "tensorflow": false,
+    "onnx": "1.23.0",
+    "transformers": false,
+    "diffusers": false,
+    "cuda": false,
+    "system_packages": [],
+    "model_files": [
+        "face2face\\models\\face_occluder.onnx",
+        "face2face\\models\\face_enhancer\\gfpgan_1.4.onnx",
+        "face2face\\models\\face_enhancer\\gpen_bfr_1024.onnx",
+        "face2face\\models\\face_enhancer\\gpen_bfr_2048.onnx",
+        "face2face\\models\\face_enhancer\\gpen_bfr_256.onnx",
+        "face2face\\models\\face_enhancer\\gpen_bfr_512.onnx",
+        "face2face\\models\\insightface\\inswapper_128.onnx",
+        "face2face\\models\\insightface\\buffalo_l\\1k3d68.onnx",
+        "face2face\\models\\insightface\\buffalo_l\\2d106det.onnx",
+        "face2face\\models\\insightface\\buffalo_l\\det_10g.onnx",
+        "face2face\\models\\insightface\\buffalo_l\\genderage.onnx",
+        "face2face\\models\\insightface\\buffalo_l\\w600k_r50.onnx"
+    ],
+    "has_env_file": false
+}

face2face/core/face2face.py

@@ -116,7 +116,7 @@ def swap(
 
         # TODO: the fucking faces in dict is a shit with swapping by pairs (with recognition)
         # Needs to be fixed.
-        if faces is not None and not isinstance(faces, dict):
+        if faces is not None and isinstance(faces, dict):
             faces = list(self.get_faces(faces).values())
 
         # read all the provided media

face2face/server.py

@@ -9,6 +9,7 @@
 import face2face
 from face2face.core.face2face import Face2Face
 from media_toolkit.utils.generator_wrapper import SimpleGeneratorWrapper
+from face2face.core.mixins._face_embedding import FileWriteableFace
 
 from face2face.settings import ALLOW_EMBEDDING_SAVE_ON_SERVER
 
@@ -78,13 +79,19 @@ def add_face(
     faces = f2f.add_face(face_name=face_name, media=image, save=save)
 
     if isinstance(faces, dict):
+        faces = {
+            face_name: FileWriteableFace(face)
+            for face_name, face in faces.items()
+        }
+
         return MediaList([
             MediaFile(file_name=f"{face_name}.npy").from_bytesio(face.to_bytes_io())
             for face_name, face in faces.items()
         ])
 
     if isinstance(faces, tuple):
-        return MediaFile(file_name=f"{face_name}.npy").from_bytesio(faces[1].to_bytes_io())
+        face = FileWriteableFace(faces[1]).to_bytes_io()
+        return MediaFile(file_name=f"{face_name}.npy").from_bytesio(face)
 
 
 @app.endpoint("/swap")

test_api_pod.py

@@ -0,0 +1,12 @@
+# debug_apipod.py (place in face2face root directory)
+import sys
+import os
+
+# Add apipod to path
+sys.path.insert(0, r'A:\projects\apipod')
+
+# Import and run the CLI
+from apipod.cli import main
+
+if __name__ == "__main__":
+    main()