check for empty image

app.py

@@ -47,33 +47,24 @@ cfg = Config(load=True, cfg_file=config_path)
 ace_infer = ACEInference(cfg)
 
 def face_swap_app(target_img, face_img):
-    """
-    Swaps the face in the target image using the provided face image via ACE++.
+    if target_img is None or face_img is None:
+        raise ValueError("Both a target image and a face image must be provided.")
 
-    Parameters:
-      target_img: The image in which you want to swap a face.
-      face_img: The reference face image to insert.
+    # (Optional) Ensure images are in RGB
+    target_img = target_img.convert("RGB")
+    face_img = face_img.convert("RGB")
 
-    Returns:
-      The output image after applying ACE++ face swapping.
-    """
-    # For ACEInference, we pass:
-    # - reference_image: the target image,
-    # - edit_image: the new face image,
-    # - edit_mask: set to None so the image processor will create it,
-    # - prompt: "Face swap" instructs the model to perform face swapping.
-    # Other parameters (output dimensions, sampler, etc.) are set here as desired.
     output_img, edit_image, change_image, mask, seed = ace_infer(
         reference_image=target_img,
         edit_image=face_img,
-        edit_mask=None,          # No manual mask provided; let ACE++ handle it
+        edit_mask=None,          # Let ACE++ generate the mask automatically
         prompt="Face swap",
         output_height=1024,
         output_width=1024,
         sampler='flow_euler',
         sample_steps=28,
         guide_scale=50,
-        seed=-1                  # Use a random seed if not specified
+        seed=-1                  # Random seed if not provided
     )
     return output_img