Update app.py

app.py

@@ -5,13 +5,16 @@ import subprocess
 import shlex
 
 async def handle_terminal(websocket, path):
+    prompt_active = False
+    prompt_response = None
+
     try:
         async for message in websocket:
-            # Extract data from WSMessage object
             command = shlex.split(message.data)
             if not command or command[0] in ['rm', 'sudo', 'halt', 'reboot']:
                 await websocket.send_str("Command not allowed")
                 continue
+
             try:
                 process = await asyncio.create_subprocess_exec(
                     *command,
@@ -19,7 +22,19 @@ async def handle_terminal(websocket, path):
                     stderr=asyncio.subprocess.PIPE
                 )
                 async for line in process.stdout:
-                    await websocket.send_str(line.decode('utf-8').rstrip())
+                    # Check if line contains a prompt
+                    line_str = line.decode('utf-8').rstrip()
+                    if line_str.endswith('[Y/n]') or line_str.endswith('[y/n]'):
+                        await websocket.send_str(line_str)
+                        prompt_active = True
+                        # Wait for user response
+                        while prompt_active:
+                            prompt_response = await websocket.receive_str()
+                            if prompt_response.lower() in ['y', 'n']:
+                                prompt_active = False
+                                await websocket.send_str("")  # Signal end of prompt
+                    else:
+                        await websocket.send_str(line_str)
                 async for line in process.stderr:
                     await websocket.send_str(line.decode('utf-8').rstrip())
                 await process.wait()
@@ -73,6 +88,7 @@ async def index(request):
             outline: none;
             width: 100%;
             caret-color: #fff;
+            display: none; /* Hidden by default, shown on prompt */
         }
     </style>
 </head>
@@ -82,18 +98,30 @@ async def index(request):
     </div>
     <input id="input" type="text" autofocus>
     <script>
-        // Use wss:// with Cloudflare handling the secure connection
         const ws = new WebSocket('wss://' + location.host + '/ws');
         const terminal = document.getElementById('terminal');
         const input = document.getElementById('input');
+        let awaitingPrompt = false;
 
         ws.onopen = () => {
             input.focus();
         };
 
         ws.onmessage = (event) => {
-            if (event.data === '') {
-                // New prompt after command execution
+            const data = event.data;
+            if (data.endsWith('[Y/n]') || data.endsWith('[y/n]')) {
+                // Display prompt and show input
+                const promptLine = document.createElement('div');
+                promptLine.className = 'line';
+                promptLine.textContent = data + ' ';
+                terminal.appendChild(promptLine);
+                input.style.display = 'block';
+                input.focus();
+                awaitingPrompt = true;
+            } else if (data === '') {
+                // End of prompt or command, hide input and show new prompt
+                input.style.display = 'none';
+                awaitingPrompt = false;
                 const newLine = document.createElement('div');
                 newLine.className = 'line';
                 newLine.textContent = 'zoyo@zoyos-MacBook-Air 10% ';
@@ -103,12 +131,11 @@ async def index(request):
                 newLine.appendChild(prompt);
                 terminal.appendChild(newLine);
                 input.value = '';
-                input.focus();
             } else {
-                // Display output
+                // Regular output
                 const output = document.createElement('div');
                 output.className = 'line';
-                output.textContent = event.data;
+                output.textContent = data;
                 terminal.appendChild(output);
             }
             terminal.scrollTop = terminal.scrollHeight;
@@ -123,7 +150,13 @@ async def index(request):
         };
 
         input.addEventListener('keydown', (event) => {
-            if (event.key === 'Enter') {
+            if (event.key === 'Enter' && awaitingPrompt) {
+                const response = input.value.trim().toLowerCase();
+                if (response === 'y' || response === 'n') {
+                    ws.send(response);
+                    input.value = '';
+                }
+            } else if (event.key === 'Enter' && !awaitingPrompt) {
                 const command = input.value.trim();
                 if (command) {
                     ws.send(command);