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"
        )