download img csv

app.py

@@ -19,7 +19,7 @@ import pandas as pd
 from itertools import zip_longest
 import inspect
 from threading import Lock
-
+import zipfile
 import warnings
 
 # Ignore SourceChangeWarning
@@ -202,17 +202,29 @@ def create_csv():
 def get_data():
     return send_from_directory('inferenced','output.csv', as_attachment=False)
 
-from flask import jsonify
 
 @app.route('/download_csv', methods=['GET'])
 def download_csv():
     try:
-        output_file_path = r"inferenced/output.csv"  # path to output CSV file
-        # Check if the file exists
-        if os.path.exists(output_file_path):
-            return send_file(output_file_path, as_attachment=True, download_name='output.csv')
+        folder_path = r"inferenced"  # path to the folder
+        
+        # Get a list of all files in the folder
+        files = [f for f in os.listdir(folder_path) if os.path.isfile(os.path.join(folder_path, f))]
+        
+        # Check if there are any files in the folder
+        if files:
+            # Create a ZIP file containing all files
+            zip_filename = 'inferenced_files.zip'
+            zip_file_path = os.path.join(folder_path, zip_filename)
+            with zipfile.ZipFile(zip_file_path, 'w') as zipf:
+                for file in files:
+                    file_path = os.path.join(folder_path, file)
+                    zipf.write(file_path, os.path.basename(file_path))
+            
+            # Send the ZIP file as an attachment
+            return send_file(zip_file_path, as_attachment=True, download_name=zip_filename)
         else:
-            return jsonify({"error": "CSV file not found"})
+            return jsonify({"error": "No files found in the folder"})
     except Exception as e:
         return jsonify({"error": f"Download failed: {str(e)}"})