Update app.py

app.py

@@ -27,6 +27,7 @@ streamlit run cell_exp_past.py
 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
@@ -106,20 +107,33 @@ if uploaded_files:
             try:
                 new_image_name = f"img_processed{file_ext}.jpg"
                 os.rename(processed_image_path, new_image_name)
+                st.session_state['processed_image_path'] = new_image_name  # Update session state
+                st.success(f"Image renamed to {new_image_name}")
             except FileNotFoundError:
                 st.error(f"{processed_image_path} not found.")
+
         else:
             st.error("Processed image not found.")
 
         if params_file and os.path.exists(params_file):
-            new_params_name = f"saved_image_parameters{file_ext}.json"
-            os.rename(params_file, new_params_name)
+            try:
+                new_params_name = f"saved_image_parameters{file_ext}.json"
+                os.rename(params_file, new_params_name)
+                st.session_state['params_file'] = new_params_name  # Update session state
+                st.success(f"Parameters file renamed to {new_params_name}")
+            except FileNotFoundError:
+                st.error(f"{params_file} not found.")
         else:
             st.error("Saved image parameters file not found.")
 
         if desc_file and os.path.exists(desc_file):
-            new_desc_name = f"saved_image_description{file_ext}.txt"
-            os.rename(desc_file, new_desc_name)
+            try:
+                new_desc_name = f"saved_image_description{file_ext}.txt"
+                os.rename(desc_file, new_desc_name)
+                st.session_state['desc_file'] = new_desc_name  # Update session state
+                st.success(f"Description file renamed to {new_desc_name}")
+            except FileNotFoundError:
+                st.error(f"{desc_file} not found.")
         else:
             st.error("Saved image description file not found.")
 
@@ -132,23 +146,28 @@ if uploaded_files:
 
         with tempfile.NamedTemporaryFile(delete=False, suffix='.zip') as zipf:
             with zipfile.ZipFile(zipf.name, 'w') as z:
+                files_added = False  # Flag to check if any file is added
+
                 if desc_file and os.path.exists(desc_file):
                     z.write(desc_file, os.path.basename(desc_file))
+                    files_added = True
                 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))
+                    files_added = True
                 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))
+                    files_added = True
                 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:
+        if os.path.getsize(zipf.name) > 0 and files_added:
             with open(zipf.name, 'rb') as f:
                 st.download_button("Download ZIP", f, "annotations.zip")
         else: