Update app.py

app.py

@@ -59,13 +59,13 @@ if uploaded_files:
             if f"edited_{idx}" not in st.session_state:
                 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()
+                st.session_state[f"preview_{idx}"] = None  # Will be set after crop
 
             # --- Layout: 2 columns ---
             col_cropper, col_preview = st.columns([1, 1])
 
             with col_cropper:
-                st.write("### Select Crop Area:")
+                st.write("### Select Crop Area (Original Image):")
                 crop_output = st_cropper(
                     st.session_state[f"cropper_{idx}"],
                     realtime_update=False,
@@ -78,32 +78,32 @@ if uploaded_files:
 
                 if st.button(f"Crop Now {idx}", key=f"crop_{idx}"):
                     st.session_state[f"preview_{idx}"] = crop_output
-                    st.session_state[f"cropper_{idx}"] = crop_output
+                    st.success("✅ Cropped! Now Edit Preview.")
 
             with col_preview:
                 st.write("### Preview and Edit:")
-                st.image(st.session_state[f"preview_{idx}"], caption=f"Preview: {uploaded_file.name}", use_container_width=True)
+                if st.session_state[f"preview_{idx}"] is not None:
+                    st.image(st.session_state[f"preview_{idx}"], caption=f"Preview: {uploaded_file.name}", use_container_width=True)
 
-                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}"]
+                    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)
 
-                angle = st.slider(f"Rotate Angle", -180, 180, 0, key=f"rotate_slider_{idx}")
+                    angle = st.slider(f"Rotate Angle", -180, 180, 0, key=f"rotate_slider_{idx}")
 
-                if st.button(f"Rotate Now {idx}", key=f"rotate_{idx}"):
-                    rotated_img = st.session_state[f"preview_{idx}"].rotate(angle, expand=True)
-                    st.session_state[f"preview_{idx}"] = rotated_img
-                    st.session_state[f"cropper_{idx}"] = rotated_img
+                    if st.button(f"Rotate Now {idx}", key=f"rotate_{idx}"):
+                        rotated_img = st.session_state[f"preview_{idx}"].rotate(angle, expand=True)
+                        st.session_state[f"preview_{idx}"] = rotated_img
 
-                if st.button(f"Reset Preview {idx}", key=f"reset_preview_{idx}"):
-                    st.session_state[f"preview_{idx}"] = original_img.copy()
-                    st.session_state[f"cropper_{idx}"] = original_img.copy()
+                    if st.button(f"Reset Preview {idx}", key=f"reset_preview_{idx}"):
+                        st.session_state[f"preview_{idx}"] = crop_output.copy()
 
-                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!")
+                    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!")
+                else:
+                    st.info("Please crop an area first to enable editing.")
 
             processed_images.append(st.session_state.get(f"edited_{idx}", original_img))