Update app.py

app.py

@@ -1,74 +1,12 @@
-import streamlit as st
-from PIL import Image
-import io
-from rembg import remove
+import uuid
 
-# --- Custom Background ---
-page_bg_img = '''
-<style>
-.stApp {
-background-image: 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 {
-color: #ffffff;
-text-shadow: 2px 2px 4px #000000;
-}
-</style>
-'''
-st.markdown(page_bg_img, unsafe_allow_html=True)
+# In your PDF generation function
+unique_filename = f"{uuid.uuid4()}.pdf"  # This generates a random unique filename
+pdf_output_path = f"/tmp/{unique_filename}"
 
-# --- App Title ---
-st.markdown("<h1 style='text-align: center; font-size: 50px;'>🧹✨ Clean Your Images & Create a Stunning PDF ✨🧹</h1>", unsafe_allow_html=True)
+# Save the PDF with this path
+pdf.output(pdf_output_path)
 
-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... ✨'):
-        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:
-        # Save images into a high-quality PDF
-        pdf_bytes = io.BytesIO()
-        image_list = [img.convert('RGB') for img in cleaned_images]  # Ensure correct mode
-        first_image = image_list.pop(0)
-
-        first_image.save(
-            pdf_bytes,
-            format="PDF",
-            save_all=True,
-            append_images=image_list,
-            resolution=300.0,  # 300 DPI for quality
-            quality=100,       # Maximum quality
-        )
-        pdf_bytes.seek(0)
-
-        st.success("✅ Images cleaned and combined into a high-quality PDF!")
-        st.download_button(
-            label="💾 Download Final PDF",
-            data=pdf_bytes,
-            file_name="cleaned_images.pdf",
-            mime="application/pdf"
-        )
+# 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")