hgzero/ai-python/app/api/v1/suggestions.py

"""AI 제안사항 SSE 엔드포인트"""
from fastapi import APIRouter
from sse_starlette.sse import EventSourceResponse
import logging
import asyncio
from typing import AsyncGenerator
from app.models import RealtimeSuggestionsResponse
from app.services.claude_service import ClaudeService
from app.services.redis_service import RedisService
from app.config import get_settings

logger = logging.getLogger(__name__)
router = APIRouter()
settings = get_settings()

# 서비스 인스턴스
claude_service = ClaudeService()


@router.get("/meetings/{meeting_id}/stream")
async def stream_ai_suggestions(meeting_id: str):
    """
    실시간 AI 제안사항 SSE 스트리밍

    Args:
        meeting_id: 회의 ID

    Returns:
        Server-Sent Events 스트림
    """
    logger.info(f"SSE 스트림 시작 - meetingId: {meeting_id}")

    async def event_generator() -> AsyncGenerator:
        """SSE 이벤트 생성기"""
        redis_service = RedisService()

        try:
            # Redis 연결
            await redis_service.connect()

            previous_count = 0

            while True:
                # 현재 세그먼트 개수 확인
                current_count = await redis_service.get_segment_count(meeting_id)

                # 임계값 이상이고, 이전보다 증가했으면 분석
                if (current_count >= settings.min_segments_for_analysis
                        and current_count > previous_count):

                    # 누적된 텍스트 조회
                    accumulated_text = await redis_service.get_accumulated_text(meeting_id)

                    if accumulated_text:
                        # Claude API로 분석
                        suggestions = await claude_service.analyze_suggestions(accumulated_text)

                        if suggestions.suggestions:
                            # SSE 이벤트 전송
                            yield {
                                "event": "ai-suggestion",
                                "id": str(current_count),
                                "data": suggestions.json()
                            }

                            logger.info(
                                f"AI 제안사항 발행 - meetingId: {meeting_id}, "
                                f"개수: {len(suggestions.suggestions)}"
                            )

                    previous_count = current_count

                # 5초마다 체크
                await asyncio.sleep(5)

        except asyncio.CancelledError:
            logger.info(f"SSE 스트림 종료 - meetingId: {meeting_id}")
            # 회의 종료 시 데이터 정리는 선택사항 (나중에 조회 필요할 수도)
            # await redis_service.cleanup_meeting_data(meeting_id)

        except Exception as e:
            logger.error(f"SSE 스트림 오류 - meetingId: {meeting_id}", exc_info=e)

        finally:
            await redis_service.disconnect()

    return EventSourceResponse(event_generator())


@router.get("/test")
async def test_endpoint():
    """테스트 엔드포인트"""
    return {"message": "AI Suggestions API is working", "port": settings.port}