from flask import Blueprint, request, jsonify
from api.services.assessment_service import AssessmentService
from flask_jwt_extended import jwt_required, get_jwt_identity

assessment_bp = Blueprint('assessment', __name__, url_prefix='/api/assessment')

@assessment_bp.route('/submit', methods=['POST'])
@jwt_required()
def submit_assessment():
    user_id = get_jwt_identity()
    data = request.get_json()
    answers = data.get('answers')
    
    if not answers or not isinstance(answers, list):
        return jsonify({'status': False, 'message': 'Invalid answers format'}), 400
        
    result = AssessmentService.calculate_risk(user_id, answers)
    if result['status']:
        return jsonify(result), 200
    return jsonify(result), 500