OneBot-Niko.py.old

"""
Copyright (C) 2019 Peter Maar

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.

"""

import watson_developer_cloud.assistant_v1 as av1
from time import sleep

DEBUG = False

VERSION = '2018-10-15'
API_KEY = '' #REMOVED
WORKSPACE_ID = '' #REMOVED


def display_response(response_lines):
    """ Displays the lines of the response """

    for line in response_lines:
        rt = line["response_type"]

        # Bot gave text, so show it
        if rt == "text":
            print(line['text'])

        # Bot 'typing' or waiting, so wait for a moment
        elif rt == "pause":
            if line['typing']:
                print('User is typing...')

            # Sleep for typing duration (or just print it if debugging mode)
            seconds = line['time'] / 1000  # Convert from ms to s
            if DEBUG:
                print(seconds, "second sleep")
            else:
                sleep(seconds)

        elif rt == "option":
            print(line['title'])
            for o in line['options']:
                print(o['label'], ": ", o['value']['input']['text'], sep="")

        # Short pause between anything, even if no 'typing' (unless debugging mode)
        if not DEBUG:
            sleep(1)

        print()  # Newline


# Create assistant object
assistant = av1.AssistantV1(
    version=VERSION,
    iam_apikey=API_KEY,
    url='https://gateway-wdc.watsonplatform.net/assistant/api'
)


userInp = ''  # Sending empty string will give back the welcome message, so initialize it to that
contextVar = None  # Initialize for scope

print("Connected. Send exit to disconnect at any time.")


# Main Loop
while userInp.lower() != 'exit':

    # Create InputData object for text to send
    userInpData = av1.InputData(userInp)

    # Send our message and store it all to 'message'
    message = assistant.message(
        workspace_id=WORKSPACE_ID,
        input=userInpData,
        context=contextVar
    )

    # Get/isolate and store the response to the message
    response = message.result

    # Get the context variable
    contextVar = response['context']

    # Get just the array of the different lines (text, pause, etc)
    lines = response["output"]["generic"]

    if DEBUG:
        print("------------------")
        print(message)
        print("------------------")
        print(response)
        print("------------------")

    display_response(lines)

    if DEBUG:
        print("------------------")

    if userInp.lower().find("bye") != -1:
        print("To disconnect, enter 'exit' (no quotes) at any time.", end="")

    userInp = input("\n>> ")