Update app/main.py

app/main.py

@@ -1,10 +1,26 @@
+import os
+import subprocess
 from flask import Flask, render_template
+from flask_socketio import SocketIO, emit
 
 app = Flask(__name__)
+app.config['SECRET_KEY'] = 'secret!'
+socketio = SocketIO(app)
 
 @app.route('/')
 def index():
     return render_template('index.html')
 
+@socketio.on('input', namespace='/terminal')
+def handle_input(data):
+    command = data['command']
+    result = execute_command(command)
+    emit('output', {'output': result})
+
+def execute_command(command):
+    result = subprocess.run(['docker', 'exec', 'terminal-website_ubuntu_1', 'bash', '-c', command],
+                            stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    return result.stdout.decode('utf-8') + result.stderr.decode('utf-8')
+
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    socketio.run(app, host='0.0.0.0', port=7860)