Update app.py

app.py

@@ -18,7 +18,6 @@ def resize_image(img, max_height=600):
 def crop_sliders(img, key):
     w, h = img.size
     state = st.session_state
-    # Initialize default values
     for name, val in zip(["left", "top", "right", "bottom"], [0, 0, w, h]):
         state.setdefault(f"{key}_{name}", val)
 
@@ -30,7 +29,6 @@ def crop_sliders(img, key):
         top = st.slider("Crop Top", 0, h - 10, state[f"{key}_top"], key=f"{key}_top_slider")
         bottom = st.slider("Crop Bottom", top + 10, h, state[f"{key}_bottom"], key=f"{key}_bottom_slider")
 
-    # Store new values
     state[f"{key}_left"] = left
     state[f"{key}_top"] = top
     state[f"{key}_right"] = right
@@ -54,12 +52,16 @@ if uploaded_files:
         st.markdown(f"---\n### 🖼️ {file.name}")
         original = Image.open(BytesIO(file_bytes)).convert("RGB")
 
-        # Init session state for this image
         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
             st.session_state[f"filename_{key}"] = file.name.rsplit(".", 1)[0]
+            w, h = original.size
+            st.session_state[f"{key}_left"] = 0
+            st.session_state[f"{key}_top"] = 0
+            st.session_state[f"{key}_right"] = w
+            st.session_state[f"{key}_bottom"] = h
 
         if mode == "Normal Mode":
             st.image(resize_image(original), use_container_width=True)
@@ -69,6 +71,7 @@ if uploaded_files:
 
         col1, col2 = st.columns([1, 1.5])
         with col1:
+            # Controls
             if st.button("🔁 Reset", key=f"reset_{key}"):
                 st.session_state[f"angle_{key}"] = 0
                 st.session_state[f"enhance_{key}"] = False
@@ -81,40 +84,39 @@ 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}")
             left, top, right, bottom = crop_sliders(original, key)
-
             st.text_input("📁 Rename this file", value=st.session_state[f"filename_{key}"], key=f"filename_{key}_input")
 
-            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("Saved for export!")
-
         with col2:
-            # Always crop first (even if user leaves default)
-            preview = st.session_state[f"original_{key}"].crop((left, top, right, bottom))
-            preview = resize_image(preview)
+            # Real-time preview
+            img = st.session_state[f"original_{key}"].crop((left, top, right, bottom))
+            img = resize_image(img)
 
             if st.session_state[f"enhance_{key}"]:
-                preview = ImageEnhance.Contrast(preview).enhance(1.5)
-                preview = ImageEnhance.Sharpness(preview).enhance(2.0)
+                img = ImageEnhance.Contrast(img).enhance(1.5)
+                img = ImageEnhance.Sharpness(img).enhance(2.0)
 
             angle = st.session_state[f"angle_{key}"]
             if angle != 0:
-                preview = preview.rotate(angle, expand=True)
+                img = img.rotate(angle, expand=True)
 
-            st.session_state[f"preview_{key}"] = preview
-            st.image(preview, caption="🎯 Preview", use_container_width=True)
+            st.session_state[f"preview_{key}"] = img
+            st.image(img, caption="🎯 Live Preview", use_container_width=True)
+
+            if st.button("✅ Save This Image", key=f"save_{key}"):
+                st.session_state[f"final_{key}"] = img.copy()
+                st.success("Saved for export!")
 
         final_img = st.session_state.get(f"final_{key}", original)
         filename = st.session_state.get(f"filename_{key}", file.name.rsplit(".", 1)[0])
         processed_images.append(final_img)
         file_names.append(filename)
 
-    # --- EXPORT SECTION ---
+    # --- Export Section ---
     st.markdown("### 🧾 Export Options")
     with st.container():
-        c1, c2 = st.columns([1, 1])
-        with c1:
-            if st.button("📦 Download ZIP"):
+        col1, col2 = st.columns([1, 1])
+        with col1:
+            if st.button("📦 Generate ZIP", key="zip_btn"):
                 zip_buffer = BytesIO()
                 with zipfile.ZipFile(zip_buffer, "w") as zipf:
                     for name, img in zip(file_names, processed_images):
@@ -122,10 +124,10 @@ if uploaded_files:
                         img.save(buffer, format="PNG")
                         zipf.writestr(f"{name}.png", buffer.getvalue())
                 zip_buffer.seek(0)
-                st.download_button("📥 Download ZIP", zip_buffer, "images.zip", mime="application/zip")
+                st.download_button("⬇️ Download ZIP", zip_buffer, "images.zip", mime="application/zip", key="zip_dl")
 
-        with c2:
-            if st.button("📄 Download PDF"):
+        with col2:
+            if st.button("📄 Generate PDF", key="pdf_btn"):
                 pdf = FPDF()
                 tmp_paths = []
                 for name, img in zip(file_names, processed_images):
@@ -134,9 +136,9 @@ if uploaded_files:
                     pdf.add_page()
                     pdf.image(path, x=10, y=10, w=190)
                     tmp_paths.append(path)
-                pdf_out = "/tmp/output.pdf"
-                pdf.output(pdf_out)
-                with open(pdf_out, "rb") as f:
-                    st.download_button("📥 Download PDF", f, "scanned.pdf", mime="application/pdf")
-                for p in tmp_paths + [pdf_out]:
+                out_path = "/tmp/final.pdf"
+                pdf.output(out_path)
+                with open(out_path, "rb") as f:
+                    st.download_button("⬇️ Download PDF", f, "scanned.pdf", mime="application/pdf", key="pdf_dl")
+                for p in tmp_paths + [out_path]:
                     os.remove(p)