|
| 1 | +import streamlit as st |
| 2 | +from PIL import Image |
| 3 | +import os |
| 4 | +from utils.watermark import add_text_watermark, add_logo_watermark |
| 5 | + |
| 6 | +st.set_page_config(page_title=" Image Watermarking Tool", layout="wide") |
| 7 | + |
| 8 | +st.title(" Image Watermarking Tool") |
| 9 | +st.write("Upload an image, add a text or logo watermark, and download the result.") |
| 10 | + |
| 11 | +os.makedirs("output", exist_ok=True) |
| 12 | + |
| 13 | +uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"]) |
| 14 | + |
| 15 | +if uploaded_file: |
| 16 | + image = Image.open(uploaded_file).convert("RGB") |
| 17 | + st.image(image, caption="Original Image", use_container_width=True) |
| 18 | + |
| 19 | + st.sidebar.header(" Settings") |
| 20 | + wm_type = st.sidebar.radio("Watermark Type", ["Text", "Logo"]) |
| 21 | + |
| 22 | + position = st.sidebar.selectbox("Position", ["Top-Left", "Top-Right", "Bottom-Left", "Bottom-Right", "Center"]) |
| 23 | + opacity = st.sidebar.slider("Opacity", 0.0, 1.0, 0.5) |
| 24 | + |
| 25 | + if wm_type == "Text": |
| 26 | + text = st.sidebar.text_input("Enter Watermark Text", "© MyBrand") |
| 27 | + font_size = st.sidebar.slider("Font Size", 20, 100, 40) |
| 28 | + |
| 29 | + if st.sidebar.button("Apply Text Watermark"): |
| 30 | + result = add_text_watermark(image, text, position, opacity, font_size) |
| 31 | + output_path = os.path.join("output", "watermarked_text.jpg") |
| 32 | + result.save(output_path) |
| 33 | + st.image(result, caption="Watermarked Image", use_container_width=True) |
| 34 | + st.download_button("Download Image", data=open(output_path, "rb"), file_name="watermarked_text.jpg") |
| 35 | + |
| 36 | + elif wm_type == "Logo": |
| 37 | + logo_file = st.sidebar.file_uploader("Upload Logo (PNG preferred)", type=["png"]) |
| 38 | + scale = st.sidebar.slider("Logo Scale", 0.05, 0.5, 0.2) |
| 39 | + |
| 40 | + if st.sidebar.button("Apply Logo Watermark"): |
| 41 | + if logo_file: |
| 42 | + logo_path = os.path.join("assets", "temp_logo.png") |
| 43 | + os.makedirs("assets", exist_ok=True) |
| 44 | + with open(logo_path, "wb") as f: |
| 45 | + f.write(logo_file.getbuffer()) |
| 46 | + |
| 47 | + result = add_logo_watermark(image, logo_path, position, opacity, scale) |
| 48 | + output_path = os.path.join("output", "watermarked_logo.jpg") |
| 49 | + result.save(output_path) |
| 50 | + st.image(result, caption="Watermarked Image", use_container_width=True) |
| 51 | + st.download_button("Download Image", data=open(output_path, "rb"), file_name="watermarked_logo.jpg") |
| 52 | + else: |
| 53 | + st.warning("Please upload a logo image.") |
0 commit comments