from __future__ import print_function
import json
import os
from watson_developer_cloud import ConversationV1
from flask import Flask, request, jsonify
from flask_restful import Resource, Api
#########################
# message
#########################
app = Flask(__name__)
api = Api(app)


context = []

class demo(Resource):
    def get(self):

        conversation = ConversationV1(
            username='325d6d5d-28b8-4f65-afe7-93d9b81805f3',
            password='Zum48rVSgqvF',
            version='2017-04-21')

        # replace with your own workspace_id
        workspace_id = '0544f6e6-3794-4c44-85ef-28ecdef148c1'
        if os.getenv("conversation_workspace_id") is not None:
            workspace_id = os.getenv("conversation_workspace_id")

        response = conversation.message(workspace_id=workspace_id, input={
            'text': 'HRA'})

        print(json.dumps(response["output"]["text"], indent=2))
        return(jsonify(response))

class demo1(Resource):

    def get(self, demo_id):
        context = None
        if context != None:
            print("hello")
        else:
            #context = None
            conversation = ConversationV1(
                                          username='325d6d5d-28b8-4f65-afe7-93d9b81805f3',
                                          password='Zum48rVSgqvF',
                                          version='2017-04-21')

                                          # replace with your own workspace_id
            workspace_id = '0544f6e6-3794-4c44-85ef-28ecdef148c1'

            #response = conversation.message(workspace_id=workspace_id,input={'text': demo_id})
            #response = conversation.message(workspace_id=workspace_id, input={'text': 'HRA'})
            # print(json.dumps(response, indent=2))

            response = conversation.message(workspace_id=workspace_id, input={'text':demo_id},context=context)
            context = response['context']
            print(context)

            #print(json.dumps(response, indent=2))
            return jsonify(context)

#print(json.dumps(response, indent=2))





api.add_resource(demo, '/demo') # Route_1
api.add_resource(demo1, '/demo/<demo_id>')

if __name__ == '__main__':
    app.run()