Update app.py

app.py

@@ -6,7 +6,6 @@ import zipfile
 import tempfile
 from pathlib import Path
 from typing import Optional, List, Dict
-import shutil  # <-- added for optional local copy
 
 import numpy as np
 import pandas as pd
@@ -26,6 +25,7 @@ from utils.layer_util import ResizeAndConcatenate
 #      Config / paths
 # =========================
 os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
+# Uncomment next line to force CPU only
 # os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
 MODEL_PATH = "./models_final/SEG459.h5"
@@ -434,7 +434,7 @@ def process_nifti_case(nifti_path: str, model, rows_acc: List[Dict], per_frame_r
     per_frame_df.insert(0, "Patient_ID", pid)
     per_frame_rows_acc.append(per_frame_df)
 
-    # Append summary row BEFORE any UI drawing
+    # -------- Append summary row BEFORE any UI drawing (so UI errors don't drop data)
     rows_acc.append({
         'Patient_ID': pid,
         'EDV_uL': EDV_uL,
@@ -450,9 +450,10 @@ def process_nifti_case(nifti_path: str, model, rows_acc: List[Dict], per_frame_r
         'PixelSpacing_col_mm': spacing['col_mm'],
     })
 
-    # GIF: ED vs ES (slices animate)
+    # GIF: ED vs ES (slices animate) — non-fatal display
     gif_path = gif_animation_for_patient_pred_only(imgs_4d, preds_4d, pid, ed_idx, es_idx, GIFS_DIR)
     try:
+        # Streamlit uses 'use_column_width'; keep it, but guard against TypeError on older builds
         st.image(gif_path, caption="Generated GIF", use_column_width=True)
     except TypeError:
         st.image(gif_path, caption="Generated GIF")
@@ -477,20 +478,8 @@ def main():
     if uploaded_zip is None:
         st.stop()
 
-    # --- Save options UI (filename + optional local copy) ---
-    st.subheader("Save options")
-    csv_download_name = st.text_input("CSV file name (for browser download)", "Results.csv")
-    st.caption("Tip: To choose a folder on download, enable your browser setting “Ask where to save each file before downloading.”")
-
-    save_local_copy = st.checkbox("Also save a copy to a specific path on this machine (advanced)")
-    local_dir = ""
-    if save_local_copy:
-        local_dir = st.text_input("Local directory path (only works on the machine running Streamlit)", os.path.expanduser("~"))
-
     if st.button("Process Data"):
-        # Show real uploaded ZIP name in spinner
-        zip_label = getattr(uploaded_zip, "name", "ZIP file")
-        with st.spinner(f"Processing {zip_label}..."):
+        with st.spinner("Processing ZIP..."):
             # Save & extract
             tmpdir = tempfile.mkdtemp()
             zpath = os.path.join(tmpdir, "upload.zip")
@@ -514,7 +503,7 @@ def main():
             model = keras.models.load_model(
                 MODEL_PATH,
                 custom_objects={
-                    'focal_tversky_loss': None,
+                    'focal_tversky_loss': None,           # loss/metrics not needed for inference
                     'dice_coef_no_bkg': None,
                     'ResizeAndConcatenate': ResizeAndConcatenate,
                     'dice_myo': None,
@@ -534,28 +523,16 @@ def main():
                 except Exception as e:
                     st.warning(f"Failed: {Path(fp).name} — {e}")
 
-            # Write merged CSV
+            # Write merged CSV and show download
             csv_path = write_all_in_one_csv(rows, per_frame_rows, CSV_DIR)
             st.success(f"Processed {len(rows)} NIfTI file(s).")
-
-            # Browser download (browser decides save location; filename is set)
             with open(csv_path, "rb") as f:
                 st.download_button(
                     label="Download CSV",
                     data=f,
-                    file_name=csv_download_name,
+                    file_name="Results.csv",
                     mime="text/csv"
                 )
 
-            # Optional local copy (server filesystem)
-            if save_local_copy and local_dir.strip():
-                try:
-                    os.makedirs(local_dir, exist_ok=True)
-                    dest = os.path.join(local_dir, csv_download_name)
-                    shutil.copyfile(csv_path, dest)
-                    st.success(f"Saved CSV to: {dest}")
-                except Exception as e:
-                    st.error(f"Could not save CSV: {e}")
-
 if __name__ == "__main__":
-    main()
+    main()