Update app.py

app.py

@@ -1,18 +1,41 @@
 import streamlit as st
-from PIL import Image, ImageEnhance
+from PIL import Image, ImageEnhance, ImageFilter
 import numpy as np
 
-# Function to apply grayscale filter
+# Helper functions for filters and enhancements
 def apply_grayscale(image):
     return image.convert("L")
 
-# Function to adjust brightness
 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("Simple Image Editor")
+st.title("Advanced Image Editor")
 
 # Upload image
 uploaded_file = st.file_uploader("Choose an image", type=["jpg", "png", "jpeg"])
@@ -24,16 +47,50 @@ if uploaded_file is not None:
     # Display original image
     st.image(image, caption="Original Image", use_column_width=True)
 
-    # Select filter to apply
-    filter_option = st.selectbox("Choose a filter", ("None", "Grayscale", "Brightness"))
+    # Sidebar for feature selection
+    st.sidebar.title("Image Editing Options")
 
-    # Apply selected filter
-    if filter_option == "Grayscale":
-        image = apply_grayscale(image)
-    elif filter_option == "Brightness":
-        factor = st.slider("Adjust brightness", 0.5, 2.0, 1.0)
+    # 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)
 
@@ -46,3 +103,4 @@ if uploaded_file is not None:
             file_name="edited_image.png",
             mime="image/png"
         )
+