Update app.py

app.py

@@ -282,6 +282,41 @@ async def execute_command(websocket, key,project_name, command, connected):
             # await websocket.send(f'data: {message}')
             websockets.broadcast(connected,json.dumps(message) )
 
+        # async for line in process.stdout:
+        #     print('sending line')
+            
+        #     event = {
+        #         "type": "terminal-data",
+        #         "data": line.strip().decode('utf-8'),
+        #     }
+        #     await send_message(event)
+            
+
+        # async for line in process.stderr:
+        #     print(f'error:{line.strip()}')
+        #     event = {
+        #         "type": "terminal-error",
+        #         "data": line.strip().decode('utf-8'),
+        #     }
+        #     await send_message(event)
+
+        await asyncio.gather(
+          handle_user_input(websocket, process),
+          read_process_output(process, websocket)
+        )
+
+        return_code = await process.wait()
+        if return_code == 0:
+            # await send_message('Code executed successfully')
+            pass
+        else:
+            # await send_message(f'error:Execution failed with return code {return_code}')
+            pass
+
+    except Exception as e:
+        await error(websocket, str(e))
+
+async def read_process_output(process, websocket):
         async for line in process.stdout:
             print('sending line')
             
@@ -289,7 +324,7 @@ async def execute_command(websocket, key,project_name, command, connected):
                 "type": "terminal-data",
                 "data": line.strip().decode('utf-8'),
             }
-            await send_message(event)
+            await websocket.send(event)
             
 
         async for line in process.stderr:
@@ -298,18 +333,35 @@ async def execute_command(websocket, key,project_name, command, connected):
                 "type": "terminal-error",
                 "data": line.strip().decode('utf-8'),
             }
-            await send_message(event)
+            await websocket.send(event)
 
-        return_code = await process.wait()
-        if return_code == 0:
-            # await send_message('Code executed successfully')
-            pass
-        else:
-            # await send_message(f'error:Execution failed with return code {return_code}')
-            pass
-
-    except Exception as e:
-        await error(websocket, str(e))
+async def handle_user_input(websocket, process):
+    while True:
+        try:
+            if process.returncode is not None:
+                break
+            message = await websocket.recv()
+            user_input = message.strip()
+            print(f"Received user input: {user_input}")
+            if not is_valid_input(user_input):
+                await websocket.send("Invalid input. Please try again.")
+                continue
+            process_input(user_input, process)
+        except websockets.ConnectionClosed:
+            print("WebSocket connection closed")
+            break
+        except Exception as e:
+            print(f"Error in input thread: {str(e)}")
+
+def process_input(user_input, process):
+    if process:
+        try:
+            process.stdin.write(user_input.encode('utf-8') + b'\n')
+            #process.stdin.flush()
+        except Exception as e:
+            print(f"Error writing to process stdin: {str(e)}")
+    else:
+        print("No process available to write to.")
 
 async def create_file_structure(websocket, data, base_path='.'):
     if data['type'] == 'folder':