Update app.py

app.py

@@ -1,12 +1,80 @@
+import streamlit as st
+from PIL import Image
+import io
 import uuid
+from rembg import remove
+from fpdf import FPDF
+import os
 
-# In your PDF generation function
-unique_filename = f"{uuid.uuid4()}.pdf"  # This generates a random unique filename
-pdf_output_path = f"/tmp/{unique_filename}"
+# --- Custom Background with Semi-transparent Overlay ---
+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");
+background-size: cover;
+background-position: center;
+background-attachment: fixed;
+}
+h1, h2, h3, p, label, div {
+color: #000000;
+}
+</style>
+'''
+st.markdown(page_bg_img, unsafe_allow_html=True)
 
-# Save the PDF with this path
-pdf.output(pdf_output_path)
+# --- App Title ---
+st.markdown("<h1 style='text-align: center; font-size: 50px;'>🧹✨ Clean Your Images & Create a Stunning PDF ✨🧹</h1>", unsafe_allow_html=True)
 
-# Provide this file to the user for download
-with open(pdf_output_path, "rb") as f:
-    st.download_button("Download PDF", f, file_name="enhanced_images.pdf", mime="application/pdf")
+st.write("#### 📸 Supported formats: **jpg, jpeg, png, webp**")
+
+# --- Upload Section ---
+uploaded_files = st.file_uploader("Upload multiple images here", type=["jpg", "jpeg", "png", "webp"], accept_multiple_files=True)
+
+# --- 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)
+
+    if cleaned_images:
+        # Create a new PDF
+        pdf = FPDF(unit="mm", format="A4")
+        
+        for img in cleaned_images:
+            # Save cleaned image temporarily
+            temp_path = f"/tmp/{uuid.uuid4()}.png"
+            img.save(temp_path)
+            
+            # Add a new page
+            pdf.add_page()
+            
+            # Calculate width/height for A4 size
+            pdf.image(temp_path, x=10, y=10, w=190)  # Adjust image width
+            
+            # 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 in