"""
=========================================
CV6 AI Trading OS
Risk Router
=========================================
"""

from fastapi import APIRouter

from app.risk.risk_manager import RiskManager
from app.risk.risk_models import (
    RiskRequest,
    RiskResponse,
)

router = APIRouter(
    prefix="/risk/engine",
    tags=["Risk Engine"]
)

manager = RiskManager()


@router.post(
    "/calculate",
    response_model=RiskResponse
)
def calculate_risk(
    request: RiskRequest
):

    result = manager.calculate(
        capital=request.capital,
        risk_percent=request.risk_percent,
        entry_price=request.entry_price,
        stop_loss=request.stop_loss,
        target_price=request.target_price
    )

    return result