Update app.py

app.py

@@ -19,6 +19,7 @@ app.config['UPLOAD_FOLDER'] = 'uploads/'
 app.config['JSON_FOLDER'] = 'JSON/'
 app.config['DATA_FOLDER'] = 'data/'
 app.config['MODELS_FOLDER'] = 'Models/'
+app.config['MAIN_FOLDER'] = os.getcwd() 
 
 # Creating Folders if not exists
 os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
@@ -375,11 +376,24 @@ def download_latest_model():
         return redirect(request.referrer)
         
 # Route to serve uploaded files
-'''
-@app.route('/uploads/<filename>')
-def uploaded_file(filename):
-    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
-'''
+@app.route('/<foldername>/zip', methods=['GET'])
+def zip_folder(foldername):
+    # Construct the folder path
+    folder_path = os.path.join(app.config['MAIN_FOLDER'], foldername)
+
+    # Check if the folder exists
+    if not os.path.exists(folder_path) or not os.path.isdir(folder_path):
+        abort(404, description="Folder not found")
+
+    # Define the zip file name and path
+    zip_filename = f"{foldername}.zip"
+    zip_path = os.path.join(app.config['MAIN_FOLDER'], zip_filename)
+
+    # Create a zip file of the folder
+    shutil.make_archive(zip_path[:-4], 'zip', folder_path)
+
+    # Send the zipped file to the client
+    return send_from_directory(app.config['MAIN_FOLDER'], zip_filename, as_attachment=True)
 
 if __name__ == '__main__':
     app.run(debug=True)