from fastapi import APIRouter, Depends

from app.clients.llm.base import LLMClient
from app.clients.storage.base import StorageClient
from app.core.dependencies import get_llm_client, get_storage_client
from app.models.image_models import ImageExtractRequest, ImageExtractResponse
from app.services import image_service

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


@router.post("/image/extract", response_model=ImageExtractResponse)
async def extract_image(
    req: ImageExtractRequest,
    llm: LLMClient = Depends(get_llm_client),
    storage: StorageClient = Depends(get_storage_client),
) -> ImageExtractResponse:
    return await image_service.extract_quads(req, llm, storage)