fix

app.py

@@ -33,19 +33,18 @@ def extract_lsb(stego_pixels, edge_mask, secret_bits_length, x=2, y=1):
         for j in range(width):
             for c in range(channels):
                 if bit_index >= secret_bits_length:
-                    break
+                    return np.array(extracted_bits, dtype=np.uint8)
                 n = x if edge_mask[i, j] else y
+                if n <= 0:
+                    continue
                 value = stego_pixels[i, j, c] & ((1 << n) - 1)
                 for b in range(n):
                     extracted_bits.append((value >> b) & 1)
                 bit_index += n
-            if bit_index >= secret_bits_length:
-                break
-        if bit_index >= secret_bits_length:
-            break
 
     return np.array(extracted_bits[:secret_bits_length], dtype=np.uint8)
 
+
 # Convert bits to image
 def from_bit_array(bit_array, shape):
     expected_bits = np.prod(shape) * 8