Update utils.py

utils.py

@@ -64,8 +64,6 @@ class WatermarkProcessor:
           except Exception as e:
                return im_name, f"Error adding watermark: {str(e)}"
 
-# utils.py의 encode와 decode 메서드만 수정
-
      def encode(self, image_path, watermark_text, metadata=None):
           """Encode watermark using simple LSB steganography"""
           try:
@@ -97,8 +95,12 @@ class WatermarkProcessor:
                     for j in range(image.shape[1]):
                          for k in range(3):
                               if data_index < len(binary_data):
-                                   # Replace LSB with watermark bit
-                                   image[i, j, k] = np.uint8((image[i, j, k] & ~1) | int(binary_data[data_index]))
+                                   pixel = int(image[i, j, k])
+                                   # Clear the LSB
+                                   pixel = pixel & 0xFE  # Clear last bit
+                                   # Set the LSB according to our data
+                                   pixel = pixel | (int(binary_data[data_index]) & 1)
+                                   image[i, j, k] = np.uint8(pixel)
                                    data_index += 1
                               else:
                                    break
@@ -161,7 +163,6 @@ class WatermarkProcessor:
           except Exception as e:
                return f"Error in decoding: {str(e)}"
 
-              
      def analyze_quality(self, original_path, watermarked_path):
           """Analyze watermark quality"""
           try: