Update app.py

app.py

@@ -73,6 +73,7 @@ from PyPDFForm import FormWrapper
 
 import os
 
+import zipfile
 # Get the current working directory
 current_folder = os.getcwd()
 #Variables Initialization
@@ -1209,6 +1210,9 @@ def handle_large_dataset(df, create_document,isDataFrame):
                 # Create a PDF for each row
                 create_pdf(row['COMPANY'], row['EMPLOYEE NAME'], row['ACCOUNT NUMBER'],directory)
             create_document = False
+            zip_name = 'output_files.zip'
+            zip_files_in_folder(directory, zip_name)
+            create_file_HF(zip_name,directory, False)
             docstatus = f"Please download the complete dataset here: <a href='https://redmindtechnologies.com/RedMindGPT/output.zip' download>Download</a>. {total_rows} documents are created successfully."
         print(sample_table)
         # 5. Return the summary and downloadable link
@@ -1268,6 +1272,19 @@ def create_pdf(cname,ename,account_number, directory):
     
     return f"{output_file_name} is created successfully."
 
+def zip_files_in_folder(folder_path, zip_name):
+    # Create a Zip file
+    folder_path="./"
+    with zipfile.ZipFile(zip_name, 'w', zipfile.ZIP_DEFLATED) as zipf:
+        # Walk through the folder
+        for root, dirs, files in os.walk(folder_path):
+            for file in files:
+                file_path = os.path.join(root, file)
+                # Add each file to the zip file
+                zipf.write(file_path, os.path.relpath(file_path, folder_path))
+
+    print(f"Zip file '{zip_name}' created successfully!")
+
 
 def directory_exists(repo_id, directory, token):
     try: