Spaces:
Running
Running
File size: 3,893 Bytes
6fdf8d3 590d753 6fdf8d3 c00cfe7 590d753 c00cfe7 6fdf8d3 590d753 6fdf8d3 |
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 |
import streamlit as st
from PIL import Image, ImageEnhance, ImageFilter, ImageDraw, ImageFont
import io
import numpy as np
import cv2
from rembg import remove # For background removal
# Title and Description
st.title("Advanced Image Enhancer and Editor")
st.write("Enhance, edit, and apply AI-powered enhancements to your images!")
# File Upload
uploaded_file = st.file_uploader("Upload an Image", type=["jpg", "jpeg", "png"])
if uploaded_file:
# Load image
image = Image.open(uploaded_file)
st.image(image, caption="Uploaded Image", use_column_width=True)
# Sidebar Tools
st.sidebar.title("Editing Tools")
# Basic Adjustments
brightness = st.sidebar.slider("Brightness", 0.1, 3.0, 1.0)
contrast = st.sidebar.slider("Contrast", 0.1, 3.0, 1.0)
sharpness = st.sidebar.slider("Sharpness", 0.1, 3.0, 1.0)
color = st.sidebar.slider("Color Saturation", 0.1, 3.0, 1.0)
# Debugging print for color value
st.write(f"Color Value: {color}")
# Filters
if st.sidebar.checkbox("Apply Filters"):
filter_option = st.sidebar.selectbox("Select a Filter", ["None", "BLUR", "DETAIL", "EDGE_ENHANCE", "SMOOTH"])
if filter_option == "BLUR":
image = image.filter(ImageFilter.BLUR)
elif filter_option == "DETAIL":
image = image.filter(ImageFilter.DETAIL)
elif filter_option == "EDGE_ENHANCE":
image = image.filter(ImageFilter.EDGE_ENHANCE)
elif filter_option == "SMOOTH":
image = image.filter(ImageFilter.SMOOTH)
# AI Filters
if st.sidebar.checkbox("Apply AI Filters"):
ai_filter = st.sidebar.selectbox("Select AI Filter", ["Cartoonize", "Sketch"])
image_array = np.array(image)
if ai_filter == "Cartoonize":
gray = cv2.cvtColor(image_array, cv2.COLOR_BGR2GRAY)
edges = cv2.adaptiveThreshold(gray, 255, cv2.ADAPTIVE_THRESH_MEAN_C, cv2.THRESH_BINARY, 9, 9)
color = cv2.bilateralFilter(image_array, 9, 250, 250)
cartoon = cv2.bitwise_and(color, color, mask=edges)
image = Image.fromarray(cartoon)
elif ai_filter == "Sketch":
gray = cv2.cvtColor(image_array, cv2.COLOR_BGR2GRAY)
inv = 255 - gray
blur = cv2.GaussianBlur(inv, (21, 21), 0)
sketch = cv2.divide(gray, 255 - blur, scale=256)
image = Image.fromarray(sketch)
# Background Removal
if st.sidebar.checkbox("Remove Background"):
image = Image.open(io.BytesIO(remove(uploaded_file.read())))
# Watermarking
if st.sidebar.checkbox("Add Watermark"):
watermark_text = st.sidebar.text_input("Watermark Text", "My Watermark")
draw = ImageDraw.Draw(image)
font = ImageFont.load_default() # Use default font if arial.ttf doesn't work
width, height = image.size
text_width, text_height = draw.textsize(watermark_text, font)
x, y = width - text_width - 10, height - text_height - 10
draw.text((x, y), watermark_text, font=font, fill=(255, 255, 255, 128))
# Enhance Image
try:
enhancer = ImageEnhance.Brightness(image)
image = enhancer.enhance(brightness)
enhancer = ImageEnhance.Contrast(image)
image = enhancer.enhance(contrast)
enhancer = ImageEnhance.Sharpness(image)
image = enhancer.enhance(sharpness)
enhancer = ImageEnhance.Color(image)
image = enhancer.enhance(color) # Enhance color saturation
except Exception as e:
st.write(f"Error during enhancement: {e}")
# Display Enhanced Image
st.image(image, caption="Enhanced Image", use_column_width=True)
# Download Button
buf = io.BytesIO()
image.save(buf, format="PNG")
byte_im = buf.getvalue()
st.download_button("Download Enhanced Image", data=byte_im, file_name="enhanced_image.png", mime="image/png") |