Update app.py

app.py

@@ -341,14 +341,14 @@ async def handle_user_input(websocket, process):
             await asyncio.sleep(0.1)
             if process.returncode is not None:
                 break
-            message = await websocket.recv()
-            user_input = json.loads(message)
-            print(user_input)
-            print(f'Received user input: {user_input["command"]["command"]}')
-            if not is_valid_input(user_input):
-                await websocket.send("Invalid input. Please try again.")
-                continue
-            process_input(user_input, process)
+            # message = await websocket.recv()
+            async for message in websocket:
+                user_input = json.loads(message)
+                print(user_input)
+                print(f'Received user input: {user_input["command"]["command"]}')
+                
+                process_input(user_input, process)
+                break
         except websockets.ConnectionClosed:
             print("WebSocket connection closed")
             break