Update app.py

app.py

@@ -279,7 +279,7 @@ async def read_process_output(process, websocket):
             }
             await websocket.send(json.dumps(event))
 
-async def handle_user_input(websocket,key, process):
+async def handle_user_input(websocket,key, process, connected):
     while True:
         try:
             await asyncio.sleep(0.5)
@@ -331,7 +331,7 @@ async def handle_user_input(websocket,key, process):
                                         
                                 
                                         await asyncio.gather(
-                                          handle_user_input(websocket,key, process),
+                                          handle_user_input(websocket,key, process, connected),
                                           read_process_output(process, websocket)
                                         )
                                 
@@ -563,7 +563,7 @@ async def exe(websocket,connected,key):
                             
                     
                             await asyncio.gather(
-                              handle_user_input(websocket,key, process),
+                              handle_user_input(websocket,key, process, connected),
                               read_process_output(process, websocket)
                             )