Update app.py

app.py

@@ -16,6 +16,11 @@ os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
 os.environ["CUDA_VISIBLE_DEVICES"] = "" # Set the GPU to use, if available
 
 
+if "artifacts" not in st.session_state:
+    st.session_state.artifacts = {}  # {"gif_bytes": ..., "csv_bytes": ...}
+if "processed" not in st.session_state:
+    st.session_state.processed = False
+
 import os
 
 output_root = os.path.join("/tmp", "DICOM_OUTPUTS")
@@ -511,6 +516,19 @@ def clear_dir(path):
     else:
         os.makedirs(path, exist_ok=True)
 
+def process_zip_and_make_artifacts(uploaded_zip):
+    # ... your existing extraction + processing ...
+    # e.g., write GIF to a BytesIO and CSV to bytes
+
+    gif_buf = BytesIO()
+    # anim.save(gif_buf, writer="pillow", format="gif"); gif_buf.seek(0)
+    # For demo, pretend we have bytes:
+    # gif_buf.write(b"..."); gif_buf.seek(0)
+
+    csv_bytes = b"col1,col2\n1,2\n3,4\n"
+
+    return gif_buf.getvalue(), csv_bytes
+
 
 
 # =========================================================
@@ -640,23 +658,25 @@ def main():
             df.to_excel(xlsx_path, index=False)
             print(f"\nCSV written:  {csv_path}")
             print(f"Excel written: {xlsx_path}")
-
-
-            with open(gif_path, "rb") as f:
-                st.download_button(
-                    label="📥 Download GIF",
-                    data=f,
-                    file_name="mask.gif",
-                    mime="image/gif"
-                )
-            
-            with open(csv_path, "rb") as f:
-                st.download_button(
-                    label="Download CSV",
-                    data=f,
-                    file_name="/Cardiac_Volumes_And_Mass_fromPredictions.csv",
-                    mime="text/csv"
-                )
+            st.session_state.processed = True
+
+    if st.session_state.processed == True:
+        
+        with open(gif_path, "rb") as f:
+            st.download_button(
+                label="📥 Download GIF",
+                data=f,
+                file_name="mask.gif",
+                mime="image/gif"
+            )
+        
+        with open(csv_path, "rb") as f:
+            st.download_button(
+                label="Download CSV",
+                data=f,
+                file_name="/Cardiac_Volumes_And_Mass_fromPredictions.csv",
+                mime="text/csv"
+            )