Update app.py

app.py

@@ -1,26 +1,37 @@
 import streamlit as st
 from PIL import Image, ImageEnhance
 from io import BytesIO
-import os
 import numpy as np
 import zipfile
+import os
 from fpdf import FPDF
 from streamlit_cropper import st_cropper
 
 # --- PAGE CONFIG ---
 st.set_page_config(page_title="Smart Doc Scanner", page_icon="🖼️", layout="centered")
 
-# --- SIMPLE TITLE ---
+# --- HEADER ---
 st.title("📄 Smart ID & Document Scanner")
 st.write("---")
 
-# --- MODE SELECTION ---
-app_mode = st.radio("Select Mode:", ["Normal Mode (No Editing)", "Advanced Mode (Edit Each Image)"])
+# --- Clear Session State on New Upload ---
+if "uploaded_files_hash" not in st.session_state:
+    st.session_state.uploaded_files_hash = None
 
-# --- FILE UPLOAD ---
 uploaded_files = st.file_uploader("Upload your Images", type=["png", "jpg", "jpeg"], accept_multiple_files=True)
 
 if uploaded_files:
+    # Create a hash based on filenames
+    uploaded_files_names = [file.name for file in uploaded_files]
+    current_hash = hash(tuple(uploaded_files_names))
+
+    if st.session_state.uploaded_files_hash != current_hash:
+        # New files uploaded, clear session
+        st.session_state.clear()
+        st.session_state.uploaded_files_hash = current_hash
+
+    app_mode = st.radio("Select Mode:", ["Normal Mode (No Editing)", "Advanced Mode (Edit Each Image)"])
+
     st.write("---")
     processed_images = []
 
@@ -31,21 +42,26 @@ if uploaded_files:
             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):
+            st.subheader(f"Editing: {uploaded_file.name}")
+
             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.session_state[f"preview_{idx}"] = original_img.copy()
 
-            st.subheader(f"Editing: {uploaded_file.name}")
+            col1, col2 = st.columns(2)
+
+            with col1:
+                st.write("### Preview:")
+                st.image(st.session_state[f"preview_{idx}"], caption=f"Preview of {uploaded_file.name}", use_container_width=True)
 
-            col_left, col_right = st.columns(2)
+            with col2:
+                st.write("### Editing Options:")
 
-            with col_left:
-                st.write("### Select Crop Area:")
                 crop_output = st_cropper(
-                    st.session_state[f"edited_{idx}"],
-                    realtime_update=False,  # Important fix!
+                    st.session_state[f"preview_{idx}"],
+                    realtime_update=False,
                     box_color='#FF9900',
                     aspect_ratio=None
                 )
@@ -54,25 +70,26 @@ if uploaded_files:
                     crop_output = Image.fromarray(crop_output)
 
                 if st.button(f"Crop Now {idx}", key=f"crop_{idx}"):
-                    st.session_state[f"edited_{idx}"] = crop_output
+                    st.session_state[f"preview_{idx}"] = crop_output
 
-                if st.button(f"Enhance {idx}", key=f"enhance_{idx}"):
-                    enhancer = ImageEnhance.Contrast(st.session_state[f"edited_{idx}"])
+                if st.button(f"Enhance Now {idx}", key=f"enhance_{idx}"):
+                    enhancer = ImageEnhance.Contrast(st.session_state[f"preview_{idx}"])
                     sharpener = ImageEnhance.Sharpness(enhancer.enhance(1.5))
-                    st.session_state[f"edited_{idx}"] = sharpener.enhance(2.0)
+                    st.session_state[f"preview_{idx}"] = sharpener.enhance(2.0)
+
+                angle = st.slider(f"Rotate Angle {uploaded_file.name}", -180, 180, 0, key=f"rotate_slider_{idx}")
 
-                angle = st.slider(f"Select Rotate Angle for {uploaded_file.name}", -180, 180, 0, key=f"rotate_slider_{idx}")
                 if st.button(f"Rotate Now {idx}", key=f"rotate_{idx}"):
-                    st.session_state[f"edited_{idx}"] = st.session_state[f"edited_{idx}"].rotate(angle, expand=True)
+                    st.session_state[f"preview_{idx}"] = st.session_state[f"preview_{idx}"].rotate(angle, expand=True)
 
-                if st.button(f"Reset {idx}", key=f"reset_{idx}"):
-                    st.session_state[f"edited_{idx}"] = original_img
+                if st.button(f"Reset Preview {idx}", key=f"reset_preview_{idx}"):
+                    st.session_state[f"preview_{idx}"] = original_img.copy()
 
-            with col_right:
-                st.write("### Preview:")
-                st.image(st.session_state[f"edited_{idx}"], caption=f"Preview: {uploaded_file.name}", use_container_width=True)
+                if st.button(f"Apply Changes {idx}", key=f"apply_{idx}"):
+                    st.session_state[f"edited_{idx}"] = st.session_state[f"preview_{idx}"].copy()
+                    st.success("✅ Changes Applied!")
 
-            processed_images.append(st.session_state[f"edited_{idx}"])
+            processed_images.append(st.session_state.get(f"edited_{idx}", original_img))
 
             st.write("---")