Update app.py

app.py

@@ -4,6 +4,8 @@ import numpy as np
 import os
 import pandas as pd
 import openpyxl
+import zipfile
+import io
 
 # Load the preprocessor
 preprocessor_path = 'modelExports/preprocessor.pkl'
@@ -324,7 +326,7 @@ elif interface == "Batch Prediction":
         st.header('Download Predictions')
         download_option = st.radio(
             "Choose how to download your predictions:",
-            ("All Models in Separate Files", "Churn and Non-Churn in Separate Files")
+            ("All Models in Separate Files", "Churn and Non-Churn in Separate Files", "Download All at Once")
         )
 
         if download_option == "All Models in Separate Files":
@@ -358,6 +360,21 @@ elif interface == "Batch Prediction":
                     file_name=f'{model_name.lower().replace(" ", "_")}_non_churn_predictions.csv',
                     mime='text/csv',
                 )
+        elif download_option == "Download All at Once":
+            # Create a zip file containing all outputs
+            zip_buffer = io.BytesIO()
+            with zipfile.ZipFile(zip_buffer, "w") as zip_file:
+                for model_name, output_df in model_outputs.items():
+                    csv_data = output_df.to_csv(index=False).encode('utf-8')
+                    zip_file.writestr(f'{model_name.lower().replace(" ", "_")}_predictions.csv', csv_data)
+            zip_buffer.seek(0)
+
+            st.download_button(
+                label="Download All Predictions as ZIP",
+                data=zip_buffer,
+                file_name='all_model_predictions.zip',
+                mime='application/zip'
+            )
     else:
         st.info('Awaiting CSV or Excel file to be uploaded.')
 
@@ -367,4 +384,4 @@ st.sidebar.write(f"Total models available: {len(models)}")
 st.sidebar.write(f"Models selected for prediction: {len(selected_models)}")
 st.sidebar.write("### Model Accuracies")
 for model, accuracy in model_accuracies.items():
-    st.sidebar.write(f"{model}: {accuracy}%")
+    st.sidebar.write(f"{model}: {accuracy}%")