Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from PIL import Image, ImageEnhance, ImageFilter
+from PIL import Image, ImageEnhance, ImageFilter, ImageDraw, ImageOps
 import numpy as np
 import cv2
 import io
@@ -61,76 +61,31 @@ def apply_edge_detection(image):
     edges = cv2.Canny(img, 100, 200)
     return Image.fromarray(edges)
 
+# Function to add border
+def add_border(image, border_color, border_width):
+    # Adding border using PIL ImageOps
+    return ImageOps.expand(image, border=border_width, fill=border_color)
+
+# Function to add frame
+def add_frame(image, frame_style="Simple"):
+    width, height = image.size
+    if frame_style == "Simple":
+        # Add simple black border
+        frame_width = 30
+        return ImageOps.expand(image, border=frame_width, fill="black")
+    elif frame_style == "Ornate":
+        # Add ornate frame (you can customize with more ornate styles)
+        frame_width = 50
+        return ImageOps.expand(image, border=frame_width, fill="gold")
+    elif frame_style == "Rounded":
+        # Add rounded frame (with radius)
+        frame_width = 30
+        image_with_frame = ImageOps.expand(image, border=frame_width, fill="blue")
+        return image_with_frame.convert("RGBA").rotate(0, resample=Image.BICUBIC)
+
 # Function to convert PIL image to bytes for download
 def pil_to_bytes(image, format="JPEG"):
     img_byte_arr = io.BytesIO()
     image.save(img_byte_arr, format=format)
     img_byte_arr.seek(0)
-    return img_byte_arr
-
-# Streamlit app
-def main():
-    st.title("Advanced Image Editor")
-
-    # Upload image
-    uploaded_image = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-
-    if uploaded_image is not None:
-        # Open the uploaded image using PIL
-        image = Image.open(uploaded_image)
-
-        # Resize the image for better side-by-side display
-        original_image_resized = image.resize((400, int(image.height * 400 / image.width)))
-
-        # Apply filters
-        st.sidebar.title("Adjust Filters")
-        
-        filter_option = st.sidebar.selectbox("Choose a filter", ("None", "Grayscale", "Brightness", "Contrast", "Saturation", "Sharpness", "Sepia", "Vignette", "Blur", "Edge Detection"))
-        
-        if filter_option == "Grayscale":
-            image = apply_grayscale(image)
-        elif filter_option == "Brightness":
-            brightness_factor = st.sidebar.slider("Brightness Factor", 0.1, 2.0, 1.0)
-            image = adjust_brightness(image, brightness_factor)
-        elif filter_option == "Contrast":
-            contrast_factor = st.sidebar.slider("Contrast Factor", 0.1, 2.0, 1.0)
-            image = adjust_contrast(image, contrast_factor)
-        elif filter_option == "Saturation":
-            saturation_factor = st.sidebar.slider("Saturation Factor", 0.1, 2.0, 1.0)
-            image = adjust_saturation(image, saturation_factor)
-        elif filter_option == "Sharpness":
-            sharpness_factor = st.sidebar.slider("Sharpness Factor", 0.1, 2.0, 1.0)
-            image = apply_sharpness(image, sharpness_factor)
-        elif filter_option == "Sepia":
-            image = apply_sepia(image)
-        elif filter_option == "Vignette":
-            image = apply_vignette(image)
-        elif filter_option == "Blur":
-            blur_factor = st.sidebar.slider("Blur Radius", 0.1, 10.0, 2.0)
-            image = apply_blur(image, blur_factor)
-        elif filter_option == "Edge Detection":
-            image = apply_edge_detection(image)
-
-        # Resize the filtered image for display
-        modified_image_resized = image.resize((400, int(image.height * 400 / image.width)))
-
-        # Create two columns to display images side by side
-        col1, col2 = st.columns(2)
-
-        # Display the original image in the first column
-        with col1:
-            st.image(original_image_resized, caption="Original Image", use_container_width=True)
-
-        # Display the modified image in the second column
-        with col2:
-            st.image(modified_image_resized, caption="Modified Image", use_container_width=True)
-
-        # Convert the edited image to bytes for download
-        img_bytes = pil_to_bytes(image)
-
-        # Button to download the edited image below the modified image
-        with col2:
-            st.download_button("Download Edited Image", img_bytes, "edited_image.jpg", "image/jpeg")
-
-if __name__ == "__main__":
-    main()
+    return img