Update app.py

app.py

@@ -1,14 +1,12 @@
 from flask import Flask, request, jsonify, render_template
-import io
-import sys
+import os
 import subprocess
 import tempfile
 import shutil
-import os
 
 app = Flask(__name__)
 
-# Create a temporary directory for user-installed packages
+# Create a temporary directory for operations
 temp_dir = tempfile.mkdtemp()
 
 @app.route("/")
@@ -17,46 +15,46 @@ def index():
 
 @app.route("/execute", methods=["POST"])
 def execute_code():
-    code = request.json.get("code", "").strip()
-    output = io.StringIO()
-    sys.stdout = output
-    sys.stderr = output
+    command = request.json.get("code", "").strip()
+    response = ""
 
     try:
-        if code.startswith("!pip install"):
-            # Handle pip install commands with temporary directory
-            package = code.split("!pip install", 1)[1].strip()
-            if package:
-                result = subprocess.run(
-                    [sys.executable, "-m", "pip", "install", package, "--target", temp_dir],
-                    stdout=subprocess.PIPE,
-                    stderr=subprocess.PIPE,
-                    text=True,
-                )
-                return jsonify({"result": result.stdout + result.stderr})
-            else:
-                return jsonify({"result": "Error: No package specified for installation."})
+        # Ensure all operations happen within the temporary directory
+        os.chdir(temp_dir)
+
+        if command.startswith("!"):
+            # Handle shell commands (e.g., git clone, cd, pip install)
+            shell_command = command[1:]
+            process = subprocess.run(
+                shell_command,
+                shell=True,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                text=True,
+            )
+            response = process.stdout + process.stderr
         else:
-            # Adjust sys.path to include temporary package directory
-            sys.path.insert(0, temp_dir)
-            exec(code)
-            result = output.getvalue()
+            # Handle Python code execution
+            process = subprocess.run(
+                ["python3", "-c", command],
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                text=True,
+            )
+            response = process.stdout + process.stderr
+
     except Exception as e:
-        result = f"Error: {e}"
-    finally:
-        # Reset stdout and stderr
-        sys.stdout = sys.__stdout__
-        sys.stderr = sys.__stderr__
+        response = f"Error: {e}"
 
-    return jsonify({"result": result})
+    return jsonify({"result": response})
 
 @app.route("/cleanup", methods=["POST"])
 def cleanup():
-    # Delete the temporary directory
+    # 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()  # Recreate for the next session
+        temp_dir = tempfile.mkdtemp()
     return jsonify({"result": "Temporary files cleaned up."})
 
 if __name__ == "__main__":