Update app.py

app.py

@@ -6,6 +6,7 @@ from zipfile import ZipFile
 import logging
 import json
 import psutil
+import multiprocessing
 from flask import Flask, request, jsonify, render_template, send_file
 
 # Configure logging
@@ -38,27 +39,32 @@ def update_cpu_usage_handler():
         logger.info(f"Updated CPU usage for {host}: {usage}%")
     return jsonify({"status": "success"})
 
-# Function to run the provided Python script using MPI
+# Function to run the provided Python script using multiprocessing
 def run_script(script_content, folder_path):
     script_path = os.path.join(folder_path, 'user_script.py')
     with open(script_path, 'w') as script_file:
         script_file.write(script_content)
     
-    output_log = tempfile.TemporaryFile(mode='w+t')
-    try:
-        # Collect all available CPUs including the local host CPU
-        total_cpus = sum(cpu['cpu_count'] for cpu in connected_cpus.values())
-        
-        # Run the script using MPI
-        result = subprocess.run(['mpiexec', '-n', str(total_cpus), 'python', script_path], cwd=folder_path, stdout=output_log, stderr=subprocess.STDOUT)
-        output_log.seek(0)
-        log_output = output_log.read()
-    except Exception as e:
-        log_output = str(e)
-    finally:
-        output_log.close()
+    def target_function(cpu_id):
+        output_log = tempfile.TemporaryFile(mode='w+t')
+        try:
+            result = subprocess.run(['python', script_path], cwd=folder_path, stdout=output_log, stderr=subprocess.STDOUT)
+            output_log.seek(0)
+            log_output = output_log.read()
+        except Exception as e:
+            log_output = str(e)
+        finally:
+            output_log.close()
+        return log_output
+
+    # Collect all available CPUs including the local host CPU
+    total_cpus = sum(cpu['cpu_count'] for cpu in connected_cpus.values())
     
-    return log_output
+    # Run the script using multiprocessing
+    with multiprocessing.Pool(total_cpus) as pool:
+        log_outputs = pool.map(target_function, range(total_cpus))
+
+    return '\n'.join(log_outputs)
 
 # Function to handle file uploads and script execution
 @app.route('/upload', methods=['POST'])