Update app.py

app.py

@@ -1,43 +1,37 @@
 from flask import Flask, render_template
 from flask_socketio import SocketIO, emit
-import subprocess
+import pty
 import os
+import subprocess
+import threading
 
 app = Flask(__name__)
-app.config['SECRET_KEY'] = 'secret!'
+app.config['SECRET_KEY'] = 'your_secret_key'
 socketio = SocketIO(app)
 
-# 设置工作目录
-WORKING_DIR = '/tmp'
+def run_shell():
+    master_fd, slave_fd = pty.openpty()
+    process = subprocess.Popen(['bash'], stdin=slave_fd, stdout=slave_fd, stderr=slave_fd, text=True)
+
+    while True:
+        output = os.read(master_fd, 1024).decode('utf-8')
+        if output:
+            socketio.emit('output', output)
+
+        # 这里可以添加一个小的延迟，避免CPU占用过高
+        socketio.sleep(0.1)
 
 @app.route('/')
 def index():
     return render_template('index.html')
 
-@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:
-            output = process.stdout.readline()
-            if output == '' and process.poll() is not None:
-                break
-            if output:
-                emit('command_output', {'output': output.strip()})
-        
-        # 捕获错误输出
-        error = process.stderr.read()
-        if error:
-            emit('command_output', {'output': error.strip()})
-        
-        # 发送命令执行结束信号
-        emit('command_output', {'output': '[Command execution finished]'})
-    except Exception as e:
-        emit('command_output', {'output': str(e)})
+@socketio.on('input')
+def handle_input(input_data):
+    # 将输入写入伪终端
+    os.write(shell_master_fd, (input_data + '\n').encode('utf-8'))
 
 if __name__ == '__main__':
-    socketio.run(app, host='0.0.0.0', port=7860, allow_unsafe_werkzeug=True)
+    shell_master_fd, shell_slave_fd = pty.openpty()
+    shell_thread = threading.Thread(target=run_shell)
+    shell_thread.start()
+    socketio.run(app, host='0.0.0.0', port=7860)