Update app.py

app.py

@@ -1,11 +1,15 @@
 from flask import Flask, render_template
 from flask_socketio import SocketIO, emit
 import subprocess
+import os
 
 app = Flask(__name__)
 app.config['SECRET_KEY'] = 'secret!'
 socketio = SocketIO(app)
 
+# 设置工作目录
+WORKING_DIR = '/code'
+
 @app.route('/')
 def index():
     return render_template('index.html')
@@ -13,6 +17,9 @@ def index():
 @socketio.on('execute_command')
 def handle_command(command):
     try:
+        # 切换到工作目录
+        os.chdir(WORKING_DIR)
+        
         # 执行命令并捕获输出
         process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
         while True: