Update app.py

app.py

@@ -1,58 +1,42 @@
 import streamlit as st
-from PIL import Image, ImageEnhance, ImageFilter, ImageOps
+from PIL import Image, ImageEnhance, ImageFilter, ImageOps, ImageDraw, ImageFont
 import numpy as np
+import io
 
-# Title
+# App Title
 st.title("Advanced Image Editor")
 
-# Sidebar Header
-st.sidebar.title("Editing Options")
-
-# Upload Image
-uploaded_file = st.sidebar.file_uploader("Upload an Image", type=["jpg", "jpeg", "png"])
+# File Uploader
+uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
 
 if uploaded_file:
-    # Load the image
-    img = Image.open(uploaded_file).convert("RGBA")  # Support transparency
+    # Load image
+    img = Image.open(uploaded_file).convert("RGBA")
     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)
+    # Sidebar Options
+    st.sidebar.title("Editing Options")
 
-    # Image Transformations
-    st.sidebar.header("Transformations")
+    # Crop Tool
+    st.sidebar.header("Crop")
     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)
+    left = int(crop_left * img.width / 100)
+    top = int(crop_top * img.height / 100)
+    right = img.width - int(crop_right * img.width / 100)
+    bottom = img.height - int(crop_bottom * img.height / 100)
+    img_cropped = img.crop((left, top, right, bottom))
 
-    rotation = st.sidebar.slider("Rotation (°)", 0, 360, 0)
-    flip_horizontal = st.sidebar.checkbox("Flip Horizontally")
-    flip_vertical = st.sidebar.checkbox("Flip Vertically")
-
-    # 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
+    # Rotation Tool
+    st.sidebar.header("Rotation")
+    rotation = st.sidebar.slider("Rotate (°)", 0, 360, 0)
     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)
 
-    # Snapchat-Like Filters
+    # Filters
     st.sidebar.header("Filters")
-    filters = st.sidebar.selectbox(
-        "Select Filter",
-        ["None", "Grayscale", "Sepia", "Negative", "Blur", "Contour", "Detail", "Emboss"]
-    )
+    filters = st.sidebar.selectbox("Choose a filter", ["None", "Grayscale", "Sepia", "Negative", "Blur", "Contour", "Detail", "Emboss"])
     if filters == "Grayscale":
         img_filtered = img_rotated.convert("L").convert("RGBA")
     elif filters == "Sepia":
@@ -72,27 +56,39 @@ if uploaded_file:
     else:
         img_filtered = img_rotated
 
-    # Enhancements
-    st.sidebar.header("Enhancements")
+    # Adjustments
+    st.sidebar.header("Adjustments")
     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)
-
+    saturation = st.sidebar.slider("Saturation", 0.5, 2.0, 1.0)
     img_final = ImageEnhance.Brightness(img_filtered).enhance(brightness)
     img_final = ImageEnhance.Contrast(img_final).enhance(contrast)
     img_final = ImageEnhance.Sharpness(img_final).enhance(sharpness)
+    img_final = ImageEnhance.Color(img_final).enhance(saturation)
+
+    # Annotations
+    st.sidebar.header("Annotations")
+    text = st.sidebar.text_input("Add Text")
+    if text:
+        font = ImageFont.load_default()
+        draw = ImageDraw.Draw(img_final)
+        draw.text((10, 10), text, fill="white", font=font)
+
+    # Resize Tool
+    st.sidebar.header("Resize")
+    new_width = st.sidebar.slider("Width", 50, img_final.width, img_final.width)
+    new_height = st.sidebar.slider("Height", 50, img_final.height, img_final.height)
+    img_resized = img_final.resize((new_width, new_height))
 
     # Display Final Image
-    st.image(img_final, caption="Edited Image", use_column_width=True)
+    st.image(img_resized, caption="Edited Image", use_column_width=True)
 
-    # Download Option
+    # Download Button
     st.sidebar.header("Download")
-    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"
-    )
+    buffer = io.BytesIO()
+    img_resized.save(buffer, format="JPEG")
+    st.sidebar.download_button("Download Edited Image", buffer.getvalue(), file_name="edited_image.jpg", mime="image/jpeg")
 else:
     st.info("Please upload an image to start editing.")
+