Update app.py

app.py

@@ -1207,6 +1207,7 @@ async def start(websocket,events):
     user = User()
     connected = {websocket}
     process_ids = []
+    
 
     join_key = secrets.token_urlsafe(12)
     JOIN[join_key] = user, connected
@@ -1261,22 +1262,34 @@ async def handler(websocket):
 
     """
     # Receive and parse the "init" event from the UI.
-    message = await websocket.recv()
-    event = json.loads(message)
-    # print(event)
-    # project_name = event["project_name"]
-    # assert event["type"] == "init"
-    if event["type"] == "init":
-        if "join" in event:
-            # Second player joins an existing game.
-            await join(websocket, event["join"])
-        else:
-            # First player starts a new game.
-            await start(websocket, message)
-    elif event["type"] == "cmd":
-        # print('executing commad')
-        # Execute a command in the project folder.
-        await execute_command(websocket, event["project_name"], event["command"])
+    try:
+    message_parts = {}
+    async for message in websocket:
+        data = json.loads(message)
+        msg_id = data['msg_id']
+        part = data['part']
+
+        if msg_id not in message_parts:
+            message_parts[msg_id] = []
+        message_parts[msg_id].append(part)
+            
+        if data['end']:
+            full_message = ''.join(message_parts[msg_id])
+            print(f"Received full message: {full_message}")
+    
+            event = json.loads(full_message)
+           
+            if event["type"] == "init":
+                if "join" in event:
+                    # Second player joins an existing game.
+                    await join(websocket, event["join"])
+                else:
+                    # First player starts a new game.
+                    await start(websocket, full_message)
+            elif event["type"] == "cmd":
+                # print('executing commad')
+                # Execute a command in the project folder.
+                await execute_command(websocket, event["project_name"], event["command"])
 
 async def main():
     # Set the stop condition when receiving SIGTERM.