Update app.py

app.py

@@ -95,6 +95,10 @@ 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")
 
@@ -123,7 +127,7 @@ if uploaded_files:
 
     with image_col:
         st.subheader("Processed Image")
-        if 'processed_img' in st.session_state:
+        if st.session_state.processed_img:
             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")
@@ -139,11 +143,9 @@ if uploaded_files:
             brightness = st.slider("Brightness", 0.0, 5.0, 1.0)
             sharpness = st.slider("Sharpness", 0.0, 2.0, 1.0)
 
-        if st.button("Apply Adjustments"):
-            processed_img = apply_enhancements(img, x, y, zoom, contrast, brightness, sharpness)
-            st.session_state.processed_img = processed_img
-            st.experimental_rerun()
-        else:
+        apply_button = st.button("Apply Adjustments")
+
+        if apply_button:
             processed_img = apply_enhancements(img, x, y, zoom, contrast, brightness, sharpness)
             st.session_state.processed_img = processed_img
 
@@ -153,16 +155,18 @@ if uploaded_files:
 
     # Save Options
     save_image = st.checkbox("Save Processed Image")
-    if save_image:
-        st.session_state.processed_img.save("image-processed.jpg")
-        st.success("Image saved as `image-processed.jpg`")
+    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}`")
 
     with st.expander("Save Options"):
         description = st.text_area("Describe the image", "")
         if st.button("Save Description"):
-            with open("saved_image_description.txt", "w") as f:
+            desc_path = "saved_image_description.txt"
+            with open(desc_path, "w") as f:
                 f.write(description)
-            st.success("Description saved as `saved_image_description.txt`")
+            st.success(f"Description saved as `{desc_path}`")
 
         if st.button("Save Image Parameters"):
             params = {
@@ -173,13 +177,18 @@ if uploaded_files:
                 "brightness": brightness,
                 "sharpness": sharpness
             }
-            with open("saved_image_parameters.json", "w") as f:
+            params_path = "saved_image_parameters.json"
+            with open(params_path, "w") as f:
                 f.write(pd.DataFrame([params]).to_json(orient="records"))
-            st.success("Image parameters saved as `saved_image_parameters.json`")
+            st.success(f"Image parameters saved as `{params_path}`")
 
         if st.button("Rename Files"):
             file_ext = str(np.random.randint(100))
-            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")
+            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)
             st.success("Files renamed successfully")