from flask import Blueprint, request, jsonify

import joblib

import os

from app.load_model import model

blueprint = Blueprint('blue_print', __name__)

# Prediction request

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

def predict():

    # Get the new value

    gender = request.json.get('gender')

    age = request.json.get('age')

    smoking = request.json.get('smoking')

    yellow_skin = request.json.get('yellow_skin')

    fatigue = request.json.get('fatigue')

    wheezing = request.json.get('wheezing')

    coughing = request.json.get('coughing')

    shortness_of_breath = request.json.get('shortness_of_breath')

    swallowing_difficulty = request.json.get('swallowing_difficulty')

    chest_pain = request.json.get('chest_pain')

    chronic_disease = request.json.get('chronic_disease')

    # Value type request

    def convert_data(data):

        if isinstance(data, int):

           return int(data)

        elif isinstance(data, str):

           return str(data)

    # Input data for prediction

    data = [

            convert_data(gender), 

            convert_data(age), 

            convert_data(smoking), 

            convert_data(yellow_skin), 

            convert_data(fatigue), 

            convert_data(wheezing), 

            convert_data(coughing), 

            convert_data(shortness_of_breath), 

            convert_data(swallowing_difficulty), 

            convert_data(chest_pain), 

            convert_data(chronic_disease)

           ]

    pred = model.predict([data])[0] # Predict new data

    proba = model.predict_proba([data])[0][pred] # Get the prediction outcome

    return jsonify({'prediction': int(pred), 'proba': float(proba)}), 200