from flask import Blueprint, request, jsonify
from api.services.portfolio_service import PortfolioService
from flask_jwt_extended import jwt_required, get_jwt_identity
from api.models.user import User

portfolio_bp = Blueprint('portfolio', __name__, url_prefix='/api/portfolio')

@portfolio_bp.route('/recommend', methods=['GET'])
@jwt_required()
def get_recommendation():
    user_id = get_jwt_identity()
    user = User.query.get(user_id)
    if not user:
        return jsonify({'status': False, 'message': 'User not found'}), 401
        
    risk_level = user.risk_level
    
    result = PortfolioService.get_recommendation(user_id, risk_level)
    if result['status']:
        return jsonify(result), 200
    return jsonify(result), 500