Spaces:
Sleeping
Sleeping
File size: 3,408 Bytes
c3f6167 c28dd2f c3f6167 c28dd2f c3f6167 c28dd2f c3f6167 c28dd2f c3f6167 c28dd2f c3f6167 c28dd2f c3f6167 c28dd2f c3f6167 c28dd2f |
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 |
import streamlit as st
from PIL import Image, ImageEnhance, ImageFilter
import numpy as np
# Helper functions for filters and enhancements
def apply_grayscale(image):
return image.convert("L")
def adjust_brightness(image, factor):
enhancer = ImageEnhance.Brightness(image)
return enhancer.enhance(factor)
def adjust_contrast(image, factor):
enhancer = ImageEnhance.Contrast(image)
return enhancer.enhance(factor)
def adjust_saturation(image, factor):
enhancer = ImageEnhance.Color(image)
return enhancer.enhance(factor)
def apply_blur(image, radius):
return image.filter(ImageFilter.GaussianBlur(radius))
def apply_edge_detection(image):
return image.filter(ImageFilter.FIND_EDGES)
def rotate_image(image, angle):
return image.rotate(angle, expand=True)
def flip_image(image, direction):
if direction == "Horizontal":
return image.transpose(Image.FLIP_LEFT_RIGHT)
elif direction == "Vertical":
return image.transpose(Image.FLIP_TOP_BOTTOM)
return image
# Streamlit UI
st.title("Advanced Image Editor")
# Upload image
uploaded_file = st.file_uploader("Choose an image", type=["jpg", "png", "jpeg"])
if uploaded_file is not None:
# Open image
image = Image.open(uploaded_file)
# Display original image
st.image(image, caption="Original Image", use_column_width=True)
# Sidebar for feature selection
st.sidebar.title("Image Editing Options")
# Basic transformations
if st.sidebar.checkbox("Crop"):
st.write("Crop feature will be implemented here.")
if st.sidebar.checkbox("Resize"):
width = st.sidebar.number_input("Width", min_value=10, value=image.width)
height = st.sidebar.number_input("Height", min_value=10, value=image.height)
image = image.resize((width, height))
if st.sidebar.checkbox("Rotate"):
angle = st.sidebar.slider("Angle", 0, 360, 0)
image = rotate_image(image, angle)
if st.sidebar.checkbox("Flip"):
flip_direction = st.sidebar.radio("Direction", ("Horizontal", "Vertical"))
image = flip_image(image, flip_direction)
# Enhancements
if st.sidebar.checkbox("Adjust Brightness"):
factor = st.sidebar.slider("Brightness", 0.5, 2.0, 1.0)
image = adjust_brightness(image, factor)
if st.sidebar.checkbox("Adjust Contrast"):
factor = st.sidebar.slider("Contrast", 0.5, 2.0, 1.0)
image = adjust_contrast(image, factor)
if st.sidebar.checkbox("Adjust Saturation"):
factor = st.sidebar.slider("Saturation", 0.5, 2.0, 1.0)
image = adjust_saturation(image, factor)
# Filters
if st.sidebar.checkbox("Grayscale"):
image = apply_grayscale(image)
if st.sidebar.checkbox("Blur"):
radius = st.sidebar.slider("Blur Radius", 1, 10, 2)
image = apply_blur(image, radius)
if st.sidebar.checkbox("Edge Detection"):
image = apply_edge_detection(image)
# Display edited image
st.image(image, caption="Edited Image", use_column_width=True)
# Provide option to download the edited image
with st.spinner('Preparing your image for download...'):
image.save("edited_image.png")
st.download_button(
label="Download Edited Image",
data=open("edited_image.png", "rb").read(),
file_name="edited_image.png",
mime="image/png"
)
|