import asyncio
import random
from fastapi import FastAPI, WebSocket, WebSocketDisconnect

app = FastAPI()

def random_8_digit():
    return random.randint(10_000_000, 99_999_999)

@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await websocket.accept()
    try:
        while True:
            number = random_8_digit()
            await websocket.send_json({"number": number})
            await asyncio.sleep(1)  # stream every second
    except WebSocketDisconnect:
        print("Client disconnected")