import streamlit as st
from PIL import Image, ImageEnhance

# App Title
st.title("Simple Image Enhancement App")
st.write("Upload an image, enhance it, and download the result!")

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

if uploaded_file:
    # Open the uploaded image
    image = Image.open(uploaded_file)
    st.image(image, caption="Uploaded Image", use_column_width=True)

    # Enhancements
    st.write("### Enhance the Image")
    
    # Brightness
    brightness = st.slider("Brightness", 0.5, 3.0, 1.0)
    enhancer = ImageEnhance.Brightness(image)
    enhanced_image = enhancer.enhance(brightness)
    
    # Contrast
    contrast = st.slider("Contrast", 0.5, 3.0, 1.0)
    enhancer = ImageEnhance.Contrast(enhanced_image)
    enhanced_image = enhancer.enhance(contrast)
    
    # Sharpness
    sharpness = st.slider("Sharpness", 0.5, 3.0, 1.0)
    enhancer = ImageEnhance.Sharpness(enhanced_image)
    enhanced_image = enhancer.enhance(sharpness)

    # Display the enhanced image
    st.image(enhanced_image, caption="Enhanced Image", use_column_width=True)

    # Download button
    st.write("### Download the Enhanced Image")
    download_button = st.download_button(
        label="Download Image",
        data=enhanced_image.tobytes(),
        file_name="enhanced_image.png",
        mime="image/png"
    )