import streamlit as st
from PIL import Image, ImageEnhance, ImageFilter, ImageOps, ImageDraw
import numpy as np
import io

# App Title
st.title("Advanced Image Editor")

# File Uploader
uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])

if uploaded_file:
    # Load image
    img = Image.open(uploaded_file).convert("RGBA")
    st.image(img, caption="Original Image", use_container_width=True)

    # Sidebar Options
    st.sidebar.title("Editing Options")

    # Crop Tool
    st.sidebar.header("Crop")
    crop_left = st.sidebar.slider("Crop Left (%)", 0, 50, 0)
    crop_top = st.sidebar.slider("Crop Top (%)", 0, 50, 0)
    crop_right = st.sidebar.slider("Crop Right (%)", 0, 50, 0)
    crop_bottom = st.sidebar.slider("Crop Bottom (%)", 0, 50, 0)
    left = int(crop_left * img.width / 100)
    top = int(crop_top * img.height / 100)
    right = img.width - int(crop_right * img.width / 100)
    bottom = img.height - int(crop_bottom * img.height / 100)
    img_cropped = img.crop((left, top, right, bottom))

    # Rotation Tool
    st.sidebar.header("Rotation")
    rotation = st.sidebar.slider("Rotate (°)", 0, 360, 0)
    img_rotated = img_cropped.rotate(rotation, expand=True)

    # Background Color Change
    st.sidebar.header("Background Color")
    bg_color = st.sidebar.color_picker("Pick a Background Color", "#FFFFFF")
    bg_layer = Image.new("RGBA", img_rotated.size, bg_color)
    img_with_bg = Image.alpha_composite(bg_layer, img_rotated)

    # Ensure RGBA transparency
    if img_with_bg.mode != "RGBA":
        img_with_bg = img_with_bg.convert("RGBA")

    # Filters
    st.sidebar.header("Filters")
    filters = st.sidebar.selectbox("Choose a filter", ["None", "Grayscale", "Sepia", "Negative", "Blur", "Contour", "Detail", "Emboss"])
    if filters == "Grayscale":
        img_filtered = img_with_bg.convert("L").convert("RGBA")
    elif filters == "Sepia":
        sepia_filter = np.array(img_with_bg)[:, :, :3]
        sepia = np.dot(sepia_filter, [0.393, 0.769, 0.189]).clip(0, 255).astype("uint8")
        img_filtered = Image.fromarray(sepia).convert("RGBA")
    elif filters == "Negative":
        img_filtered = ImageOps.invert(img_with_bg.convert("RGB")).convert("RGBA")
    elif filters == "Blur":
        img_filtered = img_with_bg.filter(ImageFilter.BLUR)
    elif filters == "Contour":
        img_filtered = img_with_bg.filter(ImageFilter.CONTOUR)
    elif filters == "Detail":
        img_filtered = img_with_bg.filter(ImageFilter.DETAIL)
    elif filters == "Emboss":
        img_filtered = img_with_bg.filter(ImageFilter.EMBOSS)
    else:
        img_filtered = img_with_bg

    # Adjustments
    st.sidebar.header("Adjustments")
    brightness = st.sidebar.slider("Brightness", 0.5, 2.0, 1.0)
    contrast = st.sidebar.slider("Contrast", 0.5, 2.0, 1.0)
    sharpness = st.sidebar.slider("Sharpness", 0.5, 2.0, 1.0)
    saturation = st.sidebar.slider("Saturation", 0.5, 2.0, 1.0)
    img_final = ImageEnhance.Brightness(img_filtered).enhance(brightness)
    img_final = ImageEnhance.Contrast(img_final).enhance(contrast)
    img_final = ImageEnhance.Sharpness(img_final).enhance(sharpness)
    img_final = ImageEnhance.Color(img_final).enhance(saturation)

    # Resize Tool
    st.sidebar.header("Resize")
    new_width = st.sidebar.slider("Width", 50, img_final.width, img_final.width)
    new_height = st.sidebar.slider("Height", 50, img_final.height, img_final.height)
    img_resized = img_final.resize((new_width, new_height))

    # Convert RGBA to RGB for JPEG format
    img_resized_rgb = img_resized.convert("RGB")

    # Display Final Image
    st.image(img_resized, caption="Edited Image", use_container_width=True)

    # Download Button
    st.sidebar.header("Download")
    buffer = io.BytesIO()
    img_resized_rgb.save(buffer, format="JPEG")
    st.sidebar.download_button("Download Edited Image", buffer.getvalue(), file_name="edited_image.jpg", mime="image/jpeg")
else:
    st.info("Please upload an image to start editing.")