from fastapi import APIRouter, Depends, Query
from sqlalchemy.orm import Session

from app.core.response import ApiResponse, ok
from app.core.user_context import UserContext, get_user_context
from app.db.session import get_db
from app.schemas.training_config import TrainingConfigOptionsResponse, TrainingConfigRecommendedResponse
from app.services.training_config_service import TrainingConfigService

router = APIRouter()


@router.get("/recommended", response_model=ApiResponse[TrainingConfigRecommendedResponse])
def get_recommended_training_config(
    case_id: int = Query(..., ge=1),
    _: UserContext = Depends(get_user_context),
    db: Session = Depends(get_db),
):
    """推荐配置信息：返回训练页默认病人初始化配置。"""
    return ok(TrainingConfigService(db).get_recommended(case_id))


@router.get("/options", response_model=ApiResponse[TrainingConfigOptionsResponse])
def get_training_config_options(
    case_id: int = Query(..., ge=1),
    _: UserContext = Depends(get_user_context),
    db: Session = Depends(get_db),
):
    """训练配置信息：返回训练页自定义病人初始化配置选项。"""
    return ok(TrainingConfigService(db).get_options(case_id))