Update app.py

app.py

@@ -20,18 +20,41 @@ async def execute_command(websocket, project_name, command):
     base_path = os.path.join(os.getcwd(), 'projects', project_name)
 
     try:
-        process = subprocess.Popen(
+        process = await asyncio.create_subprocess_shell(
             command,
             cwd=base_path,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE,
-            shell=True
+            stdin=asyncio.subprocess.PIPE,
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
+            # text=True,
         )
 
+        async def send_message(message):
+            print('sending msg')
+            await websocket.send(f'data: {message}')
+
+        async for line in process.stdout:
+            print('sending line')
+            print(line.strip())
+            # await process_input_request(line)
+            await send_message(line.strip())
+            print(line.strip())
+
+        async for line in process.stderr:
+            print(f'error:{line.strip()}')
+            await send_message(f'error:{line.strip()}')
+
+        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}')
+
         # Send the command output to the client
-        output, error = process.communicate()
-        response = f"Command executed:\n\nOutput:\n{output.decode('utf-8')}\nError:\n{error.decode('utf-8')}"
-        await websocket.send(response)
+        # output, error = process.communicate()
+        # response = f"Command executed:\n\nOutput:\n{output.decode('utf-8')}\nError:\n{error.decode('utf-8')}"
+        # await websocket.send(response)
     except Exception as e:
         await websocket.send(e)