Update app.py

app.py

@@ -7,7 +7,6 @@ import logging
 import json
 import psutil
 from flask import Flask, request, jsonify, render_template, send_file
-import threading
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -16,7 +15,7 @@ logger = logging.getLogger(__name__)
 # Initialize Flask app
 app = Flask(__name__)
 
-connected_cpus = {}
+connected_cpus = {"localhost": {"cpu_count": psutil.cpu_count(logical=False), "usage": 0.0}}
 
 # Endpoint to donate CPU resources
 @app.route('/donate_cpu', methods=['POST'])
@@ -40,14 +39,18 @@ def update_cpu_usage_handler():
     return jsonify({"status": "success"})
 
 # Function to run the provided Python script using MPI
-def run_script(script_name, folder_path):
+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
+        # 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_name], cwd=folder_path, stdout=output_log, stderr=subprocess.STDOUT)
+        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:
@@ -60,11 +63,11 @@ def run_script(script_name, folder_path):
 # Function to handle file uploads and script execution
 @app.route('/upload', methods=['POST'])
 def handle_upload():
-    if 'file' not in request.files or 'script_name' not in request.form:
-        return jsonify({"status": "error", "message": "File or script name not provided"}), 400
+    if 'file' not in request.files or 'script_content' not in request.form:
+        return jsonify({"status": "error", "message": "File or script content not provided"}), 400
     
     files = request.files.getlist('file')
-    script_name = request.form['script_name']
+    script_content = request.form['script_content']
     
     # Create a temporary directory to store uploaded files
     temp_dir = tempfile.mkdtemp()
@@ -77,7 +80,7 @@ def handle_upload():
         file_obj.save(file_path)
 
     # Run the script
-    log_output = run_script(script_name, folder_path)
+    log_output = run_script(script_content, folder_path)
     
     # Create a zip file of the entire folder (including any new files created by the script)
     zip_path = os.path.join(temp_dir, 'output_folder.zip')