light opencv enhancer strict facefusion

app.py

@@ -4,6 +4,13 @@ import cv2
 import numpy as np
 import traceback
 
+def enhance_sharp(img):
+    # strong unsharp mask via opencv - works without gfpgan
+    gauss = cv2.GaussianBlur(img, (0,0), 2)
+    sharp = cv2.addWeighted(img, 1.7, gauss, -0.7, 0)
+    # extra detail enhance on face region only is done by caller
+    return sharp
+
 @spaces.GPU
 def swap_face(source_img, target_img):
     import insightface
@@ -33,7 +40,19 @@ def swap_face(source_img, target_img):
             cv2.imwrite(out,tgt)
             return out
         res = swapper.get(tgt, t_faces[0], s_faces[0], paste_back=True)
-        out="/tmp/swapped.jpg"
+        # enhance face region only for sharpness
+        try:
+            # detect face bbox and enhance that region
+            x1,y1,x2,y2 = map(int, t_faces[0].bbox)
+            pad = 20
+            x1,y1 = max(0,x1-pad), max(0,y1-pad)
+            x2,y2 = min(res.shape[1], x2+pad), min(res.shape[0], y2+pad)
+            face_crop = res[y1:y2, x1:x2]
+            enhanced = enhance_sharp(face_crop)
+            res[y1:y2, x1:x2] = enhanced
+        except:
+            res = enhance_sharp(res)
+        out="/tmp/swapped_enhanced.jpg"
         cv2.imwrite(out,res)
         return out
     except Exception as e:
@@ -45,11 +64,11 @@ def swap_face(source_img, target_img):
         return out
 
 with gr.Blocks() as demo:
-    gr.Markdown("# FaceFusion Zero")
+    gr.Markdown("# FaceFusion Zero - Enhanced (opencv)")
     with gr.Row():
         src = gr.Image(type="filepath", label="Source")
         tgt = gr.Image(type="filepath", label="Target")
-    out = gr.Image(type="filepath", label="Result")
-    btn = gr.Button("Swap")
+    out = gr.Image(type="filepath", label="Enhanced Result")
+    btn = gr.Button("Swap + Enhance")
     btn.click(swap_face, inputs=[src, tgt], outputs=out)
 demo.launch()