Update app.py

app.py

@@ -33,7 +33,7 @@ async def execute_command(websocket, project_name, command):
         response = f"Command executed:\n\nOutput:\n{output.decode('utf-8')}\nError:\n{error.decode('utf-8')}"
         websockets.broadcast(connected, response)
     except Exception as e:
-        websockets.broadcast(connected, response)
+        websockets.broadcast(connected, e)
 
 async def create_file_structure(websocket, data, base_path='.'):
     if data['type'] == 'folder':
@@ -213,7 +213,8 @@ async def handler(websocket):
     # Receive and parse the "init" event from the UI.
     message = await websocket.recv()
     event = json.loads(message)
-    assert event["type"] == "init"
+    project_name = event["project_name"]
+    # assert event["type"] == "init"
 
     if "join" in event:
         # Second player joins an existing game.
@@ -221,10 +222,25 @@ async def handler(websocket):
     elif "watch" in event:
         # Spectator watches an existing game.
         await watch(websocket, event["watch"])
+    elif "cmd" in event:
+        
+        await execute_command(websocket, event["project_name"], event["command"])
     else:
         # First player starts a new game.
         await start(websocket, message)
 
+    # assert event["type"] == "cmd"
+
+    # if "cmd" in event:
+    #     # Second player joins an existing game.
+    #     await execute_command(websocket, project_name, event["command"])
+    # elif "watch" in event:
+    #     # Spectator watches an existing game.
+    #     await watch(websocket, event["watch"])
+    # else:
+    #     # First player starts a new game.
+    #     await start(websocket, message)
+
 
 async def main():
     # Set the stop condition when receiving SIGTERM.