Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from PIL import Image, ImageEnhance, ImageFilter, ImageOps, ImageDraw, ImageFont
+from PIL import Image, ImageEnhance, ImageFilter, ImageOps, ImageDraw
 import numpy as np
 import io
 
@@ -12,7 +12,7 @@ uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
 if uploaded_file:
     # Load image
     img = Image.open(uploaded_file).convert("RGBA")
-    st.image(img, caption="Original Image", use_column_width=True)
+    st.image(img, caption="Original Image", use_container_width=True)
 
     # Sidebar Options
     st.sidebar.title("Editing Options")
@@ -38,29 +38,33 @@ if uploaded_file:
     st.sidebar.header("Background Color")
     bg_color = st.sidebar.color_picker("Pick a Background Color", "#FFFFFF")
     bg_layer = Image.new("RGBA", img_rotated.size, bg_color)
-    img_rotated = Image.alpha_composite(bg_layer, img_rotated)
+    img_with_bg = Image.alpha_composite(bg_layer, img_rotated)
+
+    # Ensure RGBA transparency
+    if img_with_bg.mode != "RGBA":
+        img_with_bg = img_with_bg.convert("RGBA")
 
     # Filters
     st.sidebar.header("Filters")
     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")
+        img_filtered = img_with_bg.convert("L").convert("RGBA")
     elif filters == "Sepia":
-        sepia_filter = np.array(img_rotated)[:, :, :3]
+        sepia_filter = np.array(img_with_bg)[:, :, :3]
         sepia = np.dot(sepia_filter, [0.393, 0.769, 0.189]).clip(0, 255).astype("uint8")
         img_filtered = Image.fromarray(sepia).convert("RGBA")
     elif filters == "Negative":
-        img_filtered = ImageOps.invert(img_rotated.convert("RGB")).convert("RGBA")
+        img_filtered = ImageOps.invert(img_with_bg.convert("RGB")).convert("RGBA")
     elif filters == "Blur":
-        img_filtered = img_rotated.filter(ImageFilter.BLUR)
+        img_filtered = img_with_bg.filter(ImageFilter.BLUR)
     elif filters == "Contour":
-        img_filtered = img_rotated.filter(ImageFilter.CONTOUR)
+        img_filtered = img_with_bg.filter(ImageFilter.CONTOUR)
     elif filters == "Detail":
-        img_filtered = img_rotated.filter(ImageFilter.DETAIL)
+        img_filtered = img_with_bg.filter(ImageFilter.DETAIL)
     elif filters == "Emboss":
-        img_filtered = img_rotated.filter(ImageFilter.EMBOSS)
+        img_filtered = img_with_bg.filter(ImageFilter.EMBOSS)
     else:
-        img_filtered = img_rotated
+        img_filtered = img_with_bg
 
     # Adjustments
     st.sidebar.header("Adjustments")
@@ -83,7 +87,7 @@ if uploaded_file:
     img_resized_rgb = img_resized.convert("RGB")
 
     # Display Final Image
-    st.image(img_resized, caption="Edited Image", use_column_width=True)
+    st.image(img_resized, caption="Edited Image", use_container_width=True)
 
     # Download Button
     st.sidebar.header("Download")