Update app.py

app.py

@@ -1,75 +1,78 @@
-from flask import Flask, render_template, request, jsonify, send_from_directory
+from flask import Flask, request, jsonify, render_template
 import os
+import subprocess
 import tempfile
 import shutil
-import subprocess
-from gtts import gTTS
 
 app = Flask(__name__)
+BASE_DIR = tempfile.mkdtemp()  # Temporary directory for executing user scripts
+
+
+def cleanup_temp_dir():
+    """Cleans up the temporary directory."""
+    try:
+        shutil.rmtree(BASE_DIR)
+    except Exception as e:
+        print(f"Error cleaning up temp dir: {e}")
 
-# Temporary directory for running commands and storing files
-temp_dir = tempfile.mkdtemp()
-UPLOAD_FOLDER = os.path.join(temp_dir, "uploads")
-os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
 @app.route('/')
 def index():
+    """Render the main terminal interface."""
     return render_template('index.html')
 
+
 @app.route('/run', methods=['POST'])
 def run_command():
-    command = request.json.get("command", "").strip()
-    result = {"output": "", "error": None, "audio_file": None}
+    """Executes Python scripts or commands."""
+    user_input = request.json.get('code', '')
+    response = {"output": "", "error": ""}
 
-    try:
-        # If the command starts with 'python', execute as Python code
-        if command.startswith("python"):
-            script_code = command.replace("python", "").strip()
-            python_file = os.path.join(temp_dir, "temp_script.py")
-            with open(python_file, "w") as f:
-                f.write(script_code)
-            process = subprocess.run(
-                ["python3", python_file], 
-                capture_output=True, 
-                text=True,
-                cwd=temp_dir
-            )
-            result["output"] = process.stdout
-            if process.stderr:
-                result["error"] = process.stderr
-
-        # Handle shell commands like `git clone`
-        else:
-            process = subprocess.run(
-                command, 
-                shell=True, 
-                capture_output=True, 
-                text=True,
-                cwd=temp_dir
-            )
-            result["output"] = process.stdout
-            if process.stderr:
-                result["error"] = process.stderr
-
-        # Check for generated audio file
-        for file in os.listdir(temp_dir):
-            if file.endswith(".mp3"):
-                result["audio_file"] = file
+    # Save the user's Python script to a temporary file
+    temp_file_path = os.path.join(BASE_DIR, "user_script.py")
+    with open(temp_file_path, 'w') as script_file:
+        script_file.write(user_input)
 
+    # Execute the script and capture output
+    try:
+        process = subprocess.Popen(
+            ["python3", temp_file_path],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True
+        )
+        stdout, stderr = process.communicate()
+        response["output"] = stdout
+        response["error"] = stderr
     except Exception as e:
-        result["error"] = str(e)
+        response["error"] = str(e)
+
+    # Cleanup temporary file
+    if os.path.exists(temp_file_path):
+        os.remove(temp_file_path)
+
+    return jsonify(response)
+
+
+@app.route('/clean', methods=['POST'])
+def clean():
+    """Cleans the temporary directory when the user leaves."""
+    cleanup_temp_dir()
+    return jsonify({"status": "Temporary files cleaned."})
+
+
+@app.before_first_request
+def setup():
+    """Set up the temporary directory before the first request."""
+    if not os.path.exists(BASE_DIR):
+        os.makedirs(BASE_DIR)
 
-    return jsonify(result)
 
-@app.route('/download/<filename>')
-def download_file(filename):
-    return send_from_directory(temp_dir, filename, as_attachment=True)
+@app.teardown_appcontext
+def teardown(exception):
+    """Clean up after the app shuts down."""
+    cleanup_temp_dir()
 
-@app.route('/cleanup', methods=['POST'])
-def cleanup():
-    shutil.rmtree(temp_dir)
-    os.makedirs(temp_dir, exist_ok=True)
-    return jsonify({"status": "Temporary files deleted."})
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     app.run(host="0.0.0.0", port=7860)