Update app.py

app.py

@@ -1,90 +1,89 @@
+# app.py
+
 import streamlit as st
-from PIL import Image
+from PIL import Image, ImageEnhance
+import os
 import io
-import uuid
-from rembg import remove
 from fpdf import FPDF
-import os
+import uuid
+import datetime
 
-# --- Custom Background with Semi-transparent Overlay ---
-page_bg_img = '''
+# App page setup
+st.set_page_config(page_title="Image Cleaner & PDF Creator", page_icon="🖼️", layout="wide")
+
+# Set background color and slight image
+page_bg_img = """
 <style>
-.stApp {
-background: linear-gradient(rgba(255,255,255,0.7), rgba(255,255,255,0.7)), 
-url("https://images.unsplash.com/photo-1587620962725-abab7fe55159?ixlib=rb-4.0.3&auto=format&fit=crop&w=1350&q=80");
+body {
+background-image: url('https://images.unsplash.com/photo-1542281286-9e0a16bb7366');
 background-size: cover;
-background-position: center;
-background-attachment: fixed;
+color: #FFFFFF;
+}
+[data-testid="stHeader"] {
+background: rgba(0,0,0,0);
 }
-h1, h2, h3, p, label, div {
-color: #000000;
+[data-testid="stSidebar"] {
+background: rgba(255, 255, 255, 0.1);
 }
 </style>
-'''
+"""
 st.markdown(page_bg_img, unsafe_allow_html=True)
 
-# --- App Title ---
-st.markdown("<h1 style='text-align: center; font-size: 50px;'>🧹✨ Clean Your Images & Create a Stunning PDF ✨🧹</h1>", unsafe_allow_html=True)
-
-st.write("#### 📸 Supported formats: **jpg, jpeg, png, webp**")
+# Attractive Heading
+st.markdown("<h1 style='text-align: center; color: #FFFFFF;'>✨ Image Background Cleaner & PDF Converter ✨</h1>", unsafe_allow_html=True)
+st.markdown("<h4 style='text-align: center; color: #CCCCCC;'>Upload your images, clean backgrounds & download a single combined PDF</h4>", unsafe_allow_html=True)
 
-# --- Upload Section ---
+# File uploader
 uploaded_files = st.file_uploader(
-    "Upload multiple images here",
-    type=["jpg", "jpeg", "png", "webp"],
-    accept_multiple_files=True
+    "Upload multiple images (PNG, JPG, JPEG formats supported)", 
+    accept_multiple_files=True, 
+    type=["png", "jpg", "jpeg"]
 )
 
-# --- Functions ---
-def clean_image(image_bytes):
-    # Remove background
-    img_no_bg = remove(image_bytes)
-    img = Image.open(io.BytesIO(img_no_bg)).convert("RGBA")
-    
-    # Create white background
-    white_bg = Image.new("RGBA", img.size, (255, 255, 255, 255))
-    final_img = Image.alpha_composite(white_bg, img).convert("RGB")
-    
-    return final_img
-
-# --- Main Logic ---
 if uploaded_files:
     cleaned_images = []
-    with st.spinner('✨ Cleaning images... Please wait... ✨'):
-        for uploaded_file in uploaded_files:
-            img_bytes = uploaded_file.read()
-            cleaned_img = clean_image(img_bytes)
-            cleaned_images.append(cleaned_img)
 
+    for uploaded_file in uploaded_files:
+        image = Image.open(uploaded_file).convert("RGB")
+
+        # Light enhance image for better clarity
+        enhancer = ImageEnhance.Brightness(image)
+        image = enhancer.enhance(1.1)  # Slight brightness
+
+        enhancer = ImageEnhance.Contrast(image)
+        image = enhancer.enhance(1.2)  # Slight contrast
+
+        # White background to replace transparent or black
+        white_bg = Image.new("RGB", image.size, (255, 255, 255))
+        white_bg.paste(image, (0, 0), mask=None)
+
+        cleaned_images.append(white_bg)
+
+    # Convert all images into a single PDF
     if cleaned_images:
-        # Create a new PDF
-        pdf = FPDF(unit="mm", format="A4")
-        
+        pdf = FPDF(unit="pt", format=[cleaned_images[0].width, cleaned_images[0].height])
+
         for img in cleaned_images:
-            # Save cleaned image temporarily
-            temp_path = f"/tmp/{uuid.uuid4()}.png"
-            img.save(temp_path)
-            
-            # Add a new page
+            temp_buffer = io.BytesIO()
+            img.save(temp_buffer, format="JPEG")
+            temp_buffer.seek(0)
+
             pdf.add_page()
-            
-            # Calculate width/height for A4 size
-            pdf.image(temp_path, x=10, y=10, w=190)
-            
-            # Remove temporary file
-            os.remove(temp_path)
-
-        # Save final PDF
-        unique_filename = f"{uuid.uuid4()}.pdf"
-        pdf_path = f"/tmp/{unique_filename}"
-        pdf.output(pdf_path)
-
-        # Download Button
-        with open(pdf_path, "rb") as f:
-            st.success("✅ Images cleaned and combined into a high-quality PDF!")
+            pdf.image(temp_buffer, x=0, y=0, w=cleaned_images[0].width, h=cleaned_images[0].height)
+
+        # Unique file name with timestamp
+        timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
+        pdf_output_path = f"cleaned_images_{timestamp}.pdf"
+        pdf.output(pdf_output_path)
+
+        # Download button
+        with open(pdf_output_path, "rb") as f:
             st.download_button(
-                label="💾 Download Final PDF",
+                label="📄 Download Cleaned PDF",
                 data=f,
-                file_name="cleaned_images.pdf",
+                file_name=pdf_output_path,
                 mime="application/pdf"
             )
+
+        st.success("✅ Images cleaned and combined into PDF successfully!")
+