Update app.py

app.py

@@ -58,12 +58,14 @@ if uploaded_files:
             continue
 
         # --- Advanced Mode ---
+        key_base = f"base_{idx}"
         key_cropped = f"cropped_{idx}"
         key_preview = f"preview_{idx}"
         key_final = f"final_{idx}"
         key_angle = f"angle_{idx}"
 
-        if key_cropped not in st.session_state:
+        if key_base not in st.session_state:
+            st.session_state[key_base] = resized.copy()
             st.session_state[key_cropped] = resized.copy()
             st.session_state[key_preview] = resized.copy()
             st.session_state[key_final] = None
@@ -71,20 +73,20 @@ if uploaded_files:
 
         with st.expander("✂️ Crop, Enhance & Rotate", expanded=True):
             # Step 1: Crop
-            cropped = crop_with_sliders(resized, key_prefix=f"img_{idx}")
+            cropped = crop_with_sliders(st.session_state[key_base], key_prefix=f"img_{idx}")
             st.session_state[key_cropped] = cropped
 
-            # Step 2: Rotation and Enhancement Controls
+            # Step 2: Rotate + Enhance + Reset
             col1, col2, col3 = st.columns(3)
 
             with col1:
                 if st.button("🔁 Reset", key=f"reset_{idx}"):
-                    st.session_state[key_preview] = cropped.copy()
+                    st.session_state[key_preview] = st.session_state[key_base].copy()
                     st.session_state[key_angle] = 0
 
             with col2:
                 if st.button("✨ Enhance", key=f"enhance_{idx}"):
-                    enhancer = ImageEnhance.Contrast(cropped)
+                    enhancer = ImageEnhance.Contrast(st.session_state[key_cropped])
                     sharp = ImageEnhance.Sharpness(enhancer.enhance(1.5))
                     st.session_state[key_preview] = sharp.enhance(2.0)