Update app.py

app.py

@@ -1,98 +1,81 @@
-### app.py ###
 import streamlit as st
-from PIL import Image, ImageEnhance, ImageFilter
+from PIL import Image, ImageEnhance, ImageFilter, ImageDraw, ImageFont
 import numpy as np
-import cv2
-
-def apply_filters(image, filter_option, brightness, contrast, saturation, tint):
-    # Convert image to numpy array for OpenCV manipulations
-    img_array = np.array(image)
-
-    # Apply selected filter
-    if filter_option == "Gray Scale":
-        image = image.convert("L")
-    elif filter_option == "Blur":
-        image = image.filter(ImageFilter.GaussianBlur(5))
-    elif filter_option == "Sharpness":
-        image = image.filter(ImageFilter.SHARPEN)
-    elif filter_option == "Shadow":
-        image = image.filter(ImageFilter.CONTOUR)
-    elif filter_option == "Magic Eraser":
-        image = image.filter(ImageFilter.SMOOTH)
-
-    # Apply tint (using OpenCV)
-    if tint != 0:
-        img_array = cv2.addWeighted(img_array, 1, np.zeros_like(img_array, img_array.dtype), 0, tint)
-        image = Image.fromarray(img_array)
-
-    # Apply adjustments
-    enhancer = ImageEnhance.Brightness(image)
-    image = enhancer.enhance(brightness)
-
-    enhancer = ImageEnhance.Contrast(image)
-    image = enhancer.enhance(contrast)
-
-    enhancer = ImageEnhance.Color(image)
-    image = enhancer.enhance(saturation)
-
-    return image
-
-def main():
-    st.title("Image Editor")
-    st.subheader("Enhance and Edit Your Images")
-
-    uploaded_image = st.file_uploader("Upload an Image", type=["jpg", "jpeg", "png"])
-
-    if uploaded_image is not None:
-        image = Image.open(uploaded_image)
-
-        st.sidebar.header("Filters & Adjustments")
-
-        # Filters & Enhancements
-        filter_option = st.sidebar.selectbox("Select a Filter", [
-            "None", "Gray Scale", "Blur", "Sharpness", "Shadow", "Magic Eraser"
-        ])
-
-        brightness = st.sidebar.slider("Brightness", 0.5, 3.0, 1.0)
-        contrast = st.sidebar.slider("Contrast", 0.5, 3.0, 1.0)
-        saturation = st.sidebar.slider("Saturation", 0.5, 3.0, 1.0)
-        tint = st.sidebar.slider("Tint Adjustment", -50, 50, 0)
-
-        # Crop Tool
-        st.sidebar.subheader("Crop")
-        crop_left = st.sidebar.slider("Left", 0, 100, 0)
-        crop_top = st.sidebar.slider("Top", 0, 100, 0)
-        crop_right = st.sidebar.slider("Right", 0, 100, 0)
-        crop_bottom = st.sidebar.slider("Bottom", 0, 100, 0)
-
-        # Display original and edited images side by side
-        col1, col2 = st.columns(2)
-
-        with col1:
-            st.image(image, caption="Original Image", use_column_width=True)
-
-        with col2:
-            # Apply cropping
-            width, height = image.size
-            left = crop_left / 100 * width
-            top = crop_top / 100 * height
-            right = width - (crop_right / 100 * width)
-            bottom = height - (crop_bottom / 100 * height)
-            cropped_image = image.crop((left, top, right, bottom))
-
-            # Apply filters and adjustments
-            edited_image = apply_filters(cropped_image, filter_option, brightness, contrast, saturation, tint)
-            st.image(edited_image, caption="Edited Image", use_column_width=True)
-
-        if st.sidebar.button("Download Edited Image"):
-            edited_image.save("edited_image.png")
-            with open("edited_image.png", "rb") as file:
-                btn = st.download_button(
-                    label="Download",
-                    data=file,
-                    file_name="edited_image.png",
-                    mime="image/png"
-                )
-
-if __name__ == "__main__":
-    main()
+import io
+
+# Function to apply filters
+def apply_filter(img, filter_name, value):
+    if filter_name == 'Grayscale':
+        return img.convert("L")
+    elif filter_name == 'Blur':
+        return img.filter(ImageFilter.GaussianBlur(radius=value))
+    elif filter_name == 'Magic Eraser':
+        img = np.array(img)
+        img[img == 255] = 0  # Magic Eraser: remove white pixels
+        return Image.fromarray(img)
+    elif filter_name == 'Sharpness':
+        enhancer = ImageEnhance.Sharpness(img)
+        return enhancer.enhance(value)
+    elif filter_name == 'Contrast':
+        enhancer = ImageEnhance.Contrast(img)
+        return enhancer.enhance(value)
+    elif filter_name == 'Saturation':
+        enhancer = ImageEnhance.Color(img)
+        return enhancer.enhance(value)
+    elif filter_name == 'Brightness':
+        enhancer = ImageEnhance.Brightness(img)
+        return enhancer.enhance(value)
+    elif filter_name == 'Tint':
+        img = np.array(img)
+        img = np.clip(img + (value * 50), 0, 255)
+        return Image.fromarray(img)
+    else:
+        return img
+
+def crop_image(img, left, top, right, bottom):
+    return img.crop((left, top, right, bottom))
+
+# Streamlit UI components
+st.title('Image Editor')
+
+# Upload Image
+uploaded_file = st.file_uploader("Upload Image", type=["jpg", "png", "jpeg"])
+
+if uploaded_file is not None:
+    # Open the uploaded image
+    img = Image.open(uploaded_file)
+
+    # Display the original image
+    st.image(img, caption="Uploaded Image", use_column_width=True)
+
+    # Filter options
+    filter_name = st.selectbox("Choose a filter", 
+                               ["None", "Grayscale", "Blur", "Magic Eraser", "Sharpness", "Contrast", "Saturation", "Brightness", "Tint"])
+    
+    if filter_name in ["Blur", "Sharpness", "Contrast", "Saturation", "Brightness"]:
+        value = st.slider(f"Adjust {filter_name}", 0.0, 2.0, 1.0)
+    elif filter_name == "Tint":
+        value = st.slider("Adjust Tint", -1.0, 1.0, 0.0)
+    else:
+        value = 0
+    
+    if filter_name != "None":
+        img = apply_filter(img, filter_name, value)
+    
+    # Cropping
+    st.write("Crop Image")
+    left = st.slider("Left", 0, img.width, 0)
+    top = st.slider("Top", 0, img.height, 0)
+    right = st.slider("Right", 0, img.width, img.width)
+    bottom = st.slider("Bottom", 0, img.height, img.height)
+    
+    img = crop_image(img, left, top, right, bottom)
+    
+    # Display updated image
+    st.image(img, caption="Edited Image", use_column_width=True)
+
+    # Download the edited image
+    buffer = io.BytesIO()
+    img.save(buffer, format="PNG")
+    buffer.seek(0)
+    st.download_button(label="Download Image", data=buffer, file_name="edited_image.png", mime="image/png")