Update app.py

app.py

@@ -3,8 +3,11 @@ import os
 import subprocess
 import tempfile
 import shutil
+from flask_socketio import SocketIO, emit
 
 app = Flask(__name__)
+app.config['SECRET_KEY'] = 'secret!'
+socketio = SocketIO(app)
 
 # Create a temporary directory for operations
 temp_dir = tempfile.mkdtemp()
@@ -13,49 +16,56 @@ temp_dir = tempfile.mkdtemp()
 def index():
     return render_template("index.html")
 
-@app.route("/execute", methods=["POST"])
-def execute_code():
-    command = request.json.get("code", "").strip()
+@socketio.on("execute")
+def execute_command(data):
+    command = data.get("code", "").strip()
     response = ""
 
     try:
-        # Ensure all operations happen within the temporary directory
-        os.chdir(temp_dir)
+        os.chdir(temp_dir)  # Ensure all operations happen within the temporary directory
+        emit("output", {"data": f"$ {command}\n"}, broadcast=True)
 
         if command.startswith("!"):
-            # Handle shell commands (e.g., git clone, cd, pip install)
             shell_command = command[1:]
-            process = subprocess.run(
+            process = subprocess.Popen(
                 shell_command,
                 shell=True,
                 stdout=subprocess.PIPE,
                 stderr=subprocess.PIPE,
                 text=True,
             )
-            response = process.stdout + process.stderr
+            # Stream output live
+            for line in process.stdout:
+                emit("output", {"data": line}, broadcast=True)
+            for line in process.stderr:
+                emit("output", {"data": line}, broadcast=True)
+
+            process.wait()
+            emit("output", {"data": "Command execution complete.\n"}, broadcast=True)
         else:
-            # Handle Python code execution
-            process = subprocess.run(
+            # Handle Python script execution
+            process = subprocess.Popen(
                 ["python3", "-c", command],
                 stdout=subprocess.PIPE,
                 stderr=subprocess.PIPE,
                 text=True,
             )
-            response = process.stdout + process.stderr
-
+            for line in process.stdout:
+                emit("output", {"data": line}, broadcast=True)
+            for line in process.stderr:
+                emit("output", {"data": line}, broadcast=True)
+            process.wait()
+            emit("output", {"data": "Python execution complete.\n"}, broadcast=True)
     except Exception as e:
-        response = f"Error: {e}"
-
-    return jsonify({"result": response})
+        emit("output", {"data": f"Error: {e}\n"}, broadcast=True)
 
-@app.route("/cleanup", methods=["POST"])
+@socketio.on("cleanup")
 def cleanup():
-    # Clean up the temporary directory and recreate it
     global temp_dir
     if os.path.exists(temp_dir):
         shutil.rmtree(temp_dir)
         temp_dir = tempfile.mkdtemp()
-    return jsonify({"result": "Temporary files cleaned up."})
+    emit("output", {"data": "Temporary files cleaned up.\n"}, broadcast=True)
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    socketio.run(app, host="0.0.0.0", port=7860)