v1.1

app.py

@@ -64,13 +64,17 @@ def process_images(original_img, mask, logo, progress=gr.Progress()):
     """Process images: overlay logo using uploaded mask."""
     original_img = cv2.cvtColor(np.array(original_img), cv2.COLOR_RGB2BGR)
     mask = np.array(mask)
+    print(f"Logo type before conversion: {type(logo)}, dtype: {getattr(logo, 'dtype', 'N/A')}")
+    logo = np.array(logo)
+    if logo.dtype != np.uint8:
+        logo = logo.astype(np.uint8)
     logo = cv2.cvtColor(np.array(logo), cv2.COLOR_RGB2BGR)
 
 
     if len(mask.shape) == 3:
         mask = cv2.cvtColor(mask, cv2.COLOR_RGB2GRAY)
 
-    mask = (mask > 0).astype(np.uint8) * 255  # Ensure binary mask
+    mask = (mask > 0).astype(np.uint8) * 255 
 
     print(f"Mask shape: {mask.shape}, unique values: {np.unique(mask)}")