Update app.py

app.py

@@ -298,6 +298,9 @@ async def handle_user_input(websocket, process):
                 print(f'Received user input: {event}')
                 try:
                     if event["command"]["type"]=="shell":
+                            await asyncio.sleep(0.1)
+                            if process.returncode is not None:
+                                break
                             process_input(event["command"]["command"], process)
                        
                     elif event["command"]["type"]=="write":