Update app.py

app.py

@@ -1,169 +1,158 @@
-# app.py
-
 import streamlit as st
-import cv2
-import numpy as np
 from PIL import Image, ImageEnhance
+import numpy as np
 import io
+import cv2
 from fpdf import FPDF
-import datetime
-
-# -----------------------
-# Helper Functions
-# -----------------------
-
-def auto_crop_document(image):
-    """Auto-detect document and crop it."""
-    img = np.array(image)
-    gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
-    blurred = cv2.GaussianBlur(gray, (5, 5), 0)
-    edged = cv2.Canny(blurred, 75, 200)
+import tempfile
+import os
+from rembg import remove
 
-    contours, _ = cv2.findContours(edged.copy(), cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)
-    contours = sorted(contours, key=cv2.contourArea, reverse=True)[:5]
+# --- Page config ---
+st.set_page_config(page_title="Document Scanner App", page_icon="📄", layout="centered")
 
-    for c in contours:
-        peri = cv2.arcLength(c, True)
-        approx = cv2.approxPolyDP(c, 0.02 * peri, True)
-
-        if len(approx) == 4:
-            doc_cnt = approx
-            break
-    else:
-        return image  # Return original if no contour found
-
-    pts = doc_cnt.reshape(4, 2)
-    rect = np.zeros((4, 2), dtype="float32")
-
-    s = pts.sum(axis=1)
-    rect[0] = pts[np.argmin(s)]      # Top-left
-    rect[2] = pts[np.argmax(s)]      # Bottom-right
-
-    diff = np.diff(pts, axis=1)
-    rect[1] = pts[np.argmin(diff)]   # Top-right
-    rect[3] = pts[np.argmax(diff)]   # Bottom-left
-
-    (tl, tr, br, bl) = rect
+# --- Background image ---
+page_bg_img = '''
+<style>
+.stApp {
+background-image: url("https://images.unsplash.com/photo-1504150559088-41c99c17b7c6");
+background-size: cover;
+background-position: top left;
+background-repeat: no-repeat;
+background-attachment: fixed;
+}
+.block-container {
+background-color: rgba(255,255,255,0.8);
+border-radius: 10px;
+padding: 2rem;
+}
+</style>
+'''
+st.markdown(page_bg_img, unsafe_allow_html=True)
 
-    widthA = np.linalg.norm(br - bl)
-    widthB = np.linalg.norm(tr - tl)
-    maxWidth = max(int(widthA), int(widthB))
+# --- Title ---
+st.title("📄 Smart Document Scanner")
+st.write("Upload photos of ID Cards, Documents taken from mobile & get them auto-cropped, enhanced, and converted to PDF!")
 
-    heightA = np.linalg.norm(tr - br)
-    heightB = np.linalg.norm(tl - bl)
-    maxHeight = max(int(heightA), int(heightB))
+# --- Functions ---
 
-    dst = np.array([
-        [0, 0],
-        [maxWidth-1, 0],
-        [maxWidth-1, maxHeight-1],
-        [0, maxHeight-1]
-    ], dtype="float32")
+def auto_crop_image(image):
+    # Convert to grayscale
+    gray = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2GRAY)
+    blur = cv2.GaussianBlur(gray, (5,5), 0)
+    edged = cv2.Canny(blur, 50, 150)
 
-    M = cv2.getPerspectiveTransform(rect, dst)
-    warped = cv2.warpPerspective(img, M, (maxWidth, maxHeight))
+    # Find contours
+    contours, _ = cv2.findContours(edged, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+    if not contours:
+        return image
 
-    return Image.fromarray(warped)
+    # Find the biggest contour
+    c = max(contours, key=cv2.contourArea)
+    x,y,w,h = cv2.boundingRect(c)
 
-def enhance_image(img_pil):
-    """Enhance the image (brightness and contrast)."""
-    enhancer_brightness = ImageEnhance.Brightness(img_pil)
-    img_bright = enhancer_brightness.enhance(1.2)
+    # Crop and return
+    cropped = image.crop((x,y,x+w,y+h))
+    return cropped
 
-    enhancer_contrast = ImageEnhance.Contrast(img_bright)
-    img_contrast = enhancer_contrast.enhance(1.3)
+def enhance_image(image):
+    enhancer = ImageEnhance.Contrast(image)
+    image = enhancer.enhance(1.5)
 
-    return img_contrast
+    enhancer = ImageEnhance.Sharpness(image)
+    image = enhancer.enhance(2.0)
+    
+    return image
 
-def rotate_image(img_pil, angle):
-    """Rotate image."""
-    return img_pil.rotate(angle, expand=True)
+def rotate_image(image, angle):
+    return image.rotate(angle, expand=True)
 
-# -----------------------
-# Streamlit App
-# -----------------------
+def flip_image(image, mode):
+    if mode == "Horizontal":
+        return image.transpose(Image.FLIP_LEFT_RIGHT)
+    elif mode == "Vertical":
+        return image.transpose(Image.FLIP_TOP_BOTTOM)
+    return image
 
-# Set app background and style
-page_bg_img = f"""
-<style>
-.stApp {{
-background-image: url("https://images.unsplash.com/photo-1581090700227-1c065c585ab3?ixlib=rb-4.0.3&auto=format&fit=crop&w=1470&q=80");
-background-size: cover;
-background-position: center;
-}}
-h1, h2, h3, h4, h5 {{
-color: white;
-}}
-</style>
-"""
-st.markdown(page_bg_img, unsafe_allow_html=True)
+# --- Sidebar options ---
 
-# App Title
-st.title("📄 Smart Document Scanner App")
+st.sidebar.header("⚙️ Settings")
+crop_mode = st.sidebar.selectbox("Cropping Mode", ["Auto Crop", "Manual Crop"])
+apply_enhance = st.sidebar.checkbox("Enhance Image (contrast, sharpness)", value=False)
+rotate_angle = st.sidebar.selectbox("Rotate", [0, 90, 180, 270])
+flip_option = st.sidebar.selectbox("Flip", ["None", "Horizontal", "Vertical"])
 
-st.write("Upload your **ID cards or documents**. We'll detect, crop, enhance and generate a clean PDF!")
+# --- Upload images ---
 
-uploaded_files = st.file_uploader(
-    "Upload Images (JPG, JPEG, PNG supported)", 
-    type=["jpg", "jpeg", "png"], 
-    accept_multiple_files=True
-)
+uploaded_files = st.file_uploader("Upload images", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
 
 if uploaded_files:
-    final_images = []
+    st.success(f"✅ {len(uploaded_files)} file(s) uploaded successfully!")
+    cleaned_images = []
 
-    for file in uploaded_files:
-        image = Image.open(file)
-        st.subheader(f"Original Uploaded Image: {file.name}")
-        st.image(image, caption="Original Image", use_column_width=True)
+    progress = st.progress(0)
 
-        # Auto crop
-        cropped_img = auto_crop_document(image)
-        st.image(cropped_img, caption="Auto Cropped", use_column_width=True)
+    for idx, uploaded_file in enumerate(uploaded_files):
+        image = Image.open(uploaded_file).convert("RGB")
 
-        # Rotate option
-        rotate_angle = st.selectbox(f"Rotate {file.name}?", [0, 90, 180, 270], key=f"rotate_{file.name}")
+        # Apply rotation
         if rotate_angle != 0:
-            cropped_img = rotate_image(cropped_img, rotate_angle)
-            st.image(cropped_img, caption=f"Rotated by {rotate_angle}°", use_column_width=True)
-
-        # Enhance option
-        if st.button(f"Enhance {file.name}", key=f"enhance_{file.name}"):
-            enhanced_img = enhance_image(cropped_img)
-            st.image(enhanced_img, caption="Enhanced Image (After)", use_column_width=True)
-            final_images.append(enhanced_img)
-        else:
-            final_images.append(cropped_img)
+            image = rotate_image(image, rotate_angle)
 
-    # Generate PDF
-    if st.button("📥 Generate PDF"):
-        pdf = FPDF()
-        pdf.set_auto_page_break(0)
+        # Apply flip
+        if flip_option != "None":
+            image = flip_image(image, flip_option)
 
-        for img in final_images:
-            img_temp = io.BytesIO()
-            img.save(img_temp, format='JPEG')
-            img_temp.seek(0)
-
-            cover = Image.open(img_temp)
-            width, height = cover.size
-
-            # Convert pixel in mm (1px = 0.264583 mm)
-            width, height = float(width * 0.264583), float(height * 0.264583)
-
-            pdf.add_page()
-            pdf.image(img_temp, 0, 0, width, height)
-
-        pdf_output = io.BytesIO()
-        pdf.output(pdf_output)
-        pdf_output.seek(0)
-
-        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename = f"scanned_document_{timestamp}.pdf"
+        # Crop
+        if crop_mode == "Auto Crop":
+            image = auto_crop_image(image)
+        else:
+            # Manual Crop
+            st.write(f"✂️ Manually crop for {uploaded_file.name}")
+            cropped_area = st.slider(f"Adjust crop for {uploaded_file.name}", 0, 100, (5,95), key=idx)
+            width, height = image.size
+            left = width * cropped_area[0] // 100
+            right = width * cropped_area[1] // 100
+            image = image.crop((left, 0, right, height))
+
+        # Enhance if selected
+        if apply_enhance:
+            image = enhance_image(image)
+
+        st.image(image, caption=f"Processed: {uploaded_file.name}", use_container_width=True)
+
+        cleaned_images.append(image)
+
+        progress.progress((idx+1)/len(uploaded_files))
+
+    # --- Generate PDF ---
+    st.subheader("📄 Download PDF")
+
+    if st.button("Generate PDF"):
+        with tempfile.TemporaryDirectory() as tmpdirname:
+            pdf = FPDF(unit="pt", format="A4")
+            for img_idx, img in enumerate(cleaned_images):
+                temp_img_path = os.path.join(tmpdirname, f"img_{img_idx}.jpg")
+                img.save(temp_img_path, format="JPEG")
+
+                # Calculate aspect ratio
+                img_width, img_height = img.size
+                page_width = pdf.w - 40  # 20 margin each side
+                ratio = page_width / img_width
+                new_width = page_width
+                new_height = img_height * ratio
+
+                pdf.add_page()
+                pdf.image(temp_img_path, 20, 20, new_width, new_height)
+
+            pdf_output_path = os.path.join(tmpdirname, "scanned_document.pdf")
+            pdf.output(pdf_output_path)
+
+            with open(pdf_output_path, "rb") as f:
+                st.download_button(
+                    label="📥 Download Scanned PDF",
+                    data=f,
+                    file_name="scanned_document.pdf",
+                    mime="application/pdf"
+                )
 
-        st.download_button(
-            label="📄 Download Your PDF",
-            data=pdf_output,
-            file_name=filename,
-            mime="application/pdf"
-        )