Update app.py

app.py

@@ -7,31 +7,31 @@ import io
 from fpdf import FPDF
 import uuid
 import datetime
+import tempfile
+import base64
 
-# App page setup
+# Setup the app page
 st.set_page_config(page_title="Image Cleaner & PDF Creator", page_icon="🖼️", layout="wide")
 
-# Set background color and slight image
-page_bg_img = """
+# Background color (light gray/white)
+page_bg_color = """
 <style>
 body {
-background-image: url('https://images.unsplash.com/photo-1542281286-9e0a16bb7366');
-background-size: cover;
-color: #FFFFFF;
+background-color: #f5f5f5;
 }
 [data-testid="stHeader"] {
-background: rgba(0,0,0,0);
+background: rgba(255, 255, 255, 0.8);
 }
 [data-testid="stSidebar"] {
-background: rgba(255, 255, 255, 0.1);
+background: rgba(255, 255, 255, 0.9);
 }
 </style>
 """
-st.markdown(page_bg_img, unsafe_allow_html=True)
+st.markdown(page_bg_color, unsafe_allow_html=True)
 
 # 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)
+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)
 
 # File uploader
 uploaded_files = st.file_uploader(
@@ -53,9 +53,9 @@ if uploaded_files:
         enhancer = ImageEnhance.Contrast(image)
         image = enhancer.enhance(1.2)  # Slight contrast
 
-        # White background to replace transparent or black
+        # White background to replace any black background
         white_bg = Image.new("RGB", image.size, (255, 255, 255))
-        white_bg.paste(image, (0, 0), mask=None)
+        white_bg.paste(image)
 
         cleaned_images.append(white_bg)
 
@@ -63,27 +63,35 @@ if uploaded_files:
     if cleaned_images:
         pdf = FPDF(unit="pt", format=[cleaned_images[0].width, cleaned_images[0].height])
 
-        for img in cleaned_images:
-            temp_buffer = io.BytesIO()
-            img.save(temp_buffer, format="JPEG")
-            temp_buffer.seek(0)
+        # Temporary directory to save images
+        with tempfile.TemporaryDirectory() as temp_dir:
+            for idx, img in enumerate(cleaned_images):
+                img_path = os.path.join(temp_dir, f"img_{idx}.jpg")
+                img.save(img_path, format="JPEG")
 
-            pdf.add_page()
-            pdf.image(temp_buffer, x=0, y=0, w=cleaned_images[0].width, h=cleaned_images[0].height)
+                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
-        timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-        pdf_output_path = f"cleaned_images_{timestamp}.pdf"
-        pdf.output(pdf_output_path)
+            # 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
+        # Load the pdf for download button
         with open(pdf_output_path, "rb") as f:
-            st.download_button(
-                label="📄 Download Cleaned PDF",
-                data=f,
-                file_name=pdf_output_path,
-                mime="application/pdf"
-            )
-
-        st.success("✅ Images cleaned and combined into PDF successfully!")
-
+            pdf_bytes = f.read()
+            b64_pdf = base64.b64encode(pdf_bytes).decode()
+
+            download_link = f"""
+                <a href="data:application/pdf;base64,{b64_pdf}" download="{pdf_output_path}" 
+                style="
+                display:inline-block;
+                padding:12px 24px;
+                font-size:18px;
+                font-weight:bold;
+                color:white;
+                background: linear-gradient(135deg, #4CAF50 0%, #81C784 100%);
+                border-radius:10px;
+                text-decoration:none;
+                text-align:center;">
+                📄 D