Update app.py

app.py

@@ -28,6 +28,36 @@ streamlit run cell_exp_past.py
 associated files.
 '''
 
+'''
+an image processing tool that allows users to upload microscope images,
+adjust the view with zoom and enhancement controls, and save the processed
+image along with annotations. The tool uses OpenCV for image processing and
+PIL for image enhancements. The processed image can be saved locally or
+exported as a zip file containing the processed image, description, and
+parameters. The tool also provides options to rename the processed image and
+associated files.
+
+The tool consists of the following components:
+1. File Uploader: Allows users to upload microscope images in JPG or PNG format.
+2. Image Controls: Provides sliders to adjust the zoom, contrast, brightness, and sharpness of the image.
+3. Processed Image Display: Displays the processed image after applying the adjustments.
+4. Original Image Display: Displays the original image uploaded by the user.
+5. Save and Export Options: Allows users to add annotations, prepare a zip file for download, save the processed image locally, and rename the processed image and associated files.
+
+To run the tool:
+1. Save the script as `cell_exp_past.py`.
+2. Run the script in a Python environment.
+```python
+streamlit run cell_exp_past.py
+```
+3. Open the provided local URL in a web browser.
+4. Upload microscope images and adjust the image view.
+5. Apply adjustments and save the processed image with annotations.
+6. Download the processed image and annotations as a zip file.
+7. Save the processed image locally or rename the processed image and
+associated files.
+'''
+
 import streamlit as st
 from PIL import Image, ImageEnhance
 import pandas as pd
@@ -45,7 +75,7 @@ def zoom_at(img, x, y, zoom):
                     x + w / zoom2, y + h / zoom2))
     return img.resize((w, h), Image.LANCZOS)
 
-st.title("Cell Explorer")
+st.title("CLL Image Processing Tool")
 
 uploaded_files = st.file_uploader("Upload Images", accept_multiple_files=True, type="jpg")
 
@@ -53,7 +83,7 @@ if uploaded_files:
     img_index = st.selectbox("Select Image", range(len(uploaded_files)))
     x = st.slider("X Coordinate", 0, 500, 205)
     y = st.slider("Y Coordinate", 0, 500, 250)
-    zoom = st.slider("Zoom", 1.0, 10.0, 0.5)
+    zoom = st.slider("Zoom", 1, 10, 0.5)
     contrast = st.slider("Contrast", 0.0, 5.0, 1.0)
     brightness = st.slider("Brightness", 0.0, 5.0, 1.0)
     sharpness = st.slider("Sharpness", 0.0, 2.0, 1.0)
@@ -127,11 +157,30 @@ if uploaded_files:
         st.success("Files renamed successfully")
 
     if st.button("Export to ZIP"):
+        desc_file = st.session_state.get('desc_file', None)
+        params_file = st.session_state.get('params_file', None)
+        processed_image_path = st.session_state.get('processed_image_path', None)
+
         with tempfile.NamedTemporaryFile(delete=False, suffix='.zip') as zipf:
             with zipfile.ZipFile(zipf.name, 'w') as z:
-                if 'desc_file' in locals():
+                if desc_file and os.path.exists(desc_file):
                     z.write(desc_file, os.path.basename(desc_file))
-                if 'params_file' in locals():
+                else:
+                    st.warning("Description file not found and was not added to the ZIP.")
+
+                if params_file and os.path.exists(params_file):
                     z.write(params_file, os.path.basename(params_file))
-        with open(zipf.name, 'rb') as f:
-            st.download_button("Download ZIP", f, "annotations.zip")
+                else:
+                    st.warning("Parameters file not found and was not added to the ZIP.")
+
+                if processed_image_path and os.path.exists(processed_image_path):
+                    z.write(processed_image_path, os.path.basename(processed_image_path))
+                else:
+                    st.warning("Processed image not found and was not added to the ZIP.")
+
+        # Check if the ZIP file has any content
+        if os.path.getsize(zipf.name) > 0:
+            with open(zipf.name, 'rb') as f:
+                st.download_button("Download ZIP", f, "annotations.zip")
+        else:
+            st.error("No files were added to the ZIP. Please ensure files are saved before exporting.")