"""
Models API endpoints
"""

from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.ext.asyncio import AsyncSession
from typing import List
import structlog

from app.core.database import get_db
from app.core.auth import verify_token
from app.schemas.training import TrainedModelResponse
from app.services.training_service import TrainingService

logger = structlog.get_logger()
router = APIRouter()

training_service = TrainingService()

@router.get("/", response_model=List[TrainedModelResponse])
async def get_trained_models(
    user_data: dict = Depends(verify_token),
    db: AsyncSession = Depends(get_db)
):
    """Get trained models"""
    try:
        return await training_service.get_trained_models(user_data, db)
    except Exception as e:
        logger.error(f"Get trained models error: {e}")
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail="Failed to get trained models"
        )