Make facer and fpie dependencies optional for face alignment

runners/simple_runner.py

@@ -17,7 +17,18 @@ from runners.inference_runners import FSEInferenceRunner
 
 
 def extract_mask(image_path, save_dir_path, trash=0.995):
-    from models.farl.farl import Masker
+    try:
+        from models.farl.farl import Masker
+    except ImportError:
+        print("Warning: facer module not available, skipping background mask extraction")
+        # Return a dummy mask path
+        save_dir_path = Path(save_dir_path)
+        image_path = Path(image_path)
+        mask_path = save_dir_path / (image_path.stem + "_mask.jpg")
+        # Create a simple white mask (no masking)
+        mask = Image.new("1", (1024, 1024), 1)
+        mask.save(mask_path)
+        return mask_path
 
     save_dir_path = Path(save_dir_path)
     image_path = Path(image_path)
@@ -104,11 +115,16 @@ def unalign(edited_image, unalign_dict, orig_img_pth, unaligned_path):
     PIL.Image.fromarray(unaligned.astype('uint8'), 'RGB').save("edited.png")
     PIL.Image.fromarray(np.uint8(np.clip(np.rint((1 - mask) * 255), 0, 255)), 'RGB').save("mask.jpg")
 
-    subprocess.run(
+    try:
+        subprocess.run(
             ["fpie", "-s", orig_img_pth, "-m", "mask.jpg", "-t", "edited.png", "-o", unaligned_path, "-n",
              "5000", "-b", "taichi-gpu", "-g", "src"],
             check=True
         )
+    except FileNotFoundError:
+        print("Warning: fpie command not available, skipping unalign step")
+        # Just copy the edited image as the final result
+        PIL.Image.open("edited.png").save(unaligned_path)
 
 
 class SimpleRunner: