from fastapi import APIRouter, HTTPException

from app.models.training import TrainingStatus
from app.services.training_monitor import TrainingMonitor

router = APIRouter()
monitor = TrainingMonitor()


@router.get("/status", response_model=TrainingStatus)
async def get_training_status():
    """
    Get current training status including progress, loss, and learning rate
    """
    try:
        return await monitor.get_status()
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.get("/log")
async def get_training_log():
    """
    Get recent training log entries
    """
    try:
        return await monitor.get_log()
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))