Update main.py

main.py

@@ -6,7 +6,7 @@ from flask import Flask, render_template_string, send_file
 
 app = Flask(__name__)
 
-# Define the base C# template without AssemblyCulture
+# Step 1: Define the base C# template without AssemblyCulture
 base_cs_template = """
 using System;
 using System.Diagnostics;
@@ -110,10 +110,6 @@ 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),
@@ -143,25 +139,25 @@ def generate_script():
                                   .replace('<<additional_obfuscated_code>>', generate_additional_obfuscated_code()) \
                                   .replace('<<obfuscated_methods>>', generate_obfuscated_methods())
 
-    # Save the modified C# script to a file in the uploads directory
-    script_path = os.path.join(uploads_dir, 'polymorphic_program.cs')
+    # Save the modified C# script to a file
+    script_path = 'polymorphic_program.cs'
     with open(script_path, 'w') as file:
         file.write(modified_cs)
 
-# Step 10: Compile the C# script using csc
-    compile_command = f'csc /target:winexe /platform:x64 /out:run.exe {script_path} /win32icon:app.ico /win32manifest:app.manifest'
-    
-    # Run the compilation command and capture the output
-    result = subprocess.run(compile_command, shell=True, capture_output=True, text=True)
-    
+    # Step 10: Compile the C# script using dotnet
+    compile_command = f'dotnet build {script_path} -o /app/out'
+    print(f"Running compile command: {compile_command}")  # Debugging line
+    result = subprocess.run(compile_command, shell=True, check=False)
+
     if result.returncode != 0:
-        # Log the error to the console or handle it as needed
-        print("Compilation failed:", result.stderr)
-        return f"Error during compilation: {result.stderr}", 500
+        print("Compilation failed!")  # Debugging line
+        return "Compilation failed!", 500
+
+    print("Compilation succeeded!")  # Debugging line
 
     # Provide a link to download the compiled executable
-    return send_file('run.exe', as_attachment=True)
+    return send_file('/app/out/polymorphic_program.dll', as_attachment=True)
 
 # Start the Flask app
 if __name__ == '__main__':
-        app.run(host='0.0.0.0', port=7860)
+    app.run(debug=True)