"""
=========================================
CV6 AI Trading OS
Strategy Router
=========================================
"""

from fastapi import APIRouter

from app.strategy.strategy_manager import StrategyManager
from app.strategy.strategy_models import (
    StrategyRequest,
    StrategyResponse,
)

router = APIRouter(
    prefix="/strategy",
    tags=["Strategy"]
)

manager = StrategyManager()


@router.post(
    "/generate",
    response_model=StrategyResponse
)
def generate_strategy(
    request: StrategyRequest
):

    result = manager.generate_strategy(
        ema=request.ema,
        rsi=request.rsi,
        macd=request.macd,
        supertrend=request.supertrend,
        atr=request.atr,
        vwap=request.vwap,
    )

    return result