Update app.py

app.py

@@ -1,113 +1,134 @@
 import streamlit as st
-from PIL import Image, ImageEnhance
-from io import BytesIO
-import os
-import cv2
+from PIL import Image, ImageEnhance, ImageOps
 import numpy as np
-from fpdf import FPDF
-from streamlit_cropper import st_cropper
+import io
 import zipfile
+import os
+import datetime
+import cv2
 
-# --- Set Page Config ---
-st.set_page_config(page_title="Smart ID Scanner", page_icon="🪪", layout="centered")
+# Set page configuration
+st.set_page_config(
+    page_title="📄 Smart Doc Scanner",
+    page_icon="🖼️",
+    layout="centered",
+)
 
-# --- Add Background Style ---
-page_bg_img = """
+# Custom CSS for background and buttons
+page_bg_img = '''
 <style>
-body {
-background: linear-gradient(to right, #ece9e6, #ffffff);
+.stApp {
+background-image: url("https://images.unsplash.com/photo-1524995997946-a1c2e315a42f");
+background-size: cover;
+background-position: center;
+}
+h1 {
+color: #ffffff;
+text-align: center;
+text-shadow: 2px 2px #000000;
 }
 .stButton>button {
 color: white;
-background-color: #4CAF50;
-border: none;
-padding: 0.5em 1em;
-border-radius: 5px;
-}
-.stHeader, .stText, .stMarkdown {
-color: #333333;
+background-color: #0078FF;
+border-radius: 10px;
 }
 </style>
-"""
+'''
+
 st.markdown(page_bg_img, unsafe_allow_html=True)
 
-# --- App Heading ---
-st.markdown("<h1 style='text-align: center; color: #2E86C1;'>📄 Smart ID & Document Scanner</h1>", unsafe_allow_html=True)
-st.write("---")
+st.markdown("<h1>📄 Smart Doc Scanner</h1>", unsafe_allow_html=True)
 
-# --- Upload Images ---
-uploaded_files = st.file_uploader("Upload ID/Document Images 📷 (All formats supported)", type=['png', 'jpg', 'jpeg'], accept_multiple_files=True)
+# Upload multiple images
+uploaded_files = st.file_uploader(
+    "Upload Document Images (JPEG, PNG, JPG)", 
+    type=["jpg", "jpeg", "png"], 
+    accept_multiple_files=True
+)
 
 if uploaded_files:
-    cropped_images = []
+    enhance_option = st.checkbox("✨ Enhance Images (Scanner Effect)", value=True)
+    manual_crop = st.checkbox("✂️ Manual Crop")
+    auto_crop = st.checkbox("⚡ Auto Crop Background")
+    rotate_left = st.button("⏪ Rotate Left")
+    rotate_right = st.button("⏩ Rotate Right")
+    
+    processed_images = []
+
     for uploaded_file in uploaded_files:
         image = Image.open(uploaded_file).convert("RGB")
-        
-        st.subheader(f"Crop & Adjust: {uploaded_file.name}")
 
-        # --- Free style manual crop ---
-        crop_box = st_cropper(image, realtime_update=True, box_color='#FF9900', aspect_ratio=None)
-        cropped_img = crop_box.copy()
-        
-        # --- Before / After Preview ---
-        st.write("### Preview:")
-        col1, col2 = st.columns(2)
-        with col1:
-            st.image(image, caption="Original", use_container_width=True)
-        with col2:
-            st.image(cropped_img, caption="Cropped", use_container_width=True)
+        # Rotate manually if needed
+        if rotate_left:
+            image = image.rotate(90, expand=True)
+        if rotate_right:
+            image = image.rotate(-90, expand=True)
+
+        # Auto-crop (border removal)
+        if auto_crop:
+            open_cv_image = np.array(image)
+            gray = cv2.cvtColor(open_cv_image, cv2.COLOR_RGB2GRAY)
+            _, thresh = cv2.threshold(gray, 250, 255, cv2.THRESH_BINARY_INV)
+            contours, _ = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+
+            if contours:
+                x, y, w, h = cv2.boundingRect(contours[0])
+                open_cv_image = open_cv_image[y:y+h, x:x+w]
+                image = Image.fromarray(open_cv_image)
+
+        # Manual crop if user wants
+        if manual_crop:
+            st.write("Adjust cropping manually below:")
+            cropped_image = st.image(image, use_container_width=True)
+            st.info("Manual cropping not yet interactive (advanced version can be added later).")
+
+        # Enhance image
+        if enhance_option:
+            enhancer = ImageEnhance.Contrast(image)
+            image = enhancer.enhance(1.5)
+            sharpener = ImageEnhance.Sharpness(image)
+            image = sharpener.enhance(2.0)
+
+        # Display final image
+        st.image(image, caption="Processed Image", use_container_width=True)
+
+        processed_images.append(image)
+
+    if processed_images:
+        # Download processed images as zip
+        zip_buffer = io.BytesIO()
+        with zipfile.ZipFile(zip_buffer, "a", zipfile.ZIP_DEFLATED) as zf:
+            for idx, img in enumerate(processed_images):
+                img_byte_arr = io.BytesIO()
+                img.save(img_byte_arr, format='JPEG')
+                zf.writestr(f"processed_image_{idx+1}.jpg", img_byte_arr.getvalue())
         
-        # --- Enhancement Option ---
-        if st.button(f"Enhance Image: {uploaded_file.name}"):
-            with st.spinner('Enhancing...'):
-                enhancer = ImageEnhance.Contrast(cropped_img)
-                cropped_img = enhancer.enhance(1.5)
-                sharpener = ImageEnhance.Sharpness(cropped_img)
-                cropped_img = sharpener.enhance(2.0)
-                st.success("Enhanced Successfully!")
-                st.image(cropped_img, caption="Enhanced", use_container_width=True)
-
-        cropped_images.append(cropped_img)
-
-    st.write("---")
-
-    # --- Download Options ---
-    col1, col2 = st.columns(2)
-
-    # Save as PDF
-    with col1:
-        if st.button("Generate PDF 📄"):
-            with st.spinner('Generating PDF...'):
-                pdf = FPDF()
-                temp_image_paths = []
-                for img in cropped_images:
-                    temp_path = f"/tmp/{np.random.randint(0,10000)}.jpg"
-                    img.save(temp_path, format="JPEG")
-                    temp_image_paths.append(temp_path)
-                    pdf.add_page()
-                    pdf.image(temp_path, x=10, y=10, w=190)
-                
-                pdf_output_path = "/tmp/scanned_document.pdf"
-                pdf.output(pdf_output_path)
-                
-                with open(pdf_output_path, "rb") as f:
-                    st.download_button(label="📥 Download PDF", data=f, file_name="scanned_document.pdf", mime="application/pdf")
-                
-                # Clean up
-                for path in temp_image_paths:
-                    os.remove(path)
-                os.remove(pdf_output_path)
-
-    # Save all as Images in ZIP
-    with col2:
-        if st.button("Save All as Images 📷"):
-            with st.spinner('Saving Images...'):
-                zip_buffer = BytesIO()
-                with zipfile.ZipFile(zip_buffer, "a", zipfile.ZIP_DEFLATED) as zip_file:
-                    for idx, img in enumerate(cropped_images):
-                        img_bytes = BytesIO()
-                        img.save(img_bytes, format='PNG')
-                        img_bytes.seek(0)
-                        zip_file.writestr(f"scanned_image_{idx+1}.png", img_bytes.read())
-                zip_buffer.seek(0)
-                st.download_button(label="📥 Download All Images (ZIP)", data=zip_buffer, file_name="scanned_images.zip", mime="application/zip")
+        st.download_button(
+            label="📥 Download All Images (ZIP)",
+            data=zip_buffer.getvalue(),
+            file_name=f"processed_images_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.zip",
+            mime="application/zip"
+        )
+
+        # Create PDF
+        from fpdf import FPDF
+        pdf = FPDF()
+        for img in processed_images:
+            img_path = "temp_img.jpg"
+            img.save(img_path)
+            width, height = img.size
+            pdf.add_page()
+            pdf.image(img_path, x=0, y=0, w=210, h=297)
+            os.remove(img_path)
+
+        pdf_output_path = f"scanned_document_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf"
+        pdf.output(pdf_output_path)
+
+        with open(pdf_output_path, "rb") as f:
+            st.download_button(
+                label="📄 Download as PDF",
+                data=f,
+                file_name=pdf_output_path,
+                mime="application/pdf"
+            )
+        os.remove(pdf_output_path)