JasonData commited on
Commit
1e8ed33
·
verified ·
1 Parent(s): 7a32c69

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +43 -0
app.py ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import re
3
+ import time
4
+ import requests
5
+
6
+ flightFileLink = 'https://winningchatbot.s3.ap-southeast-1.amazonaws.com/SG-AD-S003'
7
+ tripFileLink= 'https://winningchatbot.s3.ap-southeast-1.amazonaws.com/SG-HR-W013'
8
+
9
+ def chat(message, history):
10
+
11
+ print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()), message)
12
+
13
+ result = requests.get(f"https://5nzypzim7yvdq2jqbodhwnxpy40glpse.lambda-url.ap-southeast-1.on.aws/?message={message}").json()
14
+ print(result)
15
+ reply = result['text']
16
+ files = result['files']
17
+
18
+ print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()), reply)
19
+
20
+ reply = re.sub(r'\【\d+\:\d+\†source\】', '', reply)
21
+
22
+ if 'file-akk9CVNRjnmc3vNk3sV203lN' in files:
23
+ reply = reply+ f"\n\n【信息源: SG-AD-S003 韦立新加坡-出差订票和住宿管理方案-修订20240129.pdf】\n({flightFileLink})"
24
+ if 'file-sHwZmYpPK67lsMX4uflSZXD1' in files:
25
+ reply = reply+ f"\n\n【信息源: SG-AD-S004 SG-HR-W013 出差制度】\n({tripFileLink})"
26
+
27
+ return reply
28
+
29
+ gr.ChatInterface(
30
+ fn=chat,
31
+ chatbot= gr.Chatbot(),
32
+ textbox= gr.Textbox(placeholder="开始聊天", container=False, scale=7),
33
+ multimodal = False,
34
+ title="韦立 HR 聊天机器人 (GPT)",
35
+ description="目前所搜集的数据可回答关于韦立的出差订票,住宿管理方案,和出差制度方面问题。",
36
+ # theme="soft",
37
+ examples=["副董事长应该订哪一类机票?", "青岛的负责订票人员是谁?", "在新加坡能订哪些酒店?"],
38
+ cache_examples = False,
39
+ autofocus = True,
40
+ retry_btn = None,
41
+ undo_btn = None,
42
+ clear_btn = None,
43
+ ).queue().launch(server_port=8080)