Update app.py

app.py

@@ -1,157 +1,99 @@
 import streamlit as st
 from PIL import Image, ImageEnhance
-import numpy as np
-import io
+from io import BytesIO
+import os
 import cv2
+import numpy as np
 from fpdf import FPDF
-import tempfile
-import os
+from streamlit_cropper import st_cropper
 
-# --- Page config ---
-st.set_page_config(page_title="Document Scanner App", page_icon="📄", layout="centered")
+# --- Set Page Config ---
+st.set_page_config(page_title="Smart ID Scanner", page_icon="🪪", layout="centered")
 
-# --- Background image ---
-page_bg_img = '''
+# --- Add Background Style ---
+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;
+body {
+background: linear-gradient(to right, #ece9e6, #ffffff);
+}
+.stButton>button {
+color: white;
+background-color: #4CAF50;
+border: none;
+padding: 0.5em 1em;
+border-radius: 5px;
 }
-.block-container {
-background-color: rgba(255,255,255,0.8);
-border-radius: 10px;
-padding: 2rem;
+.stHeader, .stText, .stMarkdown {
+color: #333333;
 }
 </style>
-'''
+"""
 st.markdown(page_bg_img, unsafe_allow_html=True)
 
-# --- 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!")
-
-# --- Functions ---
-
-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)
-
-    # Find contours
-    contours, _ = cv2.findContours(edged, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-    if not contours:
-        return image
-
-    # Find the biggest contour
-    c = max(contours, key=cv2.contourArea)
-    x,y,w,h = cv2.boundingRect(c)
-
-    # Crop and return
-    cropped = image.crop((x,y,x+w,y+h))
-    return cropped
-
-def enhance_image(image):
-    enhancer = ImageEnhance.Contrast(image)
-    image = enhancer.enhance(1.5)
-
-    enhancer = ImageEnhance.Sharpness(image)
-    image = enhancer.enhance(2.0)
-    
-    return image
-
-def rotate_image(image, angle):
-    return image.rotate(angle, expand=True)
-
-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
-
-# --- Sidebar options ---
-
-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"])
+# --- App Heading ---
+st.markdown("<h1 style='text-align: center; color: #2E86C1;'>📄 Smart ID & Document Scanner</h1>", unsafe_allow_html=True)
+st.write("---")
 
-# --- Upload images ---
-
-uploaded_files = st.file_uploader("Upload images", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
+# --- Upload Images ---
+uploaded_files = st.file_uploader("Upload ID/Document Images 📷 (All formats supported)", type=['png', 'jpg', 'jpeg'], accept_multiple_files=True)
 
 if uploaded_files:
-    st.success(f"✅ {len(uploaded_files)} file(s) uploaded successfully!")
-    cleaned_images = []
-
-    progress = st.progress(0)
-
-    for idx, uploaded_file in enumerate(uploaded_files):
+    cropped_images = []
+    for uploaded_file in uploaded_files:
         image = Image.open(uploaded_file).convert("RGB")
-
-        # Apply rotation
-        if rotate_angle != 0:
-            image = rotate_image(image, rotate_angle)
-
-        # Apply flip
-        if flip_option != "None":
-            image = flip_image(image, flip_option)
-
-        # 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.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)
+        
+        # --- 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()
+                for img in cropped_images:
+                    temp_path = f"/tmp/{np.random.randint(0,10000)}.jpg"
+                    img.save(temp_path, format="JPEG")
+                    pdf.add_page()
+                    pdf.image(temp_path, x=10, y=10, w=190)
+                    os.remove(temp_path)
+                pdf_output = BytesIO()
+                pdf.output(pdf_output)
+                st.download_button(label="📥 Download PDF", data=pdf_output.getvalue(), file_name="scanned_document.pdf", mime="application/pdf")
+    
+    # Save as Images
+    with col2:
+        if st.button("Save as JPG/PNG 📷"):
+            with st.spinner('Saving Images...'):
+                for idx, img in enumerate(cropped_images):
+                    img_output = BytesIO()
+                    img.save(img_output, format='PNG')
+                    st.download_button(label=f"📥 Download Image {idx+1}", data=img_output.getvalue(), file_name=f"scanned_image_{idx+1}.png", mime="image/png")