Update app.py

app.py

@@ -1,134 +1,144 @@
 import streamlit as st
-from PIL import Image, ImageEnhance, ImageOps
+from PIL import Image, ImageEnhance
+from io import BytesIO
+import os
 import numpy as np
-import io
 import zipfile
-import os
-import datetime
-import cv2
-
-# Set page configuration
-st.set_page_config(
-    page_title="📄 Smart Doc Scanner",
-    page_icon="🖼️",
-    layout="centered",
-)
-
-# Custom CSS for background and buttons
-page_bg_img = '''
-<style>
-.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: #0078FF;
-border-radius: 10px;
-}
-</style>
-'''
-
-st.markdown(page_bg_img, unsafe_allow_html=True)
-
-st.markdown("<h1>📄 Smart Doc Scanner</h1>", unsafe_allow_html=True)
-
-# Upload multiple images
-uploaded_files = st.file_uploader(
-    "Upload Document Images (JPEG, PNG, JPG)", 
-    type=["jpg", "jpeg", "png"], 
-    accept_multiple_files=True
-)
+from fpdf import FPDF
+from streamlit_cropper import st_cropper
+
+# --- CONFIG ---
+st.set_page_config(page_title="Smart Doc Scanner", page_icon="🖼️", layout="centered")
+
+# --- BACKGROUND STYLE ---
+def set_background():
+    page_bg_img = f"""
+    <style>
+    [data-testid="stAppViewContainer"] {{
+        background-image: url("https://images.unsplash.com/photo-1524995997946-a1c2e315a42f");
+        background-size: cover;
+    }}
+    [data-testid="stHeader"] {{
+        background: rgba(0, 0, 0, 0);
+    }}
+    </style>
+    """
+    st.markdown(page_bg_img, unsafe_allow_html=True)
+
+set_background()
+
+# --- LIGHT/DARK MODE TOGGLE ---
+mode = st.selectbox("Select Theme:", ["Light Mode", "Dark Mode"])
+if mode == "Dark Mode":
+    dark_style = """
+    <style>
+    body { background-color: #222; color: white; }
+    </style>
+    """
+    st.markdown(dark_style, unsafe_allow_html=True)
+
+# --- APP TITLE ---
+st.markdown("<h1 style='text-align: center; color: #2E86C1;'>📄 Smart ID & Document Scanner</h1>", unsafe_allow_html=True)
+st.write("---")
+
+# --- MODE SELECTION ---
+app_mode = st.radio("Select Mode:", ["Normal Mode (No Editing)", "Advanced Mode (Edit Each Image)"])
+
+# --- UPLOAD IMAGES ---
+uploaded_files = st.file_uploader("Upload your Images", type=["png", "jpg", "jpeg"], accept_multiple_files=True)
 
 if uploaded_files:
-    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")
-    
+    st.write("---")
     processed_images = []
 
-    for uploaded_file in uploaded_files:
-        image = Image.open(uploaded_file).convert("RGB")
-
-        # 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())
-        
-        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"
+    if app_mode == "Normal Mode (No Editing)":
+        for uploaded_file in uploaded_files:
+            img = Image.open(uploaded_file).convert("RGB")
+            processed_images.append(img)
+            st.image(img, caption=uploaded_file.name, use_container_width=True)
+
+    else:  # Advanced Mode
+        st.info("🔧 Edit each image manually: crop, rotate, enhance, reset.")
+        for idx, uploaded_file in enumerate(uploaded_files):
+            original_img = Image.open(uploaded_file).convert("RGB")
+            if f"edited_{idx}" not in st.session_state:
+                st.session_state[f"edited_{idx}"] = original_img
+
+            st.subheader(f"Editing: {uploaded_file.name}")
+
+            # Cropper
+            cropped_img = st_cropper(
+                st.session_state[f"edited_{idx}"],
+                realtime_update=True,
+                box_color='#FF9900',
+                aspect_ratio=None
             )
-        os.remove(pdf_output_path)
+
+            # Actions
+            col1, col2, col3, col4 = st.columns(4)
+
+            with col1:
+                if st.button(f"Enhance {idx}", key=f"enhance_{idx}"):
+                    enhancer = ImageEnhance.Contrast(cropped_img)
+                    enhanced_img = enhancer.enhance(1.5)
+                    sharpener = ImageEnhance.Sharpness(enhanced_img)
+                    st.session_state[f"edited_{idx}"] = sharpener.enhance(2.0)
+
+            with col2:
+                angle = st.slider(f"Rotate {uploaded_file.name}", -180, 180, 0, key=f"rotate_{idx}")
+                if angle != 0:
+                    st.session_state[f"edited_{idx}"] = cropped_img.rotate(angle, expand=True)
+
+            with col3:
+                if st.button(f"Reset {idx}", key=f"reset_{idx}"):
+                    st.session_state[f"edited_{idx}"] = original_img
+                    st.experimental_rerun()
+
+            with col4:
+                st.write("")
+
+            # Show Image
+            st.image(st.session_state[f"edited_{idx}"], caption=f"Edited {uploaded_file.name}", use_container_width=True)
+            processed_images.append(st.session_state[f"edited_{idx}"])
+
+            st.write("---")
+
+    # --- EXPORT OPTIONS ---
+    st.header("📥 Download Options")
+
+    col1, col2 = st.columns(2)
+
+    with col1:
+        if st.button("Download All as ZIP 📷"):
+            with st.spinner("Creating ZIP..."):
+                zip_buffer = BytesIO()
+                with zipfile.ZipFile(zip_buffer, "a", zipfile.ZIP_DEFLATED) as zip_file:
+                    for idx, img in enumerate(processed_images):
+                        img_bytes = BytesIO()
+                        img.save(img_bytes, format='PNG')
+                        img_bytes.seek(0)
+                        zip_file.writestr(f"image_{idx+1}.png", img_bytes.read())
+                zip_buffer.seek(0)
+                st.download_button(label="📥 Download ZIP", data=zip_buffer, file_name="images.zip", mime="application/zip")
+
+    with col2:
+        if st.button("Download All as PDF 📄"):
+            with st.spinner("Generating PDF..."):
+                pdf = FPDF()
+                temp_image_paths = []
+                for img in processed_images:
+                    temp_path = f"/tmp/{np.random.randint(0, 999999)}.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/converted.pdf"
+                pdf.output(pdf_output_path)
+
+                with open(pdf_output_path, "rb") as f:
+                    st.download_button(label="📥 Download PDF", data=f, file_name="converted_document.pdf", mime="application/pdf")
+
+                for path in temp_image_paths:
+                    os.remove(path)
+                os.remove(pdf_output_path)