Update app.py

app.py

@@ -1,71 +1,98 @@
 import streamlit as st
-from PIL import Image, ImageEnhance, ImageFilter
+from PIL import Image, ImageEnhance, ImageFilter, ImageOps
 import numpy as np
 
-# App title
-st.title("Image Editor App")
+# Title
+st.title("Advanced Image Editor")
 
-# Sidebar for options
+# Sidebar Header
 st.sidebar.title("Editing Options")
 
-# File uploader
-uploaded_file = st.sidebar.file_uploader("Upload an image", type=["jpg", "jpeg", "png", "bmp", "tiff"])
+# Upload Image
+uploaded_file = st.sidebar.file_uploader("Upload an Image", type=["jpg", "jpeg", "png"])
 
 if uploaded_file:
-    # Open the uploaded image
-    img = Image.open(uploaded_file)
-    st.image(img, caption="Uploaded Image", use_column_width=True)
-    
-    # Image Editing Options
+    # Load the image
+    img = Image.open(uploaded_file).convert("RGBA")  # Support transparency
+    st.image(img, caption="Original Image", use_column_width=True)
+
+    # Background Color Change
+    st.sidebar.header("Background Color")
+    bg_color = st.sidebar.color_picker("Pick a background color", "#FFFFFF")
+    background = Image.new("RGBA", img.size, bg_color)
+    img_with_bg = Image.alpha_composite(background, img)
+
+    # Image Transformations
     st.sidebar.header("Transformations")
-    resize_width = st.sidebar.slider("Resize Width", 50, img.width, img.width)
-    resize_height = st.sidebar.slider("Resize Height", 50, img.height, img.height)
-    rotation = st.sidebar.slider("Rotate", 0, 360, 0)
+    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)
+
+    rotation = st.sidebar.slider("Rotation (°)", 0, 360, 0)
     flip_horizontal = st.sidebar.checkbox("Flip Horizontally")
     flip_vertical = st.sidebar.checkbox("Flip Vertically")
-    
-    # Apply transformations
-    img_resized = img.resize((resize_width, resize_height))
-    img_rotated = img_resized.rotate(rotation, expand=True)
+
+    # Apply Crop
+    left = int(crop_left * img_with_bg.width / 100)
+    top = int(crop_top * img_with_bg.height / 100)
+    right = img_with_bg.width - int(crop_right * img_with_bg.width / 100)
+    bottom = img_with_bg.height - int(crop_bottom * img_with_bg.height / 100)
+    img_cropped = img_with_bg.crop((left, top, right, bottom))
+
+    # Apply Rotation and Flips
+    img_rotated = img_cropped.rotate(rotation, expand=True)
     if flip_horizontal:
         img_rotated = img_rotated.transpose(Image.FLIP_LEFT_RIGHT)
     if flip_vertical:
         img_rotated = img_rotated.transpose(Image.FLIP_TOP_BOTTOM)
 
-    # Image Enhancement
-    st.sidebar.header("Enhancements")
-    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)
-    
-    enhancer_brightness = ImageEnhance.Brightness(img_rotated).enhance(brightness)
-    enhancer_contrast = ImageEnhance.Contrast(enhancer_brightness).enhance(contrast)
-    enhancer_sharpness = ImageEnhance.Sharpness(enhancer_contrast).enhance(sharpness)
-
-    # Filters
+    # Snapchat-Like Filters
     st.sidebar.header("Filters")
-    filters = st.sidebar.selectbox("Select Filter", ["None", "BLUR", "CONTOUR", "DETAIL", "EDGE_ENHANCE", "SHARPEN"])
-    if filters == "BLUR":
-        img_filtered = enhancer_sharpness.filter(ImageFilter.BLUR)
-    elif filters == "CONTOUR":
-        img_filtered = enhancer_sharpness.filter(ImageFilter.CONTOUR)
-    elif filters == "DETAIL":
-        img_filtered = enhancer_sharpness.filter(ImageFilter.DETAIL)
-    elif filters == "EDGE_ENHANCE":
-        img_filtered = enhancer_sharpness.filter(ImageFilter.EDGE_ENHANCE)
-    elif filters == "SHARPEN":
-        img_filtered = enhancer_sharpness.filter(ImageFilter.SHARPEN)
+    filters = st.sidebar.selectbox(
+        "Select Filter",
+        ["None", "Grayscale", "Sepia", "Negative", "Blur", "Contour", "Detail", "Emboss"]
+    )
+    if filters == "Grayscale":
+        img_filtered = img_rotated.convert("L").convert("RGBA")
+    elif filters == "Sepia":
+        sepia_filter = np.array(img_rotated)[:, :, :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_rotated.convert("RGB")).convert("RGBA")
+    elif filters == "Blur":
+        img_filtered = img_rotated.filter(ImageFilter.BLUR)
+    elif filters == "Contour":
+        img_filtered = img_rotated.filter(ImageFilter.CONTOUR)
+    elif filters == "Detail":
+        img_filtered = img_rotated.filter(ImageFilter.DETAIL)
+    elif filters == "Emboss":
+        img_filtered = img_rotated.filter(ImageFilter.EMBOSS)
     else:
-        img_filtered = enhancer_sharpness
+        img_filtered = img_rotated
+
+    # Enhancements
+    st.sidebar.header("Enhancements")
+    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)
 
-    # Display final image
-    st.image(img_filtered, caption="Edited Image", use_column_width=True)
+    img_final = ImageEnhance.Brightness(img_filtered).enhance(brightness)
+    img_final = ImageEnhance.Contrast(img_final).enhance(contrast)
+    img_final = ImageEnhance.Sharpness(img_final).enhance(sharpness)
 
-    # Download option
+    # Display Final Image
+    st.image(img_final, caption="Edited Image", use_column_width=True)
+
+    # Download Option
     st.sidebar.header("Download")
-    img_filtered = img_filtered.convert("RGB")
-    img_bytes = img_filtered.tobytes()
-    st.sidebar.download_button("Download Edited Image", data=img_bytes, file_name="edited_image.jpg", mime="image/jpeg")
+    img_final = img_final.convert("RGB")
+    st.sidebar.download_button(
+        "Download Edited Image", 
+        data=img_final.tobytes(), 
+        file_name="edited_image.jpg", 
+        mime="image/jpeg"
+    )
 else:
-    st.info("Upload an image to start editing.")
-
+    st.info("Please upload an image to start editing.")