forked from homanp/gcp-langchain
-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
46 lines (36 loc) · 1.27 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from flask import Flask, jsonify, make_response, request
from utils import create_chain
import requests
import os
app = Flask(__name__)
@app.route("/", methods = ['POST'])
def chat():
if not request.is_json:
return make_response(
jsonify(
{"success": False,
"error": "Unexpected error, request is not in JSON format"}),
400)
try:
data = request.json
message = data["events"][0]["message"]["text"]
replyToken = data["events"][0]["replyToken"]
chatgpt_chain = create_chain()
prediction = chatgpt_chain.predict(human_input=message)
response = {
"replyToken": replyToken,
"messages":
[
{ "type": "text",
"text": prediction
}
]
}
headers = {
"Authorization": f"Bearer {os.environ['CHANEL_ACCESS_TOKEN']}",
"Content-type": "application/json"
}
resp = requests.post("https://api.line.me/v2/bot/message/reply", data=response, headers=headers)
return resp.text
except Exception as e:
return make_response(f"There is an error processing request: {e}")