Update app.py

app.py

@@ -10,6 +10,14 @@ from streamlit_cropper import st_cropper
 # --- PAGE CONFIG ---
 st.set_page_config(page_title="Smart Doc Scanner", page_icon="🖼️", layout="centered")
 
+# --- Resize Function ---
+def resize_image(img, max_width=800):
+    if img.width > max_width:
+        ratio = max_width / img.width
+        new_height = int(img.height * ratio)
+        img = img.resize((max_width, new_height), resample=Image.Resampling.LANCZOS)
+    return img
+
 # --- HEADER ---
 st.title("📄 Smart ID & Document Scanner")
 st.write("---")
@@ -21,7 +29,6 @@ if "uploaded_files_hash" not in st.session_state:
 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))
 
@@ -37,6 +44,7 @@ if uploaded_files:
     if app_mode == "Normal Mode (No Editing)":
         for uploaded_file in uploaded_files:
             img = Image.open(uploaded_file).convert("RGB")
+            img = resize_image(img)  # Always resize
             processed_images.append(img)
             st.image(img, caption=uploaded_file.name, use_container_width=True)
 
@@ -45,16 +53,13 @@ if uploaded_files:
             st.subheader(f"Editing: {uploaded_file.name}")
 
             original_img = Image.open(uploaded_file).convert("RGB")
+            original_img = resize_image(original_img)
 
-            # Resize image if it's too large
-            max_size = (800, 800)
-            if original_img.size[0] > max_size[0] or original_img.size[1] > max_size[1]:
-                original_img.thumbnail(max_size, Image.ANTIALIAS)
-
+            # Initialize session states
             if f"edited_{idx}" not in st.session_state:
-                st.session_state[f"edited_{idx}"] = original_img.copy()
-                st.session_state[f"preview_{idx}"] = original_img.copy()
+                st.session_state[f"edited_{idx}"] = original_img
                 st.session_state[f"cropper_{idx}"] = original_img.copy()
+                st.session_state[f"preview_{idx}"] = original_img.copy()
 
             # --- Layout: 2 columns ---
             col_cropper, col_preview = st.columns([1, 1])
@@ -76,14 +81,14 @@ if uploaded_files:
                     st.session_state[f"cropper_{idx}"] = crop_output
 
             with col_preview:
-                st.write("### Preview:")
+                st.write("### Preview and Edit:")
                 st.image(st.session_state[f"preview_{idx}"], caption=f"Preview: {uploaded_file.name}", use_container_width=True)
 
-                # --- Buttons next to preview ---
                 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"preview_{idx}"] = sharpener.enhance(2.0)
+                    st.session_state[f"cropper_{idx}"] = st.session_state[f"preview_{idx}"]
 
                 angle = st.slider(f"Rotate Angle", -180, 180, 0, key=f"rotate_slider_{idx}")