debug face counts

app.py

@@ -2,38 +2,68 @@ import spaces
 import gradio as gr
 import cv2
 import numpy as np
+import traceback
 
 @spaces.GPU
 def swap_face(source_img, target_img):
     import insightface
     from insightface.app import FaceAnalysis
     try:
+        # Try GPU ctx 0, fallback to CPU
         app = FaceAnalysis(name='buffalo_l')
-        app.prepare(ctx_id=0, det_size=(640,640))
+        try:
+            app.prepare(ctx_id=0, det_size=(640,640))
+        except:
+            app.prepare(ctx_id=-1, det_size=(640,640))
         import insightface.model_zoo
         swapper = insightface.model_zoo.get_model('inswapper_128.onnx', download=True)
         if swapper is None:
+            # try hf hub download
+            from huggingface_hub import hf_hub_download
+            p = hf_hub_download(repo_id="facefusion/models", filename="inswapper_128.onnx")
+            swapper = insightface.model_zoo.get_model(p)
+        src = cv2.imread(source_img)
+        tgt = cv2.imread(target_img)
+        if src is None or tgt is None:
             return target_img
-        source = cv2.imread(source_img)
-        target = cv2.imread(target_img)
-        s_faces = app.get(source)
-        t_faces = app.get(target)
-        if not s_faces or not t_faces:
-            return target_img
-        res = swapper.get(target, t_faces[0], s_faces[0], paste_back=True)
-        out_path = "/tmp/swapped.jpg"
-        cv2.imwrite(out_path, res)
-        return out_path
+        s_faces = app.get(src)
+        t_faces = app.get(tgt)
+        # debug: if fails try larger det_size
+        if len(s_faces)==0 or len(t_faces)==0:
+            app2 = FaceAnalysis(name='buffalo_l')
+            app2.prepare(ctx_id=-1, det_size=(1024,1024))
+            if len(s_faces)==0:
+                s_faces = app2.get(src)
+            if len(t_faces)==0:
+                t_faces = app2.get(tgt)
+        if len(s_faces)==0 or len(t_faces)==0:
+            # return target with debug text
+            cv2.putText(tgt, f"s:{len(s_faces)} t:{len(t_faces)}", (20,40), cv2.FONT_HERSHEY_SIMPLEX, 1, (0,0,255),2)
+            out="/tmp/debug.jpg"
+            cv2.imwrite(out,tgt)
+            return out
+        res = swapper.get(tgt, t_faces[0], s_faces[0], paste_back=True)
+        out="/tmp/swapped.jpg"
+        cv2.imwrite(out,res)
+        return out
     except Exception as e:
-        return target_img
+        tb=traceback.format_exc()[:500]
+        # return target with error text
+        try:
+            tgt2=cv2.imread(target_img)
+            cv2.putText(tgt2, str(e)[:40], (20,40), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0,0,255),2)
+            out="/tmp/err.jpg"
+            cv2.imwrite(out,tgt2)
+            return out
+        except:
+            return target_img
 
 with gr.Blocks() as demo:
-    gr.Markdown("# FaceFusion Zero (PRO)")
+    gr.Markdown("# FaceFusion Zero (PRO) - Debug")
     with gr.Row():
-        src = gr.Image(type="filepath", label="Source (your face)")
-        tgt = gr.Image(type="filepath", label="Target (flying Kathmandu)")
-    out = gr.Image(type="filepath", label="Result")
+        src = gr.Image(type="filepath", label="Source (closeup)")
+        tgt = gr.Image(type="filepath", label="Target")
+    out = gr.Image(type="filepath", label="Result (debug shows s:t face counts)")
     btn = gr.Button("Swap")
     btn.click(swap_face, inputs=[src, tgt], outputs=out)
-
 demo.launch()