WebSocket – Real-time Bidirectional Communication

# server.py ─────────────────────────────────────────────────
from fastapi import FastAPI, WebSocket, WebSocketDisconnect
import asyncio, json

app = FastAPI()

@app.websocket("/ws/analyze/{job_id}")
async def analysis_stream(websocket: WebSocket, job_id: str):
    await websocket.accept()   # WebSocket 연결 수락 (101 Switching Protocols)
    try:
        # 클라이언트로부터 분석 요청 수신
        data = await websocket.receive_json()
        prompt = data.get("prompt", "비디오를 설명해주세요")

        # 진행 상황을 실시간 스트리밍
        stages = [
            ("processing", "ffmpeg로 프레임 추출 중..."),
            ("processing", "워커 A (27B 모델) 분석 중..."),
            ("processing", "워커 B (9B 모델) 분석 중..."),
            ("done",       "분석 완료! 결과: 영상에는 고양이가 등장합니다."),
        ]
        for status, message in stages:
            await websocket.send_json({"status": status, "message": message})
            await asyncio.sleep(1)  # 실제로는 AI 처리 대기

    except WebSocketDisconnect:
        print(f"클라이언트 연결 끊김: job_id={job_id}")

# client.py ─────────────────────────────────────────────────
import asyncio, websockets, json

async def watch_analysis(job_id: str, prompt: str):
    uri = f"ws://localhost:8000/ws/analyze/{job_id}"
    async with websockets.connect(uri) as ws:
        # 분석 요청 전송
        await ws.send(json.dumps({"prompt": prompt}))
        print(f"분석 요청 전송: {prompt}")

        # 서버가 보내는 업데이트를 실시간 수신
        async for raw_message in ws:
            msg = json.loads(raw_message)
            emoji = "✅" if msg["status"] == "done" else "⏳"
            print(f"  {emoji} {msg['message']}")
            if msg["status"] == "done":
                break

asyncio.run(watch_analysis("job-123", "이 영상에 무엇이 나오나요?"))
# uvicorn server:app 실행 후 위 클라이언트 코드 별도 실행