Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import zipfile
 import pandas as pd
 import gradio as gr
 from helper import assign_main_accounts, generate_schedule, save_processed_files
@@ -6,6 +7,7 @@ from helper import assign_main_accounts, generate_schedule, save_processed_files
 # Global Constants
 UPLOAD_FOLDER = "uploads"
 PROCESSED_FOLDER = "processed"
+ZIP_FILE = "processed_files.zip"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(PROCESSED_FOLDER, exist_ok=True)
 
@@ -67,6 +69,12 @@ def generate_main_accounts():
             os.path.join(PROCESSED_FOLDER, "creators_file.xlsx"), index=False
         )
 
+        # Create a zip file containing all processed files
+        with zipfile.ZipFile(ZIP_FILE, "w") as zipf:
+            for root, _, files in os.walk(PROCESSED_FOLDER):
+                for file in files:
+                    zipf.write(os.path.join(root, file), arcname=file)
+
         return "Main accounts generated successfully!", combined_assignments, processed_creator_file
     except Exception as e:
         return f"Error during main account generation: {e}", None, None
@@ -84,7 +92,13 @@ def update_assignments(assignments_df, creators_df):
         creators_file_path = os.path.join(PROCESSED_FOLDER, "creators_file.xlsx")
         creators_df.to_excel(creators_file_path, index=False)
 
-        return "Assignments and Creator File updated successfully!", PROCESSED_FOLDER
+        # Update the ZIP file with the latest files
+        with zipfile.ZipFile(ZIP_FILE, "w") as zipf:
+            for root, _, files in os.walk(PROCESSED_FOLDER):
+                for file in files:
+                    zipf.write(os.path.join(root, file), arcname=file)
+
+        return "Assignments and Creator File updated successfully!", ZIP_FILE
     except Exception as e:
         return f"Error updating assignments: {e}", None
 
@@ -147,7 +161,7 @@ def app():
 
             generate_main_btn = gr.Button("Generate Main Accounts")
             update_btn = gr.Button("Update and Download Files")
-            download_zip_btn = gr.File(label="Download All Processed Files")
+            download_zip_btn = gr.File(label="Download Processed Files", value=ZIP_FILE)
 
             generate_main_btn.click(
                 generate_main_accounts,
@@ -177,3 +191,4 @@ if __name__ == "__main__":
 
 
 
+