router/image_embedding_router.py

from fastapi import APIRouter, HTTPException, UploadFile, File
from pydantic import BaseModel
from typing import List, Optional, Any
from io import BytesIO
from PIL import Image

# 코어 모듈에서 모델 가져오기 (싱글톤 보장)
from core.dependencies import get_image_embedding_model

router = APIRouter(tags=["Image Embedding"])

# 라우터 진입점에서 모델을 확보
image_embedding_model = get_image_embedding_model()

class ImageEmbeddingResponse(BaseModel):
    success: bool
    data: Optional[Any] = None
    msg: Optional[str] = None

@router.post("/image_to_embedding", response_model=ImageEmbeddingResponse)
async def image_to_embedding(file: UploadFile = File(...)):
    """
    이미지 파일을 업로드받아 EfficientNetV2-S 모델로 임베딩한 결과를 반환합니다.
    """
    try:
        # 이미지 파일 읽기
        contents = await file.read()
        image = Image.open(BytesIO(contents)).convert("RGB")
        
        # 이미지를 임베딩 변환. 1280 차원
        emb_vector = image_embedding_model.embed_image(image)
        
        return {"success": True, "data": {"embedding": emb_vector}, "msg": ""} 
    except Exception as e:
        return {"success": False, "data": None, "msg": str(e)}