Create app.py

app.py

@@ -0,0 +1,43 @@
+import gradio as gr
+import re
+import time 
+import requests
+
+flightFileLink = 'https://winningchatbot.s3.ap-southeast-1.amazonaws.com/SG-AD-S003' 
+tripFileLink= 'https://winningchatbot.s3.ap-southeast-1.amazonaws.com/SG-HR-W013'
+
+def chat(message, history):
+    
+  print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()), message)
+  
+  result = requests.get(f"https://5nzypzim7yvdq2jqbodhwnxpy40glpse.lambda-url.ap-southeast-1.on.aws/?message={message}").json()
+  print(result)
+  reply = result['text']
+  files = result['files']
+
+  print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()), reply)
+  
+  reply = re.sub(r'\【\d+\:\d+\†source\】', '', reply)
+
+  if 'file-akk9CVNRjnmc3vNk3sV203lN' in files:
+    reply = reply+ f"\n\n【信息源: SG-AD-S003 韦立新加坡-出差订票和住宿管理方案-修订20240129.pdf】\n({flightFileLink})"
+  if 'file-sHwZmYpPK67lsMX4uflSZXD1' in files:
+    reply = reply+ f"\n\n【信息源: SG-AD-S004 SG-HR-W013 出差制度】\n({tripFileLink})"
+
+  return reply
+
+gr.ChatInterface(
+  fn=chat,
+  chatbot= gr.Chatbot(),
+  textbox= gr.Textbox(placeholder="开始聊天", container=False, scale=7),
+  multimodal = False,
+  title="韦立 HR 聊天机器人 (GPT)",
+  description="目前所搜集的数据可回答关于韦立的出差订票，住宿管理方案，和出差制度方面问题。",
+  # theme="soft",
+  examples=["副董事长应该订哪一类机票?", "青岛的负责订票人员是谁?", "在新加坡能订哪些酒店?"],
+  cache_examples = False,
+  autofocus = True,                 
+  retry_btn = None, 
+  undo_btn = None,
+  clear_btn = None,
+).queue().launch(server_port=8080)