jarvis-server/jarvis/main.py

import flask
import lingua_franca
from flask import Flask, request, jsonify, Response

from jarvis.ia import process
from jarvis.skills import intent_manager
from jarvis.skills.entertainement.spotify import SpotifySkill
from jarvis.utils import languages_utils
from utils import config_utils, flask_utils, intents_utils, utils

app = Flask(__name__)


@app.route("/process", methods=['POST'])
def process_request():
    data = flask_utils.get_data_in_request(request)

    if 'sentence' not in data or not data['sentence']:
        flask.abort(Response('You must provide a \'sentence\' parameter (not empty aswell)!'))

    sentence = data['sentence']
    tag_for_request = process.get_tag_for_sentence(sentence)

    print("SENTENCE : " + sentence + " /// TAG : " + tag_for_request)

    # stop here if the sentence isn't understood
    if tag_for_request == 'dont_understand':
        return jsonify("I didn't get that.")

    path_of_intent = intents_utils.get_path(tag_for_request)
    path_of_intent = path_of_intent.split('/skills/')[1].replace('/', '.')
    path_of_intent = "skills." + path_of_intent + "intent"

    method = utils.import_method_from_string(path_of_intent, tag_for_request)
    return jsonify(method())


if __name__ == '__main__':
    # Load lingua franca in the memory
    # Supported : English French German Hungarian Italian Portuguese Swedish
    lingua_franca.load_language(lang=languages_utils.get_language().split("-")[0])

    # Tests
    # WikipediaSkill().register()
    # JokesSkill().register()
    SpotifySkill().register()

    intent_manager.process_handlers()

    # intent_manager.recognise("cherche sur wikipedia Elon Musk")  # TO CHECK
    # intent_manager.recognise("raconte moi une blague")  # WORKING
    # intent_manager.recognise("joue le morceau crazy crazy nights de KISS sur spotify")  # WORKING
    # intent_manager.recognise("coupe la musique")  # WORKING
    intent_manager.recognise("c'est quoi le nom de cette chanson ?")

    # start the flask server
    app.config['JSON_AS_ASCII'] = False
    app.run(port=config_utils.get_in_config("PORT"), debug=False, host='0.0.0.0', threaded=True)