Update app.py

app.py

@@ -95,10 +95,6 @@ def apply_enhancements(img, x, y, zoom, contrast, brightness, sharpness):
     enhanced_sharpness = ImageEnhance.Sharpness(enhanced_brightness).enhance(sharpness)
     return enhanced_sharpness
 
-# Initialize session state for processed image
-if 'processed_img' not in st.session_state:
-    st.session_state.processed_img = None
-
 st.set_page_config(page_title="CLL Explorer", layout="wide")
 st.title("CLL Explorer: Cell Image Analysis Prep Tool")
 
@@ -127,7 +123,7 @@ if uploaded_files:
 
     with image_col:
         st.subheader("Processed Image")
-        if st.session_state.processed_img:
+        if 'processed_img' in st.session_state:
             st.image(st.session_state.processed_img, use_column_width=True, caption="Processed Image")
         else:
             st.image(img, use_column_width=True, caption="Processed Image")
@@ -143,9 +139,10 @@ if uploaded_files:
             brightness = st.slider("Brightness", 0.0, 5.0, 1.0)
             sharpness = st.slider("Sharpness", 0.0, 2.0, 1.0)
 
-        apply_button = st.button("Apply Adjustments")
-
-        if apply_button:
+        if st.button("Apply Adjustments"):
+            processed_img = apply_enhancements(img, x, y, zoom, contrast, brightness, sharpness)
+            st.session_state.processed_img = processed_img
+        else:
             processed_img = apply_enhancements(img, x, y, zoom, contrast, brightness, sharpness)
             st.session_state.processed_img = processed_img
 
@@ -155,18 +152,16 @@ if uploaded_files:
 
     # Save Options
     save_image = st.checkbox("Save Processed Image")
-    if save_image and st.session_state.processed_img:
-        processed_image_path = "image-processed.jpg"
-        st.session_state.processed_img.save(processed_image_path)
-        st.success(f"Image saved as `{processed_image_path}`")
+    if save_image:
+        st.session_state.processed_img.save("image-processed.jpg")
+        st.success("Image saved as `image-processed.jpg`")
 
     with st.expander("Save Options"):
         description = st.text_area("Describe the image", "")
         if st.button("Save Description"):
-            desc_path = "saved_image_description.txt"
-            with open(desc_path, "w") as f:
+            with open("saved_image_description.txt", "w") as f:
                 f.write(description)
-            st.success(f"Description saved as `{desc_path}`")
+            st.success("Description saved as `saved_image_description.txt`")
 
         if st.button("Save Image Parameters"):
             params = {
@@ -177,18 +172,13 @@ if uploaded_files:
                 "brightness": brightness,
                 "sharpness": sharpness
             }
-            params_path = "saved_image_parameters.json"
-            with open(params_path, "w") as f:
+            with open("saved_image_parameters.json", "w") as f:
                 f.write(pd.DataFrame([params]).to_json(orient="records"))
-            st.success(f"Image parameters saved as `{params_path}`")
+            st.success("Image parameters saved as `saved_image_parameters.json`")
 
         if st.button("Rename Files"):
             file_ext = str(np.random.randint(100))
-            if st.session_state.processed_img:
-                new_image_name = f"img_processed{file_ext}.jpg"
-                os.rename("image-processed.jpg", new_image_name)
-            new_params_name = f"saved_image_parameters{file_ext}.json"
-            new_desc_name = f"saved_image_description{file_ext}.txt"
-            os.rename("saved_image_parameters.json", new_params_name)
-            os.rename("saved_image_description.txt", new_desc_name)
+            os.rename("image-processed.jpg", f"img_processed{file_ext}.jpg")
+            os.rename("saved_image_parameters.json", f"saved_image_parameters{file_ext}.json")
+            os.rename("saved_image_description.txt", f"saved_image_description{file_ext}.txt")
             st.success("Files renamed successfully")