Update app.py

app.py

@@ -4,9 +4,8 @@ import shutil
 from zipfile import ZipFile
 import logging
 import psutil
-from flask import Flask, request, jsonify, render_template, send_file
-import multiprocessing
 import subprocess
+from flask import Flask, request, jsonify, render_template, send_file
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -91,6 +90,34 @@ def get_cpu_info():
         logger.error(f"Error in get_cpu_info: {e}")
         return jsonify({"status": "error", "message": str(e)}), 500
 
+# Endpoint to execute commands
+@app.route('/execute_command', methods=['POST'])
+def execute_command():
+    try:
+        command = request.form['command']
+        if not command:
+            return jsonify({"status": "error", "message": "No command provided"}), 400
+        
+        # Ensure commands are executed in a safe environment
+        allowed_commands = ['pip install']
+        if not any(command.startswith(cmd) for cmd in allowed_commands):
+            return jsonify({"status": "error", "message": "Command not allowed"}), 400
+        
+        output_log = tempfile.TemporaryFile(mode='w+t')
+        try:
+            result = subprocess.run(command.split(), 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 jsonify({"status": "success", "log_output": log_output})
+    
+    except Exception as e:
+        logger.error(f"Error in execute_command: {e}")
+        return jsonify({"status": "error", "message": str(e)}), 500
+
 # Main interface
 @app.route('/')
 def index():