from fastapi import FastAPI, WebSocket
import asyncio

app = FastAPI()

@app.websocket("/ws")
async def websocket_endpoint(ws: WebSocket):
    await ws.accept()
    for i in range(10):
        await ws.send_text(f"chunk {i}")
        await asyncio.sleep(0.5)
    await ws.close()