main.py

from flask import Flask, request
import json
import numpy as np
from linebot.models import TextSendMessage, StickerSendMessage
from linebot.models.template import *
from linebot import LineBotApi
from dotenv import dotenv_values
from utils import build_chatbot

config = dotenv_values('.env')

user_history = dict()

line_bot_api = LineBotApi(config["LINE_ACCESS_TOKEN"])
app = Flask(__name__)

@app.route('/')
def index():
    return "Hello World!"

@app.route('/webhook', methods=['POST'])
def callback():
    json_line = request.get_json(force=False,cache=False)
    json_line = json.dumps(json_line)
    decoded = json.loads(json_line)
    no_event = len(decoded['events'])
    for i in range(no_event):
        event = decoded['events'][i]
        event_handle(event)
    return '',200

def event_handle(event):
    try:
        userId = event['source']['userId']
    except:
        print('error cannot get userId')
        return ''

    try:
        rtoken = event['replyToken']
    except:
        print('error cannot get rtoken')
        return ''
    try:
        msgId = event["message"]["id"]
        msgType = event["message"]["type"]
    except:
        print('error cannot get msgID, and msgType')
        sk_id = np.random.randint(1,17)
        replyObj = StickerSendMessage(package_id=str(1),sticker_id=str(sk_id))
        line_bot_api.reply_message(rtoken, replyObj)
        return ''

    if msgType == "text":
        msg = str(event["message"]["text"])
        userId = event['source']['userId']

        ai_msg = build_chatbot(userId, msg, user_history)
        ai_replyObj = TextSendMessage(text=ai_msg)
        line_bot_api.reply_message(rtoken, ai_replyObj)
    else:
        sk_id = np.random.randint(1,17)
        replyObj = StickerSendMessage(package_id=str(1),sticker_id=str(sk_id))
        line_bot_api.reply_message(rtoken, replyObj)
    return ''

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