Update utils.py

utils.py

@@ -3,12 +3,14 @@ import subprocess
 import random
 import string
 from datetime import datetime
-from flask import jsonify, send_file, current_app, url_for
+from flask import Flask, jsonify, send_file, current_app, url_for
 import shutil
 import tempfile
 import requests
 import json
 
+app = Flask(__name__)
+
 BASE_DIR = os.path.abspath(os.path.dirname(__file__))
 UPLOAD_FOLDER = os.path.join(BASE_DIR, "uploads")
 PE_FOLDER = os.path.join(BASE_DIR, "pe")
@@ -18,6 +20,8 @@ OBFUSCATOR_SCRIPT = os.path.join(BASE_DIR, "Obfus", "main.ps1")
 UPLOAD_URL = 'https://ambelo-benjamin.hf.space/upload'
 POWERSHELL_FILE_PATH = os.path.join(PE_FOLDER, "powershell.ps1")
 
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+
 def generate_random_string(length=8):
     return ''.join(random.choices(string.ascii_letters + string.digits, k=length))
 
@@ -190,7 +194,8 @@ def replace_url_in_exe(file_path, old_url, new_url, old_string, new_string):
     except Exception as e:
         raise Exception(f"An error occurred: {e}")
 
-def process_request(request):
+@app.route('/process', methods=['POST'])
+def process_request():
     temp_dir = None  # Initialize temp_dir to be used in the finally block
     try:
         # Save the incoming binary file
@@ -272,4 +277,11 @@ def process_request(request):
     finally:
         # Clean up temporary directories and files
         if temp_dir and os.path.exists(temp_dir):
-            shutil.rmtree(temp_dir, ignore_errors=True)
+            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)
+
+if __name__ == '__main__':
+    app.run(debug=True)