Update utils.py

utils.py

@@ -94,8 +94,12 @@ class WatermarkProcessor:
                     for j in range(image.shape[1]):
                          for k in range(3):
                               if data_index < len(binary_secret):
-                                   # Clear LSB and add data bit
-                                   image[i, j, k] = (image[i, j, k] & ~1) | int(binary_secret[data_index])
+                                   # Get current pixel value
+                                   pixel = image[i, j, k]
+                                   # Clear the LSB by setting it to 0 and add the watermark bit
+                                   pixel = (pixel & 0xFE) | (int(binary_secret[data_index]) & 0x01)
+                                   # Ensure value stays within uint8 range
+                                   image[i, j, k] = np.uint8(pixel)
                                    data_index += 1
                               else:
                                    break
@@ -115,6 +119,14 @@ class WatermarkProcessor:
      def decode(self, image_path):
           """Decode watermark using simple LSB steganography"""
           try:
+               # Try PNG metadata method first
+               try:
+                    im = Image.open(image_path)
+                    if "TXT" in im.info:
+                         return im.info["TXT"]
+               except:
+                    pass
+
                image = cv2.imread(image_path)
                if image is None:
                     raise ValueError("Could not read image file")