Update app.py

app.py

@@ -1281,15 +1281,14 @@ def create_pdf(cname,ename,account_number, directory):
 
 def zip_files_in_folder(directory, output_zip):
     # Initialize the ZIP file
-    directory = "./"
     with zipfile.ZipFile(output_zip, 'w', zipfile.ZIP_DEFLATED) as zipf:
-        # Walk through the directory and add PDF files to the ZIP
-        for root, _, files in os.listdir(directory):
-            for file in files:
-                if file.endswith(".pdf"):  # Only include PDF files
-                    file_path = os.path.join(root, file)
-                    # Add file to the ZIP archive
-                    zipf.write(file_path, file)
+        # List all files in the directory
+        for file in os.listdir(directory):
+            file_path = os.path.join(directory, file)
+            # Check if it's a PDF file and a regular file (not a directory)
+            if file.endswith(".pdf") and os.path.isfile(file_path):
+                # Add file to the ZIP archive
+                zipf.write(file_path, file)  # Save with its filename in the ZIP
     print(f"ZIP file created: {output_zip}")
 
 def directory_exists(repo_id, directory, token):