Update app/main.py

app/main.py

@@ -14,16 +14,21 @@ def index():
 @socketio.on('input', namespace='/terminal')
 def handle_input(data):
     command = data['command']
-    print(f"Received command: {command}") # Debugging log
+    print(f"Received command: {command}")  # Debugging log
     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)
-    output = result.stdout.decode('utf-8') + result.stderr.decode('utf-8')
-    print(f"Command output: {output}") # Debugging log
-    return output
+    try:
+        result = subprocess.run(['docker', 'exec', 'terminal-website_ubuntu_1', 'bash', '-c', command],
+                                stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True)
+        output = result.stdout.decode('utf-8') + result.stderr.decode('utf-8')
+        print(f"Command output: {output}")  # Debugging log
+        return output
+    except subprocess.CalledProcessError as e:
+        error_output = e.stdout.decode('utf-8') + e.stderr.decode('utf-8')
+        print(f"Command error: {error_output}")  # Debugging log
+        return error_output
 
 if __name__ == '__main__':
     socketio.run(app, host='0.0.0.0', port=7860)