from flask import Blueprint, request, jsonify

from src.prediction_system import get_prediction, healthcheck

import json

model_controller = Blueprint('controller', __name__)

@model_controller.route('/predict', methods=['POST'])

def prediction_endpoint():

    if not request.is_json:

        return jsonify({'error': 'Invalid JSON data'}), 400

    data = json.loads(request.data)

    if 'array'in data:

        response = get_prediction(data)

        return jsonify(response)

    else:

        return jsonify({'error': 'Invalid or missing "array" field'}), 400

@model_controller.route('/healthcheck', methods=['GET'])

def healthcheck_endpoint():

    return jsonify({'model': healthcheck()}), 400