from flask import Flask, request, Response
import ibm_watson
import json

app = Flask(__name__)

API_KEY = "lV-3uzZNYNss6dVzCF1hHrsZzZaNxzoBdgtgHc-Ntx7Q"
W_URL = "https://gateway.watsonplatform.net/assistant/api"
AID = "9847758a-7dfa-430b-b67b-1ac18d65cc50"


@app.route("/message", methods=["POST"])
def login(*args, **kwargs):
    if request.method == "POST":
        msg = request.get_json(force=True).get("message", "")
        return process_message(msg)


def process_message(msg):
    service = ibm_watson.AssistantV2(
        iam_apikey=API_KEY, version="2019-02-28", url=W_URL
    )

    session_id = service.create_session(assistant_id=AID).get_result()["session_id"]

    response = service.message(
        assistant_id=AID,
        session_id=session_id,
        input={"message_type": "text", "text": msg},
    ).get_result()

    return Response(json.dumps(response), status=200)