Update app.py

app.py

@@ -34,6 +34,12 @@ if uploaded_file:
     rotation = st.sidebar.slider("Rotate (°)", 0, 360, 0)
     img_rotated = img_cropped.rotate(rotation, expand=True)
 
+    # Background Color Change
+    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)
+
     # Filters
     st.sidebar.header("Filters")
     filters = st.sidebar.selectbox("Choose a filter", ["None", "Grayscale", "Sepia", "Negative", "Blur", "Contour", "Detail", "Emboss"])
@@ -67,27 +73,22 @@ if uploaded_file:
     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))
 
+    # Convert RGBA to RGB for JPEG format
+    img_resized_rgb = img_resized.convert("RGB")
+
     # Display Final Image
     st.image(img_resized, caption="Edited Image", use_column_width=True)
 
     # Download Button
     st.sidebar.header("Download")
     buffer = io.BytesIO()
-    img_resized.save(buffer, format="JPEG")
+    img_resized_rgb.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.")