Update app.py

app.py

@@ -2,18 +2,18 @@
 
 import streamlit as st
 from PIL import Image, ImageEnhance
-import os
-import io
 from fpdf import FPDF
-import uuid
-import datetime
+import io
+import os
 import tempfile
 import base64
+from rembg import remove
+import datetime
 
 # Setup the app page
-st.set_page_config(page_title="Image Cleaner & PDF Creator", page_icon="🖼️", layout="wide")
+st.set_page_config(page_title="AI Image Cleaner & PDF Creator", page_icon="🧹", layout="wide")
 
-# Background color (light gray/white)
+# Light background
 page_bg_color = """
 <style>
 body {
@@ -30,8 +30,8 @@ background: rgba(255, 255, 255, 0.9);
 st.markdown(page_bg_color, unsafe_allow_html=True)
 
 # Attractive Heading
-st.markdown("<h1 style='text-align: center; color: #333333;'>✨ Image Background Cleaner & PDF Converter ✨</h1>", unsafe_allow_html=True)
-st.markdown("<h4 style='text-align: center; color: #555555;'>Upload your images, clean them & get a crystal-clear PDF easily!</h4>", unsafe_allow_html=True)
+st.markdown("<h1 style='text-align: center; color: #333333;'>🧹 AI Image Background Cleaner & PDF Converter 🖼️</h1>", unsafe_allow_html=True)
+st.markdown("<h4 style='text-align: center; color: #555555;'>Upload your images ➔ AI removes background ➔ Download crystal-clear PDF!</h4>", unsafe_allow_html=True)
 
 # File uploader
 uploaded_files = st.file_uploader(
@@ -44,18 +44,28 @@ if uploaded_files:
     cleaned_images = []
 
     for uploaded_file in uploaded_files:
-        image = Image.open(uploaded_file).convert("RGB")
+        input_image = Image.open(uploaded_file).convert("RGB")
+        
+        # Remove background using rembg AI
+        buffered = io.BytesIO()
+        input_image.save(buffered, format="PNG")
+        buffered = buffered.getvalue()
+
+        output = remove(buffered)
+
+        # Create cleaned image
+        cleaned_image = Image.open(io.BytesIO(output)).convert("RGBA")
 
-        # Light enhance image for better clarity
-        enhancer = ImageEnhance.Brightness(image)
-        image = enhancer.enhance(1.1)  # Slight brightness
+        # Put cleaned subject over white background
+        white_bg = Image.new("RGB", cleaned_image.size, (255, 255, 255))
+        white_bg.paste(cleaned_image, mask=cleaned_image.split()[3])
 
-        enhancer = ImageEnhance.Contrast(image)
-        image = enhancer.enhance(1.2)  # Slight contrast
+        # Optional slight enhance
+        enhancer = ImageEnhance.Brightness(white_bg)
+        white_bg = enhancer.enhance(1.05)
 
-        # White background to replace any black background
-        white_bg = Image.new("RGB", image.size, (255, 255, 255))
-        white_bg.paste(image)
+        enhancer = ImageEnhance.Contrast(white_bg)
+        white_bg = enhancer.enhance(1.1)
 
         cleaned_images.append(white_bg)
 
@@ -72,7 +82,7 @@ if uploaded_files:
                 pdf.add_page()
                 pdf.image(img_path, x=0, y=0, w=cleaned_images[0].width, h=cleaned_images[0].height)
 
-            # Unique file name with timestamp
+            # Unique filename
             timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
             pdf_output_path = f"cleaned_images_{timestamp}.pdf"
             pdf.output(pdf_output_path)