from flask import Blueprint, request 
from flask_restful import Api, Resource
from chatBot.common.utils import getAnswerLlama, getAnswerGpt

resources = Blueprint('resources', __name__)
api = Api(resources)

class Gpt(Resource):
    def get(self):
        question = request.json['question']

        return {'answer': getAnswerGpt(question)}
api.add_resource(Gpt, '/gpt')


class Llama(Resource):
    def get(self):
        question = request.json['question']
        return {'answer': getAnswerLlama(question)}
api.add_resource(Llama, '/llama')