Update utils.py

utils.py

@@ -209,11 +209,6 @@ def create_encrypted_7z_archive(file_path, password):
     
     return archive_path
     
-@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:
@@ -284,16 +279,15 @@ def process_request(request):
         # Create an encrypted 7z archive
         archive_path = create_encrypted_7z_archive(modified_pe_path, 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_filename": archive_filename,
+        # Return the .7z file and password in the response
+        return send_file(
+            archive_path,
+            as_attachment=True,
+            download_name=os.path.basename(archive_path),
+            mimetype='application/octet-stream'
+        ), {
             "password": password
-        })
+        }
 
     except Exception as e:
         current_app.logger.error(f"An error occurred: {str(e)}")