Update main.py

main.py

@@ -160,7 +160,7 @@ async def exec_command(command):
 async def send_prompt(websocket):
     # Send initial prompt to client
     prompt = await get_prompt()
-    await websocket.send(prompt)
+    await websocket.send_text(prompt)
 
 @app.websocket("/shell")
 async def websocket_endpoint(websocket: WebSocket):
@@ -173,14 +173,14 @@ async def websocket_endpoint(websocket: WebSocket):
             command = await websocket.receive_text()
 
             if command == "exit":
-                await websocket.send("Goodbye!")
+                await websocket.send_text("Goodbye!")
                 break
 
             # Execute command
             output = await exec_command(command)
 
             # Send output to client
-            await websocket.send(output)
+            await websocket.send_text(output)
 
             # Send updated prompt to client
             await send_prompt(websocket)