Update app.py

app.py

@@ -6,42 +6,43 @@ import zipfile
 import os
 from fpdf import FPDF
 
-st.set_page_config(page_title="Smart Doc Scanner", page_icon="🖼️", layout="wide")
+st.set_page_config(page_title="Smart Doc Scanner", layout="wide")
 
-# --- Resize image ---
-def resize_image(img, max_height=500):
+# Resize for faster preview (non-destructive)
+def resize_image(img, max_height=600):
     if img.height > max_height:
         ratio = max_height / img.height
-        img = img.resize((int(img.width * ratio), max_height), Image.Resampling.LANCZOS)
+        return img.resize((int(img.width * ratio), max_height), Image.Resampling.LANCZOS)
     return img
 
-# --- Crop with slider ---
-def crop_sliders(img, key):
+def crop_sliders(img, key, compact=False):
     w, h = img.size
     state = st.session_state
-
     if f"{key}_left" not in state:
         state[f"{key}_left"] = 0
         state[f"{key}_top"] = 0
         state[f"{key}_right"] = w
         state[f"{key}_bottom"] = h
 
-    left = st.slider("Crop Left", 0, w - 10, state[f"{key}_left"], key=f"{key}_left_slider")
-    top = st.slider("Crop Top", 0, h - 10, state[f"{key}_top"], key=f"{key}_top_slider")
-    right = st.slider("Crop Right", left + 10, w, state[f"{key}_right"], key=f"{key}_right_slider")
-    bottom = st.slider("Crop Bottom", top + 10, h, state[f"{key}_bottom"], key=f"{key}_bottom_slider")
+    slider_args = {"label_visibility": "visible"} if compact else {}
+    with st.container():
+        st.markdown("#### ✂️ Crop", help=None)
+        col1, col2 = st.columns(2)
+        with col1:
+            left = st.slider("Crop Left", 0, w - 10, state[f"{key}_left"], key=f"{key}_left_slider", **slider_args)
+            right = st.slider("Crop Right", left + 10, w, state[f"{key}_right"], key=f"{key}_right_slider", **slider_args)
+        with col2:
+            top = st.slider("Crop Top", 0, h - 10, state[f"{key}_top"], key=f"{key}_top_slider", **slider_args)
+            bottom = st.slider("Crop Bottom", top + 10, h, state[f"{key}_bottom"], key=f"{key}_bottom_slider", **slider_args)
 
-    # Update stored values
     state[f"{key}_left"] = left
     state[f"{key}_top"] = top
     state[f"{key}_right"] = right
     state[f"{key}_bottom"] = bottom
-
     return (left, top, right, bottom)
 
-# --- App UI ---
 st.title("📄 Smart ID & Document Scanner")
-uploaded_files = st.file_uploader("📤 Upload Images", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
+uploaded_files = st.file_uploader("📤 Upload Images", type=["png", "jpg", "jpeg"], accept_multiple_files=True)
 
 if uploaded_files:
     mode = st.radio("Mode", ["Normal Mode", "Advanced Mode (Edit Each)"])
@@ -51,22 +52,18 @@ if uploaded_files:
         key = f"img_{idx}"
         st.markdown(f"---\n### 🖼️ {file.name}")
         original = Image.open(file).convert("RGB")
-        resized = resize_image(original)
-
         if f"original_{key}" not in st.session_state:
             st.session_state[f"original_{key}"] = original
             st.session_state[f"angle_{key}"] = 0
             st.session_state[f"enhance_{key}"] = False
 
         if mode == "Normal Mode":
-            st.image(resized, use_container_width=True)
+            st.image(resize_image(original), use_container_width=True)
             processed_images.append(original)
             continue
 
-        # --- Advanced Mode ---
-        col_edit, col_preview = st.columns([1, 1.5])
-
-        with col_edit:
+        col1, col2 = st.columns([1, 1.5])
+        with col1:
             if st.button("🔁 Reset", key=f"reset_{key}"):
                 st.session_state[f"angle_{key}"] = 0
                 st.session_state[f"enhance_{key}"] = False
@@ -77,56 +74,59 @@ if uploaded_files:
 
             st.slider("🌀 Rotate", -180, 180, st.session_state[f"angle_{key}"], key=f"angle_{key}")
             st.checkbox("✨ Enhance", value=st.session_state[f"enhance_{key}"], key=f"enhance_{key}")
-            st.markdown("#### ✂️ Crop")
-            left, top, right, bottom = crop_sliders(st.session_state[f"original_{key}"], key)
+            left, top, right, bottom = crop_sliders(original, key, compact=True)
 
             if st.button("✅ Save This Image", key=f"save_{key}"):
                 st.session_state[f"final_{key}"] = st.session_state[f"preview_{key}"].copy()
-                st.success("Image saved for export!")
+                st.success("Saved for export!")
+
+        with col2:
+            preview = st.session_state[f"original_{key}"].copy()
+            preview = preview.crop((left, top, right, bottom))
+            preview = resize_image(preview)  # faster rendering
 
-        with col_preview:
-            img = st.session_state[f"original_{key}"].copy()
-            img = img.crop((left, top, right, bottom))
             if st.session_state[f"enhance_{key}"]:
-                img = ImageEnhance.Contrast(img).enhance(1.5)
-                img = ImageEnhance.Sharpness(img).enhance(2.0)
-            if st.session_state[f"angle_{key}"] != 0:
-                img = img.rotate(st.session_state[f"angle_{key}"], expand=True)
-            st.session_state[f"preview_{key}"] = img
-            st.image(img, caption="🎯 Preview", use_container_width=True)
+                preview = ImageEnhance.Contrast(preview).enhance(1.5)
+                preview = ImageEnhance.Sharpness(preview).enhance(2.0)
+
+            angle = st.session_state[f"angle_{key}"]
+            if angle != 0:
+                preview = preview.rotate(angle, expand=True)
+
+            st.session_state[f"preview_{key}"] = preview
+            st.image(preview, caption="🎯 Preview", use_container_width=True)
 
-        final = st.session_state.get(f"final_{key}", original)
-        processed_images.append(final)
+        final_img = st.session_state.get(f"final_{key}", original)
+        processed_images.append(final_img)
 
-    # --- Export Section ---
     st.markdown("---")
     st.header("📥 Export Options")
     col1, col2 = st.columns(2)
 
     with col1:
-        if st.button("📦 Download as ZIP"):
+        if st.button("📦 Download ZIP"):
             zip_buffer = BytesIO()
-            with zipfile.ZipFile(zip_buffer, "w") as zf:
+            with zipfile.ZipFile(zip_buffer, "w") as zipf:
                 for i, img in enumerate(processed_images):
                     buffer = BytesIO()
                     img.save(buffer, format="PNG")
-                    zf.writestr(f"image_{i+1}.png", buffer.getvalue())
+                    zipf.writestr(f"image_{i+1}.png", buffer.getvalue())
             zip_buffer.seek(0)
-            st.download_button("📥 Download ZIP", zip_buffer, file_name="images.zip", mime="application/zip")
+            st.download_button("📥 Download ZIP", zip_buffer, "images.zip", mime="application/zip")
 
     with col2:
-        if st.button("📄 Download as PDF"):
+        if st.button("📄 Download PDF"):
             pdf = FPDF()
-            temp_files = []
+            tmp_files = []
             for img in processed_images:
                 path = f"/tmp/{np.random.randint(0, 999999)}.jpg"
-                img.save(path, format="JPEG")
+                img.save(path, "JPEG")
                 pdf.add_page()
                 pdf.image(path, x=10, y=10, w=190)
-                temp_files.append(path)
-            out_path = "/tmp/output.pdf"
-            pdf.output(out_path)
-            with open(out_path, "rb") as f:
-                st.download_button("📥 Download PDF", f, file_name="scanned.pdf", mime="application/pdf")
-            for path in temp_files + [out_path]:
+                tmp_files.append(path)
+            output_pdf = "/tmp/output.pdf"
+            pdf.output(output_pdf)
+            with open(output_pdf, "rb") as f:
+                st.download_button("📥 Download PDF", f, "scanned.pdf", "application/pdf")
+            for path in tmp_files + [output_pdf]:
                 os.remove(path)