Update main.py

main.py

@@ -6,7 +6,7 @@ from flask import Flask, render_template_string, send_file
 
 app = Flask(__name__)
 
-# Step 1: Define the base C# template without AssemblyCulture
+# Define the base C# template without AssemblyCulture
 base_cs_template = """
 using System;
 using System.Diagnostics;
@@ -110,6 +110,10 @@ def index():
 
 @app.route('/generate', methods=['POST'])
 def generate_script():
+    # Create uploads directory if it doesn't exist
+    uploads_dir = 'uploads'
+    os.makedirs(uploads_dir, exist_ok=True)
+
     # Step 7: Generate the randomized assembly information using meaningful words
     assembly_info = {
         'title': random.choice(titles),
@@ -139,25 +143,18 @@ def generate_script():
                                   .replace('<<additional_obfuscated_code>>', generate_additional_obfuscated_code()) \
                                   .replace('<<obfuscated_methods>>', generate_obfuscated_methods())
 
-    # Use an absolute path to prevent permission errors
-    script_path = os.path.join('/app/uploads', 'polymorphic_program.cs')
-    try:
-        with open(script_path, 'w') as file:
-            file.write(modified_cs)
-    except PermissionError:
-        return "Permission denied. Could not write the script file.", 500
+    # Save the modified C# script to a file in the uploads directory
+    script_path = os.path.join(uploads_dir, 'polymorphic_program.cs')
+    with open(script_path, 'w') as file:
+        file.write(modified_cs)
 
     # Step 10: Compile the C# script using csc
-    try:
-        compile_command = ['csc', '/target:winexe', '/platform:x64', '/out:run.exe', script_path, '/win32icon:app.ico', '/win32manifest:app.manifest']
-        subprocess.run(compile_command, shell=False, check=True)
-    except subprocess.CalledProcessError as e:
-        return f"Compilation failed: {e}", 500
+    compile_command = f'csc /target:winexe /platform:x64 /out:run.exe {script_path} /win32icon:app.ico /win32manifest:app.manifest'
+    subprocess.run(compile_command, shell=True, check=False)
 
     # Provide a link to download the compiled executable
     return send_file('run.exe', as_attachment=True)
 
-
 # Start the Flask app
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+        app.run(host='0.0.0.0', port=7860)