ensure image is a png

webgui.py

@@ -137,6 +137,20 @@ pipe = Audio2VideoPipeline(
     scheduler=scheduler,
 ).to("cuda", dtype=weight_dtype)
 
+def ensure_png(image_path):
+    # Load the image with Pillow
+    with Image.open(image_path) as img:
+        # Check if the image is already a PNG
+        if img.format != "PNG":
+            # Convert and save as PNG
+            png_path = os.path.splitext(image_path)[0] + ".png"
+            img.save(png_path, format="PNG")
+            print(f"Image converted to PNG and saved as {png_path}")
+            return png_path
+        else:
+            print("Image is already a PNG.")
+            return image_path
+
 def select_face(det_bboxes, probs):
     ## max face from faces that the prob is above 0.8
     ## box: xyxy
@@ -158,7 +172,9 @@ def process_video(uploaded_img, uploaded_audio, width, height, length, seed, fac
     else:
         generator = torch.manual_seed(random.randint(100, 1000000))
 
-    #### face musk prepare
+    uploaded_img = ensure_png(uploaded_img)
+
+    #### face mask prepare
     face_img = cv2.imread(uploaded_img)
     
     # Get the original dimensions