File size: 5,141 Bytes
319dbb7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
import streamlit as st
from PIL import Image, ImageDraw, ImageFont
import io
import numpy as np
def watermark_():
# Function to create a visible watermark on an image
def add_visible_watermark(image, watermark_text, size, opacity):
width, height = image.size
font_size = int(size)
try:
font = ImageFont.truetype("arial.ttf", font_size)
except IOError:
font = ImageFont.load_default()
# Create watermark layer
# Create watermark layer
watermark_layer = Image.new("RGBA", image.size, (255, 255, 255, 0))
watermark_draw = ImageDraw.Draw(watermark_layer)
# Calculate center position
text_width, text_height = watermark_draw.textbbox((0, 0), watermark_text, font=font)[2:]
x = (width - text_width) / 2
y = (height - text_height) / 2
# Add shadow
shadow_color = (0, 0, 0, int(opacity * 0.3))
watermark_draw.text((x + 3, y + 3), watermark_text, font=font, fill=shadow_color)
# Add main watermark in a slant line
watermark_color = (255, 255, 255, int(opacity))
for i in range(-width, width, text_width + 20):
watermark_draw.text((x + i, y + i), watermark_text, font=font, fill=watermark_color)
# Combine the watermark layer with the original image
combined_image = Image.alpha_composite(image.convert("RGBA"), watermark_layer).convert("RGB")
return combined_image
# Function to embed an invisible watermark in an image
def add_invisible_watermark(image, watermark_text):
binary_watermark = ''.join(format(ord(char), '08b') for char in watermark_text)
img_array = np.array(image)
flat_image = img_array.flatten()
for i in range(len(binary_watermark)):
flat_image[i] = flat_image[i] & ~1 | int(binary_watermark[i])
watermarked_image = flat_image.reshape(img_array.shape)
return Image.fromarray(watermarked_image.astype(np.uint8))
# Function to detect an invisible watermark in an image
def detect_invisible_watermark(image, watermark_length):
img_array = np.array(image)
flat_image = img_array.flatten()
binary_watermark = ''.join(str(flat_image[i] & 1) for i in range(watermark_length * 8))
watermark = ''.join(chr(int(binary_watermark[i:i + 8], 2)) for i in range(0, len(binary_watermark), 8))
return watermark
# Streamlit UI
def main():
st.title("Visible and Invisible Watermarking Tool")
col1, col2 = st.tabs(["Visible","Invisible"])
with col1:
st.header("Visible Watermark")
watermark_text = st.text_input("Enter Watermark Text", key="visible_text")
uploaded_file = st.file_uploader("Upload an Image", type=["png", "jpg", "jpeg"], key="visible_upload")
if uploaded_file and (watermark_text):
image = Image.open(uploaded_file).convert("RGB")
size = st.slider("Select Font Size", 10, 200, 50, key="font_size")
opacity = st.slider("Select Opacity", 0, 255, 128, key="opacity")
watermarked_image = add_visible_watermark(
image, watermark_text, size, opacity
)
st.image(watermarked_image, caption="Watermarked Image")
buffer = io.BytesIO()
watermarked_image.save(buffer, format="PNG")
st.download_button("Download Watermarked Image", buffer.getvalue(), "visible_watermarked_image.png")
with col2:
st.header("Invisible Watermark")
watermark_text = st.text_input("Enter Invisible Watermark Text", key="invisible_text")
uploaded_file = st.file_uploader("Upload an Image", type=["png", "jpg", "jpeg"], key="invisible_upload")
if uploaded_file and watermark_text:
image = Image.open(uploaded_file).convert("RGB")
watermarked_image = add_invisible_watermark(image, watermark_text)
st.image(watermarked_image, caption="Image with Invisible Watermark")
buffer = io.BytesIO()
watermarked_image.save(buffer, format="PNG")
st.download_button("Download Image with Invisible Watermark", buffer.getvalue(), "invisible_watermarked_image.png")
if st.checkbox("Check Invisible Watermark"):
uploaded_file = st.file_uploader("Upload an Image to Check Watermark", type=["png", "jpg", "jpeg"], key="check_upload")
watermark_length = st.number_input("Enter Watermark Length", min_value=1, step=1, key="watermark_length")
if uploaded_file and watermark_length:
image = Image.open(uploaded_file).convert("RGB")
extracted_watermark = detect_invisible_watermark(image, watermark_length)
st.write("Extracted Watermark:", extracted_watermark)
if __name__ == "__main__":
main()
|