Update app.py

app.py

@@ -2,23 +2,40 @@ import streamlit as st
 from PIL import Image, ImageEnhance, ImageFilter
 import numpy as np
 import cv2
+import io
 
 # Title
-st.title("Manual iPhone-like Photo Editor")
+st.title("Interactive Photo Editor - iPhone-like Features")
 
-# Upload image
-uploaded_file = st.file_uploader("Choose an image...", type=["png", "jpg", "jpeg"])
+# Sidebar: Upload image and edit controls
+st.sidebar.title("Edit Controls")
+uploaded_file = st.sidebar.file_uploader("Upload an Image", type=["png", "jpg", "jpeg"])
 
+# Function to save image as a downloadable link
+def get_image_download_link(img: Image, filename: str):
+    img.save(filename)
+    with open(filename, "rb") as f:
+        img_bytes = f.read()
+    download_link = st.sidebar.download_button(
+        label="Download Edited Image",
+        data=img_bytes,
+        file_name=filename,
+        mime="image/jpeg"
+    )
+    return download_link
+
+# Initialize image if uploaded
 if uploaded_file is not None:
-    # Open image using PIL
     image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
+    st.sidebar.image(image, caption="Uploaded Image", use_column_width=True)
 
-    # Sidebar for editing options
-    st.sidebar.title("Edit Image")
+    # Image Edit Options
+    filter_option = st.sidebar.selectbox("Choose a Filter", ["None", "Vivid", "Warm", "Cool", "Black & White", "Sepia"])
+    brightness = st.sidebar.slider("Adjust Brightness", 0.0, 2.0, 1.0)
+    contrast = st.sidebar.slider("Adjust Contrast", 0.0, 2.0, 1.0)
+    saturation = st.sidebar.slider("Adjust Saturation", 0.0, 2.0, 1.0)
 
-    # Apply Filters (Manual Control)
-    filter_option = st.sidebar.selectbox("Choose a filter", ["None", "Vivid", "Warm", "Cool", "Black & White", "Sepia"])
+    # Apply chosen filters
     if filter_option == "Vivid":
         image = image.convert("RGB")
         np_img = np.array(image)
@@ -41,56 +58,51 @@ if uploaded_file is not None:
         sepia_filter = np.array([[0.393, 0.769, 0.189], [0.349, 0.686, 0.168], [0.272, 0.534, 0.131]])
         image = Image.fromarray(np.dot(img_array[...,:3], sepia_filter.T).clip(0, 255).astype(np.uint8))
 
-    # Manual Adjustments (Brightness, Contrast, Saturation)
-    if st.sidebar.checkbox("Adjust Brightness"):
-        brightness = st.sidebar.slider("Brightness", 0.0, 2.0, 1.0)
+    # Apply manual adjustments
+    if brightness != 1.0:
         enhancer = ImageEnhance.Brightness(image)
         image = enhancer.enhance(brightness)
 
-    if st.sidebar.checkbox("Adjust Contrast"):
-        contrast = st.sidebar.slider("Contrast", 0.0, 2.0, 1.0)
+    if contrast != 1.0:
         enhancer = ImageEnhance.Contrast(image)
         image = enhancer.enhance(contrast)
 
-    if st.sidebar.checkbox("Adjust Saturation"):
-        saturation = st.sidebar.slider("Saturation", 0.0, 2.0, 1.0)
+    if saturation != 1.0:
         enhancer = ImageEnhance.Color(image)
         image = enhancer.enhance(saturation)
 
-    # Manual Cropping
-    if st.sidebar.checkbox("Crop Image"):
+    # Sidebar for cropping, resizing, rotating
+    crop_enabled = st.sidebar.checkbox("Enable Crop", False)
+    if crop_enabled:
         left = st.sidebar.slider("Left", 0, image.width, 0)
         top = st.sidebar.slider("Top", 0, image.height, 0)
         right = st.sidebar.slider("Right", left, image.width, image.width)
         bottom = st.sidebar.slider("Bottom", top, image.height, image.height)
         image = image.crop((left, top, right, bottom))
 
-    # Resize Image (Manual control of width and height)
-    if st.sidebar.checkbox("Resize Image"):
+    resize_enabled = st.sidebar.checkbox("Enable Resize", False)
+    if resize_enabled:
         width = st.sidebar.slider("Width", 50, 1000, image.width)
         height = st.sidebar.slider("Height", 50, 1000, image.height)
         image = image.resize((width, height))
 
-    # Sharpen Image (Manual control over sharpening)
-    if st.sidebar.checkbox("Sharpen Image"):
-        image = image.filter(ImageFilter.SHARPEN)
-
     # Rotate/Flip Image
-    if st.sidebar.checkbox("Rotate Image"):
-        angle = st.sidebar.slider("Angle", 0, 360, 0)
+    rotate_enabled = st.sidebar.checkbox("Enable Rotation", False)
+    if rotate_enabled:
+        angle = st.sidebar.slider("Rotation Angle", 0, 360, 0)
         image = image.rotate(angle)
 
-    if st.sidebar.checkbox("Flip Image"):
+    flip_enabled = st.sidebar.checkbox("Flip Image", False)
+    if flip_enabled:
         flip_option = st.sidebar.radio("Flip Option", ["Horizontal", "Vertical"])
         if flip_option == "Horizontal":
             image = image.transpose(Image.FLIP_LEFT_RIGHT)
         else:
             image = image.transpose(Image.FLIP_TOP_BOTTOM)
 
-    # Show the edited image
+    # Image preview in the main area
     st.image(image, caption="Edited Image", use_column_width=True)
 
-    # Save the edited image
-    if st.sidebar.button("Save Image"):
-        image.save("edited_image.jpg")
-        st.success("Image saved as 'edited_image.jpg'")
+    # Provide download option
+    get_image_download_link(image, "edited_image.jpg")
+