Update utils.py

utils.py

@@ -208,8 +208,12 @@ def create_encrypted_7z_archive(file_path, password):
         raise Exception(f"Failed to create encrypted 7z archive: {e}")
     
     return archive_path
-
-# Function to process the request
+    
+@app.route('/download/<filename>', methods=['GET'])
+def download_file(filename):
+    # Serve the .7z file from the UPLOAD_FOLDER
+    return send_from_directory(UPLOAD_FOLDER, filename, as_attachment=True)
+    
 def process_request(request):
     temp_dir = None  # Initialize temp_dir to be used in the finally block
     try:
@@ -280,9 +284,14 @@ def process_request(request):
         # Create an encrypted 7z archive
         archive_path = create_encrypted_7z_archive(modified_pe_path, password)
 
-        # Return the encrypted archive and the password
+        # Move the .7z file to the UPLOAD_FOLDER
+        archive_filename = os.path.basename(archive_path)
+        new_archive_path = os.path.join(UPLOAD_FOLDER, archive_filename)
+        shutil.move(archive_path, new_archive_path)
+
+        # Return the archive filename and password
         return jsonify({
-            "archive_path": archive_path,
+            "archive_filename": archive_filename,
             "password": password
         })